Liver Protection Hediff Gene

LiverProtectionHediffGene.cs

public class LiverProtectionHediffGene : Gene
{
    public static Lazy<HediffDef> LiverProtectionHediff = new(() => DefDatabase<HediffDef>.GetNamed("Your_Cool_Hediff"));

    public override void PostRemove()
    {
        if (GetHediff() is { } firstHediffOfDef)
            pawn.health.RemoveHediff(firstHediffOfDef);
        base.PostRemove();
    }

    public override void PostAdd()
    {
        base.PostAdd();
        ApplyHediff();
    }

    public override void Tick()
    {
        base.Tick();
        if (!pawn.Spawned || !pawn.IsHashIntervalTick(GenTicks.TickLongInterval) || Rand.Chance(0.75f) || GetHediff() != null) return;
        if (Rand.Chance(0.5f)) ApplyHediff();
    }

    public Hediff GetHediff() => pawn.health.hediffSet.GetFirstHediffOfDef(LiverProtectionHediff.Value);

    public void ApplyHediff()
    {
        if (GetHediff() != null || GetLiver() is not {} liver) return;
        Hediff liverHediff = HediffMaker.MakeHediff(LiverProtectionHediff.Value, pawn, liver);
        pawn.health.AddHediff(liverHediff);
    }

    public BodyPartRecord GetLiver()
    {
        foreach (BodyPartRecord notMissingPart in pawn.health.hediffSet?.GetNotMissingParts() ?? new List<BodyPartRecord>())
        {
            if (notMissingPart.def.tags.Contains(BodyPartTagDefOf.BloodFiltrationLiver))
                return notMissingPart;
        }
        return null;
    }
}