Core/BattlePets: Added default cooldown to revive battle pets

Port From (https://github.com/TrinityCore/TrinityCore/commit/1a846415493289fc3852e56c82514d85c421267c)
2021-12-24 20:41:51 -05:00
parent 8ff2e2d75b
commit 5de09a5046
5 changed files with 67 additions and 10 deletions
@@ -1015,6 +1015,25 @@ namespace Game.Entities
            m_lastPotionId = 0;
        }

+        public void UpdateReviveBattlePetCooldown()
+        {
+            SpellInfo reviveBattlePetSpellInfo = Global.SpellMgr.GetSpellInfo(SharedConst.SpellReviveBattlePets, Difficulty.None);
+
+            if (reviveBattlePetSpellInfo != null && HasSpell(SharedConst.SpellReviveBattlePets))
+            {
+                var remainingCooldown = GetSpellHistory().GetRemainingCategoryCooldown(reviveBattlePetSpellInfo);
+                if (remainingCooldown > TimeSpan.Zero)
+                {
+                    if (remainingCooldown < SharedConst.ReviveBattlePetsCooldown)
+                        GetSpellHistory().ModifyCooldown(reviveBattlePetSpellInfo, SharedConst.ReviveBattlePetsCooldown - remainingCooldown);
+                }
+                else
+                {
+                    GetSpellHistory().StartCooldown(reviveBattlePetSpellInfo, 0, null, false, SharedConst.ReviveBattlePetsCooldown);
+                }
+            }
+        }
+
        public bool CanUseMastery()
        {
            ChrSpecializationRecord chrSpec = CliDB.ChrSpecializationStorage.LookupByKey(GetPrimarySpecialization());