first commit

NitroxPatcher/Patches/Dynamic/BaseHullStrength_CrushDamageUpdate_Patch.cs

@@ -0,0 +1,59 @@
+using System.Collections.Generic;
+using System.Reflection;
+using System.Reflection.Emit;
+using HarmonyLib;
+using NitroxClient.GameLogic;
+using NitroxClient.MonoBehaviours;
+using NitroxModel.DataStructures;
+using NitroxModel.DataStructures.GameLogic.Entities.Bases;
+using NitroxModel.Helper;
+using NitroxModel_Subnautica.DataStructures;
+
+namespace NitroxPatcher.Patches.Dynamic;
+
+public sealed partial class BaseHullStrength_CrushDamageUpdate_Patch : NitroxPatch, IDynamicPatch
+{
+    internal static readonly MethodInfo TARGET_METHOD = Reflect.Method((BaseHullStrength t) => t.CrushDamageUpdate());
+
+    public static bool Prefix(BaseHullStrength __instance)
+    {
+        return __instance.TryGetNitroxId(out NitroxId baseId) && Resolve<SimulationOwnership>().HasAnyLockType(baseId);
+    }
+
+    /*
+     * }
+     * ErrorMessage.AddMessage(Language.main.GetFormat<float>("BaseHullStrDamageDetected", this.totalStrength));
+     * BroadcastChange(this, random);           <------ Inserted line
+     */
+    public static IEnumerable<CodeInstruction> Transpiler(IEnumerable<CodeInstruction> instructions)
+    {
+        // We add instructions right before the ret which is equivalent to inserting at last offset
+        return new CodeMatcher(instructions).End()
+                                            .InsertAndAdvance(new CodeInstruction(OpCodes.Ldarg_0))
+                                            .InsertAndAdvance(new CodeInstruction(OpCodes.Ldloc_0))
+                                            .Insert(new CodeInstruction(OpCodes.Call, Reflect.Method(() => BroadcastChange(default, default))))
+                                            .InstructionEnumeration();
+    }
+
+    public static void BroadcastChange(BaseHullStrength baseHullStrength, LiveMixin liveMixin)
+    {
+        if (!baseHullStrength.TryGetNitroxId(out NitroxId baseId))
+        {
+            return;
+        }
+        BaseCell baseCell = liveMixin.GetComponent<BaseCell>();
+        Int3 relativeCell = baseCell.cell - baseHullStrength.baseComp.anchor;
+        BaseLeakManager baseLeakManager = baseHullStrength.gameObject.EnsureComponent<BaseLeakManager>();
+        if (!baseLeakManager.TryGetAbsoluteCellId(baseCell.cell, out NitroxId leakId))
+        {
+            leakId = new();
+        }
+        baseLeakManager.EnsureLeak(relativeCell, leakId, liveMixin.health);
+
+        if (!Resolve<LiveMixinManager>().IsRemoteHealthChanging)
+        {
+            BaseLeakEntity baseLeakEntity = new(liveMixin.health, relativeCell.ToDto(), leakId, baseId);
+            Resolve<Entities>().BroadcastEntitySpawnedByClient(baseLeakEntity, true);
+        }
+    }
+}