rjw-genes/Source/Genes/Life_Force/Abilities/AbilityUtility.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Verse.Sound;
using Verse;
using RimWorld;
using rjw;
using rjw.Modules.Interactions.Helpers;
using rjw.Modules.Interactions.Enums;

namespace RJW_Genes
{
    public class AbilityUtility
    {
        public static void PussyHeal(SexProps props)
        {
            if (InteractionHelper.GetWithExtension(props.dictionaryKey).DominantHasFamily(GenitalFamily.Vagina) || InteractionHelper.GetWithExtension(props.dictionaryKey).SubmissiveHasFamily(GenitalFamily.Vagina))
            {
                Pawn pawn = props.pawn;
                Pawn partner = props.partner;
                FloatRange tendQualityRange;
                tendQualityRange.min = 0.4f;
                tendQualityRange.max = 0.8f;
                if (GeneUtility.IsPussyHealer(pawn))
                {
                    Heal(partner, tendQualityRange);
                }
                if (GeneUtility.IsPussyHealer(partner))
                {
                    Heal(pawn, tendQualityRange);
                }
            }
        }

        public static bool Heal(Pawn pawn, FloatRange tendQualityRange)
        {
            bool any_wound_tended = false;
            List<Hediff> hediffs = pawn.health.hediffSet.hediffs;
            for (int i = hediffs.Count - 1; i >= 0; i--)
            {
                if ((hediffs[i] is Hediff_Injury || hediffs[i] is Hediff_MissingPart) && hediffs[i].TendableNow(false))
                {
                    hediffs[i].Tended(tendQualityRange.RandomInRange, tendQualityRange.TrueMax, 1);
                    any_wound_tended = true;
                }
            }
            return any_wound_tended;
        }

        public static float LifeForceCost(Ability ability)
        {
            if (ability.comps != null)
            {
                using (List<AbilityComp>.Enumerator enumerator = ability.comps.GetEnumerator())
                {
                    while (enumerator.MoveNext())
                    {
                        CompAbilityEffect_LifeForceCost compAbilityEffect_HemogenCost;
                        if ((compAbilityEffect_HemogenCost = (enumerator.Current as CompAbilityEffect_LifeForceCost)) != null)
                        {
                            return compAbilityEffect_HemogenCost.Props.fertilinCost;
                        }
                    }
                }
            }
            return 0f;
        }
    }
}