rjw-genes/Source/Genes/ExtraGenitalia/Gene_NoVagina.cs

using Verse;
using rjw;
using RimWorld;

namespace RJW_Genes
{
    public class Gene_NoVagina : RJW_Gene
    {

        internal Hediff removed_vagina;
        
        public override void PostMake()
        {
            base.PostMake();

            // Vaginas are only removed for female pawns!
            if (GenderUtility.IsFemale(pawn) && removed_vagina == null)
            {
                RemoveButStoreVagina();
            }
        }
        
        public override void PostAdd()
        {
            base.PostAdd();

            // Vaginas are only removed for female pawns!
            if (GenderUtility.IsFemale(pawn) && removed_vagina == null)
            {
                RemoveButStoreVagina();
            }
        }

        public override void PostRemove()
        {
            base.PostRemove();
            if(removed_vagina != null)    
                pawn.health.AddHediff(removed_vagina);
        }

        internal void RemoveButStoreVagina()
        {
            var partBPR = Genital_Helper.get_genitalsBPR(pawn);
            Hediff vaginaToRemove = Genital_Helper.get_AllPartsHediffList(pawn).FindLast(x => Genital_Helper.is_vagina(x));

            if(vaginaToRemove != null)
            {
                removed_vagina = vaginaToRemove;
                pawn.health.RemoveHediff(vaginaToRemove);
            }
        }

    }
}