using RimWorld;
using UnityEngine;
using Verse;

namespace RJW_Menstruation
{
    public class CompProperties_InducedOvulator : CompProperties_Menstruation
    {
        public CompProperties_InducedOvulator()
        {
            compClass = typeof(HediffComp_InducedOvulator);
        }

    }

    public class HediffComp_InducedOvulator : HediffComp_Menstruation
    {
        public override string GetCurStageDesc
        {
            get
            {
                switch (CurrentVisibleStage)
                {
                    case Stage.Follicular:
                        return Translations.Stage_Follicular_Induced_Desc + (EggHealth < 1f ? Translations.Stage_Climacteric_Desc : "");
                    default:
                        return base.GetCurStageDesc;
                }
            }
        }

        // In an induced ovulator, about 65% of eggs ovulated will become pregnancies, so the expected lifetime supply
        // is the number of pregnancies they could have, with some extra to be sure.
        // An IUD or a poor fertility mate could run this down quicker. Oh well.
        protected override float RaceCyclesPerYear()
        {
            // Don't bother with breeding season, since so much time is planned to be spent pregnant.
            float pregnanciesPerYear = GenDate.DaysPerYear / Mathf.Max(1, Pawn.def.race.gestationPeriodDays);

            return 2 * pregnanciesPerYear / Configurations.ImplantationChanceDefault;
        }

        // There's really no good way to estimate the number of times it's been induced, so this is all we can do
        protected override int PawnEggsUsed(float pawnCyclesElapsed, float avglittersize)
        {
            return Mathf.CeilToInt((Pawn.relations?.ChildrenCount ?? 0) / Configurations.ImplantationChanceDefault);
        }

        protected override void GoOvulatoryStage()
        {
            estrusflag = false;
            GoNextStage(Stage.Luteal);
        }

        protected override void AfterCumIn(Pawn cummer)
        {
            base.AfterCumIn(cummer);
            switch (curStage)
            {
                case Stage.Follicular:
                    GoNextStage(Stage.Ovulatory);
                    break;
            }
        }

        public override bool IsDangerDay
        {
            get
            {
                if (Pawn.HasIUD()) return false;

                switch (curStage)
                {
                    case Stage.Follicular:
                    case Stage.Ovulatory:
                        return true;
                    case Stage.Luteal:
                        return IsEggExist && curStageHrs < EggLifespanHours;
                    default:
                        return false;
                }
            }
        }

        protected override bool ShouldBeInEstrus()
        {
            if (!loaded)
                Initialize();
            switch (curStage)
            {
                case Stage.Follicular:
                    return curStageHrs > currentIntervalHours - Props.estrusDaysBeforeOvulation * 24;
                case Stage.Ovulatory:
                    return true;
                case Stage.Luteal:
                    return IsEggExist && curStageHrs < EggLifespanHours;
                default:
                    return false;
            }
        }
    }
}