crystall-punk-14/Content.Server/_CP14/MagicRitualTrigger/CP14RitualTriggerSystem.Sacrifice.cs

using Content.Shared._CP14.MagicRitual;
using Content.Shared._CP14.MagicRitualTrigger.Triggers;
using Content.Shared.Humanoid;
using Content.Shared.Mobs;
using Content.Shared.Whitelist;

namespace Content.Server._CP14.MagicRitualTrigger;


public partial class CP14RitualTriggerSystem
{
    [Dependency] private readonly EntityWhitelistSystem _whitelist = default!;

    private void InitializeSacrifice()
    {
        SubscribeLocalEvent<MobStateChangedEvent>(OnMobStateChanged);
    }

    private void OnMobStateChanged(MobStateChangedEvent ev)
    {
        if (ev.NewMobState != MobState.Dead)
            return;

        var deathXform = Transform(ev.Target);

        SacrificeSpecies(ev, deathXform);
        SacrificeWhitelist(ev, deathXform);
    }

    private void SacrificeSpecies(MobStateChangedEvent ev, TransformComponent deathXform)
    {

        if (!TryComp<HumanoidAppearanceComponent>(ev.Target, out var humanoid))
            return;

        var query = EntityQueryEnumerator<CP14RitualSacrificeSpeciesTriggerComponent, CP14MagicRitualPhaseComponent, TransformComponent>();
        while (query.MoveNext(out var uid, out var sacrifice, out var phase, out var xform))
        {
            if (!deathXform.Coordinates.TryDistance(EntityManager, xform.Coordinates, out var distance))
                continue;

            foreach (var trigger in sacrifice.Triggers)
            {
                if (distance > trigger.Range)
                    continue;

                if (trigger.Edge is null)
                    continue;

                if (trigger.Species != humanoid.Species)
                    continue;

                TriggerRitualPhase((uid, phase), trigger.Edge.Value.Target);
            }
        }
    }

    private void SacrificeWhitelist(MobStateChangedEvent ev, TransformComponent deathXform)
    {
        var query = EntityQueryEnumerator<CP14RitualSacrificeWhitelistTriggerComponent, CP14MagicRitualPhaseComponent, TransformComponent>();
        while (query.MoveNext(out var uid, out var sacrifice, out var phase, out var xform))
        {
            if (!deathXform.Coordinates.TryDistance(EntityManager, xform.Coordinates, out var distance))
                continue;

            foreach (var trigger in sacrifice.Triggers)
            {
                if (distance > trigger.Range)
                    continue;

                if (trigger.Edge is null)
                    continue;

                var entProto = MetaData(ev.Target).EntityPrototype;

                if (entProto is null)
                    continue;

                if (!_whitelist.IsValid(trigger.Whitelist, ev.Target))
                    continue;

                TriggerRitualPhase((uid, phase), trigger.Edge.Value.Target);
            }
        }
    }
}