first commit

NitroxPatcher/Patches/Dynamic/Builder_TryPlace_Patch.cs

@@ -0,0 +1,106 @@
+using System.Collections.Generic;
+using System.Reflection;
+using HarmonyLib;
+using NitroxClient.Communication.Abstract;
+using NitroxClient.GameLogic.Spawning.Bases;
+using NitroxClient.MonoBehaviours;
+using NitroxClient.MonoBehaviours.Cyclops;
+using NitroxClient.Unity.Helper;
+using NitroxModel.DataStructures;
+using NitroxModel.DataStructures.GameLogic.Entities.Bases;
+using NitroxModel.Helper;
+using NitroxModel.Packets;
+using NitroxPatcher.PatternMatching;
+using UnityEngine;
+using static System.Reflection.Emit.OpCodes;
+
+namespace NitroxPatcher.Patches.Dynamic;
+
+public sealed partial class Builder_TryPlace_Patch : NitroxPatch, IDynamicPatch
+{
+    public static readonly MethodInfo TARGET_METHOD = Reflect.Method(() => Builder.TryPlace());
+
+    public static readonly InstructionsPattern AddInstructionPattern1 = new()
+    {
+        Ldloc_0,
+        Ldc_I4_0,
+        Ldc_I4_1,
+        new() { OpCode = Callvirt, Operand = new(nameof(Constructable), nameof(Constructable.SetState)) },
+        { Pop, "Insert1" }
+    };
+
+    public static readonly List<CodeInstruction> InstructionsToAdd1 = new()
+    {
+        new(Ldloc_0),
+        new(Call, Reflect.Method(() => GhostCreated(default)))
+    };
+
+    public static readonly InstructionsPattern AddInstructionPattern2 = new()
+    {
+        Ldloc_S,
+        Ldloc_3,
+        Ldloc_S,
+        Or,
+        { new() { OpCode = Callvirt, Operand = new(nameof(Constructable), nameof(Constructable.SetIsInside)) }, "Insert2" }
+    };
+
+    public static readonly List<CodeInstruction> InstructionsToAdd2 = new()
+    {
+        TARGET_METHOD.Ldloc<Constructable>(),
+        new(Call, Reflect.Method(() => GhostCreated(default)))
+    };
+
+    public static IEnumerable<CodeInstruction> Transpiler(MethodBase original, IEnumerable<CodeInstruction> instructions) =>
+        instructions.Transform(AddInstructionPattern1, (label, instruction) =>
+        {
+            if (label.Equals("Insert1"))
+            {
+                return InstructionsToAdd1;
+            }
+            return null;
+        }).Transform(AddInstructionPattern2, (label, instruction) =>
+        {
+            if (label.Equals("Insert2"))
+            {
+                return InstructionsToAdd2;
+            }
+            return null;
+        });
+
+    public static void GhostCreated(Constructable constructable)
+    {
+        GameObject ghostObject = constructable.gameObject;
+
+        NitroxId parentId = null;
+        if (ghostObject.TryGetComponentInParent(out SubRoot subRoot, true) && (subRoot.isBase || subRoot.isCyclops) &&
+            subRoot.TryGetNitroxId(out NitroxId entityId))
+        {
+            parentId = entityId;
+        }
+
+        // Assign a NitroxId to the ghost now
+        NitroxId ghostId = new();
+        NitroxEntity.SetNewId(ghostObject, ghostId);
+        if (constructable is ConstructableBase constructableBase)
+        {
+            GhostEntity ghost = GhostEntitySpawner.From(constructableBase);
+            ghost.Id = ghostId;
+            ghost.ParentId = parentId;
+            Resolve<IPacketSender>().Send(new PlaceGhost(ghost));
+        }
+        else
+        {
+            ModuleEntitySpawner.MoveToGlobalRoot(ghostObject);
+            
+            ModuleEntity module = ModuleEntitySpawner.From(constructable);
+            module.Id = ghostId;
+            module.ParentId = parentId;
+            Resolve<IPacketSender>().Send(new PlaceModule(module));
+
+            if (constructable.transform.parent && constructable.transform.parent.TryGetComponent(out NitroxCyclops nitroxCyclops) && nitroxCyclops.Virtual)
+            {
+                nitroxCyclops.Virtual.ReplicateConstructable(constructable);
+            }
+        }
+    }
+}