WulaFallenEmpireRW/Source/WulaFallenEmpire/ThingComp/WULA_PsychicRitual/PsychicRitualToil_GatherForInvocation_Wula.cs

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

namespace WulaFallenEmpire
{
    public class PsychicRitualToil_GatherForInvocation_Wula : PsychicRitualToil_Multiplex
    {
        protected PsychicRitualToil_Goto fallbackToil;
        protected PsychicRitualGraph invokerToil;
        protected PsychicRitualToil_Goto invokerFinalToil;
        private static List<Pawn> blockingPawns = new List<Pawn>(16);

        protected PsychicRitualToil_GatherForInvocation_Wula() { }

        protected PsychicRitualToil_GatherForInvocation_Wula(PsychicRitualDef_WulaBase def, PsychicRitualToil_Goto fallbackToil, PsychicRitualGraph invokerToil)
            : base(new Dictionary<PsychicRitualRoleDef, PsychicRitualToil> { { def.InvokerRole, invokerToil } }, fallbackToil)
        {
            this.fallbackToil = fallbackToil;
            this.invokerToil = invokerToil;
            invokerFinalToil = (PsychicRitualToil_Goto)invokerToil.GetToil(invokerToil.ToilCount - 1);
        }

        public PsychicRitualToil_GatherForInvocation_Wula(PsychicRitual psychicRitual, PsychicRitualDef_WulaBase def, IReadOnlyDictionary<PsychicRitualRoleDef, List<IntVec3>> rolePositions)
            : this(def, FallbackToil(psychicRitual, def, rolePositions), InvokerToil(def, rolePositions))
        {
        }

        public override void ExposeData()
        {
            base.ExposeData();
            Scribe_References.Look(ref fallbackToil, "fallbackToil");
            Scribe_References.Look(ref invokerToil, "invokerToil");
            Scribe_References.Look(ref invokerFinalToil, "invokerFinalToil");
        }

        public override string GetReport(PsychicRitual psychicRitual, PsychicRitualGraph parent)
        {
            blockingPawns.Clear();
            blockingPawns.AddRange(fallbackToil.BlockingPawns);
            if (invokerToil.CurrentToil == invokerFinalToil)
            {
                blockingPawns.AddRange(invokerFinalToil.BlockingPawns);
            }
            else
            {
                blockingPawns.AddRange(invokerFinalToil.ControlledPawns(psychicRitual));
            }
            string text = "PsychicRitualToil_GatherForInvocation_Report".Translate();
            string text2 = blockingPawns.Select((Pawn pawn) => pawn.LabelShortCap).ToCommaList();
            return text + ": " + text2;
        }

        public static PsychicRitualToil_Goto FallbackToil(PsychicRitual psychicRitual, PsychicRitualDef_WulaBase def, IReadOnlyDictionary<PsychicRitualRoleDef, List<IntVec3>> rolePositions)
        {
            return new PsychicRitualToil_Goto(rolePositions.Slice(rolePositions.Keys.Except(def.InvokerRole)));
        }

        public static PsychicRitualGraph InvokerToil(PsychicRitualDef_WulaBase def, IReadOnlyDictionary<PsychicRitualRoleDef, List<IntVec3>> rolePositions)
        {
            return new PsychicRitualGraph(InvokerGatherPhaseToils(def, rolePositions))
            {
                willAdvancePastLastToil = false
            };
        }

        public static IEnumerable<PsychicRitualToil> InvokerGatherPhaseToils(PsychicRitualDef_WulaBase def, IReadOnlyDictionary<PsychicRitualRoleDef, List<IntVec3>> rolePositions)
        {
            if (def.RequiredOffering != null)
            {
                yield return new PsychicRitualToil_GatherOfferings(def.InvokerRole, def.RequiredOffering);
            }
            if (def.TargetRole != null)
            {
                yield return new PsychicRitualToil_CarryAndGoto(def.InvokerRole, def.TargetRole, rolePositions);
            }
            yield return new PsychicRitualToil_Goto(rolePositions.Slice(def.InvokerRole));
        }
    }
}