From d777d183c521201c46a68dd1ecf431cbc21317bc Mon Sep 17 00:00:00 2001
From: Walter Pagani <paganiwalter@gmail.com>
Date: Mon, 11 Mar 2024 05:00:48 -0300
Subject: [PATCH] chore. Add group loot (#30)

* Add group loot

* fix. loot in group

* Distribute the gold obtained, in a group or alone
---
 src/AoeLoot_SC.cpp | 97 ++++++++++++++++------------------------------
 1 file changed, 33 insertions(+), 64 deletions(-)

diff --git a/src/AoeLoot_SC.cpp b/src/AoeLoot_SC.cpp
index d4819e0..cd31688 100644
--- a/src/AoeLoot_SC.cpp
+++ b/src/AoeLoot_SC.cpp
@@ -15,7 +15,6 @@
  * with this program. If not, see <http://www.gnu.org/licenses/>.
  */
 
-#include "Log.h"
 #include "ScriptMgr.h"
 #include "Config.h"
 #include "Chat.h"
@@ -28,7 +27,6 @@ enum AoeLootString
     AOE_ITEM_IN_THE_MAIL
 };
 
-typedef std::map<uint32, uint32> AOEContainer;
 
 class AoeLoot_Player : public PlayerScript
 {
@@ -55,87 +53,58 @@ class AoeLoot_Player : public PlayerScript
         if (!_enable)
             return;
 
+        std::list<Creature*> deadCreatures;
+
         float range = sConfigMgr->GetOption<float>("AOELoot.Range", 30.0);
 
         uint32 gold = 0;
 
-        std::list<Creature*> deadCreatures;
-        deadCreatures.clear();
-        AOEContainer aoeLoot;
-        player->GetDeadCreatureListInGrid(deadCreatures, range, false);
+        player->GetDeadCreatureListInGrid(deadCreatures, range);
 
         for (auto& _creature : deadCreatures)
         {
-            if (player->GetGroup())
-            {
-                if (player->GetGroup()->GetMembersCount() > 1)
-                {
-                    if (_creature->IsDungeonBoss() || _creature->isWorldBoss())
-                        continue;
-                }
-                else if (player->GetGroup()->GetMembersCount() == 1)
-                {
-                    player->GetGroup()->SetLootMethod(FREE_FOR_ALL);
-                }
-            }
+            ObjectGuid lootGuid = player->GetLootGUID();
+            Loot* loot = &_creature->loot;
+            gold += loot->gold;
+            loot->gold = 0;
+            uint8 maxSlot = loot->GetMaxSlotInLootFor(player);
 
-            if (player == _creature->GetLootRecipient() && (_creature->HasDynamicFlag(UNIT_DYNFLAG_LOOTABLE)))
+            for (uint32 lootSlot = 0; lootSlot < maxSlot; ++lootSlot)
             {
-                Loot* loot = &_creature->loot;
-                gold += loot->gold;
-                loot->gold = 0;
+                InventoryResult msg;
+                LootItem* lootItem = player->StoreLootItem(lootSlot, loot, msg);
 
-                for (auto const& item : loot->items)
+                if (msg != EQUIP_ERR_OK && lootGuid.IsItem() && loot->loot_type != LOOT_CORPSE)
                 {
-                    if (loot->items.size() > 1 && (loot->items[item.itemIndex].itemid == loot->items[item.itemIndex + 1].itemid))
-                        continue;
-
-                    ItemTemplate const* itemTemplate = sObjectMgr->GetItemTemplate(item.itemid);
-
-                    if (itemTemplate->MaxCount != 1)
-                    {
-                        aoeLoot[item.itemid] += (uint32)item.count;
-                    }
-                    else
-                    {
-                        if (!player->HasItemCount(item.itemid, 1, true))
-                            aoeLoot[item.itemid] = 1;
-                    }
-                }
+                    lootItem->is_looted = true;
+                    loot->NotifyItemRemoved(lootItem->itemIndex);
+                    loot->unlootedCount--;
 
-                for (auto const& item : loot->quest_items)
-                {
-                    aoeLoot[item.itemid] += (uint32)item.count;
+                    player->SendItemRetrievalMail(lootItem->itemid, lootItem->count);
                 }
+            }
 
-                player->SendLootRelease(player->GetLootGUID());
-
-                if (!loot->empty())
-                {
-                    if (!_creature->IsAlive())
-                    {
-                        _creature->AllLootRemovedFromCorpse();
-                        _creature->RemoveFlag(UNIT_DYNAMIC_FLAGS, UNIT_DYNFLAG_LOOTABLE);
-                        loot->clear();
-                    }
-                }
-                else
-                {
-                    _creature->RemoveFlag(UNIT_DYNAMIC_FLAGS, UNIT_DYNFLAG_LOOTABLE);
+            if (loot->isLooted())
+            {
+                // skip pickpocketing loot for speed, skinning timer reduction is no-op in fact
+                if (!_creature->IsAlive())
                     _creature->AllLootRemovedFromCorpse();
-                }
-            }
-        }
 
-        for (auto const& [itemId, count] : aoeLoot)
-        {
-            if (!player->AddItem(itemId, count))
+                _creature->RemoveDynamicFlag(UNIT_DYNFLAG_LOOTABLE);
+                loot->clear();
+            }
+            else
             {
-                if (sConfigMgr->GetOption<bool>("AOELoot.MailEnable", true))
+                // if the round robin player release, reset it.
+                if (player->GetGUID() == loot->roundRobinPlayer)
                 {
-                    player->SendItemRetrievalMail(itemId, count);
-                    ChatHandler(player->GetSession()).SendSysMessage(AOE_ITEM_IN_THE_MAIL);
+                    loot->roundRobinPlayer.Clear();
+
+                    if (Group* group = player->GetGroup())
+                        group->SendLooter(_creature, nullptr);
                 }
+                // force dynflag update to update looter and lootable info
+                _creature->ForceValuesUpdateAtIndex(UNIT_DYNAMIC_FLAGS);
             }
         }