Transmogrification.cpp

/*
Transmogrification 3.3.5a - Gossip Menu
By Rochet2
Now with DB implemented

ScriptName for NPC:
NPC_Transmogrify

ScriptLoader.cpp:
void AddSC_NPC_Transmogrify();
AddSC_NPC_Transmogrify();
*/

#include "ScriptPCH.h"

class NPC_Transmogrify : public CreatureScript
{
public:
    NPC_Transmogrify() : CreatureScript("NPC_Transmogrify") { }

    bool OnGossipHello(Player* pPlayer, Creature* pUnit)
    {
        pPlayer->PlayerTalkClass->ClearMenus();
        for (uint8 Slot = EQUIPMENT_SLOT_START; Slot < EQUIPMENT_SLOT_END; Slot++)
        {
            if (Item* pItem = pPlayer->GetItemByPos(INVENTORY_SLOT_BAG_0, Slot))
            {
                uint32 Quality = pItem->GetTemplate()->Quality;
                if(Quality == ITEM_QUALITY_UNCOMMON || Quality == ITEM_QUALITY_RARE || Quality == ITEM_QUALITY_EPIC || Quality == ITEM_QUALITY_HEIRLOOM)
                    if(char* SlotName = GetSlotName(Slot))
                        pPlayer->ADD_GOSSIP_ITEM(GOSSIP_ICON_TRAINER, SlotName, EQUIPMENT_SLOT_END, Slot);
            }
        }
        pPlayer->ADD_GOSSIP_ITEM_EXTENDED(GOSSIP_ICON_INTERACT_1, "Remove transmogrifications", EQUIPMENT_SLOT_END+3, 0, "Remove transmogrifications from all equipped items?", 0, false);
        pPlayer->ADD_GOSSIP_ITEM(GOSSIP_ICON_TALK, "Update Menu", EQUIPMENT_SLOT_END+1, 0);
        pPlayer->SEND_GOSSIP_MENU(DEFAULT_GOSSIP_MESSAGE, pUnit->GetGUID());
        return true;
    }

    bool OnGossipSelect(Player* pPlayer, Creature* pUnit, uint32 sender, uint32 uiAction)
    {
        pPlayer->PlayerTalkClass->ClearMenus();
        if(sender == EQUIPMENT_SLOT_END)
        {
            if (Item* OLD = pPlayer->GetItemByPos(INVENTORY_SLOT_BAG_0, uiAction))
            {
                uint32 GUID = pPlayer->GetGUIDLow();
                Items[GUID].clear();
                uint32 limit = 0;
                for (uint8 i = INVENTORY_SLOT_ITEM_START; i < INVENTORY_SLOT_ITEM_END; i++)
                {
                    if(limit > 30)
                        break;
                    if (Item* pItem = pPlayer->GetItemByPos(INVENTORY_SLOT_BAG_0, i))
                        if(IsSuitable(pItem, OLD, pPlayer))
                            if(Items[GUID].find(pItem->GetTemplate()->DisplayInfoID) == Items[GUID].end())
                            {
                                limit++;
                                Items[GUID][pItem->GetTemplate()->DisplayInfoID] = pItem;
                                pPlayer->ADD_GOSSIP_ITEM_EXTENDED(GOSSIP_ICON_INTERACT_1, pItem->GetTemplate()->Name1, uiAction, pItem->GetTemplate()->DisplayInfoID, AreYouSure(uiAction, pItem), 0, false);
                            }
                }

                for (uint8 i = INVENTORY_SLOT_BAG_START; i < INVENTORY_SLOT_BAG_END; i++)
                {
                    if (Bag* pBag = pPlayer->GetBagByPos(i))
                        for (uint32 j = 0; j < pBag->GetBagSize(); j++)
                        {
                            if(limit > 30)
                                break;
                            if (Item* pItem = pPlayer->GetItemByPos(i, j))
                                if(IsSuitable(pItem, OLD, pPlayer))
                                    if(Items[GUID].find(pItem->GetTemplate()->DisplayInfoID) == Items[GUID].end())
                                    {
                                        limit++;
                                        Items[GUID][pItem->GetTemplate()->DisplayInfoID] = pItem;
                                        pPlayer->ADD_GOSSIP_ITEM_EXTENDED(GOSSIP_ICON_INTERACT_1, pItem->GetTemplate()->Name1, uiAction, pItem->GetTemplate()->DisplayInfoID, AreYouSure(uiAction, pItem), 0, false);
                                    }
                        }
                }

                if(Items[GUID].empty())
                {
                    pPlayer->GetSession()->SendNotification("You have no suitable items in your inventory");
                    OnGossipHello(pPlayer, pUnit);
                    return true;
                }
                pPlayer->ADD_GOSSIP_ITEM(GOSSIP_ICON_TALK, "Back..", EQUIPMENT_SLOT_END+1, 0);
                pPlayer->SEND_GOSSIP_MENU(DEFAULT_GOSSIP_MESSAGE, pUnit->GetGUID());
            }
            else
                OnGossipHello(pPlayer, pUnit);
        }
        else if(sender == EQUIPMENT_SLOT_END+1)
            OnGossipHello(pPlayer, pUnit);
        else if(sender == EQUIPMENT_SLOT_END+2)
            pPlayer->PlayerTalkClass->SendCloseGossip();
        else if(sender == EQUIPMENT_SLOT_END+3)
        {
            for (uint8 Slot = EQUIPMENT_SLOT_START; Slot < EQUIPMENT_SLOT_END; Slot++)
                if (Item* pItem = pPlayer->GetItemByPos(INVENTORY_SLOT_BAG_0, Slot))
                {
                    if(pItem->FakeEntry || pItem->FakeOwner)
                    {
                        CharacterDatabase.PExecute("UPDATE item_instance SET FakeEntry = 0, FakeOwner = 0 WHERE guid = %u", pItem->GetGUIDLow());
                        pItem->FakeEntry = NULL;
                        pItem->FakeOwner = NULL;
                    }
                    pPlayer->SetUInt32Value(PLAYER_VISIBLE_ITEM_1_ENTRYID + (Slot * 2), pItem->GetEntry());
                }
                OnGossipHello(pPlayer, pUnit);
        }
        else
        {
            uint32 GUID = pPlayer->GetGUIDLow();
            Item* OLD = pPlayer->GetItemByPos(INVENTORY_SLOT_BAG_0, sender);
            if(!OLD || Items[GUID].find(uiAction) == Items[GUID].end() || !IsSuitable(Items[GUID][uiAction], OLD, pPlayer))
                pPlayer->GetSession()->SendNotification("There is no suitable item in the slot");
            else
            {
                uint32 FakeEntry = Items[GUID][uiAction]->GetEntry();
                CharacterDatabase.PExecute("UPDATE item_instance SET FakeEntry = %u, FakeOwner = %u WHERE guid = %u", FakeEntry, pPlayer->GetGUIDLow(), OLD->GetGUIDLow());
                OLD->FakeEntry = FakeEntry;
                OLD->FakeOwner = pPlayer->GetGUIDLow();
                // pPlayer->SetVisibleItemSlot(sender, OLD); // No need to use this, useless checking
                pPlayer->UpdateUInt32Value(PLAYER_VISIBLE_ITEM_1_ENTRYID + (sender * 2), FakeEntry); // use this instead
                pPlayer->GetSession()->SendAreaTriggerMessage("%s transmogrified", GetSlotName(sender));
            }
            Items[GUID].clear();
            OnGossipHello(pPlayer, pUnit);
        }
        return true;
    }
private:
    std::map<uint64, std::map<uint32, Item*> > Items; // Items[GUID][DISPLAY] = item

    char * GetSlotName(uint8 slot)
    {
        switch(slot)
        {
        case EQUIPMENT_SLOT_HEAD      : return "Head";
        case EQUIPMENT_SLOT_SHOULDERS : return "Shoulders";
        case EQUIPMENT_SLOT_BODY      : return "Shirt";
        case EQUIPMENT_SLOT_CHEST     : return "Chest";
        case EQUIPMENT_SLOT_WAIST     : return "Waist";
        case EQUIPMENT_SLOT_LEGS      : return "Legs";
        case EQUIPMENT_SLOT_FEET      : return "Feet";
        case EQUIPMENT_SLOT_WRISTS    : return "Wrists";
        case EQUIPMENT_SLOT_HANDS     : return "Hands";
        case EQUIPMENT_SLOT_BACK      : return "Back";
        case EQUIPMENT_SLOT_MAINHAND  : return "Main hand";
        case EQUIPMENT_SLOT_OFFHAND   : return "Off hand";
        case EQUIPMENT_SLOT_RANGED    : return "Ranged";
        case EQUIPMENT_SLOT_TABARD    : return "Tabard";
        default: return NULL;
        }
    }

    bool IsSuitable(Item* pItem, Item* OLD, Player* pPlayer)
    {
        if(pItem->GetTemplate()->DisplayInfoID != OLD->GetTemplate()->DisplayInfoID)
            if(pPlayer->CanUseItem(pItem, false) == EQUIP_ERR_OK)
            {
                uint32 Quality = pItem->GetTemplate()->Quality;
                if(Quality == ITEM_QUALITY_UNCOMMON || Quality == ITEM_QUALITY_RARE || Quality == ITEM_QUALITY_EPIC || Quality == ITEM_QUALITY_HEIRLOOM) // Heirlooms also accepted? Remember to change in Gossip Hello too!
                {
                    uint32 NClass = pItem->GetTemplate()->Class;
                    uint32 OClass = OLD->GetTemplate()->Class;
                    uint32 NSubClass = pItem->GetTemplate()->SubClass;
                    uint32 OSubClass = OLD->GetTemplate()->SubClass;
                    if(NClass == OClass) // not possibly the best structure here, but atleast I got my head around this
                        if(NClass == ITEM_CLASS_WEAPON)
                        {
                            if(NSubClass == OSubClass || ((NSubClass == ITEM_SUBCLASS_WEAPON_BOW || NSubClass == ITEM_SUBCLASS_WEAPON_GUN || NSubClass == ITEM_SUBCLASS_WEAPON_CROSSBOW) && (OSubClass == ITEM_SUBCLASS_WEAPON_BOW || OSubClass == ITEM_SUBCLASS_WEAPON_GUN || OSubClass == ITEM_SUBCLASS_WEAPON_CROSSBOW)))
                            {
                                return true;
                            }
                        }
                        else if(NClass == ITEM_CLASS_ARMOR)
                            if(NSubClass == OSubClass && pItem->GetTemplate()->InventoryType == OLD->GetTemplate()->InventoryType)
                            {
                                return true;
                            }
                }
            }
            return false;
    }
    std::string AreYouSure(uint8 slot, Item* pItem)
    {
        std::string msg = "Transmogrify ";
        msg += GetSlotName(slot);
        msg += "\nTo ";
        msg += pItem->GetTemplate()->Name1;
        return msg;
    }
};

void AddSC_NPC_Transmogrify()
{
    new NPC_Transmogrify();
}