// Copyright (c) CypherCore <http://github.com/CypherCore> All rights reserved.
// Licensed under the Gnu General Public License. See License file in the project root for full license information.

using Framework.Constants;
using Framework.Dynamic;
using Game.AI;
using Game.Entities;
using Game.Maps;
using Game.Scripting;
using Game.Spells;
using System;
using System.Collections.Generic;

namespace Scripts.Spells.Warlock;

struct SpellIds
{
    public const uint AbsoluteCorruption = 196103;
    public const uint Agony = 980;
    public const uint Backdraft = 196406;
    public const uint BackdraftProc = 117828;
    public const uint BilescourgeBombers = 267211;
    public const uint BilescourgeBombersMissile = 267212;
    public const uint BilescourgeBombersAreatrigger = 282248;
    public const uint ChannelDemonfireActivator = 228312;
    public const uint ChannelDemonfireDamage = 281362;
    public const uint ChannelDemonfireSelector = 196449;
    public const uint ConflagrateDebuff = 265931;
    public const uint ConflagrateEnergize = 245330;
    public const uint CorruptionDamage = 146739;
    public const uint CreateHealthstone = 23517;
    public const uint CurseOfExhaustion = 334275;
    public const uint DeathsEmbrace = 453189;
    public const uint DemonboltEnergize = 280127;
    public const uint DemonicCircleAllowCast = 62388;
    public const uint DemonicCircleSummon = 48018;
    public const uint DemonicCircleTeleport = 48020;
    public const uint DevourMagicHeal = 19658;
    public const uint DoomEnergize = 193318;
    public const uint DrainSoulEnergize = 205292;
    public const uint Flameshadow = 37379;
    public const uint GlyphOfDemonTraining = 56249;
    public const uint GlyphOfSoulSwap = 56226;
    public const uint GlyphOfSuccubus = 56250;
    public const uint ImmolatePeriodic = 157736;
    public const uint ImprovedHealthFunnelBuffR1 = 60955;
    public const uint ImprovedHealthFunnelBuffR2 = 60956;
    public const uint ImprovedHealthFunnelR1 = 18703;
    public const uint ImprovedHealthFunnelR2 = 18704;
    public const uint IncubusPact = 365355;
    public const uint PerpetualUnstabilityDamage = 459461;
    public const uint PerpetualUnstabilityTalent = 459376;
    public const uint PyrogenicsDebuff = 387096;
    public const uint PyrogenicsTalent = 387095;
    public const uint RainOfFire = 5740;
    public const uint RainOfFireDamage = 42223;
    public const uint RoaringBlaze = 205184;
    public const uint SeedOfCorruptionDamage = 27285;
    public const uint SeedOfCorruptionGeneric = 32865;
    public const uint ShadowburnEnergize = 245731;
    public const uint ShadowBoltEnergize = 194192;
    public const uint Shadowflame = 37378;
    public const uint SiphonLifeHeal = 453000;
    public const uint SoulFireEnergize = 281490;
    public const uint SoulSwapCdMarker = 94229;
    public const uint SoulSwapDotMarker = 92795;
    public const uint SoulSwapModCost = 92794;
    public const uint SoulSwapOverride = 86211;
    public const uint SoulshatterEffect = 32835;
    public const uint StrengthenPactIncubus = 366325;
    public const uint StrengthenPactSuccubus = 366323;
    public const uint SuccubusPact = 365360;
    public const uint SummonIncubus = 365349;
    public const uint SummonSuccubus = 712;
    public const uint UnstableAfflictionDamage = 196364;
    public const uint UnstableAfflictionEnergize = 31117;
    public const uint VileTaintDamage = 386931;
    public const uint VolatileAgonyDamage = 453035;
    public const uint VolatileAgonyTalent = 453034;
    public const uint WitherPeriodic = 445474;
    public const uint WitherTalent = 445465;

    public const uint GenReplenishment = 57669;
    public const uint PriestShadowWordDeath = 32409;

    public const uint VisualWarlockBilescourgeBombersCrash = 75806;
}

// 146739 - Corruption
[Script] // 445474 - Wither
class spell_warl_absolute_corruption : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellEffect((SpellIds.AbsoluteCorruption, 0));
    }

    public override bool Load()
    {
        return GetCaster().HasAura(SpellIds.AbsoluteCorruption);
    }

    void HandleApply(uint effIndex)
    {
        Aura absoluteCorruption = GetCaster().GetAura(SpellIds.AbsoluteCorruption);
        if (absoluteCorruption != null)
        {
            TimeSpan duration = GetHitUnit().IsPvP()
                ? TimeSpan.FromSeconds(absoluteCorruption.GetSpellInfo().GetEffect(0).CalcValue())
                : TimeSpan.FromSeconds(-1);

            GetHitAura().SetMaxDuration((int)duration.TotalMilliseconds);
            GetHitAura().SetDuration((int)duration.TotalMilliseconds);
        }
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleApply, 0, SpellEffectName.ApplyAura));
    }
}

[Script] // Called by 17962 - Conflagrate
class spell_warl_backdraft : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.Backdraft, SpellIds.BackdraftProc);
    }

    public override bool Load()
    {
        return GetCaster().HasAura(SpellIds.Backdraft);
    }

    void HandleAfterCast()
    {
        Unit caster = GetCaster();
        caster.CastSpell(caster, SpellIds.BackdraftProc, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringSpell = GetSpell()
        });
    }

    public override void Register()
    {
        AfterCast.Add(new(HandleAfterCast));
    }
}

[Script] // 710 - Banish
class spell_warl_banish : SpellScript
{
    void HandleBanish(SpellMissInfo missInfo)
    {
        if (missInfo != SpellMissInfo.Immune)
            return;

        Unit target = GetHitUnit();
        if (target != null)
        {
            // Casting Banish on a banished target will remove applied aura
            Aura banishAura = target.GetAura(GetSpellInfo().Id, GetCaster().GetGUID());
            if (banishAura != null)
                banishAura.Remove();
        }
    }

    public override void Register()
    {
        BeforeHit.Add(new(HandleBanish));
    }
}

[Script] // 267211 - Bilescourge Bombers
class spell_warl_bilescourge_bombers : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.BilescourgeBombersAreatrigger);
    }

    void HandleHit(uint effIndex)
    {
        GetCaster().CastSpell(GetCaster().GetPosition(), SpellIds.BilescourgeBombersAreatrigger, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringSpell = GetSpell()
        });
    }

    public override void Register()
    {
        OnEffectHit.Add(new(HandleHit, 0, SpellEffectName.CreateAreaTrigger));
    }
}

class BilescourgeBombersEvent(Unit caster, Position srcPos, Position destPos) : BasicEvent()
{
    public override bool Execute(ulong time, uint diff)
    {
        caster.SendPlayOrphanSpellVisual(srcPos, destPos, SpellIds.VisualWarlockBilescourgeBombersCrash, 0.5f, true);
        caster.CastSpell(destPos, SpellIds.BilescourgeBombersMissile);
        return true;
    }
}

[Script] // 15141 - Bilescourge Bombers
class at_warl_bilescourge_bombers(AreaTrigger areaTrigger) : AreaTriggerAI(areaTrigger)
{
    static byte MaxTicks = 12;

    public override void OnCreate(Spell creatingSpell)
    {
        Unit caster = at.GetCaster();
        if (caster == null)
            return;

        AreaTrigger targetAt = caster.GetAreaTrigger(SpellIds.BilescourgeBombers);
        if (targetAt == null)
            return;

        int tickRate = at.GetTotalDuration() / MaxTicks;

        for (byte i = 1; i <= 12; i++)
            caster.m_Events.AddEventAtOffset(new BilescourgeBombersEvent(caster, at.GetPosition(), targetAt.GetPosition()), TimeSpan.FromSeconds(tickRate * i));
    }
}

[Script] // 111400 - Burning Rush
class spell_warl_burning_rush : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellEffect((spellInfo.Id, 1));
    }

    SpellCastResult CheckApplyAura()
    {
        Unit caster = GetCaster();

        if (caster.GetHealthPct() <= (float)GetEffectInfo(1).CalcValue(caster))
        {
            SetCustomCastResultMessage(SpellCustomErrors.YouDontHaveEnoughHealth);
            return SpellCastResult.CustomError;
        }

        return SpellCastResult.SpellCastOk;
    }

    public override void Register()
    {
        OnCheckCast.Add(new(CheckApplyAura));
    }
}

[Script] // 111400 - Burning Rush
class spell_warl_burning_rush_aura : AuraScript
{
    void PeriodicTick(AuraEffect aurEff)
    {
        if (GetTarget().GetHealthPct() <= (float)aurEff.GetAmount())
        {
            PreventDefaultAction();
            Remove();
        }
    }

    public override void Register()
    {
        OnEffectPeriodic.Add(new(PeriodicTick, 1, AuraType.PeriodicDamagePercent));
    }
}

[Script] // 152108 - Cataclysm
class spell_warl_cataclysm : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.ImmolatePeriodic);
    }

    void HandleHit(uint effIndex)
    {
        GetCaster().CastSpell(GetHitUnit(), SpellIds.ImmolatePeriodic, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringSpell = GetSpell()
        });
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleHit, 0, SpellEffectName.SchoolDamage));
    }
}

[Script] // 228312 - Immolate (attached to 157736 - Immolate and 445474 - Wither)
class spell_warl_channel_demonfire_activator : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.ChannelDemonfireActivator);
    }

    void ApplyEffect(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        Unit caster = GetCaster();
        if (caster != null)
            caster.CastSpell(caster, SpellIds.ChannelDemonfireActivator, new CastSpellExtraArgs()
            {
                TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
                SpellValueOverrides = { new(SpellValueMod.Duration, GetDuration()) }
            });
    }

    void RemoveEffect(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        Unit caster = GetCaster();
        if (caster == null)
            return;

        UnitAuraCheck check = new(true, GetId(), caster.GetGUID());
        UnitSearcher searcher = new(caster, check);
        Cell.VisitAllObjects(caster, searcher, 100.0f);

        if (searcher.GetResult() == null)
            caster.RemoveAurasDueToSpell(SpellIds.ChannelDemonfireActivator);
    }

    public override void Register()
    {
        OnEffectApply.Add(new(ApplyEffect, 0, AuraType.PeriodicDamage, AuraEffectHandleModes.RealOrReapplyMask));
        OnEffectRemove.Add(new(RemoveEffect, 0, AuraType.PeriodicDamage, AuraEffectHandleModes.Real));
    }
}

[Script] // 196447 - Channel Demonfire
class spell_warl_channel_demonfire_periodic : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.ChannelDemonfireSelector);
    }

    void HandleEffectPeriodic(AuraEffect aurEff)
    {
        Unit caster = GetCaster();
        if (caster != null)
        {
            caster.CastSpell(caster, SpellIds.ChannelDemonfireSelector, new CastSpellExtraArgs()
            {
                TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
                TriggeringAura = aurEff
            });
        }
    }

    public override void Register()
    {
        OnEffectPeriodic.Add(new(HandleEffectPeriodic, 0, AuraType.PeriodicDummy));
    }
}

[Script] // 196449 - Channel Demonfire
class spell_warl_channel_demonfire_selector : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.ChannelDemonfireDamage, SpellIds.ImmolatePeriodic, SpellIds.WitherTalent, SpellIds.ImmolatePeriodic);
    }

    void FilterTargets(List<WorldObject> targets)
    {
        uint auraFilter = GetCaster().HasAura(SpellIds.WitherTalent)
            ? SpellIds.WitherPeriodic
            : SpellIds.ImmolatePeriodic;
        targets.RemoveAll(new UnitAuraCheck(false, auraFilter, GetCaster().GetGUID()));
    }

    void HandleDamage(uint effIndex)
    {
        GetCaster().CastSpell(GetHitUnit(), SpellIds.ChannelDemonfireDamage, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringSpell = GetSpell()
        });
    }

    public override void Register()
    {
        OnObjectAreaTargetSelect.Add(new(FilterTargets, 0, Targets.UnitDestAreaEnemy));
        OnEffectLaunchTarget.Add(new(HandleDamage, 0, SpellEffectName.Dummy));
    }
}

[Script] // 116858 - Chaos Bolt
class spell_warl_chaos_bolt : SpellScript
{
    public override bool Load()
    {
        return GetCaster().IsPlayer();
    }

    void HandleDummy(uint effIndex)
    {
        SetHitDamage(GetHitDamage() + MathFunctions.CalculatePct(GetHitDamage(), GetCaster().ToPlayer().m_activePlayerData.SpellCritPercentage));
    }

    void CalcCritChance(Unit victim, ref float critChance)
    {
        critChance = 100.0f;
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleDummy, 0, SpellEffectName.SchoolDamage));
        OnCalcCritChance.Add(new(CalcCritChance));
    }
}

[Script] // 77220 - Mastery: Chaotic Energies
class spell_warl_chaotic_energies : AuraScript
{
    void HandleAbsorb(AuraEffect aurEff, DamageInfo dmgInfo, ref uint absorbAmount)
    {
        AuraEffect aura = GetEffect(1);
        if (aura == null || !GetTargetApplication().HasEffect(1))
        {
            PreventDefaultAction();
            return;
        }

        // You take ${$s2/3}% reduced damage
        float damageReductionPct = (float)(aura.GetAmount()) / 3;
        // plus a random amount of up to ${$s2/3}% additional reduced damage
        damageReductionPct += RandomHelper.FRand(0.0f, damageReductionPct);

        absorbAmount = MathFunctions.CalculatePct(dmgInfo.GetDamage(), damageReductionPct);
    }

    public override void Register()
    {
        OnEffectAbsorb.Add(new(HandleAbsorb, 2));
    }
}

[Script] // 17962 - Conflagrate
class spell_warl_conflagrate : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.ConflagrateEnergize);
    }

    void HandleAfterCast(uint effIndex)
    {
        GetCaster().CastSpell(GetCaster(), SpellIds.ConflagrateEnergize, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringSpell = GetSpell()
        });
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleAfterCast, 0, SpellEffectName.SchoolDamage));
    }
}

[Script] // 6201 - Create Healthstone
class spell_warl_create_healthstone : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.CreateHealthstone);
    }

    public override bool Load()
    {
        return GetCaster().IsPlayer();
    }

    void HandleScriptEffect(uint effIndex)
    {
        GetCaster().CastSpell(GetCaster(), SpellIds.CreateHealthstone, true);
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleScriptEffect, 0, SpellEffectName.ScriptEffect));
    }
}

[Script] // 108416 - Dark Pact
class spell_warl_dark_pact : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellEffect((spellInfo.Id, 1), (spellInfo.Id, 2));
    }

    void CalculateAmount(AuraEffect aurEff, ref int amount, ref bool canBeRecalculated)
    {
        canBeRecalculated = false;
        Unit caster = GetCaster();
        if (caster != null)
        {
            float extraAmount = caster.SpellBaseDamageBonusDone(GetSpellInfo().GetSchoolMask()) * 2.5f;
            ulong absorb = caster.CountPctFromCurHealth(GetEffectInfo(1).CalcValue(caster));
            caster.SetHealth((ulong)(caster.GetHealth() - absorb));
            amount = (int)(MathFunctions.CalculatePct(absorb, GetEffectInfo(2).CalcValue(caster)) + extraAmount);
        }
    }

    public override void Register()
    {
        DoEffectCalcAmount.Add(new(CalculateAmount, 0, AuraType.SchoolAbsorb));
    }
}

class spell_warl_deaths_embrace_impl
{
    public static void HandleDamageOrHealingCalculation(Unit caster, Unit target, ref float pctMod, uint inreaseEffect, uint healthLimitEffect)
    {
        Aura deathsEmbrace = caster.GetAura(SpellIds.DeathsEmbrace, ObjectGuid.Empty, ObjectGuid.Empty, 1u << (int)inreaseEffect | 1u << (int)healthLimitEffect);
        if (deathsEmbrace == null)
            return;

        if (!target.HealthBelowPct(deathsEmbrace.GetEffect(healthLimitEffect).GetAmount()))
            return;

        MathFunctions.AddPct(ref pctMod, deathsEmbrace.GetEffect(inreaseEffect).GetAmount());
    }
}

[Script] // Called by 324540 - Malefic Rapture
class spell_warl_deaths_embrace : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellEffect((SpellIds.DeathsEmbrace, 3));
    }

    void HandleDamageCalculation(SpellEffectInfo spellEffectInfo, Unit victim, ref int damage, ref int flatMod, ref float pctMod)
    {
        spell_warl_deaths_embrace_impl.HandleDamageOrHealingCalculation(GetCaster(), victim, ref pctMod, 2, 3);
    }

    public override void Register()
    {
        CalcDamage.Add(new(HandleDamageCalculation));
    }
}

[Script] // Called by 980 - Agony, 146739 - Corruption and 316099 - Unstable Affliction
class spell_warl_deaths_embrace_dots : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellEffect((SpellIds.DeathsEmbrace, 3));
    }

    void CalculateDamage(AuraEffect aurEff, Unit victim, ref int damage, ref int flatMod, ref float pctMod)
    {
        Unit caster = GetCaster();
        if (caster != null)
            spell_warl_deaths_embrace_impl.HandleDamageOrHealingCalculation(caster, victim, ref pctMod, 2, 3);
    }

    public override void Register()
    {
        DoEffectCalcDamageAndHealing.Add(new(CalculateDamage, SpellConst.EffectAll, AuraType.PeriodicDamage));
    }
}

[Script] // 234153 - Drain Life
class spell_warl_deaths_embrace_drain_life : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellEffect((SpellIds.DeathsEmbrace, 1));
    }

    void CalculateHeal(AuraEffect aurEff, Unit victim, ref int damage, ref int flatMod, ref float pctMod)
    {
        Unit caster = GetCaster();
        if (caster == null)
            return;

        if (caster != victim)   // check who is being targeted, this hook is called for both damage and healing of PeriodicLeech
            return;

        spell_warl_deaths_embrace_impl.HandleDamageOrHealingCalculation(caster, caster, ref pctMod, 0, 1);
    }

    public override void Register()
    {
        DoEffectCalcDamageAndHealing.Add(new(CalculateHeal, 0, AuraType.PeriodicLeech));
    }
}

[Script] // 264178 - Demonbolt
class spell_warl_demonbolt : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.DemonboltEnergize);
    }

    void HandleAfterCast()
    {
        GetCaster().CastSpell(GetCaster(), SpellIds.DemonboltEnergize, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringSpell = GetSpell()
        });
    }

    public override void Register()
    {
        AfterCast.Add(new(HandleAfterCast));
    }
}

[Script] // 48018 - Demonic Circle: Summon
class spell_warl_demonic_circle_summon : AuraScript
{
    void HandleRemove(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        // If effect is removed by expire remove the summoned demonic circle too.
        if (!mode.HasAnyFlag(AuraEffectHandleModes.Reapply))
            GetTarget().RemoveGameObject(GetId(), true);

        GetTarget().RemoveAura(SpellIds.DemonicCircleAllowCast);
    }

    void HandleDummyTick(AuraEffect aurEff)
    {
        GameObject circle = GetTarget().GetGameObject(GetId());
        if (circle != null)
        {
            // Here we check if player is in demonic circle teleport range, if so add
            // WarlockDemonicCircleAllowCast; allowing him to cast the WarlockDemonicCircleTeleport.
            // If not in range remove the WarlockDemonicCircleAllowCast.

            SpellInfo spellInfo = Global.SpellMgr.GetSpellInfo(SpellIds.DemonicCircleTeleport, GetCastDifficulty());

            if (GetTarget().IsWithinDist(circle, spellInfo.GetMaxRange(true)))
            {
                if (!GetTarget().HasAura(SpellIds.DemonicCircleAllowCast))
                    GetTarget().CastSpell(GetTarget(), SpellIds.DemonicCircleAllowCast, true);
            }
            else
                GetTarget().RemoveAura(SpellIds.DemonicCircleAllowCast);
        }
    }

    public override void Register()
    {
        OnEffectRemove.Add(new(HandleRemove, 0, AuraType.PeriodicDummy, AuraEffectHandleModes.RealOrReapplyMask));
        OnEffectPeriodic.Add(new(HandleDummyTick, 0, AuraType.PeriodicDummy));
    }
}

[Script] // 48020 - Demonic Circle: Teleport
class spell_warl_demonic_circle_teleport : AuraScript
{
    void HandleTeleport(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        Player player = GetTarget().ToPlayer();
        if (player != null)
        {
            GameObject circle = player.GetGameObject(SpellIds.DemonicCircleSummon);
            if (circle != null)
            {
                player.NearTeleportTo(circle.GetPositionX(), circle.GetPositionY(), circle.GetPositionZ(), circle.GetOrientation());
                player.RemoveMovementImpairingAuras(false);
            }
        }
    }

    public override void Register()
    {
        OnEffectApply.Add(new(HandleTeleport, 0, AuraType.MechanicImmunity, AuraEffectHandleModes.Real));
    }
}

[Script] // 67518, 19505 - Devour Magic
class spell_warl_devour_magic : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.GlyphOfDemonTraining, SpellIds.DevourMagicHeal)
            && ValidateSpellEffect((spellInfo.Id, 1));
    }

    void OnSuccessfulDispel(uint effIndex)
    {
        Unit caster = GetCaster();
        CastSpellExtraArgs args = new();
        args.TriggerFlags = TriggerCastFlags.FullMask;
        args.AddSpellMod(SpellValueMod.BasePoint0, GetEffectInfo(1).CalcValue(caster));

        caster.CastSpell(caster, SpellIds.DevourMagicHeal, args);

        // Glyph of Felhunter
        Unit owner = caster.GetOwner();
        if (owner != null)
            if (owner.GetAura(SpellIds.GlyphOfDemonTraining) != null)
                owner.CastSpell(owner, SpellIds.DevourMagicHeal, args);
    }

    public override void Register()
    {
        OnEffectSuccessfulDispel.Add(new(OnSuccessfulDispel, 0, SpellEffectName.Dispel));
    }
}

[Script] // 603 - Doom
class spell_warl_doom : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.DoomEnergize);
    }

    void HandleEffectPeriodic(AuraEffect aurEff)
    {
        Unit caster = GetCaster();
        if (caster != null)
            caster.CastSpell(caster, SpellIds.DoomEnergize, true);
    }

    public override void Register()
    {
        OnEffectPeriodic.Add(new(HandleEffectPeriodic, 0, AuraType.PeriodicDamage));
    }
}

[Script] // 198590 - Drain Soul
class spell_warl_drain_soul : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.DrainSoulEnergize)
            && ValidateSpellEffect((spellInfo.Id, 2));
    }

    void HandleRemove(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        if (GetTargetApplication().GetRemoveMode() != AuraRemoveMode.Death)
            return;

        Unit caster = GetCaster();
        if (caster != null)
            caster.CastSpell(caster, SpellIds.DrainSoulEnergize, true);
    }

    void CalculateDamage(AuraEffect aurEff, Unit victim, ref int damage, ref int flatMod, ref float pctMod)
    {
        if (victim.HealthBelowPct(GetEffectInfo(2).CalcValue(GetCaster())))
            MathFunctions.AddPct(ref pctMod, GetEffectInfo(1).CalcValue(GetCaster()));
    }

    public override void Register()
    {
        AfterEffectRemove.Add(new(HandleRemove, 0, AuraType.PeriodicDamage, AuraEffectHandleModes.Real));
        DoEffectCalcDamageAndHealing.Add(new(CalculateDamage, 0, AuraType.PeriodicDamage));
    }
}

[Script] // 48181 - Haunt
class spell_warl_haunt : AuraScript
{
    void HandleRemove(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        if (GetTargetApplication().GetRemoveMode() == AuraRemoveMode.Death)
        {
            Unit caster = GetCaster();
            if (caster != null)
                caster.GetSpellHistory().ResetCooldown(GetId(), true);
        }
    }

    public override void Register()
    {
        OnEffectRemove.Add(new(HandleRemove, 1, AuraType.ModSchoolMaskDamageFromCaster, AuraEffectHandleModes.Real));
    }
}

[Script] // 755 - Health Funnel
class spell_warl_health_funnel : AuraScript
{
    void ApplyEffect(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        Unit caster = GetCaster();
        if (caster == null)
            return;

        Unit target = GetTarget();
        if (caster.HasAura(SpellIds.ImprovedHealthFunnelR2))
            target.CastSpell(target, SpellIds.ImprovedHealthFunnelBuffR2, true);
        else if (caster.HasAura(SpellIds.ImprovedHealthFunnelR1))
            target.CastSpell(target, SpellIds.ImprovedHealthFunnelBuffR1, true);
    }

    void RemoveEffect(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        Unit target = GetTarget();
        target.RemoveAurasDueToSpell(SpellIds.ImprovedHealthFunnelBuffR1);
        target.RemoveAurasDueToSpell(SpellIds.ImprovedHealthFunnelBuffR2);
    }

    void OnPeriodic(AuraEffect aurEff)
    {
        Unit caster = GetCaster();
        if (caster == null)
            return;
        //! Hack for self damage, is not blizz :/
        uint damage = (uint)caster.CountPctFromMaxHealth(aurEff.GetBaseAmount());

        Player modOwner = caster.GetSpellModOwner();
        if (modOwner != null)
            modOwner.ApplySpellMod(GetSpellInfo(), SpellModOp.PowerCost0, ref damage);

        SpellNonMeleeDamage damageInfo = new(caster, caster, GetSpellInfo(), GetAura().GetSpellVisual(), GetSpellInfo().SchoolMask, GetAura().GetCastId());
        damageInfo.periodicLog = true;
        damageInfo.damage = damage;
        caster.DealSpellDamage(damageInfo, false);
        caster.SendSpellNonMeleeDamageLog(damageInfo);
    }

    public override void Register()
    {
        OnEffectApply.Add(new(ApplyEffect, 0, AuraType.ObsModHealth, AuraEffectHandleModes.Real));
        OnEffectRemove.Add(new(RemoveEffect, 0, AuraType.ObsModHealth, AuraEffectHandleModes.Real));
        OnEffectPeriodic.Add(new(OnPeriodic, 0, AuraType.ObsModHealth));
    }
}

[Script] // 6262 - Healthstone
class spell_warl_healthstone_heal : SpellScript
{
    void HandleOnHit()
    {
        int heal = (int)MathFunctions.CalculatePct(GetCaster().GetCreateHealth(), GetHitHeal());
        SetHitHeal(heal);
    }

    public override void Register()
    {
        OnHit.Add(new(HandleOnHit));
    }
}

[Script] // 348 - Immolate
class spell_warl_immolate : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.ImmolatePeriodic);
    }

    void HandleOnEffectHit(uint effIndex)
    {
        GetCaster().CastSpell(GetHitUnit(), SpellIds.ImmolatePeriodic, GetSpell());
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleOnEffectHit, 0, SpellEffectName.SchoolDamage));
    }
}

// Called by 316099 - Unstable Affliction
[Script] // 459376 - Perpetual Unstability
class spell_warl_perpetual_unstability : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.PerpetualUnstabilityTalent, SpellIds.PerpetualUnstabilityDamage);
    }

    void TriggerExplosion()
    {
        Unit caster = GetCaster();
        Unit target = GetHitUnit();

        AuraEffect perpetualUnstability = caster.GetAuraEffect(SpellIds.PerpetualUnstabilityTalent, 0);
        if (perpetualUnstability != null)
        {
            Aura unstableAfflictionAura = target.GetAura(GetSpellInfo().Id, caster.GetGUID());
            if (unstableAfflictionAura != null)
            {
                TimeSpan maxUnstableAfflictionDuration = TimeSpan.FromSeconds(perpetualUnstability.GetAmount());
                if (TimeSpan.FromSeconds(unstableAfflictionAura.GetDuration()) <= maxUnstableAfflictionDuration)
                    caster.CastSpell(target, SpellIds.PerpetualUnstabilityDamage, new CastSpellExtraArgs()
                        .SetTriggerFlags(TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError)
                        .SetTriggeringSpell(GetSpell()));
            }
        }
    }

    public override void Register()
    {
        OnHit.Add(new(TriggerExplosion));
    }
}

[Script] // 387095 - Pyrogenics
class spell_warl_pyrogenics : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.PyrogenicsDebuff);
    }

    void HandleProc(AuraEffect aurEff, ProcEventInfo procInfo)
    {
        GetTarget().CastSpell(procInfo.GetActionTarget(), SpellIds.PyrogenicsDebuff, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringAura = aurEff
        });
    }

    public override void Register()
    {
        OnEffectProc.Add(new(HandleProc, 0, AuraType.AddFlatModifierBySpellLabel));
    }
}

// 5740 - Rain of Fire
[Script] /// Updated 11.0.2
class spell_warl_rain_of_fire : AuraScript
{
    void HandleDummyTick(AuraEffect aurEff)
    {
        List<AreaTrigger> rainOfFireAreaTriggers = GetTarget().GetAreaTriggers(SpellIds.RainOfFire);
        List<ObjectGuid> targetsInRainOfFire = new();

        foreach (AreaTrigger rainOfFireAreaTrigger in rainOfFireAreaTriggers)
        {
            var insideTargets = rainOfFireAreaTrigger.GetInsideUnits();
            targetsInRainOfFire.AddRange(insideTargets);
        }

        foreach (ObjectGuid insideTargetGuid in targetsInRainOfFire)
        {
            Unit insideTarget = Global.ObjAccessor.GetUnit(GetTarget(), insideTargetGuid);
            if (insideTarget != null)
                if (!GetTarget().IsFriendlyTo(insideTarget))
                    GetTarget().CastSpell(insideTarget, SpellIds.RainOfFireDamage, true);
        }
    }

    public override void Register()
    {
        OnEffectPeriodic.Add(new(HandleDummyTick, 2, AuraType.PeriodicDummy));
    }
}

[Script] // 366330 - Random Sayaad
class spell_warl_random_sayaad : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.SuccubusPact, SpellIds.IncubusPact);
    }

    void HandleDummy(uint effIndex)
    {
        Unit caster = GetCaster();

        caster.RemoveAurasDueToSpell(SpellIds.SuccubusPact);
        caster.RemoveAurasDueToSpell(SpellIds.IncubusPact);

        Player player = GetCaster().ToPlayer();
        if (player == null)
            return;

        Pet pet = player.GetPet();
        if (pet != null)
        {
            if (pet.IsPetSayaad())
                pet.DespawnOrUnsummon();
        }
    }

    public override void Register()
    {
        OnEffectHit.Add(new(HandleDummy, 0, SpellEffectName.Dummy));
    }
}

[Script] // Called by 17962 - Conflagrate
class spell_warl_roaring_blaze : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.RoaringBlaze, SpellIds.ConflagrateDebuff);
    }

    public override bool Load()
    {
        return GetCaster().HasAura(SpellIds.RoaringBlaze);
    }

    void HandleDummy(uint effIndex)
    {
        GetCaster().CastSpell(GetHitUnit(), SpellIds.ConflagrateDebuff, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringSpell = GetSpell()
        });
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleDummy, 0, SpellEffectName.SchoolDamage));
    }
}

// 366323 - Strengthen Pact - Succubus
// 366325 - Strengthen Pact - Incubus
[Script] // 366222 - Summon Sayaad
class spell_warl_sayaad_precast_disorientation : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SharedConst.SpellPetSummoningDisorientation);
    }

    // Note: this is a special case in which the warlock's minion pet must also cast Summon Disorientation at the beginning since this is only handled by SpellEffectSummonPet in Spell::CheckCast.
    public override void OnPrecast()
    {
        Player player = GetCaster().ToPlayer();
        if (player == null)
            return;

        Pet pet = player.GetPet();
        if (pet != null)
        {
            pet.CastSpell(pet, SharedConst.SpellPetSummoningDisorientation, new CastSpellExtraArgs(TriggerCastFlags.FullMask)
                .SetOriginalCaster(pet.GetGUID())
                .SetTriggeringSpell(GetSpell()));
        }
    }

    public override void Register()
    {
    }
}

[Script] // 6358 - Seduction (Special Ability)
class spell_warl_seduction : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.GlyphOfSuccubus, SpellIds.PriestShadowWordDeath);
    }

    void HandleScriptEffect(uint effIndex)
    {
        Unit caster = GetCaster();
        Unit target = GetHitUnit();
        if (target != null)
        {
            if (caster.GetOwner() != null && caster.GetOwner().HasAura(SpellIds.GlyphOfSuccubus))
            {
                target.RemoveAurasByType(AuraType.PeriodicDamage, ObjectGuid.Empty, target.GetAura(SpellIds.PriestShadowWordDeath)); // Sw:D shall not be removed.
                target.RemoveAurasByType(AuraType.PeriodicDamagePercent);
                target.RemoveAurasByType(AuraType.PeriodicLeech);
            }
        }
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleScriptEffect, 0, SpellEffectName.ApplyAura));
    }
}

[Script] // 27285 - Seed of Corruption (damage)
class spell_warl_seed_of_corruption : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.CorruptionDamage);
    }

    void HandleHit(uint effIndex)
    {
        GetCaster().CastSpell(GetHitUnit(), SpellIds.CorruptionDamage, true);
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleHit, 0, SpellEffectName.SchoolDamage));
    }
}

[Script]
class spell_warl_seed_of_corruption_dummy : SpellScript
{
    void RemoveVisualMissile(ref WorldObject target)
    {
        target = null;
    }

    void SelectTarget(List<WorldObject> targets)
    {
        if (targets.Count < 2)
            return;

        if (!GetExplTargetUnit().HasAura(GetSpellInfo().Id, GetCaster().GetGUID()))
        {
            // primary target doesn't have seed, keep it
            targets.Clear();
            targets.Add(GetExplTargetUnit());
        }
        else
        {
            // primary target has seed, select random other target with no seed
            targets.RemoveAll(new UnitAuraCheck(true, GetSpellInfo().Id, GetCaster().GetGUID()));
            if (!targets.Empty())
                targets.RandomResize(1);
            else
                targets.Add(GetExplTargetUnit());
        }
    }

    public override void Register()
    {
        OnObjectTargetSelect.Add(new(RemoveVisualMissile, 0, Targets.UnitTargetEnemy));
        OnObjectAreaTargetSelect.Add(new(SelectTarget, 1, Targets.UnitDestAreaEnemy));
        OnObjectAreaTargetSelect.Add(new(SelectTarget, 2, Targets.UnitDestAreaEnemy));
    }
}

[Script] // 27243 - Seed of Corruption
class spell_warl_seed_of_corruption_dummy_aura : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.SeedOfCorruptionDamage);
    }

    void OnPeriodic(AuraEffect aurEff)
    {
        Unit caster = GetCaster();
        if (caster != null)
            caster.CastSpell(GetTarget(), SpellIds.SeedOfCorruptionDamage, aurEff);
    }

    void CalculateBuffer(AuraEffect aurEff, ref int amount, ref bool canBeRecalculated)
    {
        Unit caster = GetCaster();
        if (caster == null)
            return;

        amount = caster.SpellBaseDamageBonusDone(GetSpellInfo().GetSchoolMask()) * GetEffectInfo(0).CalcValue(caster) / 100;
    }

    void HandleProc(AuraEffect aurEff, ProcEventInfo eventInfo)
    {
        PreventDefaultAction();

        DamageInfo damageInfo = eventInfo.GetDamageInfo();
        if (damageInfo == null)
            return;

        Unit caster = GetCaster();
        if (caster == null)
            return;

        if (damageInfo.GetAttacker() == null || damageInfo.GetAttacker() != caster)
            return;

        // other seed explosions detonate this instantly, no matter what damage amount is
        if (damageInfo.GetSpellInfo() == null || damageInfo.GetSpellInfo().Id != SpellIds.SeedOfCorruptionDamage)
        {
            int amount = (int)(aurEff.GetAmount() - damageInfo.GetDamage());
            if (amount > 0)
            {
                aurEff.SetAmount(amount);
                if (!GetTarget().HealthBelowPctDamaged(1, damageInfo.GetDamage()))
                    return;
            }
        }

        Remove();

        caster.CastSpell(eventInfo.GetActionTarget(), SpellIds.SeedOfCorruptionDamage, aurEff);
    }

    public override void Register()
    {
        OnEffectPeriodic.Add(new(OnPeriodic, 1, AuraType.PeriodicDamage));
        DoEffectCalcAmount.Add(new(CalculateBuffer, 2, AuraType.Dummy));
        OnEffectProc.Add(new(HandleProc, 2, AuraType.Dummy));
    }
}

// 32863 - Seed of Corruption
// 36123 - Seed of Corruption
// 38252 - Seed of Corruption
// 39367 - Seed of Corruption
// 44141 - Seed of Corruption
// 70388 - Seed of Corruption
[Script] // Monster spells, triggered only on amount drop (not on death)
class spell_warl_seed_of_corruption_generic : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.SeedOfCorruptionGeneric);
    }

    void HandleProc(AuraEffect aurEff, ProcEventInfo eventInfo)
    {
        PreventDefaultAction();
        DamageInfo damageInfo = eventInfo.GetDamageInfo();
        if (damageInfo == null || damageInfo.GetDamage() == 0)
            return;

        int amount = (int)(aurEff.GetAmount() - damageInfo.GetDamage());
        if (amount > 0)
        {
            aurEff.SetAmount(amount);
            return;
        }

        Remove();

        Unit caster = GetCaster();
        if (caster == null)
            return;

        caster.CastSpell(eventInfo.GetActionTarget(), SpellIds.SeedOfCorruptionGeneric, aurEff);
    }

    public override void Register()
    {
        OnEffectProc.Add(new(HandleProc, 1, AuraType.Dummy));
    }
}

[Script] // 17877 - Shadowburn
class spell_warl_shadowburn : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.ShadowburnEnergize)
            && ValidateSpellEffect((spellInfo.Id, 3));
    }

    void HandleEnergize()
    {
        if (GetHitUnit().IsAlive())
            return;

        // killing target with current spell doesn't apply the aura (apply/remove scripts don't execute)
        // but we can use the fact that it still gets created and immediately marked as removed to detect that case
        Aura hitAura = GetHitAura(false, true);
        if (hitAura == null || !hitAura.IsRemoved())
            return;

        TryEnergize(GetCaster()?.ToPlayer(), GetHitUnit(), GetSpellInfo(), GetSpell(), null);
    }

    void CalcCritChance(Unit victim, ref float critChance)
    {
        if (victim.HealthBelowPct(GetEffectInfo(3).CalcValue(GetCaster())))
            critChance += GetEffectInfo(2).CalcValue(GetCaster());
    }

    public override void Register()
    {
        AfterHit.Add(new(HandleEnergize));
        OnCalcCritChance.Add(new(CalcCritChance));
    }

    public static void TryEnergize(Player caster, Unit target, SpellInfo spellInfo, Spell triggeringSpell, AuraEffect triggeringAura)
    {
        if (caster == null)
            return;

        if (caster.IsHonorOrXPTarget(target))
        {
            caster.CastSpell(caster, SpellIds.ShadowburnEnergize, new CastSpellExtraArgs()
            {
                TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
                TriggeringSpell = triggeringSpell,
                TriggeringAura = triggeringAura
            });

            caster.GetSpellHistory().RestoreCharge(spellInfo.ChargeCategoryId);
        }
    }
}

[Script]
class spell_warl_shadowburn_aura : AuraScript
{
    void RemoveEffect(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        if (GetTargetApplication().GetRemoveMode() != AuraRemoveMode.Death)
            return;

        spell_warl_shadowburn.TryEnergize(GetCaster()?.ToPlayer(), GetTarget(), GetSpellInfo(), null, aurEff);
    }

    public override void Register()
    {
        AfterEffectRemove.Add(new(RemoveEffect, 1, AuraType.Dummy, AuraEffectHandleModes.Real));
    }
}

[Script] // 686 - Shadow Bolt
class spell_warl_shadow_bolt : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.ShadowBoltEnergize);
    }

    void HandleAfterCast()
    {
        GetCaster().CastSpell(GetCaster(), SpellIds.ShadowBoltEnergize, true);
    }

    public override void Register()
    {
        AfterCast.Add(new(HandleAfterCast));
    }
}

[Script] // 422054 - Shadow Invocation
class spell_warl_shadow_invocation : AuraScript
{
    void HandleProc(ProcEventInfo eventInfo)
    {
        Unit caster = eventInfo.GetActor();
        Unit target = eventInfo.GetActionTarget();
        caster.m_Events.AddEventAtOffset(new BilescourgeBombersEvent(caster, caster.GetPosition(), target.GetPosition()), TimeSpan.FromSeconds(500));
    }

    public override void Register()
    {
        OnProc.Add(new(HandleProc));
    }
}

[Script] // 452999 - Siphon Life
class spell_warl_siphon_life : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.SiphonLifeHeal);
    }

    void HandleProc(AuraEffect aurEff, ProcEventInfo eventInfo)
    {
        DamageInfo damageInfo = eventInfo.GetDamageInfo();
        if (damageInfo == null || damageInfo.GetDamage() == 0)
            return;

        Unit caster = GetTarget();
        caster.CastSpell(caster, SpellIds.SiphonLifeHeal, new CastSpellExtraArgs(aurEff)
            .AddSpellMod(SpellValueMod.BasePoint0, (int)MathFunctions.CalculatePct(damageInfo.GetDamage(), aurEff.GetAmount())));
    }

    public override void Register()
    {
        OnEffectProc.Add(new(HandleProc, 1, AuraType.Dummy));
    }
}

[Script] // 6353 - Soul Fire
class spell_warl_soul_fire : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.SoulFireEnergize, SpellIds.WitherTalent, SpellIds.WitherPeriodic, SpellIds.ImmolatePeriodic);
    }

    void HandleTriggers(uint effIndex)
    {
        Unit caster = GetCaster();

        caster.CastSpell(caster, SpellIds.SoulFireEnergize, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringSpell = GetSpell()
        });

        uint periodicDamage = GetCaster().HasAura(SpellIds.WitherTalent)
            ? SpellIds.WitherPeriodic
            : SpellIds.ImmolatePeriodic;
        caster.CastSpell(GetHitUnit(), periodicDamage, new CastSpellExtraArgs()
        {
            TriggerFlags = TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError,
            TriggeringSpell = GetSpell()
        });
    }

    public override void Register()
    {
        OnEffectLaunchTarget.Add(new(HandleTriggers, 0, SpellEffectName.SchoolDamage));
    }
}

[Script] // 86121 - Soul Swap
class spell_warl_soul_swap : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.GlyphOfSoulSwap, SpellIds.SoulSwapCdMarker, SpellIds.SoulSwapOverride);
    }

    void HandleHit(uint effIndex)
    {
        GetCaster().CastSpell(GetCaster(), SpellIds.SoulSwapOverride, true);
        GetHitUnit().CastSpell(GetCaster(), SpellIds.SoulSwapDotMarker, true);
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleHit, 0, SpellEffectName.SchoolDamage));
    }
}

[Script] // 86211 - Soul Swap Override - Also acts as a dot container
class spell_warl_soul_swap_override : AuraScript
{
    List<uint> _dotList = new();
    Unit _swapCaster;

    //! Forced to, pure virtual functions must have a body when linking
    public override void Register() { }

    public void AddDot(uint id) { _dotList.Add(id); }
    public List<uint> GetDotList() { return _dotList; }
    public Unit GetOriginalSwapSource() { return _swapCaster; }
    public void SetOriginalSwapSource(Unit victim) { _swapCaster = victim; }
}

[Script] //! Soul Swap Copy Spells - 92795 - Simply copies spell IDs.
class spell_warl_soul_swap_dot_marker : SpellScript
{
    void HandleHit(uint effIndex)
    {
        Unit swapVictim = GetCaster();
        Unit warlock = GetHitUnit();
        if (warlock == null || swapVictim == null)
            return;

        var appliedAuras = swapVictim.GetAppliedAuras();
        spell_warl_soul_swap_override swapSpellScript = null;
        Aura swapOverrideAura = warlock.GetAura(SpellIds.SoulSwapOverride);
        if (swapOverrideAura != null)
            swapSpellScript = swapOverrideAura.GetScript<spell_warl_soul_swap_override>();

        if (swapSpellScript == null)
            return;

        FlagArray128 classMask = GetEffectInfo().SpellClassMask;

        foreach (var (key, app) in appliedAuras)
        {
            SpellInfo spellProto = app.GetBase().GetSpellInfo();
            if (app.GetBase().GetCaster() == warlock)
                if (spellProto.SpellFamilyName == SpellFamilyNames.Warlock && (spellProto.SpellFamilyFlags & classMask))
                    swapSpellScript.AddDot(key);
        }

        swapSpellScript.SetOriginalSwapSource(swapVictim);
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleHit, 0, SpellEffectName.Dummy));
    }
}

[Script] // 86213 - Soul Swap Exhale
class spell_warl_soul_swap_exhale : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.SoulSwapModCost, SpellIds.SoulSwapOverride);
    }

    SpellCastResult CheckCast()
    {
        Unit currentTarget = GetExplTargetUnit();
        Unit swapTarget = null;
        Aura swapOverride = GetCaster().GetAura(SpellIds.SoulSwapOverride);
        if (swapOverride != null)
        {
            spell_warl_soul_swap_override swapScript = swapOverride.GetScript<spell_warl_soul_swap_override>();
            if (swapScript != null)
                swapTarget = swapScript.GetOriginalSwapSource();
        }

        // Soul Swap Exhale can't be cast on the same target than Soul Swap
        if (swapTarget != null && currentTarget != null && swapTarget == currentTarget)
            return SpellCastResult.BadTargets;

        return SpellCastResult.SpellCastOk;
    }

    void OnEffectHit(uint effIndex)
    {
        GetCaster().CastSpell(GetCaster(), SpellIds.SoulSwapModCost, true);
        bool hasGlyph = GetCaster().HasAura(SpellIds.GlyphOfSoulSwap);

        List<uint> dotList = new();
        Unit swapSource = null;
        Aura swapOverride = GetCaster().GetAura(SpellIds.SoulSwapOverride);
        if (swapOverride != null)
        {
            spell_warl_soul_swap_override swapScript = swapOverride.GetScript<spell_warl_soul_swap_override>();
            if (swapScript == null)
                return;
            dotList = swapScript.GetDotList();
            swapSource = swapScript.GetOriginalSwapSource();
        }

        if (dotList.Empty())
            return;

        foreach (var spellId in dotList)
        {
            GetCaster().AddAura(spellId, GetHitUnit());
            if (!hasGlyph && swapSource != null)
                swapSource.RemoveAurasDueToSpell(spellId);
        }

        // Remove Soul Swap Exhale buff
        GetCaster().RemoveAurasDueToSpell(SpellIds.SoulSwapOverride);

        if (hasGlyph) // Add a cooldown on Soul Swap if caster has the glyph
            GetCaster().CastSpell(GetCaster(), SpellIds.SoulSwapCdMarker, false);
    }

    public override void Register()
    {
        OnCheckCast.Add(new(CheckCast));
        OnEffectHitTarget.Add(new(OnEffectHit, 0, SpellEffectName.SchoolDamage));
    }
}

[Script] // 29858 - Soulshatter
class spell_warl_soulshatter : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.SoulshatterEffect);
    }

    void HandleDummy(uint effIndex)
    {
        Unit caster = GetCaster();
        Unit target = GetHitUnit();
        if (target != null)
            if (target.GetThreatManager().IsThreatenedBy(caster, true))
                caster.CastSpell(target, SpellIds.SoulshatterEffect, true);
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleDummy, 0, SpellEffectName.Dummy));
    }
}

[Script] // 366323 - Strengthen Pact - Succubus
class spell_warl_strengthen_pact_succubus : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.SuccubusPact, SpellIds.SummonSuccubus);
    }

    void HandleDummy(uint effIndex)
    {
        Unit caster = GetCaster();

        caster.CastSpell(null, SpellIds.SuccubusPact, TriggerCastFlags.FullMask);
        caster.CastSpell(null, SpellIds.SummonSuccubus, TriggerCastFlags.FullMask);
    }

    public override void Register()
    {
        OnEffectHit.Add(new(HandleDummy, 0, SpellEffectName.Dummy));
    }
}

[Script] // 366325 - Strengthen Pact - Incubus
class spell_warl_strengthen_pact_incubus : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.IncubusPact, SpellIds.SummonIncubus);
    }

    void HandleDummy(uint effIndex)
    {
        Unit caster = GetCaster();

        caster.CastSpell(null, SpellIds.IncubusPact, TriggerCastFlags.FullMask);
        caster.CastSpell(null, SpellIds.SummonIncubus, TriggerCastFlags.FullMask);
    }

    public override void Register()
    {
        OnEffectHit.Add(new(HandleDummy, 0, SpellEffectName.Dummy));
    }
}

[Script] // 366222 - Summon Sayaad
class spell_warl_summon_sayaad : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.SummonSuccubus, SpellIds.SummonIncubus);
    }

    void HandleDummy(uint effIndex)
    {
        GetCaster().CastSpell(null, RandomHelper.randChance(50) ? SpellIds.SummonSuccubus : SpellIds.SummonIncubus, TriggerCastFlags.FullMask);
    }

    public override void Register()
    {
        OnEffectHit.Add(new(HandleDummy, 0, SpellEffectName.Dummy));
    }
}

// 37377 - Shadowflame
// 39437 - Shadowflame Hellfire and RoF
[Script("spell_warl_t4_2p_bonus_shadow", SpellIds.Flameshadow)]
[Script("spell_warl_t4_2p_bonus_fire", SpellIds.Shadowflame)]
class spell_warl_t4_2p_bonus(uint trigger) : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(trigger);
    }

    void HandleProc(AuraEffect aurEff, ProcEventInfo eventInfo)
    {
        PreventDefaultAction();
        Unit caster = eventInfo.GetActor();
        caster.CastSpell(caster, trigger, aurEff);
    }

    public override void Register()
    {
        OnEffectProc.Add(new(HandleProc, 0, AuraType.Dummy));
    }
}

[Script] // 316099 - Unstable Affliction
class spell_warl_unstable_affliction : AuraScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.UnstableAfflictionDamage, SpellIds.UnstableAfflictionEnergize);
    }

    void HandleDispel(DispelInfo dispelInfo)
    {
        Unit caster = GetCaster();
        if (caster == null)
            return;

        AuraEffect removedEffect = GetEffect(1);
        if (removedEffect == null)
            return;

        int damage = (int)(GetEffectInfo(0).CalcValue(caster, null, GetUnitOwner()) / 100.0f * removedEffect.CalculateEstimatedAmount(caster, removedEffect.GetAmount()).Value);
        caster.CastSpell(dispelInfo.GetDispeller(), SpellIds.UnstableAfflictionDamage, new CastSpellExtraArgs()
            .AddSpellMod(SpellValueMod.BasePoint0, damage)
            .SetTriggerFlags(TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError));
    }

    void HandleRemove(AuraEffect aurEff, AuraEffectHandleModes mode)
    {
        if (GetTargetApplication().GetRemoveMode() != AuraRemoveMode.Death)
            return;

        GetCaster().CastSpell(GetCaster(), SpellIds.UnstableAfflictionEnergize, true);
    }

    public override void Register()
    {
        AfterDispel.Add(new(HandleDispel));
        OnEffectRemove.Add(new(HandleRemove, 1, AuraType.PeriodicDamage, AuraEffectHandleModes.Real));
    }
}

[Script] // 278350 - Vile Taint
class spell_warl_vile_taint : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.Agony, SpellIds.CurseOfExhaustion, SpellIds.VileTaintDamage);
    }

    void HandleScriptEffect(uint effIndex)
    {
        Unit caster = GetCaster();
        CastSpellTargetArg target = GetHitUnit();

        CastSpellExtraArgs args = new();
        args.SetTriggerFlags(TriggerCastFlags.IgnoreGCD | TriggerCastFlags.IgnorePowerCost
            | TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError);
        args.SetTriggeringSpell(GetSpell());

        caster.CastSpell(target, SpellIds.Agony, args);
        caster.CastSpell(target, SpellIds.CurseOfExhaustion, args);
        caster.CastSpell(target, SpellIds.VileTaintDamage, args);
    }

    public override void Register()
    {
        OnEffectHitTarget.Add(new(HandleScriptEffect, 0, SpellEffectName.Dummy));
    }
}

// Called by 980 - Agony
[Script] // 453034 - Volatile Agony
class spell_warl_volatile_agony : SpellScript
{
    public override bool Validate(SpellInfo spellInfo)
    {
        return ValidateSpellInfo(SpellIds.VolatileAgonyTalent, SpellIds.VolatileAgonyDamage);
    }

    void TriggerExplosion()
    {
        Unit caster = GetCaster();
        Unit target = GetHitUnit();

        AuraEffect volatileAgony = caster.GetAuraEffect(SpellIds.VolatileAgonyTalent, 0);
        if (volatileAgony != null)
        {
            Aura agonyAura = target.GetAura(GetSpellInfo().Id, caster.GetGUID());
            if (agonyAura != null)
            {
                TimeSpan maxAgonyDuration = TimeSpan.FromSeconds(volatileAgony.GetAmount());
                if (TimeSpan.FromSeconds(agonyAura.GetDuration()) <= maxAgonyDuration)
                    caster.CastSpell(target, SpellIds.VolatileAgonyDamage, new CastSpellExtraArgs()
                        .SetTriggerFlags(TriggerCastFlags.IgnoreCastInProgress | TriggerCastFlags.DontReportCastError)
                        .SetTriggeringSpell(GetSpell()));
            }
        }
    }

    public override void Register()
    {
        OnHit.Add(new(TriggerExplosion));
    }
}