Sync UIBeltWindow.OnReverseButtonClick

NebulaModTeam · Feb 1, 2023 · 439d302 · 439d302
1 parent a0f232b
commit 439d302
Show file tree

Hide file tree

Showing 3 changed files with 76 additions and 0 deletions.
diff --git a/NebulaModel/Packets/Factory/Belt/BeltReversePacket.cs b/NebulaModel/Packets/Factory/Belt/BeltReversePacket.cs
@@ -0,0 +1,15 @@
+namespace NebulaModel.Packets.Factory.Belt
+{
+    public class BeltReversePacket
+    {
+        public int BeltId { get; set; }
+        public int PlanetId { get; set; }
+
+        public BeltReversePacket() { }
+        public BeltReversePacket(int beltId, int planetId)
+        {
+            BeltId = beltId;
+            PlanetId = planetId;
+        }
+    }
+}
diff --git a/NebulaNetwork/PacketProcessors/Factory/Belt/BeltReverseProcessor.cs b/NebulaNetwork/PacketProcessors/Factory/Belt/BeltReverseProcessor.cs
@@ -0,0 +1,40 @@
+using NebulaAPI;
+using NebulaModel.Packets.Factory.Belt;
+
+namespace NebulaNetwork.PacketProcessors.Factory.Belt
+{
+    [RegisterPacketProcessor]
+    internal class BeltReverseProcessor : BasePacketProcessor<BeltReversePacket>
+    {
+        public override void ProcessPacket(BeltReversePacket packet, INebulaConnection conn)
+        {
+            PlanetFactory factory = GameMain.galaxy.PlanetById(packet.PlanetId).factory;
+            if (factory == null)
+                return;
+
+            using (NebulaModAPI.MultiplayerSession.Factories.IsIncomingRequest.On())
+            {
+                NebulaModAPI.MultiplayerSession.Factories.EventFactory = factory;
+                NebulaModAPI.MultiplayerSession.Factories.TargetPlanet = packet.PlanetId;
+                if (NebulaModAPI.MultiplayerSession.LocalPlayer.IsHost)
+                {
+                    // Load planet model
+                    NebulaModAPI.MultiplayerSession.Factories.AddPlanetTimer(packet.PlanetId);
+                }
+
+                UIBeltWindow beltWindow = UIRoot.instance.uiGame.beltWindow;
+                beltWindow._Close(); // close the window first to avoid changing unwant variable when setting beltId
+                PlanetFactory tmpFactory = beltWindow.factory;
+                int tmpBeltId = beltWindow.beltId;
+                beltWindow.factory = factory;
+                beltWindow.beltId = packet.BeltId;
+                beltWindow.OnReverseButtonClick(0);
+                beltWindow.factory = tmpFactory;
+                beltWindow.beltId = tmpBeltId;
+
+                NebulaModAPI.MultiplayerSession.Factories.EventFactory = null;
+                NebulaModAPI.MultiplayerSession.Factories.TargetPlanet = NebulaModAPI.PLANET_NONE;
+            }
+        }
+    }
+}
diff --git a/NebulaPatcher/Patches/Dynamic/UIBeltWindow_Patch.cs b/NebulaPatcher/Patches/Dynamic/UIBeltWindow_Patch.cs
@@ -0,0 +1,21 @@
+using HarmonyLib;
+using NebulaModel.Packets.Factory.Belt;
+using NebulaWorld;
+
+namespace NebulaPatcher.Patches.Dynamic
+{
+    [HarmonyPatch(typeof(UIBeltWindow))]
+    internal class UIBeltWindow_Patch
+    {
+        [HarmonyPostfix]
+        [HarmonyPatch(typeof(UIBeltWindow), nameof(UIBeltWindow.OnReverseButtonClick))]
+        public static void OnReverseButtonClick_Postfix(UIBeltWindow __instance)
+        {
+            // Notify others about belt direction reverse
+            if (Multiplayer.IsActive && !Multiplayer.Session.Factories.IsIncomingRequest.Value)
+            {
+                Multiplayer.Session.Network.SendPacketToLocalStar(new BeltReversePacket(__instance.beltId, __instance.factory.planetId));
+            }
+        }
+    }
+}