using RimWorld;
using System.Collections.Generic;
using System.Linq;
using Verse;
using Verse.AI;

namespace RJW_Menstruation
{
    public class JobDriver_VaginaWashing : JobDriver
    {
        const int excretingTime = 300;//ticks - 120 = 2 real seconds, 3 in-game minutes

        public override bool TryMakePreToilReservations(bool errorOnFailed)
        {
            return pawn.Reserve(pawn, job, 1, -1, null, errorOnFailed);
        }

        protected override IEnumerable<Toil> MakeNewToils()
        {
            List<HediffComp_Menstruation> comps = pawn.GetMenstruationComps().ToList();
            this.FailOn(delegate
            {
                return comps.All(comp => comp.TotalCumPercent < 0.001);
            });
            Toil excreting = Toils_General.Wait(excretingTime, TargetIndex.None);//duration of 

            excreting.WithProgressBarToilDelay(TargetIndex.A);
            yield return excreting;
            yield return new Toil()
            {
                initAction = delegate ()
                {
                    foreach (HediffComp_Menstruation comp in comps)
                        comp.CumOut(null, 0.5f);
                    if (comps.Any(comp => comp.TotalCumPercent > 0.001)) JumpToToil(excreting);
                }
            };
            //yield return excreting;
            yield break;
        }
    }

    public class JobDriver_MilkSelf : JobDriver
    {
        protected float progress = 0;
        protected float MilkingTime
        {
            get
            {
                return 250f * Fullness + 50f;
            }
        }
        protected virtual float Fullness
        {
            get
            {
                return comp?.Fullness ?? 0;
            }
        }

        private CompMilkable comp;

        public override bool TryMakePreToilReservations(bool errorOnFailed)
        {
            return pawn.Reserve(pawn, job, 1, -1, null, errorOnFailed);
        }

        protected virtual void PreMakeNewToils()
        {
            comp = pawn.GetComp<CompMilkable>();
        }

        protected override IEnumerable<Toil> MakeNewToils()
        {
            PreMakeNewToils();
            this.FailOnDespawnedNullOrForbidden(TargetIndex.A);
            this.FailOnNotCasualInterruptible(TargetIndex.A);
            Toil milking = new Toil
            {
                initAction = delegate ()
                {
                    pawn.pather.StopDead();
                },
                tickAction = MilkingTick
            };
            milking.AddFinishAction(Finish);
            milking.defaultCompleteMode = ToilCompleteMode.Never;
            milking.WithProgressBar(TargetIndex.A, () => progress / MilkingTime);
            yield return milking;
            yield break;
        }

        protected void MilkingTick()
        {
            progress += pawn.GetStatValue(StatDefOf.AnimalGatherSpeed);
            if (progress > MilkingTime)
            {
                Gathered();
                pawn.jobs.EndCurrentJob(JobCondition.Succeeded);
            }
            PostTickAction();
        }

        protected virtual void Gathered()
        {
            pawn.GetComp<CompMilkable>().Gathered(pawn);
        }

        protected virtual void Finish()
        {
            if (pawn.CurJobDef == JobDefOf.Wait_MaintainPosture)
            {
                pawn.jobs.EndCurrentJob(JobCondition.InterruptForced);
            }
        }

        protected virtual void PostTickAction()
        {
        }


    }


}