Make bots break CC effects on themselves.

Author: ratkosrb

src/game/PlayerBots/BattleBotAI.cpp

@@ -813,7 +813,10 @@ void BattleBotAI::UpdateAI(uint32 const diff)
     }
 
     if (me->HasUnitState(UNIT_STATE_CAN_NOT_REACT_OR_LOST_CONTROL))
+    {
+        BreakCrowdControlEffects();
         return;
+    }
 
     if (me->GetCurrentSpell(CURRENT_AUTOREPEAT_SPELL))
     {

src/game/PlayerBots/CombatBotBaseAI.cpp

@@ -3012,17 +3012,19 @@ SpellCastResult CombatBotBaseAI::CastWeaponBuff(SpellEntry const* pSpellEntry, E
     return spell->prepare(std::move(targets), nullptr);
 }
 
-void CombatBotBaseAI::UseTrinketEffects()
+bool CombatBotBaseAI::UseTrinketEffects(bool onlyToBreakCC)
 {
     if (Item* pItem = me->GetItemByPos(INVENTORY_SLOT_BAG_0, EQUIPMENT_SLOT_TRINKET1))
-        if (UseItemEffect(pItem))
-            return;
+        if (UseItemEffect(pItem, onlyToBreakCC))
+            return true;
     if (Item* pItem = me->GetItemByPos(INVENTORY_SLOT_BAG_0, EQUIPMENT_SLOT_TRINKET2))
-        if (UseItemEffect(pItem))
-            return;
+        if (UseItemEffect(pItem, onlyToBreakCC))
+            return true;
+
+    return false;
 }
 
-bool CombatBotBaseAI::UseItemEffect(Item* pItem)
+bool CombatBotBaseAI::UseItemEffect(Item* pItem, bool onlyToBreakCC)
 {
     ItemPrototype const* pProto = pItem->GetProto();
     for (auto const& itr : pProto->Spells)
@@ -3033,6 +3035,9 @@ bool CombatBotBaseAI::UseItemEffect(Item* pItem)
             {
                 if (me->IsSpellReady(*pSpellEntry, pProto))
                 {
+                    if (onlyToBreakCC && !pSpellEntry->HasAttribute(SPELL_ATTR_EX_IMMUNITY_PURGES_EFFECT))
+                        continue;
+
                     if (pSpellEntry->IsPositiveSpell())
                         return me->CastSpell(me, pSpellEntry, false, pItem) == SPELL_CAST_OK;
                     else if (me->GetVictim())
@@ -3044,6 +3049,109 @@ bool CombatBotBaseAI::UseItemEffect(Item* pItem)
     return false;
 }
 
+void CombatBotBaseAI::BreakCrowdControlEffects()
+{
+    if (UseTrinketEffects(true))
+        return;
+
+    switch (me->GetClass())
+    {
+        case CLASS_PALADIN:
+        {
+            if (m_spells.paladin.pDivineShield &&
+                CanTryToCastSpell(me, m_spells.paladin.pDivineShield))
+            {
+                if (DoCastSpell(me, m_spells.paladin.pDivineShield) == SPELL_CAST_OK)
+                {
+                    if (m_role != ROLE_HEALER)
+                    {
+                        me->m_Events.AddLambdaEventAtOffset([player = me, spellId = m_spells.paladin.pDivineShield->Id]()
+                        {
+                            if (player->GetHealthPercent() > 75.0f && player->GetAttackers().size() < 3)
+                                player->RemoveAurasDueToSpellByCancel(spellId);
+                        }, 1 * IN_MILLISECONDS);
+                    }
+                    return;
+                }
+                   
+            }
+            break;
+        }
+        case CLASS_MAGE:
+        {
+            if (me->HasUnitState(UNIT_STATE_STUNNED) && m_spells.mage.pBlink &&
+                CanTryToCastSpell(me, m_spells.mage.pBlink))
+            {
+                if (DoCastSpell(me, m_spells.mage.pBlink) == SPELL_CAST_OK)
+                    return;
+            }
+            if (m_spells.mage.pIceBlock &&
+                CanTryToCastSpell(me, m_spells.mage.pIceBlock))
+            {
+                if (DoCastSpell(me, m_spells.mage.pIceBlock) == SPELL_CAST_OK)
+                {
+                    me->m_Events.AddLambdaEventAtOffset([player = me, spellId = m_spells.mage.pIceBlock->Id]()
+                    {
+                        if (player->GetHealthPercent() > 75.0f && player->GetAttackers().size() < 3)
+                            player->RemoveAurasDueToSpellByCancel(spellId);
+                    }, 1 * IN_MILLISECONDS);
+                    return;
+                }
+            }
+            break;
+        }
+        case CLASS_DRUID:
+        {
+            bool polymorphed = false;
+            auto const& auraList = me->GetAurasByType(SPELL_AURA_MOD_CONFUSE);
+            for (auto const& pAura : auraList)
+            {
+                if (pAura->GetSpellProto()->Mechanic == MECHANIC_POLYMORPH)
+                {
+                    polymorphed = true;
+                    break;
+                }
+            }
+
+            if (polymorphed)
+            {
+                SpellEntry const* pShapeshift = nullptr;
+
+                if (m_role == ROLE_TANK && m_spells.druid.pBearForm && CanTryToCastSpell(me, m_spells.druid.pBearForm))
+                    pShapeshift = m_spells.druid.pBearForm;
+                else if (m_role == ROLE_MELEE_DPS && m_spells.druid.pCatForm && CanTryToCastSpell(me, m_spells.druid.pCatForm))
+                    pShapeshift = m_spells.druid.pCatForm;
+                else if (m_role == ROLE_RANGE_DPS && m_spells.druid.pMoonkinForm && CanTryToCastSpell(me, m_spells.druid.pMoonkinForm))
+                    pShapeshift = m_spells.druid.pMoonkinForm;
+                else
+                {
+                    for (auto const& pSpell : m_spells.raw.spells)
+                    {
+                        if (pSpell && pSpell->HasAura(SPELL_AURA_MOD_SHAPESHIFT) && CanTryToCastSpell(me, pSpell))
+                        {
+                            pShapeshift = pSpell;
+                            break;
+                        }
+                    }
+                }
+
+                if (pShapeshift && DoCastSpell(me, pShapeshift) == SPELL_CAST_OK)
+                {
+                    if (m_role == ROLE_HEALER)
+                    {
+                        me->m_Events.AddLambdaEventAtOffset([player = me, spellId = pShapeshift->Id]()
+                        {
+                            player->RemoveAurasDueToSpellByCancel(spellId);
+                        }, 1 * IN_MILLISECONDS);
+                    }
+                    return;
+                }
+            }
+            break;
+        }
+    }
+}
+
 bool CombatBotBaseAI::IsWearingShield(Player* pPlayer) const
 {
     Item* pItem = pPlayer->GetItemByPos(INVENTORY_SLOT_BAG_0, EQUIPMENT_SLOT_OFFHAND);

src/game/PlayerBots/CombatBotBaseAI.h

@@ -135,8 +135,9 @@ class CombatBotBaseAI : public PlayerBotAI
 
     bool SummonShamanTotems();
     SpellCastResult CastWeaponBuff(SpellEntry const* pSpellEntry, EquipmentSlots slot);
-    void UseTrinketEffects();
-    bool UseItemEffect(Item* pItem);
+    bool UseTrinketEffects(bool onlyToBreakCC = false);
+    bool UseItemEffect(Item* pItem, bool onlyToBreakCC = false);
+    void BreakCrowdControlEffects();
 
     virtual void UpdateInCombatAI() = 0;
     virtual void UpdateOutOfCombatAI() = 0;

src/game/PlayerBots/PartyBotAI.cpp

@@ -699,7 +699,10 @@ void PartyBotAI::UpdateAI(uint32 const diff)
         me->RemoveSpellsCausingAura(SPELL_AURA_FEIGN_DEATH);
 
     if (me->HasUnitState(UNIT_STATE_CAN_NOT_REACT_OR_LOST_CONTROL))
+    {
+        BreakCrowdControlEffects();
         return;
+    }
 
     if (me->IsDead())
     {