Merge branch 'main' of https://github.com/Tourswen/WulaFallenEmpireRW

Source/WulaFallenEmpire/MechanitorPatch.cs

@@ -5,37 +5,51 @@ using RimWorld;
 using Verse;
 using HarmonyLib; // 引入Harmony库
 
-// 定义一个新的HediffComp，用于标记可以赋予机械师能力的Hediff
-public class HediffComp_MakesMechanitor : HediffComp
+namespace WulaFallenEmpire
 {
-    // 这个组件本身不需要任何特殊逻辑，它的存在就是标记
-}
-
-// Harmony Patch类，用于修改MechanitorUtility.ShouldBeMechanitor方法
-[HarmonyPatch(typeof(MechanitorUtility), "ShouldBeMechanitor")]
-public static class MechanitorShouldBeMechanitorPatch
-{
-    // Postfix方法将在原始方法执行后运行
-    // originalResult 是原始方法的返回值
-    // pawn 是原始方法的参数
-    public static void Postfix(Pawn pawn, ref bool __result)
+    // 定义一个新的HediffCompProperties，用作标记，以赋予机械师能力
+    public class HediffCompProperties_MakesMechanitor : HediffCompProperties
     {
-        // 如果原始方法已经返回true，则无需进一步检查
-        if (__result)
+        // 这个类本身不需要任何逻辑，它的存在就是为了在XML中被引用作为标记
+        public HediffCompProperties_MakesMechanitor()
         {
-            return;
+            // compClass必须指向一个有效的HediffComp类。
+            // 由于我们只想用这个Properties作为标记，我们可以指向一个通用的、空的HediffComp。
+            // 但更简洁的方法是直接在补丁里检查Properties本身。
+            // 为了避免运行时错误，我们暂时指向一个基础的HediffComp。
+            // 实际上，在下面的补丁逻辑中，我们不会实例化这个compClass。
+            this.compClass = typeof(HediffComp);
         }
+    }
 
-        // 检查Biotech DLC是否激活且Pawn属于玩家安全派系
-        if (ModsConfig.BiotechActive && pawn.Faction.IsPlayerSafe())
+
+    // Harmony Patch类，用于修改MechanitorUtility.ShouldBeMechanitor方法
+    [HarmonyPatch(typeof(MechanitorUtility), "ShouldBeMechanitor")]
+    public static class MechanitorShouldBeMechanitorPatch
+    {
+        // Postfix方法将在原始方法执行后运行
+        // originalResult 是原始方法的返回值
+        // pawn 是原始方法的参数
+        public static void Postfix(Pawn pawn, ref bool __result)
         {
-            // 遍历Pawn的所有Hediff，检查是否存在HediffComp_MakesMechanitor组件
-            foreach (Hediff hediff in pawn.health.hediffSet.hediffs)
+            // 如果原始方法已经返回true，则无需进一步检查
+            if (__result)
             {
-                if (hediff.TryGetComp<HediffComp_MakesMechanitor>() != null)
+                return;
+            }
+
+            // 检查Biotech DLC是否激活且Pawn属于玩家安全派系
+            if (ModsConfig.BiotechActive && pawn.Faction.IsPlayerSafe())
+            {
+                // 遍历Pawn的所有Hediff
+                foreach (Hediff hediff in pawn.health.hediffSet.hediffs)
                 {
-                    __result = true; // 如果找到，则将结果设置为true
-                    return;
+                    // 检查Hediff的定义中是否包含HediffCompProperties_MakesMechanitor
+                    if (hediff.def.comps?.Any(c => c is HediffCompProperties_MakesMechanitor) ?? false)
+                    {
+                        __result = true; // 如果找到，则将结果设置为true
+                        return;
+                    }
                 }
             }
         }