Scripts/Spells: Update Haunt script

Port From (https://github.com/TrinityCore/TrinityCore/commit/4e7d6d590fe7629935a22f238e120db9428045a4)
2024-11-10 13:20:41 -05:00
parent 46643bd178
commit 204991d266
1 changed files with 7 additions and 8 deletions
@@ -400,22 +400,21 @@ namespace Scripts.Spells.Warlock
    }

    [Script] // 48181 - Haunt
-    class spell_warl_haunt : SpellScript
+    class spell_warl_haunt : AuraScript
    {
-        void HandleAfterHit()
+        void HandleRemove(AuraEffect aurEff, AuraEffectHandleModes mode)
        {
-            Aura aura = GetHitAura();
-            if (aura != null)
+            if (GetTargetApplication().GetRemoveMode() == AuraRemoveMode.Death)
            {
-                AuraEffect aurEff = aura.GetEffect(1);
-                if (aurEff != null)
-                    aurEff.SetAmount(MathFunctions.CalculatePct(GetHitDamage(), aurEff.GetAmount()));
+                Unit caster = GetCaster();
+                if (caster != null)
+                    caster.GetSpellHistory().ResetCooldown(GetId(), true);
            }
        }

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