refactor(CheatEnabler): split FactoryPatch into focused classes

2026-08-07 14:50:15 +08:00 · 2026-06-23 03:47:18 +08:00
parent 06015d51d0
commit eae4ba0376
10 changed files with 1820 additions and 1815 deletions
@@ -0,0 +1,58 @@
+using HarmonyLib;
+using UXAssist.Common;
+
+namespace CheatEnabler.Patches.Factory;
+
+internal class ArchitectMode : PatchImpl<ArchitectMode>
+{
+    private static bool[] _canBuildItems;
+
+    protected override void OnEnable()
+    {
+        var factory = GameMain.mainPlayer?.factory;
+        if (factory?.planet?.data != null)
+        {
+            FactoryPatch.ArrivePlanet(factory);
+        }
+    }
+
+    [HarmonyPrefix]
+    [HarmonyPatch(typeof(StorageComponent), nameof(StorageComponent.TakeTailItems), [typeof(int), typeof(int), typeof(int), typeof(bool)],
+        [ArgumentType.Ref, ArgumentType.Ref, ArgumentType.Out, ArgumentType.Normal])]
+    [HarmonyPatch(typeof(StorageComponent), nameof(StorageComponent.TakeTailItems), [typeof(int), typeof(int), typeof(int[]), typeof(int), typeof(bool)],
+        [ArgumentType.Ref, ArgumentType.Ref, ArgumentType.Normal, ArgumentType.Out, ArgumentType.Normal])]
+    public static bool TakeTailItemsPatch(StorageComponent __instance, int itemId)
+    {
+        if (__instance == null || GameMain.mainPlayer == null || __instance.id != GameMain.mainPlayer.package.id) return true;
+        if (itemId <= 0) return true;
+        if (_canBuildItems == null)
+        {
+            DoInit();
+        }
+
+        return itemId >= 12000 || !_canBuildItems[itemId];
+    }
+
+    [HarmonyPostfix]
+    [HarmonyPatch(typeof(StorageComponent), nameof(StorageComponent.GetItemCount), typeof(int))]
+    public static void GetItemCountPatch(StorageComponent __instance, int itemId, ref int __result)
+    {
+        if (__result > 99) return;
+        if (__instance == null || GameMain.mainPlayer == null || __instance.id != GameMain.mainPlayer.package.id) return;
+        if (itemId <= 0) return;
+        if (_canBuildItems == null)
+        {
+            DoInit();
+        }
+        if (itemId < 12000 && _canBuildItems[itemId]) __result = 100;
+    }
+
+    private static void DoInit()
+    {
+        _canBuildItems = new bool[12000];
+        foreach (var ip in LDB.items.dataArray)
+        {
+            if ((ip.Type == EItemType.Logistics || ip.CanBuild) && ip.ID < 12000) _canBuildItems[ip.ID] = true;
+        }
+    }
+}