Remove world buffs when the strategy gets removed

cmangos · Apr 7, 2024 · 0e83bda · 0e83bda
1 parent e1d94ef
commit 0e83bda
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/playerbot/strategy/generic/NonCombatStrategy.cpp b/playerbot/strategy/generic/NonCombatStrategy.cpp
@@ -62,3 +62,19 @@ void WorldBuffStrategy::InitNonCombatTriggers(std::list<TriggerNode*>& triggers)
         "need world buff",
         NextAction::array(0, new NextAction("world buff", 1.0f), NULL)));
 }
+
+void WorldBuffStrategy::OnStrategyRemoved(BotState state)
+{
+    // Remove world buffs
+    Player* bot = ai->GetBot();
+    if (bot)
+    {
+        for (auto& wb : sPlayerbotAIConfig.worldBuffs)
+        {
+            if (bot->HasAura(wb.spellId))
+            {
+                bot->RemoveAurasDueToSpell(wb.spellId);
+            }
+        }
+    }
+}
diff --git a/playerbot/strategy/generic/NonCombatStrategy.h b/playerbot/strategy/generic/NonCombatStrategy.h
@@ -53,5 +53,6 @@ namespace ai
 
     protected:
         virtual void InitNonCombatTriggers(std::list<TriggerNode*>& triggers) override;
+        void OnStrategyRemoved(BotState state) override;
     };
 }