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