Files
mod-playerbots/src/ChatHelper.cpp
2022-09-26 17:20:59 -06:00

570 lines
17 KiB
C++

/*
* Copyright (C) 2016+ AzerothCore <www.azerothcore.org>, released under GNU GPL v2 license, you may redistribute it and/or modify it under version 2 of the License, or (at your option), any later version.
*/
#include "ChatHelper.h"
#include "AiFactory.h"
#include "Playerbots.h"
#include "SpellInfo.h"
std::map<std::string, uint32> ChatHelper::consumableSubClasses;
std::map<std::string, uint32> ChatHelper::tradeSubClasses;
std::map<std::string, uint32> ChatHelper::itemQualities;
std::map<std::string, uint32> ChatHelper::projectileSubClasses;
std::map<std::string, uint32> ChatHelper::slots;
std::map<std::string, uint32> ChatHelper::skills;
std::map<std::string, ChatMsg> ChatHelper::chats;
std::map<uint8, std::string> ChatHelper::classes;
std::map<uint8, std::string> ChatHelper::races;
std::map<uint8, std::map<uint8, std::string> > ChatHelper::specs;
template<class T>
static bool substrContainsInMap(std::string const searchTerm, std::map<std::string, T> searchIn)
{
for (typename std::map<std::string, T>::iterator i = searchIn.begin(); i != searchIn.end(); ++i)
{
std::string const term = i->first;
if (term.size() > 1 && searchTerm.find(term) != std::string::npos)
return true;
}
return false;
}
ChatHelper::ChatHelper(PlayerbotAI* botAI) : PlayerbotAIAware(botAI)
{
itemQualities["poor"] = ITEM_QUALITY_POOR;
itemQualities["gray"] = ITEM_QUALITY_POOR;
itemQualities["normal"] = ITEM_QUALITY_NORMAL;
itemQualities["white"] = ITEM_QUALITY_NORMAL;
itemQualities["uncommon"] = ITEM_QUALITY_UNCOMMON;
itemQualities["green"] = ITEM_QUALITY_UNCOMMON;
itemQualities["rare"] = ITEM_QUALITY_RARE;
itemQualities["blue"] = ITEM_QUALITY_RARE;
itemQualities["epic"] = ITEM_QUALITY_EPIC;
itemQualities["violet"] = ITEM_QUALITY_EPIC;
itemQualities["legendary"] = ITEM_QUALITY_LEGENDARY;
itemQualities["yellow"] = ITEM_QUALITY_LEGENDARY;
consumableSubClasses["potion"] = ITEM_SUBCLASS_POTION;
consumableSubClasses["elixir"] = ITEM_SUBCLASS_ELIXIR;
consumableSubClasses["flask"] = ITEM_SUBCLASS_FLASK;
consumableSubClasses["scroll"] = ITEM_SUBCLASS_SCROLL;
consumableSubClasses["food"] = ITEM_SUBCLASS_FOOD;
consumableSubClasses["bandage"] = ITEM_SUBCLASS_BANDAGE;
consumableSubClasses["enchant"] = ITEM_SUBCLASS_CONSUMABLE_OTHER;
projectileSubClasses["arrows"] = ITEM_SUBCLASS_ARROW;
projectileSubClasses["bullets"] = ITEM_SUBCLASS_BULLET;
//tradeSubClasses["cloth"] = ITEM_SUBCLASS_CLOTH;
//tradeSubClasses["leather"] = ITEM_SUBCLASS_LEATHER;
//tradeSubClasses["metal"] = ITEM_SUBCLASS_METAL_STONE;
//tradeSubClasses["stone"] = ITEM_SUBCLASS_METAL_STONE;
//tradeSubClasses["ore"] = ITEM_SUBCLASS_METAL_STONE;
//tradeSubClasses["meat"] = ITEM_SUBCLASS_MEAT;
//tradeSubClasses["herb"] = ITEM_SUBCLASS_HERB;
//tradeSubClasses["elemental"] = ITEM_SUBCLASS_ELEMENTAL;
//tradeSubClasses["disenchants"] = ITEM_SUBCLASS_ENCHANTING;
//tradeSubClasses["enchanting"] = ITEM_SUBCLASS_ENCHANTING;
//tradeSubClasses["gems"] = ITEM_SUBCLASS_JEWELCRAFTING;
//tradeSubClasses["jewels"] = ITEM_SUBCLASS_JEWELCRAFTING;
//tradeSubClasses["jewelcrafting"] = ITEM_SUBCLASS_JEWELCRAFTING;
slots["head"] = EQUIPMENT_SLOT_HEAD;
slots["neck"] = EQUIPMENT_SLOT_NECK;
slots["shoulder"] = EQUIPMENT_SLOT_SHOULDERS;
slots["shirt"] = EQUIPMENT_SLOT_BODY;
slots["chest"] = EQUIPMENT_SLOT_CHEST;
slots["waist"] = EQUIPMENT_SLOT_WAIST;
slots["legs"] = EQUIPMENT_SLOT_LEGS;
slots["feet"] = EQUIPMENT_SLOT_FEET;
slots["wrist"] = EQUIPMENT_SLOT_WRISTS;
slots["hands"] = EQUIPMENT_SLOT_HANDS;
slots["finger 1"] = EQUIPMENT_SLOT_FINGER1;
slots["finger 2"] = EQUIPMENT_SLOT_FINGER2;
slots["trinket 1"] = EQUIPMENT_SLOT_TRINKET1;
slots["trinket 2"] = EQUIPMENT_SLOT_TRINKET2;
slots["back"] = EQUIPMENT_SLOT_BACK;
slots["main hand"] = EQUIPMENT_SLOT_MAINHAND;
slots["off hand"] = EQUIPMENT_SLOT_OFFHAND;
slots["ranged"] = EQUIPMENT_SLOT_RANGED;
slots["tabard"] = EQUIPMENT_SLOT_TABARD;
skills["first aid"] = SKILL_FIRST_AID;
skills["fishing"] = SKILL_FISHING;
skills["cooking"] = SKILL_COOKING;
skills["alchemy"] = SKILL_ALCHEMY;
skills["enchanting"] = SKILL_ENCHANTING;
skills["engineering"] = SKILL_ENGINEERING;
skills["leatherworking"] = SKILL_LEATHERWORKING;
skills["blacksmithing"] = SKILL_BLACKSMITHING;
skills["tailoring"] = SKILL_TAILORING;
skills["herbalism"] = SKILL_HERBALISM;
skills["mining"] = SKILL_MINING;
skills["skinning"] = SKILL_SKINNING;
skills["jewelcrafting"] = SKILL_JEWELCRAFTING;
chats["party"] = CHAT_MSG_PARTY;
chats["p"] = CHAT_MSG_PARTY;
chats["guild"] = CHAT_MSG_GUILD;
chats["g"] = CHAT_MSG_GUILD;
chats["raid"] = CHAT_MSG_RAID;
chats["r"] = CHAT_MSG_RAID;
chats["whisper"] = CHAT_MSG_WHISPER;
chats["w"] = CHAT_MSG_WHISPER;
classes[CLASS_DRUID] = "druid";
specs[CLASS_DRUID][0] = "balance";
specs[CLASS_DRUID][1] = "feral combat";
specs[CLASS_DRUID][2] = "restoration";
classes[CLASS_HUNTER] = "hunter";
specs[CLASS_HUNTER][0] = "beast mastery";
specs[CLASS_HUNTER][1] = "marksmanship";
specs[CLASS_HUNTER][2] = "survival";
classes[CLASS_MAGE] = "mage";
specs[CLASS_MAGE][0] = "arcane";
specs[CLASS_MAGE][1] = "fire";
specs[CLASS_MAGE][2] = "frost";
classes[CLASS_PALADIN] = "paladin";
specs[CLASS_PALADIN][0] = "holy";
specs[CLASS_PALADIN][1] = "protection";
specs[CLASS_PALADIN][2] = "retribution";
classes[CLASS_PRIEST] = "priest";
specs[CLASS_PRIEST][0] = "discipline";
specs[CLASS_PRIEST][1] = "holy";
specs[CLASS_PRIEST][2] = "shadow";
classes[CLASS_ROGUE] = "rogue";
specs[CLASS_ROGUE][0] = "assasination";
specs[CLASS_ROGUE][1] = "combat";
specs[CLASS_ROGUE][2] = "subtlety";
classes[CLASS_SHAMAN] = "shaman";
specs[CLASS_SHAMAN][0] = "elemental";
specs[CLASS_SHAMAN][1] = "enhancement";
specs[CLASS_SHAMAN][2] = "restoration";
classes[CLASS_WARLOCK] = "warlock";
specs[CLASS_WARLOCK][0] = "affliction";
specs[CLASS_WARLOCK][1] = "demonology";
specs[CLASS_WARLOCK][2] = "destruction";
classes[CLASS_WARRIOR] = "warrior";
specs[CLASS_WARRIOR][0] = "arms";
specs[CLASS_WARRIOR][1] = "fury";
specs[CLASS_WARRIOR][2] = "protection";
classes[CLASS_DEATH_KNIGHT] = "dk";
specs[CLASS_DEATH_KNIGHT][0] = "blood";
specs[CLASS_DEATH_KNIGHT][1] = "frost";
specs[CLASS_DEATH_KNIGHT][2] = "unholy";
races[RACE_DWARF] = "Dwarf";
races[RACE_GNOME] = "Gnome";
races[RACE_HUMAN] = "Human";
races[RACE_NIGHTELF] = "Night Elf";
races[RACE_ORC] = "Orc";
races[RACE_TAUREN] = "Tauren";
races[RACE_TROLL] = "Troll";
races[RACE_UNDEAD_PLAYER] = "Undead";
races[RACE_BLOODELF] = "Blood Elf";
races[RACE_DRAENEI] = "Draenei";
}
std::string const ChatHelper::formatMoney(uint32 copper)
{
std::ostringstream out;
if (!copper)
{
out << "0";
return out.str();
}
uint32 gold = uint32(copper / 10000);
copper -= (gold * 10000);
uint32 silver = uint32(copper / 100);
copper -= (silver * 100);
bool space = false;
if (gold > 0)
{
out << gold << "g";
space = true;
}
if (silver > 0 && gold < 50)
{
if (space)
out << " ";
out << silver << "s";
space = true;
}
if (copper > 0 && gold < 10)
{
if (space)
out << " ";
out << copper << "c";
}
return out.str();
}
uint32 ChatHelper::parseMoney(std::string const text)
{
// if user specified money in ##g##s##c format
std::string acum = "";
uint32 copper = 0;
for (uint8 i = 0; i < text.length(); i++)
{
if (text[i] == 'g')
{
copper += (atol(acum.c_str()) * 100 * 100);
acum = "";
}
else if (text[i] == 'c')
{
copper += atol(acum.c_str());
acum = "";
}
else if (text[i] == 's')
{
copper += (atol(acum.c_str()) * 100);
acum = "";
}
else if (text[i] == ' ')
break;
else if (text[i] >= 48 && text[i] <= 57)
acum += text[i];
else
{
copper = 0;
break;
}
}
return copper;
}
ItemIds ChatHelper::parseItems(std::string const text)
{
ItemIds itemIds;
uint8 pos = 0;
while (true)
{
uint32 i = text.find("Hitem:", pos);
if (i == -1)
break;
pos = i + 6;
uint32 endPos = text.find(':', pos);
if (endPos == -1)
break;
std::string const idC = text.substr(pos, endPos - pos);
uint32 id = atol(idC.c_str());
pos = endPos;
if (id)
itemIds.insert(id);
}
return itemIds;
}
std::string const ChatHelper::FormatQuest(Quest const* quest)
{
std::ostringstream out;
out << "|cFFFFFF00|Hquest:" << quest->GetQuestId() << ':' << quest->GetQuestLevel() << "|h[" << quest->GetTitle() << "]|h|r";
return out.str();
}
std::string const ChatHelper::FormatGameobject(GameObject* go)
{
std::ostringstream out;
out << "|cFFFFFF00|Hfound:" << go->GetGUID().GetRawValue() << ":" << go->GetEntry() << ":" << "|h[" << go->GetGOInfo()->name << "]|h|r";
return out.str();
}
std::string const ChatHelper::FormatWorldobject(WorldObject* wo)
{
std::ostringstream out;
out << "|cFFFFFF00|Hfound:" << wo->GetGUID().GetRawValue() << ":" << wo->GetEntry() << ":" << "|h[";
out << (wo->ToGameObject() ? ((GameObject*)wo)->GetGOInfo()->name : wo->GetName()) << "]|h|r";
return out.str();
}
std::string const ChatHelper::FormatWorldEntry(int32 entry)
{
CreatureTemplate const* cInfo = nullptr;
GameObjectTemplate const* gInfo = nullptr;
if (entry > 0)
cInfo = sObjectMgr->GetCreatureTemplate(entry);
else
gInfo = sObjectMgr->GetGameObjectTemplate(entry * -1);
std::ostringstream out;
out << "|cFFFFFF00|Hentry:" << abs(entry) << ":" << "|h[";
if (entry < 0 && gInfo)
out << gInfo->name;
else if (entry > 0 && cInfo)
out << cInfo->Name;
else
out << "unknown";
out << "]|h|r";
return out.str();
}
std::string const ChatHelper::FormatSpell(SpellInfo const* spellInfo)
{
std::ostringstream out;
out << "|cffffffff|Hspell:" << spellInfo->Id << "|h[" << spellInfo->SpellName[LOCALE_enUS] << "]|h|r";
return out.str();
}
std::string const ChatHelper::FormatItem(ItemTemplate const* proto, uint32 count, uint32 total)
{
char color[32];
sprintf(color, "%x", ItemQualityColors[proto->Quality]);
std::ostringstream out;
out << "|c" << color << "|Hitem:" << proto->ItemId
<< ":0:0:0:0:0:0:0" << "|h[" << proto->Name1
<< "]|h|r";
if (count > 1)
out << "x" << count;
if (total > 0)
out << " (" << total << ")";
return out.str();
}
std::string const ChatHelper::FormatQItem(uint32 itemId)
{
char color[32];
sprintf(color, "%x", ItemQualityColors[0]);
std::ostringstream out;
out << "|c" << color << "|Hitem:" << itemId << ":0:0:0:0:0:0:0"
<< "|h[item"
<< "]|h|r";
return out.str();
}
ChatMsg ChatHelper::parseChat(std::string const text)
{
if (chats.find(text) != chats.end())
return chats[text];
return CHAT_MSG_SYSTEM;
}
std::string const ChatHelper::FormatChat(ChatMsg chat)
{
switch (chat)
{
case CHAT_MSG_GUILD:
return "guild";
case CHAT_MSG_PARTY:
return "party";
case CHAT_MSG_WHISPER:
return "whisper";
case CHAT_MSG_RAID:
return "raid";
}
return "unknown";
}
uint32 ChatHelper::parseSpell(std::string const text)
{
PlayerbotChatHandler handler(botAI->GetBot());
return handler.extractSpellId(text);
}
GuidVector ChatHelper::parseGameobjects(std::string const text)
{
GuidVector gos;
// Link format
// |cFFFFFF00|Hfound:" << guid << ':' << entry << ':' << "|h[" << gInfo->name << "]|h|r";
// |cFFFFFF00|Hfound:9582:1731|h[Copper Vein]|h|r
uint8 pos = 0;
while (true)
{
// extract GO guid
uint32 i = text.find("Hfound:", pos); // base H = 11
if (i == -1) // break if error
break;
pos = i + 7; //start of window in text 11 + 7 = 18
uint32 endPos = text.find(':', pos); // end of window in text 22
if (endPos == -1) //break if error
break;
std::istringstream stream(text.substr(pos, endPos - pos));
uint64 guid;
stream >> guid;
// extract GO entry
pos = endPos + 1;
endPos = text.find(':', pos); // end of window in text
if (endPos == -1) //break if error
break;
std::string const entryC = text.substr(pos, endPos - pos); // get std::string const within window i.e entry
uint32 entry = atol(entryC.c_str()); // convert ascii to float
ObjectGuid lootCurrent = ObjectGuid(guid);
if (guid)
gos.push_back(lootCurrent);
}
return gos;
}
std::string const ChatHelper::FormatQuestObjective(std::string const name, uint32 available, uint32 required)
{
std::ostringstream out;
out << "|cFFFFFFFF" << name << (available >= required ? "|c0000FF00: " : "|c00FF0000: ")
<< available << "/" << required << "|r";
return out.str();
}
uint32 ChatHelper::parseItemQuality(std::string const text)
{
if (itemQualities.find(text) == itemQualities.end())
return MAX_ITEM_QUALITY;
return itemQualities[text];
}
bool ChatHelper::parseItemClass(std::string const text, uint32* itemClass, uint32* itemSubClass)
{
if (text == "questitem")
{
*itemClass = ITEM_CLASS_QUEST;
*itemSubClass = ITEM_SUBCLASS_QUEST;
return true;
}
if (consumableSubClasses.find(text) != consumableSubClasses.end())
{
*itemClass = ITEM_CLASS_CONSUMABLE;
*itemSubClass = consumableSubClasses[text];
return true;
}
if (tradeSubClasses.find(text) != tradeSubClasses.end())
{
*itemClass = ITEM_CLASS_TRADE_GOODS;
*itemSubClass = tradeSubClasses[text];
return true;
}
if (projectileSubClasses.find(text) != projectileSubClasses.end())
{
*itemClass = ITEM_CLASS_PROJECTILE;
*itemSubClass = projectileSubClasses[text];
return true;
}
return false;
}
uint32 ChatHelper::parseSlot(std::string const text)
{
if (slots.find(text) != slots.end())
return slots[text];
return EQUIPMENT_SLOT_END;
}
bool ChatHelper::parseable(std::string const text)
{
return text.find("|H") != std::string::npos || text == "questitem" || text == "ammo" || substrContainsInMap<uint32>(text, consumableSubClasses) ||
substrContainsInMap<uint32>(text, tradeSubClasses) || substrContainsInMap<uint32>(text, itemQualities) || substrContainsInMap<uint32>(text, slots) ||
substrContainsInMap<ChatMsg>(text, chats) || substrContainsInMap<uint32>(text, skills) || parseMoney(text) > 0;
}
std::string const ChatHelper::FormatClass(Player* player, int8 spec)
{
uint8 cls = player->getClass();
std::ostringstream out;
out << specs[cls][spec] << " (";
std::map<uint8, uint32> tabs = AiFactory::GetPlayerSpecTabs(player);
uint32 c0 = tabs[0];
uint32 c1 = tabs[1];
uint32 c2 = tabs[2];
out << (c0 ? "|h|cff00ff00" : "") << c0 << "|h|cffffffff/";
out << (c1 ? "|h|cff00ff00" : "") << c1 << "|h|cffffffff/";
out << (c2 ? "|h|cff00ff00" : "") << c2 << "|h|cffffffff";
out << ")|r " << classes[cls];
return out.str();
}
std::string const ChatHelper::FormatClass(uint8 cls)
{
return classes[cls];
}
std::string const ChatHelper::FormatRace(uint8 race)
{
return races[race];
}
uint32 ChatHelper::parseSkill(std::string const text)
{
if (skills.find(text) != skills.end())
return skills[text];
return SKILL_NONE;
}
std::string const ChatHelper::FormatSkill(uint32 skill)
{
for (std::map<std::string, uint32>::iterator i = skills.begin(); i != skills.end(); ++i)
{
if (i->second == skill)
return i->first;
}
return "";
}
std::string const ChatHelper::FormatBoolean(bool flag)
{
return flag ? "|cff00ff00ON|r" : "|cffffff00OFF|r";
}
void ChatHelper::eraseAllSubStr(std::string& mainStr, std::string const toErase)
{
size_t pos = std::string::npos;
// Search for the substring in std::string const in a loop untill nothing is found
while ((pos = mainStr.find(toErase)) != std::string::npos)
{
// If found then erase it from std::string
mainStr.erase(pos, toErase.length());
}
}