zc

Source/WulaFallenEmpire/HarmonyPatches/WULA_MechUnit/Patch_RomanceFix.cs

@@ -0,0 +1,130 @@
+// File: Patch_RomanceFix.cs
+using HarmonyLib;
+using RimWorld;
+using System.Collections.Generic;
+using Verse;
+
+namespace WulaFallenEmpire
+{
+    /// <summary>
+    /// 修复浪漫关系菜单相关的空引用异常
+    /// </summary>
+    public static class RomancePatches
+    {
+        /// <summary>
+        /// 补丁：防止对机甲单位显示浪漫菜单
+        /// </summary>
+        [HarmonyPatch(typeof(FloatMenuOptionProvider_Romance))]
+        [HarmonyPatch("GetSingleOptionFor")]
+        public static class Patch_FloatMenuOptionProvider_Romance_GetSingleOptionFor
+        {
+            [HarmonyPrefix]
+            public static bool Prefix(Pawn clickedPawn, ref FloatMenuOption __result)
+            {
+                // 如果是机甲单位，直接返回null
+                if (clickedPawn is Wulamechunit)
+                {
+                    __result = null;
+                    return false; // 跳过原始方法
+                }
+                
+                // 额外检查：如果clickedPawn没有story组件，也跳过
+                if (clickedPawn?.story == null)
+                {
+                    __result = null;
+                    return false;
+                }
+                
+                return true; // 继续执行原始方法
+            }
+        }
+        
+        /// <summary>
+        /// 补丁：防止在爱情关系检查中出现空引用
+        /// </summary>
+        [HarmonyPatch(typeof(LovePartnerRelationUtility))]
+        [HarmonyPatch("ExistingLovePartners")]
+        public static class Patch_LovePartnerRelationUtility_ExistingLovePartners
+        {
+            [HarmonyPrefix]
+            public static bool Prefix(Pawn pawn, bool allowDead, ref List<DirectPawnRelation> __result)
+            {
+                // 如果pawn是机甲单位，返回空列表
+                if (pawn is Wulamechunit)
+                {
+                    __result = new List<DirectPawnRelation>();
+                    return false; // 跳过原始方法
+                }
+                
+                // 如果pawn没有story组件，返回空列表
+                if (pawn?.story == null)
+                {
+                    __result = new List<DirectPawnRelation>();
+                    return false;
+                }
+                
+                return true; // 继续执行原始方法
+            }
+        }
+        
+        /// <summary>
+        /// 补丁：防止浪漫关系配对检查中的空引用
+        /// </summary>
+        [HarmonyPatch(typeof(RelationsUtility))]
+        [HarmonyPatch("RomanceEligiblePair")]
+        public static class Patch_RelationsUtility_RomanceEligiblePair
+        {
+            [HarmonyPrefix]
+            public static bool Prefix(Pawn initiator, Pawn target, bool forOpinionExplanation, ref AcceptanceReport __result)
+            {
+                // 如果任一pawn是机甲单位，返回拒绝
+                if (initiator is Wulamechunit || target is Wulamechunit)
+                {
+                    __result = new AcceptanceReport("DD_MechCannotRomance".Translate());
+                    return false; // 跳过原始方法
+                }
+                
+                // 如果任一pawn没有story组件，返回拒绝
+                if (initiator?.story == null || target?.story == null)
+                {
+                    __result = new AcceptanceReport("DD_NoStoryComponent".Translate());
+                    return false;
+                }
+                
+                return true; // 继续执行原始方法
+            }
+        }
+        
+        /// <summary>
+        /// 补丁：防止浪漫关系检查中的空引用
+        /// </summary>
+        [HarmonyPatch(typeof(RelationsUtility))]
+        [HarmonyPatch("RomanceOption")]
+        public static class Patch_RelationsUtility_RomanceOption
+        {
+            [HarmonyPrefix]
+            public static bool Prefix(Pawn initiator, Pawn romanceTarget, ref FloatMenuOption option, ref float chance, ref bool __result)
+            {
+                // 如果任一pawn是机甲单位，返回false
+                if (initiator is Wulamechunit || romanceTarget is Wulamechunit)
+                {
+                    __result = false;
+                    option = null;
+                    chance = 0f;
+                    return false; // 跳过原始方法
+                }
+                
+                // 如果任一pawn没有story组件，返回false
+                if (initiator?.story == null || romanceTarget?.story == null)
+                {
+                    __result = false;
+                    option = null;
+                    chance = 0f;
+                    return false;
+                }
+                
+                return true; // 继续执行原始方法
+            }
+        }
+    }
+}