Scripts/Priest: Fixed Lightwell getting removed in first damage

* Spell should drop charge, only when damage > 30% of total health.

1 files changed, 4 insertions, 5 deletions

diff --git a/src/server/scripts/Spells/spell_priest.cpp b/src/server/scripts/Spells/spell_priest.cpp
index 43865fa3f54..7566035e923 100644
--- a/src/server/scripts/Spells/spell_priest.cpp
+++ b/src/server/scripts/Spells/spell_priest.cpp
@@ -810,7 +810,7 @@ class spell_pri_lightwell_renew : public SpellScriptLoader
                 _remainingAmount = GetTarget()->CountPctFromMaxHealth(30);
             }
 
-            void HandleProc(AuraEffect const* /*aurEff*/, ProcEventInfo& eventInfo)
+            void CheckDropCharge(ProcEventInfo& eventInfo)
             {
                 DamageInfo* damageInfo = eventInfo.GetDamageInfo();
                 if (!damageInfo)
@@ -818,20 +818,19 @@ class spell_pri_lightwell_renew : public SpellScriptLoader
 
                 uint32 damage = damageInfo->GetDamage();
                 if (_remainingAmount <= damage)
-                {
-                    DropCharge(AURA_REMOVE_BY_ENEMY_SPELL);
                     return;
-                }
 
                 _remainingAmount -= damage;
+                // prevent drop charge
+                PreventDefaultAction();
             }
 
             void Register() override
             {
+                DoPrepareProc += AuraProcFn(spell_pri_lightwell_renew_AuraScript::CheckDropCharge);
                 DoEffectCalcAmount += AuraEffectCalcAmountFn(spell_pri_lightwell_renew_AuraScript::CalculateAmount, EFFECT_0, SPELL_AURA_PERIODIC_HEAL);
 
                 AfterEffectApply += AuraEffectApplyFn(spell_pri_lightwell_renew_AuraScript::InitializeAmount, EFFECT_0, SPELL_AURA_PERIODIC_HEAL, AURA_EFFECT_HANDLE_REAL_OR_REAPPLY_MASK);
-                OnEffectProc += AuraEffectProcFn(spell_pri_lightwell_renew_AuraScript::HandleProc, EFFECT_0, SPELL_AURA_PERIODIC_HEAL);
             }
 
             uint32 _remainingAmount = 0;