Exchange packets between server and client

CriticalPathAnalyzer.sln.DotSettings.user

@@ -0,0 +1,2 @@
+<wpf:ResourceDictionary xml:space="preserve" xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml" xmlns:s="clr-namespace:System;assembly=mscorlib" xmlns:ss="urn:shemas-jetbrains-com:settings-storage-xaml" xmlns:wpf="http://schemas.microsoft.com/winfx/2006/xaml/presentation">
+	<s:String x:Key="/Default/CodeInspection/Highlighting/SweaWarningsMode/@EntryValue">DoNotShowAndRun</s:String></wpf:ResourceDictionary>

CriticalPathAnalyzer/CriticalPathAnalyzer.csproj

@@ -56,9 +56,21 @@
         <Reference Include="LogicWorld.SharedCode">
             <HintPath>..\LogicWorld\Logic_World_Data\Managed\LogicWorld.SharedCode.dll</HintPath>
         </Reference>
+        <Reference Include="LogicWorld.Networking">
+            <HintPath>..\LogicWorld\Logic_World_Data\Managed\LogicWorld.Networking.dll</HintPath>
+        </Reference>
         <Reference Include="UnityEngine.CoreModule">
             <HintPath>..\LogicWorld\Logic_World_Data\Managed\UnityEngine.CoreModule.dll</HintPath>
         </Reference>
+        <Reference Include="Server">
+            <HintPath>..\LogicWorld\Server\Server.dll</HintPath>
+        </Reference>
+        <Reference Include="LogicAPI.Server">
+            <HintPath>..\LogicWorld\Server\LogicAPI.Server.dll</HintPath>
+        </Reference>
+        <Reference Include="MessagePack.Annotations">
+            <HintPath>..\LogicWorld\Server\MessagePack.Annotations.dll</HintPath>
+        </Reference>
     </ItemGroup>
     <ItemGroup>
         <ProjectReference Include="..\EccsLogicWorldAPI\EccsLogicWorldAPI.csproj"/>

CriticalPathAnalyzer/CriticalPathAnalyzer/manifest.jecs

@@ -1,7 +1,7 @@
 ID: CriticalPathAnalyzer
 Name: CriticalPathAnalyzer 
 Author: D4VID
-Version: 0.0.1
+Version: 0.0.2
 Priority: 0
 Dependencies:
     - HarmonyForLogicWorld

CriticalPathAnalyzer/CriticalPathAnalyzer/src/client/CriticalPathAnalyzerClient.cs

@@ -1,17 +1,20 @@
-
-using CriticalPathAnalyzer.Client.Keybindings;
+using CriticalPathAnalyzer.Client.Keybindings;
+using CriticalPathAnalyzer.Client.Network;
+using CriticalPathAnalyzer.Client.Tool;
+using CriticalPathAnalyzer.Shared.Packets.S2C;
 using EccsLogicWorldAPI.Client.Injectors;
 using FancyInput;
 using LogicAPI.Client;
 using LogicLog;
 
 namespace CriticalPathAnalyzer.Client {
-
     public class CriticalPathAnalyzerClient : ClientMod {
         public static ILogicLogger LoggerInstance;
+
         protected override void Initialize() {
-         LoggerInstance = Logger;
             Logger.Info("CriticalPathAnalyzer mod loading");
+            LoggerInstance = Logger;
+            CriticalPathAnalyzerTool.Init(Logger);
 
             // Register keybindings in the settings menu
             CustomInput.Register<CriticalPathAnalyzerContext, CriticalPathAnalyzerTrigger>("CriticalPathAnalyzer");
@@ -22,7 +25,15 @@ namespace CriticalPathAnalyzer.Client {
             // Inject a hook into the game code to check for our keybindings
             KeybindingsInjector.Inject();
             
+            // Inject packet handlers
+			RawPacketHandlerInjector.addPacketHandler(new AnnounceModPacketHandler());
+			RawPacketHandlerInjector.addPacketHandler(new AnalyzePathResponseHandler());
+
             Logger.Info("CriticalPathAnalyzer mod loaded");
         }
+
+        public static void OnAnalyzePathResponse(AnalyzePathResponse response) {
+            LoggerInstance.Info($"Got response from server: {response.Message}");
+        }
     }
 }

CriticalPathAnalyzer/CriticalPathAnalyzer/src/client/CriticalPathAnalyzerGameState.cs

@@ -1,6 +1,6 @@
 using System.Collections.Generic;
 using CriticalPathAnalyzer.Client.Keybindings;
-using CriticalPathAnalyzer.Client.tool;
+using CriticalPathAnalyzer.Client.Tool;
 using FancyInput;
 using LogicUI;
 using LogicWorld.GameStates;

CriticalPathAnalyzer/CriticalPathAnalyzer/src/client/network/AnalyzePathResponseHandler.cs

@@ -0,0 +1,10 @@
+using CriticalPathAnalyzer.Shared.Packets.S2C;
+using LogicWorld.SharedCode.Networking;
+
+namespace CriticalPathAnalyzer.Client.Network {
+    public class AnalyzePathResponseHandler : PacketHandler<AnalyzePathResponse> {
+        public override void Handle(AnalyzePathResponse packet, HandlerContext context) {
+            CriticalPathAnalyzerClient.OnAnalyzePathResponse(packet);
+        }
+    }
+}

CriticalPathAnalyzer/CriticalPathAnalyzer/src/client/network/AnnounceModPacketHandler.cs

@@ -0,0 +1,10 @@
+using CriticalPathAnalyzer.Shared.Packets.S2C;
+using LogicWorld.SharedCode.Networking;
+
+namespace CriticalPathAnalyzer.Client.Network {
+    public class AnnounceModPacketHandler : PacketHandler<AnnounceModPresence> {
+        public override void Handle(AnnounceModPresence packet, HandlerContext context) {
+            CriticalPathAnalyzerClient.LoggerInstance.Info($"Mod is supported on the server: version={packet.Version}");
+        }
+    }
+}

CriticalPathAnalyzer/CriticalPathAnalyzer/src/client/tool/CriticalPathAnalyzerTool.cs

@@ -1,12 +1,24 @@
+using System;
+using System.Collections.Generic;
+using CriticalPathAnalyzer.Shared.Packets.C2S;
 using LogicAPI.Data;
+using LogicLog;
 using LogicWorld.Interfaces;
 using LogicWorld.Physics;
 using LogicWorld.Players;
 
-namespace CriticalPathAnalyzer.Client.tool {
+#pragma warning disable CS8073 // The result of the expression is always the same since a value of this type is never equal to 'null'
+
+namespace CriticalPathAnalyzer.Client.Tool {
     public class CriticalPathAnalyzerTool {
-        private static PegAddress _startPegAddress;
-        private static PegAddress _endPegAddress;
+        private static ILogicLogger _logger;
+
+        private static PegAddress? _startPegAddress;
+        private static PegAddress? _endPegAddress;
+
+        public static void Init(ILogicLogger logger) {
+            _logger = logger;
+        }
 
         private static PegAddress? RayCastPeg() {
             // Ray-cast into the world to find what the player is looking at
@@ -37,6 +49,9 @@ namespace CriticalPathAnalyzer.Client.tool {
             PegAddress? pegAddress = RayCastPeg();
             if (pegAddress != null) {
                 _startPegAddress = pegAddress.Value;
+                if (_startPegAddress != null && _endPegAddress != null) {
+                    CalculateCriticalPath();
+                }
             }
         }
 
@@ -45,6 +60,41 @@ namespace CriticalPathAnalyzer.Client.tool {
             PegAddress? pegAddress = RayCastPeg();
             if (pegAddress != null) {
                 _endPegAddress = pegAddress.Value;
+                if (_startPegAddress != null && _endPegAddress != null) {
+                    CalculateCriticalPath();
+                }
+            }
+        }
+
+        private static void CalculateCriticalPath() {
+            Instances.SendData.Send(new AnalyzePathRequest {
+                RequestGuid = Guid.NewGuid(),
+                StartPegAddress = _startPegAddress.Value,
+                EndPegAddress = _endPegAddress.Value,
+            });
+            HashSet<WireAddress> wires = Instances.MainWorld.Data.LookupPegWires(_startPegAddress.Value);
+            _logger.Info($"Start peg has {wires.Count} wires");
+
+            ComponentType andGate = Instances.MainWorld.ComponentTypes.GetComponentType("MHG.AndGate");
+            ComponentType xorGate = Instances.MainWorld.ComponentTypes.GetComponentType("MHG.XorGate");
+
+            foreach (WireAddress wireAddress in wires) {
+                Wire wire = Instances.MainWorld.Data.Lookup(wireAddress);
+                PegAddress nextPeg = wire.Point1 == _startPegAddress ? wire.Point2 : wire.Point1;
+                if (nextPeg.PegType != PegType.Input) {
+                    // ignore
+                    continue;
+                }
+
+                IComponentInWorld component = Instances.MainWorld.Data.Lookup(nextPeg.ComponentAddress);
+                ComponentType type = component.Data.Type;
+                if (type == andGate) {
+                    _logger.Info($"Connected to AND gate");
+                } else if (type == xorGate) {
+                    _logger.Info($"Connected to XOR gate");
+                } else {
+                    _logger.Info($"Connected to unknown gate");
+                }
             }
         }
     }

CriticalPathAnalyzer/CriticalPathAnalyzer/src/server/CriticalPathAnalyzerServer.cs

@@ -0,0 +1,69 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using CriticalPathAnalyzer.Server.Network;
+using CriticalPathAnalyzer.Shared.Packets.S2C;
+using EccsLogicWorldAPI.Server.Injectors;
+using EccsLogicWorldAPI.Shared.PacketWrapper;
+using LogicAPI.Data;
+using LogicAPI.Networking;
+using LogicAPI.Networking.Packets.Initialization;
+using LogicAPI.Server;
+using LogicAPI.Server.Networking;
+using LogicAPI.Server.Networking.ClientVerification;
+using LogicLog;
+using LogicWorld.Server;
+using LogicWorld.SharedCode.Networking;
+
+namespace CriticalPathAnalyzer.Server {
+    public class CriticalPathAnalyzerServer : ServerMod, IClientVerifier {
+        private const string ModId = "CriticalPathAnalyzer";
+        private const string ModVersion = "0.0.2";
+
+        private NetworkServer _networkServer;
+        private readonly Dictionary<string, bool> _playerHasWireTracer = new Dictionary<string, bool>();
+
+        public static ILogicLogger LoggerInstance;
+
+        protected override void Initialize() {
+            Logger.Info("CriticalPathAnalyzer server mod loading");
+            LoggerInstance = Logger;
+
+            _networkServer = Program.Get<NetworkServer>();
+            if (_networkServer == null) {
+                throw new Exception("Could not get Service 'NetworkServer'.");
+            }
+
+            // Inject verifier:
+            RawJoinVerifierInjector.addVerifier(this);
+            PacketHandlerManager.getCustomPacketHandler<ClientLoadedWorldPacket>()
+                .addHandlerToEnd(new ClientJoinedPacketHandler(this));
+            RawPacketHandlerInjector.addPacketHandler(new CriticalPathAnalyzerRequestHandler(this));
+        }
+
+        public void Verify(VerificationContext ctx) {
+            bool hasMod = ctx.ApprovalPacket.ClientMods.Contains(ModId);
+            string playerName = ctx.PlayerID.Name;
+            _playerHasWireTracer[playerName] = hasMod;
+            LoggerInstance.Info($"Verifying player {playerName}: hasMod={hasMod}");
+        }
+
+        public void PlayerJoined(PlayerData playerData, HandlerContext context) {
+            _playerHasWireTracer.TryGetValue(playerData.Name, out bool hasWireTracer);
+            if (hasWireTracer) {
+                _networkServer.Send(context.Sender, new AnnounceModPresence() {
+                    Version = ModVersion
+                });
+            }
+        }
+
+        public void AnalyzePath(Connection sender, Guid packetRequestGuid, PegAddress start, PegAddress end) {
+            LoggerInstance.Info("Got AnalyzePath request");
+            AnalyzePathResponse response = new AnalyzePathResponse() {
+                RequestGuid = packetRequestGuid,
+                Message = "Lmao Yeet",
+            };
+            _networkServer.Send(sender, response);
+        }
+    }
+}

CriticalPathAnalyzer/CriticalPathAnalyzer/src/server/network/AnalyzePathRequestHandler.cs

@@ -0,0 +1,16 @@
+using CriticalPathAnalyzer.Shared.Packets.C2S;
+using LogicWorld.SharedCode.Networking;
+
+namespace CriticalPathAnalyzer.Server.Network {
+    public class CriticalPathAnalyzerRequestHandler : PacketHandler<AnalyzePathRequest> {
+        private readonly CriticalPathAnalyzerServer _criticalPathAnalyzerServer;
+        public CriticalPathAnalyzerRequestHandler(CriticalPathAnalyzerServer criticalPathAnalyzerServer) {
+            _criticalPathAnalyzerServer = criticalPathAnalyzerServer;
+        }
+
+        public override void Handle(AnalyzePathRequest packet, HandlerContext context) {
+            _criticalPathAnalyzerServer.AnalyzePath(context.Sender, packet.RequestGuid, packet.StartPegAddress,
+                packet.EndPegAddress);
+        }
+    }
+}

CriticalPathAnalyzer/CriticalPathAnalyzer/src/server/network/ClientJoinedPacketHandler.cs

@@ -0,0 +1,16 @@
+using EccsLogicWorldAPI.Shared.PacketWrapper;
+using LogicAPI.Networking.Packets.Initialization;
+using LogicWorld.SharedCode.Networking;
+
+namespace CriticalPathAnalyzer.Server.Network {
+    public class ClientJoinedPacketHandler : CustomPacketHandler<ClientLoadedWorldPacket> {
+        private readonly CriticalPathAnalyzerServer _criticalPathAnalyzerServer;
+        public ClientJoinedPacketHandler(CriticalPathAnalyzerServer criticalPathAnalyzerServer) {
+            _criticalPathAnalyzerServer = criticalPathAnalyzerServer;
+        }
+
+        public override void handle(ref bool isCancelled, ref ClientLoadedWorldPacket packet, HandlerContext context) {
+            _criticalPathAnalyzerServer.PlayerJoined(packet.PlayerData, context);
+        }
+    }
+}

CriticalPathAnalyzer/CriticalPathAnalyzer/src/shared/packets/c2s/AnalyzePathRequest.cs

@@ -0,0 +1,13 @@
+using System;
+using LogicAPI.Data;
+using LogicAPI.Networking.Packets;
+using MessagePack;
+
+namespace CriticalPathAnalyzer.Shared.Packets.C2S {
+    [MessagePackObject]
+    public class AnalyzePathRequest : Packet {
+        [Key(0)] public Guid RequestGuid;
+        [Key(1)] public PegAddress StartPegAddress;
+        [Key(2)] public PegAddress EndPegAddress;
+    }
+}

CriticalPathAnalyzer/CriticalPathAnalyzer/src/shared/packets/s2c/AnalyzePathResponse.cs

@@ -0,0 +1,11 @@
+using System;
+using LogicAPI.Networking.Packets;
+using MessagePack;
+
+namespace CriticalPathAnalyzer.Shared.Packets.S2C {
+    [MessagePackObject]
+    public class AnalyzePathResponse : Packet {
+        [Key(0)] public Guid RequestGuid;
+        [Key(1)] public string Message;
+    }
+}

CriticalPathAnalyzer/CriticalPathAnalyzer/src/shared/packets/s2c/AnnounceModPresence.cs

@@ -0,0 +1,9 @@
+using LogicAPI.Networking.Packets;
+using MessagePack;
+
+namespace CriticalPathAnalyzer.Shared.Packets.S2C {
+    [MessagePackObject]
+    public class AnnounceModPresence : Packet {
+        [Key(0)] public string Version;
+    }
+}