-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrequire.lua
128 lines (106 loc) · 3.04 KB
/
require.lua
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
119
120
121
122
123
124
125
126
127
128
--- usage:
-- require = require"require".require
-- :o)
local error, ipairs, newproxy, type = error, ipairs, newproxy, type
local t_concat = table.concat
--- Helpers
local function checkstring(s)
local t = type(s)
if t == "string" then
return s
elseif t == "number" then
return tostring(s)
else
error("bad argument #1 to 'require' (string expected, got "..t..")", 3)
end
end
--- for Lua 5.1
local package, p_loaded = package, package.loaded
local sentinel do
local function errhandler() error("the require() sentinel can't be indexed or updated", 2) end
sentinel = newproxy and newproxy() or setmetatable({}, {__index = errhandler, __newindex = errhandler, __metatable = false})
end
local function require51 (name)
name = checkstring(name)
if p_loaded[name] == sentinel then
error("loop or previous error loading module '"..name.."'", 2)
end
local module = p_loaded[name]
if module then return module end
local msg = {}
local loader
for _, searcher in ipairs(package.loaders) do
loader = searcher(name)
if type(loader) == "function" then break end
if type(loader) == "string" then
-- `loader` is actually an error message
msg[#msg + 1] = loader
end
loader = nil
end
if loader == nil then
error("module '" .. name .. "' not found: "..t_concat(msg), 2)
end
p_loaded[name] = sentinel
local res = loader(name)
if res ~= nil then
module = res
elseif p_loaded[name] == sentinel or not p_loaded[name] then
module = true
else
module = p_loaded[name]
end
p_loaded[name] = module
return module
end
--- for Lua 5.2
local function require52 (name)
name = checkstring(name)
local module = p_loaded[name]
if module then return module end
local msg = {}
local loader, param
for _, searcher in ipairs(package.searchers) do
loader, param = searcher(name)
if type(loader) == "function" then break end
if type(loader) == "string" then
-- `loader` is actually an error message
msg[#msg + 1] = loader
end
loader = nil
end
if loader == nil then
error("module '" .. name .. "' not found: "..t_concat(msg), 2)
end
local res = loader(name, param)
if res ~= nil then
module = res
elseif not p_loaded[name] then
module = true
else
module = p_loaded[name]
end
p_loaded[name] = module
return module
end
local module = {
VERSION = "0.1.7",
require51 = require51,
require52 = require52
}
if _VERSION == "Lua 5.1" then module.require = require51 end
if _VERSION == "Lua 5.2" then module.require = require52 end
--- rerequire :o)
for _, o in ipairs{
{"rerequiredefault", require},
{"rerequire", module.require},
{"rerequire51", require51},
{"rerequire52", require52}
} do
local rereq, req = o[1], o[2]
module[rereq] = function(name)
p_loaded[name] = nil
return req(name)
end
end
return module