first commit

NitroxPatcher/Patches/Dynamic/EnergMixin_SpawnDefaultAsync_Patch.cs

@@ -0,0 +1,33 @@
+using System.Collections.Generic;
+using System.Reflection;
+using System.Reflection.Emit;
+using HarmonyLib;
+using NitroxModel.Helper;
+
+namespace NitroxPatcher.Patches.Dynamic;
+
+/// <summary>
+/// When items are spawned in Subnautica will automatically add batteries to them async.  This is challenging to deal with because it is
+/// difficult to discriminate between these async/default additions and a user purposfully placing batteries into an object. Instead, we
+/// disable the default behavior and allow Nitrox to always spawn the batteries.  This guarentees we can capture the ids correctly.
+/// </summary>
+public sealed partial class EnergyMixin_SpawnDefaultAsync_Patch : NitroxPatch, IDynamicPatch
+{
+    public static readonly MethodInfo TARGET_METHOD = AccessTools.EnumeratorMoveNext(Reflect.Method((EnergyMixin t) => t.SpawnDefaultAsync(default(float), default(TaskResult<bool>))));
+
+    public static IEnumerable<CodeInstruction> Transpiler(MethodBase original, IEnumerable<CodeInstruction> instructions)
+    {
+
+        // Blanks out the generated MoveNext() and replaces it with:
+        //
+        //      result.set(false);
+        //      return false;
+        //
+        yield return new CodeInstruction(OpCodes.Ldarg_0);
+        yield return new CodeInstruction(OpCodes.Ldfld, TARGET_METHOD.DeclaringType.GetField("result", BindingFlags.Instance | BindingFlags.Public));
+        yield return new CodeInstruction(OpCodes.Ldc_I4_0);
+        yield return new CodeInstruction(OpCodes.Callvirt, Reflect.Method((TaskResult<bool> result) => result.Set(default)));
+        yield return new CodeInstruction(OpCodes.Ldc_I4_0);
+        yield return new CodeInstruction(OpCodes.Ret);
+    }
+}