1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
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;
}
|
