Fix: add checks to CCDuration adjustment (#127)

azerothcore · May 24, 2023 · b492409 · b492409
1 parent c23e5ab
commit b492409
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/src/AutoBalance.cpp b/src/AutoBalance.cpp
@@ -813,9 +813,18 @@ class AutoBalance_UnitScript : public UnitScript
     }
 
     void OnAuraApply(Unit* unit, Aura* aura) override {
-        uint32 auraDuration = _Modifier_CCDuration(unit, aura->GetCaster(), aura);
-        aura->SetMaxDuration(auraDuration);
-        aura->SetDuration(auraDuration);
+        // Only if this aura has a duration
+        if (aura->GetDuration() > 0 || aura->GetMaxDuration() > 0)
+        {
+            uint32 auraDuration = _Modifier_CCDuration(unit, aura->GetCaster(), aura);
+
+            // only update if we decided to change it
+            if (auraDuration != (float)aura->GetDuration())
+            {
+                aura->SetMaxDuration(auraDuration);
+                aura->SetDuration(auraDuration);
+            }
+        }
     }
 
     uint32 _Modifer_DealDamage(Unit* target, Unit* attacker, uint32 damage)
@@ -861,6 +870,10 @@ class AutoBalance_UnitScript : public UnitScript
                      && target->GetMap()->IsBattleground())))
             return originalDuration;
 
+        // if the CCDurationMultiplier isn't set on this enemy
+        if (!((float)caster->CustomData.GetDefault<AutoBalanceCreatureInfo>("AutoBalanceCreatureInfo")->CCDurationMultiplier > 0))
+            return originalDuration;
+
         // if the aura was cast by a pet or summon, return the original duration
         if ((caster->IsHunterPet() || caster->IsPet() || caster->IsSummon()) && caster->IsControlledByPlayer())
             return originalDuration;