暂存

2025-09-01 15:16:51 +08:00
parent 4807f40860
commit d06e39fdf5
2 changed files with 76 additions and 0 deletions
--- a/Source/WulaFallenEmpire/Harmony/MapLifecycle_Patches.cs
+++ b/Source/WulaFallenEmpire/Harmony/MapLifecycle_Patches.cs
@@ -0,0 +1,48 @@
+using System;
+using HarmonyLib;
+using RimWorld.Planet;
+using Verse;
+
+namespace WulaFallenEmpire
+{
+    [HarmonyPatch(typeof(MapParent), "ShouldRemoveMapNow")]
+    [HarmonyPriority(600)]
+    public static class MapParent_ShouldRemoveMapNow_Patch
+    {
+        public static void Postfix(ref bool __result, MapParent __instance)
+        {
+            if (!__result)
+            {
+                return;
+            }
+            try
+            {
+                if (__instance.HasMap && WulaMapProtectionHelper.ShouldProtectMap(__instance.Map))
+                {
+                    __result = false;
+                }
+            }
+            catch (Exception arg)
+            {
+                Log.Error($"[WULA] Error in MapParent_ShouldRemoveMapNow_Patch: {arg}");
+            }
+        }
+    }
+
+    [HarmonyPatch(typeof(Game), "DeinitAndRemoveMap")]
+    [HarmonyPatch(new Type[] { typeof(Map), typeof(bool) })]
+    [HarmonyPriority(600)]
+    public static class Game_DeinitAndRemoveMap_Patch
+    {
+        [HarmonyPrefix]
+        private static bool PreventMapRemoval(Map map)
+        {
+            if (WulaMapProtectionHelper.ShouldProtectMap(map))
+            {
+                Log.Message("[WULA] Map destruction prevented by WulaMapProtectionHelper at Game.DeinitAndRemoveMap level.");
+                return false; // 返回 false 来阻止原始方法的执行
+            }
+            return true; // 返回 true 来继续执行原始方法
+        }
+    }
+}