#include "stdafx.h" int g_hMLuaLangpack; CMPlugin g_plugin; PLUGININFOEX pluginInfoEx = { sizeof(PLUGININFOEX), __PLUGIN_NAME, PLUGIN_MAKE_VERSION(__MAJOR_VERSION, __MINOR_VERSION, __RELEASE_NUM, __BUILD_NUM), __DESCRIPTION, __AUTHOR, __COPYRIGHT, __AUTHORWEB, UNICODE_AWARE, // {27d41d81-991f-4dc6-8749-b0321c87e694} { 0x27d41d81, 0x991f, 0x4dc6,{ 0x87, 0x49, 0xb0, 0x32, 0x1c, 0x87, 0xe6, 0x94 } } }; static int ScriptsCompare(const CMLuaScript* p1, const CMLuaScript* p2) { return mir_wstrcmpi(p1->GetName(), p2->GetName()); } CMPlugin::CMPlugin() : PLUGIN(MODULENAME, pluginInfoEx), L(nullptr), m_scripts(1, ScriptsCompare) { MUUID muidLast = MIID_LAST; g_hMLuaLangpack = GetPluginLangId(muidLast, 0); RegisterProtocol(PROTOTYPE_FILTER); CreatePluginService(MS_LUA_CALL, &CMPlugin::Call); CreatePluginService(MS_LUA_EXEC, &CMPlugin::Exec); CreatePluginService(MS_LUA_EVAL, &CMPlugin::Eval); } void CMPlugin::LoadLuaScripts() { CMLuaScriptLoader::Load(L, m_scripts); } void CMPlugin::UnloadLuaScripts() { for (auto &script : m_scripts.rev_iter()) { script->Unload(); delete m_scripts.removeItem(&script); } } void CMPlugin::LoadLua() { Log("Loading lua engine"); L = luaL_newstate(); Log("Loading standard modules"); luaL_openlibs(L); lua_atpanic(L, luaM_atpanic); CMLuaFunctionLoader::Load(L); CMLuaModuleLoader::Load(L); CMLuaVariablesLoader::Load(L); LoadLuaScripts(); } void CMPlugin::UnloadLua() { UnloadLuaScripts(); if (L != nullptr) { KillObjectEventHooks(L); KillObjectServices(L); Log("Unloading lua engine"); lua_close(L); } KillModuleIcons(this); KillModuleSounds(this); KillModuleMenus(this); KillModuleHotkeys(this); } void CMPlugin::ReloadLuaScripts() { UnloadLuaScripts(); LoadLuaScripts(); } /***********************************************/ int CMPlugin::OnOptionsInit(WPARAM wParam, LPARAM) { OPTIONSDIALOGPAGE odp = {}; odp.flags = ODPF_BOLDGROUPS | ODPF_UNICODE | ODPF_DONTTRANSLATE; odp.szGroup.w = LPGENW("Services"); odp.szTitle.w = L"Lua"; odp.szTab.w = LPGENW("Scripts"); odp.pDialog = new CMLuaOptionsMain(*this); g_plugin.addOptions(wParam, &odp); odp.szTab.w = LPGENW("Evaluate"); odp.pDialog = new CMLuaEvaluateOptions(*this); g_plugin.addOptions(wParam, &odp); return 0; } int CMPlugin::OnModulesLoaded(WPARAM, LPARAM) { g_hCLibsFolder = FoldersRegisterCustomPathT(MODULENAME, "CLibsFolder", MIRLUA_PATHT, TranslateT("C libs folder")); g_hScriptsFolder = FoldersRegisterCustomPathT(MODULENAME, "ScriptsFolder", MIRLUA_PATHT, TranslateT("Scripts folder")); HookPluginEvent(ME_OPT_INITIALISE, &CMPlugin::OnOptionsInit); return 0; } int CMPlugin::Load() { LoadIcons(); LoadNetlib(); LoadLua(); HookPluginEvent(ME_SYSTEM_MODULESLOADED, &CMPlugin::OnModulesLoaded); return 0; } int CMPlugin::Unload() { UnloadLua(); UnloadNetlib(); return 0; } /***********************************************/ static int mlua_call(lua_State *L) { const char *module = luaL_checkstring(L, -3); const char *function = luaL_checkstring(L, -2); if (module && module[0]) { lua_getglobal(L, "require"); lua_pushstring(L, module); lua_pcall(L, 1, 1, 0); lua_getfield(L, -1, function); lua_replace(L, -2); } else lua_getglobal(L, function); lua_pcall(L, 0, 1, 0); return 1; } INT_PTR CMPlugin::Call(WPARAM wParam, LPARAM lParam) { const wchar_t *module = (const wchar_t*)wParam; const wchar_t *function = (const wchar_t*)lParam; lua_pushstring(L, T2Utf(module)); lua_pushstring(L, T2Utf(function)); lua_pushcfunction(L, mlua_call); CMLuaEnvironment env(L); wchar_t *result = env.Call(); env.Unload(); return (INT_PTR)result; } INT_PTR CMPlugin::Eval(WPARAM, LPARAM lParam) { const wchar_t *script = (const wchar_t*)lParam; CMLuaEnvironment env(L); wchar_t *result = env.Eval(script); env.Unload(); return (INT_PTR)result; } INT_PTR CMPlugin::Exec(WPARAM, LPARAM lParam) { const wchar_t *path = (const wchar_t*)lParam; CMLuaEnvironment env(L); wchar_t *result = env.Exec(path); env.Unload(); return (INT_PTR)result; }