71 lines
1.5 KiB
Lua
71 lines
1.5 KiB
Lua
--- Checks uses of undeclared global variables.
|
|
-- All global variables must be 'declared' through a regular assignment
|
|
-- (even assigning nil will do) in a main chunk before being used
|
|
-- anywhere or assigned to inside a function.
|
|
-- @module pl.strict
|
|
|
|
require 'debug'
|
|
local getinfo, error, rawset, rawget = debug.getinfo, error, rawset, rawget
|
|
local handler,hooked
|
|
|
|
local mt = getmetatable(_G)
|
|
if mt == nil then
|
|
mt = {}
|
|
setmetatable(_G, mt)
|
|
elseif mt.hook then
|
|
hooked = true
|
|
end
|
|
|
|
-- predeclaring _PROMPT keeps the Lua Interpreter happy
|
|
mt.__declared = {_PROMPT=true}
|
|
|
|
local function what ()
|
|
local d = getinfo(3, "S")
|
|
return d and d.what or "C"
|
|
end
|
|
|
|
mt.__newindex = function (t, n, v)
|
|
if not mt.__declared[n] then
|
|
local w = what()
|
|
if w ~= "main" and w ~= "C" then
|
|
error("assign to undeclared variable '"..n.."'", 2)
|
|
end
|
|
mt.__declared[n] = true
|
|
end
|
|
rawset(t, n, v)
|
|
end
|
|
|
|
handler = function(t,n)
|
|
if not mt.__declared[n] and what() ~= "C" then
|
|
error("variable '"..n.."' is not declared", 2)
|
|
end
|
|
return rawget(t, n)
|
|
end
|
|
|
|
function package.strict (mod)
|
|
local mt = getmetatable(mod)
|
|
if mt == nil then
|
|
mt = {}
|
|
setmetatable(mod, mt)
|
|
end
|
|
mt.__declared = {}
|
|
mt.__newindex = function(t, n, v)
|
|
mt.__declared[n] = true
|
|
rawset(t, n, v)
|
|
end
|
|
mt.__index = function(t,n)
|
|
if not mt.__declared[n] then
|
|
error("variable '"..n.."' is not declared", 2)
|
|
end
|
|
return rawget(t, n)
|
|
end
|
|
end
|
|
|
|
if not hooked then
|
|
mt.__index = handler
|
|
else
|
|
mt.hook(handler)
|
|
end
|
|
|
|
|