1 files changed, 101 insertions, 0 deletions
diff --git a/plugins/MirLua/src/environment.cpp b/plugins/MirLua/src/environment.cpp
new file mode 100644
index 0000000000..4a82e63231
--- /dev/null
+++ b/plugins/MirLua/src/environment.cpp
@@ -0,0 +1,101 @@
+#include "stdafx.h"
+
+#define MT_ENVIRONMENT "ENVIRONMENT"
+
+CMLuaEnvironment::CMLuaEnvironment(lua_State *L)
+	: L(L)
+{
+	MUUID muidLast = MIID_LAST;
+	m_id = GetPluginLangId(muidLast, 0);
+}
+
+CMLuaEnvironment::~CMLuaEnvironment()
+{
+	KillModuleIcons(m_id);
+	KillModuleSounds(m_id);
+	KillModuleMenus(m_id);
+	KillModuleHotkeys(m_id);
+
+	KillObjectEventHooks(this);
+	KillObjectServices(this);
+
+	for (auto &it : m_hookRefs)
+		luaL_unref(L, LUA_REGISTRYINDEX, it.second);
+
+	for (auto &it : m_serviceRefs)
+		luaL_unref(L, LUA_REGISTRYINDEX, it.second);
+}
+
+CMLuaEnvironment* CMLuaEnvironment::GetEnvironment(lua_State *L)
+{
+	if (!luaM_getenv(L))
+		return nullptr;
+
+	lua_rawgeti(L, -1, NULL);
+	CMLuaEnvironment *env = (CMLuaEnvironment*)lua_touserdata(L, -1);
+	lua_pop(L, 3);
+
+	return env;
+}
+
+int CMLuaEnvironment::GetEnvironmentId(lua_State *L)
+{
+	CMLuaEnvironment *script = GetEnvironment(L);
+	return script != nullptr
+		? script->GetId()
+		: hMLuaLangpack;
+}
+
+int CMLuaEnvironment::GetId() const
+{
+	return m_id;
+}
+
+void CMLuaEnvironment::AddHookRef(HANDLE h, int ref)
+{
+	m_hookRefs[h] = ref;
+}
+
+void CMLuaEnvironment::ReleaseHookRef(HANDLE h)
+{
+	auto it = m_hookRefs.find(h);
+	if (it != m_hookRefs.end())
+		luaL_unref(L, LUA_REGISTRYINDEX, it->second);
+}
+
+void CMLuaEnvironment::AddServiceRef(HANDLE h, int ref)
+{
+	m_serviceRefs[h] = ref;
+}
+
+void CMLuaEnvironment::ReleaseServiceRef(HANDLE h)
+{
+	auto it = m_serviceRefs.find(h);
+	if (it != m_serviceRefs.end())
+		luaL_unref(L, LUA_REGISTRYINDEX, it->second);
+}
+
+void CMLuaEnvironment::CreateEnvironmentTable()
+{
+	lua_createtable(L, 1, 1);
+	lua_pushlightuserdata(L, this);
+	lua_rawseti(L, -2, NULL);
+	lua_pushvalue(L, -1);
+	lua_setfield(L, -2, "_G");
+	lua_createtable(L, 0, 2);
+	lua_pushliteral(L, MT_ENVIRONMENT);
+	lua_setfield(L, -2, "__metatable");
+	lua_getglobal(L, "_G");
+	lua_setfield(L, -2, "__index");
+	lua_setmetatable(L, -2);
+}
+
+bool CMLuaEnvironment::Load()
+{
+	luaL_checktype(L, -1, LUA_TFUNCTION);
+
+	CreateEnvironmentTable();
+	lua_setupvalue(L, -2, 1);
+
+	return lua_pcall(L, 0, 1, 0) == LUA_OK;
+}