using System.Collections.Generic;
using LicentiaLabs;
using rjw;
using Verse;

namespace RJW_PlasticSurgeries
{
    public abstract class Recipe_Surgery_Vaginoplasty : Recipe_Plastic_Surgery
    {
        protected override BodyPartRecord GetPartCandidate(Pawn pawn)
        {
            return Genital_Helper.get_genitalsBPR(pawn);
        }

        protected override bool HasPart(Pawn pawn, List<Hediff> hediffs)
        {
            return Genital_Helper.has_vagina(pawn, hediffs);
        }

        protected override List<Hediff> GetHediffs(Pawn pawn)
        {
            return pawn.GetGenitalsList().FindAll(Genital_Helper.is_vagina);
        }
        
        protected override (HediffDef, float) GetLicentiaDamage() =>
            (Licentia.HediffDefs.StretchTear, 0.2f);
    }

    public class Recipe_Surgery_Vaginoplasty_Micro : Recipe_Surgery_Vaginoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.1f, true);
        }
    }

    public class Recipe_Surgery_Vaginoplasty_Tight : Recipe_Surgery_Vaginoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.3f, true);
        }
    }

    public class Recipe_Surgery_Vaginoplasty_Average : Recipe_Surgery_Vaginoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.5f, true);
        }
    }

    public class Recipe_Surgery_Vaginoplasty_Accomodating : Recipe_Surgery_Vaginoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.7f, true);
        }
    }

    public class Recipe_Surgery_Vaginoplasty_Cavernous : Recipe_Surgery_Vaginoplasty
    {
        protected override void SurgeryResult(Pawn pawn)
        {
            SurgeryX(pawn, 0.9f, true);
        }
    }
}