first commit

NitroxPatcher/Patches/Dynamic/Creature_ChooseBestAction_Patch.cs

@@ -0,0 +1,43 @@
+using System.Reflection;
+using NitroxClient.GameLogic;
+using NitroxModel.DataStructures;
+using NitroxModel.Helper;
+
+namespace NitroxPatcher.Patches.Dynamic;
+
+/// <summary>
+/// For players without lock: apply remote creature actions and prevent the original call.
+/// For players with lock: broadcast new creature actions
+/// </summary>
+public sealed partial class Creature_ChooseBestAction_Patch : NitroxPatch, IDynamicPatch
+{
+    public static readonly MethodInfo TARGET_METHOD = Reflect.Method((Creature t) => t.ChooseBestAction(default));
+
+    public static bool Prefix(Creature __instance, out NitroxId __state, ref CreatureAction __result)
+    {
+        if (!__instance.TryGetIdOrWarn(out __state) || Resolve<SimulationOwnership>().HasAnyLockType(__state))
+        {
+            return true;
+        }
+
+        // If we have received any order
+        if (Resolve<AI>().TryGetActionForCreature(__instance, out CreatureAction action))
+        {
+            __result = action;
+        }
+        return false;
+    }
+
+    public static void Postfix(Creature __instance, bool __runOriginal, NitroxId __state, ref CreatureAction __result)
+    {
+        if (!__runOriginal || __state == null)
+        {
+            return;
+        }
+
+        if (Resolve<SimulationOwnership>().HasAnyLockType(__state))
+        {
+            Resolve<AI>().BroadcastNewAction(__state, __instance, __result);
+        }
+    }
+}