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

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

namespace WulaFallenEmpire
{
    /// <summary>
    /// Custom CompProperties for our ritual spot. It no longer needs a tag.
    /// </summary>
    public class CompProperties_WulaRitualSpot : CompProperties
    {
        public CompProperties_WulaRitualSpot()
        {
            this.compClass = typeof(CompWulaRitualSpot);
        }
    }

    /// <summary>
    /// The core component for the custom ritual spot. Generates its own gizmos
    /// by specifically looking for Defs that inherit from our custom PsychicRitualDef_Wula base class.
    /// </summary>
    public class CompWulaRitualSpot : ThingComp
    {
        public CompProperties_WulaRitualSpot Props => (CompProperties_WulaRitualSpot)this.props;

        public override IEnumerable<Gizmo> CompGetGizmosExtra()
        {
            foreach (Gizmo gizmo in base.CompGetGizmosExtra())
            {
                yield return gizmo;
            }

            // Find all rituals that are of our custom base class type.
            foreach (PsychicRitualDef_Wula ritualDef in DefDatabase<PsychicRitualDef_Wula>.AllDefs)
            {
                if (ritualDef.Visible)
                {
                    Command_Action command_Action = new Command_Action();
                    command_Action.defaultLabel = ritualDef.LabelCap.Resolve();
                    command_Action.defaultDesc = ritualDef.description;
                    command_Action.icon = ritualDef.uiIcon;
                    command_Action.action = delegate
                    {
                        // Mimic vanilla initialization
                        TargetInfo target = new TargetInfo(this.parent);
                        PsychicRitualRoleAssignments assignments = ritualDef.BuildRoleAssignments(target);
                        PsychicRitualCandidatePool candidatePool = ritualDef.FindCandidatePool();
                        ritualDef.InitializeCast(this.parent.Map);
                        Find.WindowStack.Add(new Dialog_BeginPsychicRitual(ritualDef, candidatePool, assignments, this.parent.Map));
                    };

                    // Corrected check for cooldown and other requirements
                    AcceptanceReport acceptanceReport = Find.PsychicRitualManager.CanInvoke(ritualDef, this.parent.Map);
                    if (!acceptanceReport.Accepted)
                    {
                        command_Action.Disable(acceptanceReport.Reason.CapitalizeFirst());
                    }

                    yield return command_Action;
                }
            }
        }
    }
}