From 204991d2665b119c4db1d184abc51ed218343b09 Mon Sep 17 00:00:00 2001
From: Hondacrx <prelude_20032002@yahoo.com>
Date: Sun, 10 Nov 2024 13:20:41 -0500
Subject: [PATCH] Scripts/Spells: Update Haunt script

Port From (https://github.com/TrinityCore/TrinityCore/commit/4e7d6d590fe7629935a22f238e120db9428045a4)
---
 Source/Scripts/Spells/Warlock.cs | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/Source/Scripts/Spells/Warlock.cs b/Source/Scripts/Spells/Warlock.cs
index 3197db593..1b41672b4 100644
--- a/Source/Scripts/Spells/Warlock.cs
+++ b/Source/Scripts/Spells/Warlock.cs
@@ -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));
         }
     }