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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
#include "stdafx.h"
#define MT_SCRIPT "SCRIPT"
CMLuaScript::CMLuaScript(lua_State *L, const wchar_t *path)
: CMLuaEnvironment(L), status(None), unloadRef(LUA_NOREF)
{
mir_wstrcpy(filePath, path);
fileName = wcsrchr(filePath, '\\') + 1;
wchar_t *dot = wcsrchr(fileName, '.');
size_t length = mir_wstrlen(fileName) - mir_wstrlen(dot) + 1;
ptrW name((wchar_t*)mir_calloc(sizeof(wchar_t) * (length + 1)));
mir_wstrncpy(name, fileName, length);
moduleName = mir_utf8encodeW(name);
}
CMLuaScript::CMLuaScript(const CMLuaScript &script)
: CMLuaEnvironment(script.L), status(None), unloadRef(LUA_NOREF)
{
mir_wstrcpy(filePath, script.filePath);
fileName = mir_wstrdup(script.fileName);
moduleName = mir_strdup(script.moduleName);
}
CMLuaScript::~CMLuaScript()
{
if (status == Loaded)
{
lua_rawgeti(L, LUA_REGISTRYINDEX, unloadRef);
if (lua_isfunction(L, -1))
luaM_pcall(L);
lua_pushnil(L);
lua_rawsetp(L, LUA_REGISTRYINDEX, this);
status = None;
}
luaL_getsubtable(L, LUA_REGISTRYINDEX, "_LOADED");
lua_pushnil(L);
lua_setfield(L, -2, moduleName);
lua_pop(L, 1);
mir_free(moduleName);
}
const char* CMLuaScript::GetModuleName() const
{
return moduleName;
}
const wchar_t* CMLuaScript::GetFilePath() const
{
return filePath;
}
const wchar_t* CMLuaScript::GetFileName() const
{
return fileName;
}
CMLuaScript::Status CMLuaScript::GetStatus() const
{
return status;
}
bool CMLuaScript::Load()
{
status = Failed;
if (luaL_loadfile(L, _T2A(filePath))) {
ReportError(L);
return false;
}
if (!CMLuaEnvironment::Load()) {
ReportError(L);
return false;
}
status = Loaded;
if (lua_isnoneornil(L, -1))
return true;
luaL_getsubtable(L, LUA_REGISTRYINDEX, "_LOADED");
lua_getfield(L, -1, moduleName);
if (!lua_toboolean(L, -1)) {
lua_pop(L, 1);
lua_pushvalue(L, -2);
lua_setfield(L, -2, moduleName);
lua_pop(L, 1);
}
else
lua_remove(L, -2);
if (!lua_istable(L, -1))
return true;
lua_getfield(L, -1, "Load");
if (lua_isfunction(L, -1))
luaM_pcall(L);
else
lua_pop(L, 1);
lua_getfield(L, -1, "Unload");
if (lua_isfunction(L, -1)) {
lua_pushvalue(L, -1);
unloadRef = luaL_ref(L, LUA_REGISTRYINDEX);
}
lua_pop(L, 1);
lua_pop(L, 1);
return true;
}
|
