#include "addons/lua_engine.hpp" #include "game/game_handler.hpp" #include "game/entity.hpp" #include "core/logger.hpp" extern "C" { #include #include #include } namespace wowee::addons { // Retrieve GameHandler pointer stored in Lua registry static game::GameHandler* getGameHandler(lua_State* L) { lua_getfield(L, LUA_REGISTRYINDEX, "wowee_game_handler"); auto* gh = static_cast(lua_touserdata(L, -1)); lua_pop(L, 1); return gh; } // WoW-compatible print() — outputs to chat window instead of stdout static int lua_wow_print(lua_State* L) { int nargs = lua_gettop(L); std::string result; for (int i = 1; i <= nargs; i++) { if (i > 1) result += '\t'; // Lua 5.1: use lua_tostring (luaL_tolstring is 5.3+) if (lua_isstring(L, i) || lua_isnumber(L, i)) { const char* s = lua_tostring(L, i); if (s) result += s; } else if (lua_isboolean(L, i)) { result += lua_toboolean(L, i) ? "true" : "false"; } else if (lua_isnil(L, i)) { result += "nil"; } else { result += lua_typename(L, lua_type(L, i)); } } auto* gh = getGameHandler(L); if (gh) { game::MessageChatData msg; msg.type = game::ChatType::SYSTEM; msg.language = game::ChatLanguage::UNIVERSAL; msg.message = result; gh->addLocalChatMessage(msg); } LOG_INFO("[Lua] ", result); return 0; } // WoW-compatible message() — same as print for now static int lua_wow_message(lua_State* L) { return lua_wow_print(L); } // Helper: get player Unit from game handler static game::Unit* getPlayerUnit(lua_State* L) { auto* gh = getGameHandler(L); if (!gh) return nullptr; auto entity = gh->getEntityManager().getEntity(gh->getPlayerGuid()); if (!entity) return nullptr; return dynamic_cast(entity.get()); } // Helper: resolve "player", "target", "focus", "pet" unit IDs to entity static game::Unit* resolveUnit(lua_State* L, const char* unitId) { auto* gh = getGameHandler(L); if (!gh || !unitId) return nullptr; std::string uid(unitId); for (char& c : uid) c = static_cast(std::tolower(static_cast(c))); uint64_t guid = 0; if (uid == "player") guid = gh->getPlayerGuid(); else if (uid == "target") guid = gh->getTargetGuid(); else if (uid == "focus") guid = gh->getFocusGuid(); else if (uid == "pet") guid = gh->getPetGuid(); else return nullptr; if (guid == 0) return nullptr; auto entity = gh->getEntityManager().getEntity(guid); if (!entity) return nullptr; return dynamic_cast(entity.get()); } // --- WoW Unit API --- static int lua_UnitName(lua_State* L) { const char* uid = luaL_optstring(L, 1, "player"); auto* unit = resolveUnit(L, uid); if (unit && !unit->getName().empty()) { lua_pushstring(L, unit->getName().c_str()); } else { lua_pushstring(L, "Unknown"); } return 1; } static int lua_UnitHealth(lua_State* L) { const char* uid = luaL_optstring(L, 1, "player"); auto* unit = resolveUnit(L, uid); lua_pushnumber(L, unit ? unit->getHealth() : 0); return 1; } static int lua_UnitHealthMax(lua_State* L) { const char* uid = luaL_optstring(L, 1, "player"); auto* unit = resolveUnit(L, uid); lua_pushnumber(L, unit ? unit->getMaxHealth() : 0); return 1; } static int lua_UnitPower(lua_State* L) { const char* uid = luaL_optstring(L, 1, "player"); auto* unit = resolveUnit(L, uid); lua_pushnumber(L, unit ? unit->getPower() : 0); return 1; } static int lua_UnitPowerMax(lua_State* L) { const char* uid = luaL_optstring(L, 1, "player"); auto* unit = resolveUnit(L, uid); lua_pushnumber(L, unit ? unit->getMaxPower() : 0); return 1; } static int lua_UnitLevel(lua_State* L) { const char* uid = luaL_optstring(L, 1, "player"); auto* unit = resolveUnit(L, uid); lua_pushnumber(L, unit ? unit->getLevel() : 0); return 1; } static int lua_UnitExists(lua_State* L) { const char* uid = luaL_optstring(L, 1, "player"); auto* unit = resolveUnit(L, uid); lua_pushboolean(L, unit != nullptr); return 1; } static int lua_UnitIsDead(lua_State* L) { const char* uid = luaL_optstring(L, 1, "player"); auto* unit = resolveUnit(L, uid); lua_pushboolean(L, unit && unit->getHealth() == 0); return 1; } static int lua_UnitClass(lua_State* L) { const char* uid = luaL_optstring(L, 1, "player"); auto* gh = getGameHandler(L); auto* unit = resolveUnit(L, uid); if (unit && gh) { static const char* kClasses[] = {"", "Warrior","Paladin","Hunter","Rogue","Priest", "Death Knight","Shaman","Mage","Warlock","","Druid"}; uint8_t classId = 0; // For player, use character data; for others, use UNIT_FIELD_BYTES_0 std::string uidStr(uid); for (char& c : uidStr) c = static_cast(std::tolower(static_cast(c))); if (uidStr == "player") classId = gh->getPlayerClass(); const char* name = (classId < 12) ? kClasses[classId] : "Unknown"; lua_pushstring(L, name); lua_pushstring(L, name); // WoW returns localized + English lua_pushnumber(L, classId); return 3; } lua_pushstring(L, "Unknown"); lua_pushstring(L, "Unknown"); lua_pushnumber(L, 0); return 3; } // --- Player/Game API --- static int lua_GetMoney(lua_State* L) { auto* gh = getGameHandler(L); lua_pushnumber(L, gh ? static_cast(gh->getMoneyCopper()) : 0.0); return 1; } static int lua_IsInGroup(lua_State* L) { auto* gh = getGameHandler(L); lua_pushboolean(L, gh && gh->isInGroup()); return 1; } static int lua_IsInRaid(lua_State* L) { auto* gh = getGameHandler(L); lua_pushboolean(L, gh && gh->isInGroup() && gh->getPartyData().groupType == 1); return 1; } static int lua_GetPlayerMapPosition(lua_State* L) { auto* gh = getGameHandler(L); if (gh) { const auto& mi = gh->getMovementInfo(); lua_pushnumber(L, mi.x); lua_pushnumber(L, mi.y); return 2; } lua_pushnumber(L, 0); lua_pushnumber(L, 0); return 2; } // --- Action API --- static int lua_SendChatMessage(lua_State* L) { auto* gh = getGameHandler(L); if (!gh) return 0; const char* msg = luaL_checkstring(L, 1); const char* chatType = luaL_optstring(L, 2, "SAY"); // language arg (3) ignored — server determines language const char* target = luaL_optstring(L, 4, ""); std::string typeStr(chatType); for (char& c : typeStr) c = static_cast(std::toupper(static_cast(c))); game::ChatType ct = game::ChatType::SAY; if (typeStr == "SAY") ct = game::ChatType::SAY; else if (typeStr == "YELL") ct = game::ChatType::YELL; else if (typeStr == "PARTY") ct = game::ChatType::PARTY; else if (typeStr == "GUILD") ct = game::ChatType::GUILD; else if (typeStr == "OFFICER") ct = game::ChatType::OFFICER; else if (typeStr == "RAID") ct = game::ChatType::RAID; else if (typeStr == "WHISPER") ct = game::ChatType::WHISPER; else if (typeStr == "BATTLEGROUND") ct = game::ChatType::BATTLEGROUND; std::string targetStr(target && *target ? target : ""); gh->sendChatMessage(ct, msg, targetStr); return 0; } static int lua_CastSpellByName(lua_State* L) { auto* gh = getGameHandler(L); if (!gh) return 0; const char* name = luaL_checkstring(L, 1); if (!name || !*name) return 0; // Find highest rank of spell by name (same logic as /cast) std::string nameLow(name); for (char& c : nameLow) c = static_cast(std::tolower(static_cast(c))); uint32_t bestId = 0; int bestRank = -1; for (uint32_t sid : gh->getKnownSpells()) { std::string sn = gh->getSpellName(sid); for (char& c : sn) c = static_cast(std::tolower(static_cast(c))); if (sn != nameLow) continue; int rank = 0; const std::string& rk = gh->getSpellRank(sid); if (!rk.empty()) { std::string rkl = rk; for (char& c : rkl) c = static_cast(std::tolower(static_cast(c))); if (rkl.rfind("rank ", 0) == 0) { try { rank = std::stoi(rkl.substr(5)); } catch (...) {} } } if (rank > bestRank) { bestRank = rank; bestId = sid; } } if (bestId != 0) { uint64_t target = gh->hasTarget() ? gh->getTargetGuid() : 0; gh->castSpell(bestId, target); } return 0; } static int lua_IsSpellKnown(lua_State* L) { auto* gh = getGameHandler(L); uint32_t spellId = static_cast(luaL_checknumber(L, 1)); lua_pushboolean(L, gh && gh->getKnownSpells().count(spellId)); return 1; } static int lua_GetSpellCooldown(lua_State* L) { auto* gh = getGameHandler(L); if (!gh) { lua_pushnumber(L, 0); lua_pushnumber(L, 0); return 2; } // Accept spell name or ID uint32_t spellId = 0; if (lua_isnumber(L, 1)) { spellId = static_cast(lua_tonumber(L, 1)); } else { const char* name = luaL_checkstring(L, 1); std::string nameLow(name); for (char& c : nameLow) c = static_cast(std::tolower(static_cast(c))); for (uint32_t sid : gh->getKnownSpells()) { std::string sn = gh->getSpellName(sid); for (char& c : sn) c = static_cast(std::tolower(static_cast(c))); if (sn == nameLow) { spellId = sid; break; } } } float cd = gh->getSpellCooldown(spellId); lua_pushnumber(L, 0); // start time (not tracked precisely, return 0) lua_pushnumber(L, cd); // duration remaining return 2; } static int lua_HasTarget(lua_State* L) { auto* gh = getGameHandler(L); lua_pushboolean(L, gh && gh->hasTarget()); return 1; } // --- WoW Utility Functions --- // strsplit(delimiter, str) — WoW's string split static int lua_strsplit(lua_State* L) { const char* delim = luaL_checkstring(L, 1); const char* str = luaL_checkstring(L, 2); if (!delim[0]) { lua_pushstring(L, str); return 1; } int count = 0; std::string s(str); size_t pos = 0; while (pos <= s.size()) { size_t found = s.find(delim[0], pos); if (found == std::string::npos) { lua_pushstring(L, s.substr(pos).c_str()); count++; break; } lua_pushstring(L, s.substr(pos, found - pos).c_str()); count++; pos = found + 1; } return count; } // strtrim(str) — remove leading/trailing whitespace static int lua_strtrim(lua_State* L) { const char* str = luaL_checkstring(L, 1); std::string s(str); size_t start = s.find_first_not_of(" \t\r\n"); size_t end = s.find_last_not_of(" \t\r\n"); lua_pushstring(L, (start == std::string::npos) ? "" : s.substr(start, end - start + 1).c_str()); return 1; } // wipe(table) — clear all entries from a table static int lua_wipe(lua_State* L) { luaL_checktype(L, 1, LUA_TTABLE); // Remove all integer keys int len = static_cast(lua_objlen(L, 1)); for (int i = len; i >= 1; i--) { lua_pushnil(L); lua_rawseti(L, 1, i); } // Remove all string keys lua_pushnil(L); while (lua_next(L, 1) != 0) { lua_pop(L, 1); // pop value lua_pushvalue(L, -1); // copy key lua_pushnil(L); lua_rawset(L, 1); // table[key] = nil } lua_pushvalue(L, 1); return 1; } // date(format) — safe date function (os.date was removed) static int lua_wow_date(lua_State* L) { const char* fmt = luaL_optstring(L, 1, "%c"); time_t now = time(nullptr); struct tm* tm = localtime(&now); char buf[256]; strftime(buf, sizeof(buf), fmt, tm); lua_pushstring(L, buf); return 1; } // time() — current unix timestamp static int lua_wow_time(lua_State* L) { lua_pushnumber(L, static_cast(time(nullptr))); return 1; } // Stub for GetTime() — returns elapsed seconds static int lua_wow_gettime(lua_State* L) { static auto start = std::chrono::steady_clock::now(); auto now = std::chrono::steady_clock::now(); double elapsed = std::chrono::duration(now - start).count(); lua_pushnumber(L, elapsed); return 1; } LuaEngine::LuaEngine() = default; LuaEngine::~LuaEngine() { shutdown(); } bool LuaEngine::initialize() { if (L_) return true; L_ = luaL_newstate(); if (!L_) { LOG_ERROR("LuaEngine: failed to create Lua state"); return false; } // Open safe standard libraries (no io, os, debug, package) luaopen_base(L_); luaopen_table(L_); luaopen_string(L_); luaopen_math(L_); // Remove unsafe globals from base library const char* unsafeGlobals[] = { "dofile", "loadfile", "load", "collectgarbage", "newproxy", nullptr }; for (const char** g = unsafeGlobals; *g; ++g) { lua_pushnil(L_); lua_setglobal(L_, *g); } registerCoreAPI(); registerEventAPI(); LOG_INFO("LuaEngine: initialized (Lua 5.1)"); return true; } void LuaEngine::shutdown() { if (L_) { lua_close(L_); L_ = nullptr; LOG_INFO("LuaEngine: shut down"); } } void LuaEngine::setGameHandler(game::GameHandler* handler) { gameHandler_ = handler; if (L_) { lua_pushlightuserdata(L_, handler); lua_setfield(L_, LUA_REGISTRYINDEX, "wowee_game_handler"); } } void LuaEngine::registerCoreAPI() { // Override print() to go to chat lua_pushcfunction(L_, lua_wow_print); lua_setglobal(L_, "print"); // WoW API stubs lua_pushcfunction(L_, lua_wow_message); lua_setglobal(L_, "message"); lua_pushcfunction(L_, lua_wow_gettime); lua_setglobal(L_, "GetTime"); // Unit API static const struct { const char* name; lua_CFunction func; } unitAPI[] = { {"UnitName", lua_UnitName}, {"UnitHealth", lua_UnitHealth}, {"UnitHealthMax", lua_UnitHealthMax}, {"UnitPower", lua_UnitPower}, {"UnitPowerMax", lua_UnitPowerMax}, {"UnitLevel", lua_UnitLevel}, {"UnitExists", lua_UnitExists}, {"UnitIsDead", lua_UnitIsDead}, {"UnitClass", lua_UnitClass}, {"GetMoney", lua_GetMoney}, {"IsInGroup", lua_IsInGroup}, {"IsInRaid", lua_IsInRaid}, {"GetPlayerMapPosition", lua_GetPlayerMapPosition}, {"SendChatMessage", lua_SendChatMessage}, {"CastSpellByName", lua_CastSpellByName}, {"IsSpellKnown", lua_IsSpellKnown}, {"GetSpellCooldown", lua_GetSpellCooldown}, {"HasTarget", lua_HasTarget}, // Utilities {"strsplit", lua_strsplit}, {"strtrim", lua_strtrim}, {"wipe", lua_wipe}, {"date", lua_wow_date}, {"time", lua_wow_time}, }; for (const auto& [name, func] : unitAPI) { lua_pushcfunction(L_, func); lua_setglobal(L_, name); } // WoW aliases lua_getglobal(L_, "string"); lua_getfield(L_, -1, "format"); lua_setglobal(L_, "format"); lua_pop(L_, 1); // pop string table // tinsert/tremove aliases lua_getglobal(L_, "table"); lua_getfield(L_, -1, "insert"); lua_setglobal(L_, "tinsert"); lua_getfield(L_, -1, "remove"); lua_setglobal(L_, "tremove"); lua_pop(L_, 1); // pop table // SlashCmdList table — addons register slash commands here lua_newtable(L_); lua_setglobal(L_, "SlashCmdList"); // SLASH_* globals will be set by addons, dispatched by the /run command handler } // ---- Event System ---- // Lua-side: WoweeEvents table holds { ["EVENT_NAME"] = { handler1, handler2, ... } } // RegisterEvent("EVENT", handler) adds a handler function // UnregisterEvent("EVENT", handler) removes it static int lua_RegisterEvent(lua_State* L) { const char* eventName = luaL_checkstring(L, 1); luaL_checktype(L, 2, LUA_TFUNCTION); // Get or create the WoweeEvents table lua_getglobal(L, "__WoweeEvents"); if (lua_isnil(L, -1)) { lua_pop(L, 1); lua_newtable(L); lua_pushvalue(L, -1); lua_setglobal(L, "__WoweeEvents"); } // Get or create the handler list for this event lua_getfield(L, -1, eventName); if (lua_isnil(L, -1)) { lua_pop(L, 1); lua_newtable(L); lua_pushvalue(L, -1); lua_setfield(L, -3, eventName); } // Append the handler function to the list int len = static_cast(lua_objlen(L, -1)); lua_pushvalue(L, 2); // push the handler function lua_rawseti(L, -2, len + 1); lua_pop(L, 2); // pop handler list + WoweeEvents return 0; } static int lua_UnregisterEvent(lua_State* L) { const char* eventName = luaL_checkstring(L, 1); luaL_checktype(L, 2, LUA_TFUNCTION); lua_getglobal(L, "__WoweeEvents"); if (lua_isnil(L, -1)) { lua_pop(L, 1); return 0; } lua_getfield(L, -1, eventName); if (lua_isnil(L, -1)) { lua_pop(L, 2); return 0; } // Remove matching handler from the list int len = static_cast(lua_objlen(L, -1)); for (int i = 1; i <= len; i++) { lua_rawgeti(L, -1, i); if (lua_rawequal(L, -1, 2)) { lua_pop(L, 1); // Shift remaining elements down for (int j = i; j < len; j++) { lua_rawgeti(L, -1, j + 1); lua_rawseti(L, -2, j); } lua_pushnil(L); lua_rawseti(L, -2, len); break; } lua_pop(L, 1); } lua_pop(L, 2); return 0; } void LuaEngine::registerEventAPI() { lua_pushcfunction(L_, lua_RegisterEvent); lua_setglobal(L_, "RegisterEvent"); lua_pushcfunction(L_, lua_UnregisterEvent); lua_setglobal(L_, "UnregisterEvent"); // Create the events table lua_newtable(L_); lua_setglobal(L_, "__WoweeEvents"); } void LuaEngine::fireEvent(const std::string& eventName, const std::vector& args) { if (!L_) return; lua_getglobal(L_, "__WoweeEvents"); if (lua_isnil(L_, -1)) { lua_pop(L_, 1); return; } lua_getfield(L_, -1, eventName.c_str()); if (lua_isnil(L_, -1)) { lua_pop(L_, 2); return; } int handlerCount = static_cast(lua_objlen(L_, -1)); for (int i = 1; i <= handlerCount; i++) { lua_rawgeti(L_, -1, i); if (!lua_isfunction(L_, -1)) { lua_pop(L_, 1); continue; } // Push arguments: event name first, then extra args lua_pushstring(L_, eventName.c_str()); for (const auto& arg : args) { lua_pushstring(L_, arg.c_str()); } int nargs = 1 + static_cast(args.size()); if (lua_pcall(L_, nargs, 0, 0) != 0) { const char* err = lua_tostring(L_, -1); LOG_ERROR("LuaEngine: event '", eventName, "' handler error: ", err ? err : "(unknown)"); lua_pop(L_, 1); } } lua_pop(L_, 2); // pop handler list + WoweeEvents } bool LuaEngine::dispatchSlashCommand(const std::string& command, const std::string& args) { if (!L_) return false; // Check each SlashCmdList entry: for key NAME, check SLASH_NAME1, SLASH_NAME2, etc. lua_getglobal(L_, "SlashCmdList"); if (!lua_istable(L_, -1)) { lua_pop(L_, 1); return false; } std::string cmdLower = command; for (char& c : cmdLower) c = static_cast(std::tolower(static_cast(c))); lua_pushnil(L_); while (lua_next(L_, -2) != 0) { // Stack: SlashCmdList, key, handler if (!lua_isfunction(L_, -1) || !lua_isstring(L_, -2)) { lua_pop(L_, 1); continue; } const char* name = lua_tostring(L_, -2); // Check SLASH_1 through SLASH_9 for (int i = 1; i <= 9; i++) { std::string globalName = "SLASH_" + std::string(name) + std::to_string(i); lua_getglobal(L_, globalName.c_str()); if (lua_isstring(L_, -1)) { std::string slashStr = lua_tostring(L_, -1); for (char& c : slashStr) c = static_cast(std::tolower(static_cast(c))); if (slashStr == cmdLower) { lua_pop(L_, 1); // pop global // Call the handler with args lua_pushvalue(L_, -1); // copy handler lua_pushstring(L_, args.c_str()); if (lua_pcall(L_, 1, 0, 0) != 0) { LOG_ERROR("LuaEngine: SlashCmdList['", name, "'] error: ", lua_tostring(L_, -1)); lua_pop(L_, 1); } lua_pop(L_, 3); // pop handler, key, SlashCmdList return true; } } lua_pop(L_, 1); // pop global } lua_pop(L_, 1); // pop handler, keep key for next iteration } lua_pop(L_, 1); // pop SlashCmdList return false; } bool LuaEngine::executeFile(const std::string& path) { if (!L_) return false; int err = luaL_dofile(L_, path.c_str()); if (err != 0) { const char* errMsg = lua_tostring(L_, -1); std::string msg = errMsg ? errMsg : "(unknown error)"; LOG_ERROR("LuaEngine: error loading '", path, "': ", msg); if (gameHandler_) { game::MessageChatData errChat; errChat.type = game::ChatType::SYSTEM; errChat.language = game::ChatLanguage::UNIVERSAL; errChat.message = "|cffff4040[Lua Error] " + msg + "|r"; gameHandler_->addLocalChatMessage(errChat); } lua_pop(L_, 1); return false; } return true; } bool LuaEngine::executeString(const std::string& code) { if (!L_) return false; int err = luaL_dostring(L_, code.c_str()); if (err != 0) { const char* errMsg = lua_tostring(L_, -1); std::string msg = errMsg ? errMsg : "(unknown error)"; LOG_ERROR("LuaEngine: script error: ", msg); if (gameHandler_) { game::MessageChatData errChat; errChat.type = game::ChatType::SYSTEM; errChat.language = game::ChatLanguage::UNIVERSAL; errChat.message = "|cffff4040[Lua Error] " + msg + "|r"; gameHandler_->addLocalChatMessage(errChat); } lua_pop(L_, 1); return false; } return true; } } // namespace wowee::addons