From 5afcb0c390c3170b752e34c8f96b94e0a54cc15b Mon Sep 17 00:00:00 2001
From: SokyranTheDragon <36712560+SokyranTheDragon@users.noreply.github.com>
Date: Fri, 16 Aug 2024 04:48:14 +0200
Subject: [PATCH] Compat for Vanilla Races Expanded - Insector (#459)

A few methods weren't synced, but should be handled by MP itself due to calling `TryTakeOrderedJob`.
---
 Source/Mods/VanillaRacesInsector.cs | 174 ++++++++++++++++++++++++++++
 1 file changed, 174 insertions(+)
 create mode 100644 Source/Mods/VanillaRacesInsector.cs

diff --git a/Source/Mods/VanillaRacesInsector.cs b/Source/Mods/VanillaRacesInsector.cs
new file mode 100644
index 0000000..58407ef
--- /dev/null
+++ b/Source/Mods/VanillaRacesInsector.cs
@@ -0,0 +1,174 @@
+using System;
+using System.Collections;
+using System.Collections.Generic;
+using System.Linq;
+using HarmonyLib;
+using Multiplayer.API;
+using Verse;
+
+namespace Multiplayer.Compat;
+
+/// <summary>Vanilla Races Expanded - Insector by Oskar Potocki, Taranchuk, Sarg</summary>
+/// <see href="https://github.com/Vanilla-Expanded/VanillaRacesExpanded-Insector"/>
+/// <see href="https://steamcommunity.com/sharedfiles/filedetails/?id=3260509684"/>
+[MpCompatFor("vanillaracesexpanded.insector")]
+public class VanillaRacesInsector
+{
+    #region Fields
+
+    // GameComponent_Genelines
+    private static AccessTools.FieldRef<GameComponent> genelinesInstanceField;
+    private static AccessTools.FieldRef<GameComponent, IList> genelinesListField;
+    private static FastInvokeHandler createGenelineMethod;
+    private static FastInvokeHandler addGenelineMethod;
+
+    // Geneline
+    private static AccessTools.FieldRef<object, int> genelineIdField;
+    private static FastInvokeHandler editGenelineMethod;
+
+    #endregion
+
+    #region Main patch
+
+    public VanillaRacesInsector(ModContentPack mod)
+    {
+        MpCompatPatchLoader.LoadPatch(this);
+
+        // A few methods aren't synced, as they call TryTakeOrderedJob - general MP syncing handles those
+
+        #region Genelines
+
+        {
+            var type = AccessTools.TypeByName("VanillaRacesExpandedInsector.GameComponent_Genelines");
+            // Setup field refs
+            genelinesInstanceField = AccessTools.StaticFieldRefAccess<GameComponent>(AccessTools.DeclaredField(type, "Instance"));
+            genelinesListField = AccessTools.FieldRefAccess<IList>(type, "genelines");
+            // Setup methods
+            createGenelineMethod = MethodInvoker.GetHandler(AccessTools.DeclaredMethod(type, "CreateGeneline"));
+            addGenelineMethod = MethodInvoker.GetHandler(AccessTools.DeclaredMethod(type, "AddGeneline"));
+            // Remove geneline and all pawns that have it.
+            MP.RegisterSyncMethod(type, "DeleteGeneline");
+
+            type = AccessTools.TypeByName("VanillaRacesExpandedInsector.Geneline");
+            // Setup field refs
+            genelineIdField = AccessTools.FieldRefAccess<int>(type, "id");
+            // Setup methods
+            editGenelineMethod = MethodInvoker.GetHandler(AccessTools.DeclaredMethod(type, "EditGeneline"));
+            // Change a pawn's geneline and turn them into a metapod
+            MP.RegisterSyncMethod(type, "AddPawnWithMetapod");
+
+            // Metapod gizmos
+            MP.RegisterSyncMethodLambda(AccessTools.TypeByName("VanillaRacesExpandedInsector.Metapod"), nameof(Thing.GetGizmos), 0).SetDebugOnly();
+        }
+
+        #endregion
+
+        #region Pregnancy
+
+        {
+            var types = new[]
+            {
+                "VanillaRacesExpandedInsector.HediffComp_ChestburstPregnancyVictim",
+                "VanillaRacesExpandedInsector.HediffComp_ChestburstPregnancyVictimHidden",
+            };
+
+            // Dev progress pregnancy
+            foreach (var typeName in types)
+                MP.RegisterSyncMethodLambda(AccessTools.TypeByName(typeName), nameof(HediffComp.CompGetGizmos), 0).SetDebugOnly();
+
+            // Self impregnate
+            MP.RegisterSyncMethod(AccessTools.DeclaredMethod("VanillaRacesExpandedInsector.HediffComp_Parthenogenesis:TryParthenogenesis"));
+        }
+
+        #endregion
+
+        #region RNG
+
+        {
+            // RNG calls after ShouldSpawnMotesAt
+            PatchingUtilities.PatchPushPopRand("VanillaRacesExpandedInsector.IncomingSmoker:ThrowBlackSmoke");
+        }
+
+        #endregion
+    }
+
+    #endregion
+
+    #region Sync Workers
+
+    [MpCompatSyncWorker("VanillaRacesExpandedInsector.Geneline")]
+    private static void SyncGeneline(SyncWorker sync, ref object geneline)
+    {
+        if (sync.isWriting)
+        {
+            if (geneline != null)
+                sync.Write(genelineIdField(geneline));
+            else
+                sync.Write(-1);
+        }
+        else
+        {
+            var id = sync.Read<int>();
+            if (id < 0)
+                return;
+
+            foreach (var g in genelinesListField(genelinesInstanceField()))
+            {
+                if (genelineIdField(g) == id)
+                {
+                    geneline = g;
+                    break;
+                }
+            }
+        }
+    }
+
+    #endregion
+
+    #region Genelines
+
+    [MpCompatSyncMethod]
+    private static void SyncedCreateGeneline(List<GeneDef> selectedGenes, string genelineName)
+    {
+        var genelineGameComp = genelinesInstanceField();
+        var geneline = createGenelineMethod(genelineGameComp);
+        editGenelineMethod(geneline, selectedGenes, genelineName);
+        addGenelineMethod(genelineGameComp, geneline);
+    }
+
+    [MpCompatSyncMethod]
+    private static void SyncedEditGeneline(int genelineId, List<GeneDef> selectedGenes, string genelineName)
+    {
+        foreach (var g in genelinesListField(genelinesInstanceField()))
+        {
+            if (genelineId == genelineIdField(g))
+            {
+                editGenelineMethod(g, selectedGenes, genelineName);
+                return;
+            }
+        }
+
+        Log.Warning($"Trying to modify geneline with ID {genelineId}, but it did not exist. Was it removed?");
+    }
+
+    [MpCompatPrefix("VanillaRacesExpandedInsector.Window_EditGeneline", "Accept")]
+    private static bool PreAcceptEditGeneline(Window __instance, Action ___callback, object ___geneline, IList ___selectedGenes, string ___genelineName)
+    {
+        if (!MP.IsInMultiplayer)
+            return true;
+
+        var genes = ___selectedGenes.Cast<GeneDef>().ToList();
+
+        // No callback, the geneline is being modified
+        if (___callback == null)
+            SyncedEditGeneline(genelineIdField(___geneline), genes, ___genelineName);
+        // Callback, the geneline is being created
+        else
+            SyncedCreateGeneline(genes, ___genelineName);
+
+        __instance.Close();
+        return false;
+    }
+
+    #endregion
+}
\ No newline at end of file