Merge pull request #544 from starfi5h/pr-milestone

Add milestone syncing

Author: starfi5h

NebulaModel/Packets/Statistics/MilestoneUnlockPacket.cs

@@ -0,0 +1,19 @@
+namespace NebulaModel.Packets.Statistics
+{
+    public class MilestoneUnlockPacket
+    {
+        public int Id { get; set; }
+        public long UnlockTick { get; set; }
+        public int PatternId { get; set; }
+        public long[] Parameters { get; set; }
+
+        public MilestoneUnlockPacket() { }
+        public MilestoneUnlockPacket(int id, long unlockTick, int patternId, long[] parameters)
+        {
+            Id = id;
+            UnlockTick = unlockTick;
+            PatternId = patternId;
+            Parameters = parameters;
+        }
+    }
+}

NebulaNetwork/PacketProcessors/Statistics/MilestoneUnlockProcessor.cs

@@ -0,0 +1,44 @@
+using NebulaAPI;
+using NebulaModel.Networking;
+using NebulaModel.Packets;
+using NebulaModel.Packets.Statistics;
+using NebulaWorld;
+
+namespace NebulaNetwork.PacketProcessors.Statistics
+{
+    [RegisterPacketProcessor]
+    internal class MilestoneUnlockProcessor : PacketProcessor<MilestoneUnlockPacket>
+    {
+        public override void ProcessPacket(MilestoneUnlockPacket packet, NebulaConnection conn)
+        {
+            IPlayerManager playerManager = Multiplayer.Session.Network.PlayerManager;
+            bool valid = true;
+
+            if (IsHost)
+            {
+                INebulaPlayer player = playerManager.GetPlayer(conn);
+                if (player != null)
+                {
+                    playerManager.SendPacketToOtherPlayers(packet, player);
+                }
+                else
+                {
+                    valid = false;
+                }
+            }
+
+            if (valid)
+            {
+                using (Multiplayer.Session.Statistics.IsIncomingRequest.On())
+                {
+                    if (GameMain.data.milestoneSystem.milestoneDatas.TryGetValue(packet.Id, out MilestoneData milestoneData))
+                    {
+                        milestoneData.journalData.patternId = packet.PatternId;
+                        milestoneData.journalData.parameters = packet.Parameters;
+                        GameMain.data.milestoneSystem.UnlockMilestone(packet.Id, packet.UnlockTick);
+                    }
+                }
+            }
+        }
+    }
+}

NebulaPatcher/Patches/Dynamic/GameData_Patch.cs

@@ -4,6 +4,7 @@
 using NebulaModel.Networking;
 using NebulaModel.Packets.Logistics;
 using NebulaModel.Packets.Players;
+using NebulaModel.Packets.Statistics;
 using NebulaPatcher.Patches.Transpilers;
 using NebulaWorld;
 using NebulaWorld.Warning;

NebulaPatcher/Patches/Dynamic/MilestoneSystem_Patch.cs

@@ -12,14 +12,29 @@ internal class MilestoneSystem_Patch
         [HarmonyPatch(nameof(MilestoneSystem.SetForNewGame))]
         public static void SetForNewGame_Postfix()
         {
-            // Do not run if it is not multiplayer and the player is not a client
-            if (!Multiplayer.IsActive || Multiplayer.Session.LocalPlayer.IsHost)
+            if (Multiplayer.IsActive && !Multiplayer.Session.LocalPlayer.IsHost)
+            {
+                // Request milestone data
+                Log.Info($"Requesting MilestoneData from the server");
+                Multiplayer.Session.Network.SendPacket(new MilestoneDataRequest());
+            }
+        }
+
+        [HarmonyPostfix]
+        [HarmonyPatch(nameof(MilestoneSystem.UnlockMilestone))]
+        public static void UnlockMilestone_Postfix(MilestoneSystem __instance, int id, long unlockTick)
+        {
+            if (!Multiplayer.IsActive || Multiplayer.Session.Statistics.IsIncomingRequest)
             {
                 return;
             }
-            // Request milestone data
-            Log.Info($"Requesting MilestoneData from the server");
-            Multiplayer.Session.Network.SendPacket(new MilestoneDataRequest());
+            
+            if (__instance.milestoneDatas.TryGetValue(id, out MilestoneData milestoneData))
+            {
+                int patternId = milestoneData.journalData.patternId;
+                long[] parameters = milestoneData.journalData.parameters;
+                Multiplayer.Session.Network.SendPacket(new MilestoneUnlockPacket(id, unlockTick, patternId, parameters));
+            }
         }
     }
 }