using Content.Server.Chat.Systems;
using Content.Shared._Goobstation.DelayedDeath;
using Content.Shared.Medical;
using Content.Shared.Mobs;
using Content.Shared.Mobs.Components;
using Content.Shared.Mobs.Systems;
using Content.Shared.Popups;

namespace Content.Server._Shitmed.DelayedDeath;

public partial class DelayedDeathSystem : EntitySystem
{
    [Dependency] private readonly ChatSystem _chat = default!; // Goobstation
    [Dependency] private readonly MobStateSystem _mobState = default!;
    [Dependency] private readonly SharedPopupSystem _popupSystem = default!; // Goobstation

    public override void Initialize()
    {
        base.Initialize();

        SubscribeLocalEvent<DelayedDeathComponent, TargetBeforeDefibrillatorZapsEvent>(OnDefibZap);
    }

    public override void Update(float frameTime)
    {
        base.Update(frameTime);

        using var query = EntityQueryEnumerator<DelayedDeathComponent, MobStateComponent>();
        while (query.MoveNext(out var ent, out var comp, out var mob))
        {
            comp.DeathTimer += frameTime;

            if (comp.DeathTimer >= comp.DeathTime && !_mobState.IsDead(ent, mob))
            {
                // go crit then dead so deathgasp can happen
                _mobState.ChangeMobState(ent, MobState.Critical, mob);
                _mobState.ChangeMobState(ent, MobState.Dead, mob);

                // goob code
                var ev = new DelayedDeathEvent(ent, PreventRevive: comp.PreventAllRevives);
                RaiseLocalEvent(ent, ref ev);

                if (ev.Cancelled)
                {
                    RemCompDeferred<DelayedDeathComponent>(ent);
                    continue;
                }

                if (!string.IsNullOrWhiteSpace(comp.DeathMessageId)) // Goobstation
                    _popupSystem.PopupEntity(Loc.GetString(comp.DeathMessageId), ent, ent, PopupType.LargeCaution);
            }
        }
    }

    private void OnDefibZap(Entity<DelayedDeathComponent> ent, ref TargetBeforeDefibrillatorZapsEvent args)
    {
        // can't defib someone without a heart or brain pal
        args.Cancel();

        var failPopup = Loc.GetString(ent.Comp.DefibFailMessageId); // Goobstation
        _chat.TrySendInGameICMessage(args.Defib, failPopup, InGameICChatType.Speak, true);
    }
}