Bots prefer buff food (#278)

Author: bozimmerman

src/modules/Bots/playerbot/strategy/ItemVisitors.h

@@ -1,5 +1,8 @@
 #pragma once
 
+#include "DBCStore.h"
+#include "DBCStores.h"
+
 char * strstri (const char* str1, const char* str2);
 
 namespace ai
@@ -280,6 +283,15 @@ namespace ai
         map<uint32, int> count;
     };
 
+    inline bool IsBuffFood(const ItemPrototype* proto)
+    {
+        if (proto->Class != ITEM_CLASS_CONSUMABLE ||
+            proto->Spells[0].SpellCategory != SPELLCATEGORY_FOOD)
+            return false;
+        SpellEntry const* sp = sSpellStore.LookupEntry(proto->Spells[0].SpellId);
+        return sp && ((sp->AttributesEx2 & SPELL_ATTR_EX2_FOOD_BUFF) || sp->Effect[1] != 0 || sp->Effect[2] != 0);
+    }
+
     class FindFoodVisitor : public FindUsableItemVisitor
     {
     public:
@@ -297,6 +309,36 @@ namespace ai
         uint32 spellCategory;
     };
 
+    class FindBuffFoodVisitor : public FindUsableItemVisitor
+    {
+    public:
+        FindBuffFoodVisitor(Player* bot) : FindUsableItemVisitor(bot) {}
+
+        virtual bool Accept(const ItemPrototype* proto)
+        {
+            return IsBuffFood(proto);
+        }
+    };
+
+    inline bool HasFoodBuff(Player* bot, const list<Item*>& buffFoods)
+    {
+        for (Item* item : buffFoods)
+        {
+            SpellEntry const* sp = sSpellStore.LookupEntry(item->GetProto()->Spells[0].SpellId);
+            if (!sp)
+                continue;
+            if (bot->HasAura(sp->Id))
+                return true;
+            for (int i = 1; i < MAX_EFFECT_INDEX; ++i)
+            {
+                uint32 triggerSpell = sp->EffectTriggerSpell[i];
+                if (triggerSpell && bot->HasAura(triggerSpell))
+                    return true;
+            }
+        }
+        return false;
+    }
+
     class FindConjuredFoodVisitor : public FindUsableItemVisitor
     {
     public:

src/modules/Bots/playerbot/strategy/actions/InventoryAction.cpp

@@ -278,6 +278,13 @@ list<Item*> InventoryAction::parseItems(string text)
         found.insert(visitor.GetResult().begin(), visitor.GetResult().end());
     }
 
+    if (text == "buff food")
+    {
+        FindBuffFoodVisitor visitor(bot);
+        IterateItems(&visitor, ITERATE_ITEMS_IN_BAGS);
+        found.insert(visitor.GetResult().begin(), visitor.GetResult().end());
+    }
+
     if (text == "drink")
     {
         FindFoodVisitor visitor(bot, SPELLCATEGORY_DRINK);

src/modules/Bots/playerbot/strategy/actions/NonCombatActions.h

@@ -3,6 +3,7 @@
 #include "../Action.h"
 #include "UseItemAction.h"
 #include "../../PlayerbotAIConfig.h"
+#include "../ItemVisitors.h"
 
 namespace ai
 {
@@ -55,7 +56,13 @@ namespace ai
             if (ai->IsEating())
                 return true;
 
-            bool result = UseItemAction::Execute(event);
+            bool result = false;
+            list<Item*> buffFoods = AI_VALUE2(list<Item*>, "inventory items", "buff food");
+            if (!buffFoods.empty() && !HasFoodBuff(bot, buffFoods))
+                result = UseItemAuto(*buffFoods.begin());
+            if (!result)
+                result = UseItemAction::Execute(event);
+
             if (result)
                 ai->SetEating();
             return result;