moonscript/moon/init.lua

134 lines
2.6 KiB
Lua
Raw Normal View History

2011-10-27 07:03:38 +00:00
if not moon or not moon.inject then
module("moon", package.seeall)
end
local util = require("moonscript.util")
2011-11-15 03:16:26 +00:00
local lua = {
debug = debug
}
2011-10-27 07:03:38 +00:00
dump = util.dump
2011-11-21 01:52:19 +00:00
p = function(...)
return print(dump(...))
end
2011-11-15 03:16:26 +00:00
debug = {
upvalue = function(fn, k, v)
local upvalues = { }
local i = 1
while true do
local name = lua.debug.getupvalue(fn, i)
if name == nil then
break
end
upvalues[name] = i
i = i + 1
end
if not upvalues[k] then
error("Failed to find upvalue: " .. tostring(k))
end
if not v then
local _, value = lua.debug.getupvalue(fn, upvalues[k])
return value
else
return lua.debug.setupvalue(fn, upvalues[k], v)
end
end
}
run_with_scope = function(fn, scope, ...)
2011-10-27 07:03:38 +00:00
local old_env = getfenv(fn)
local env = setmetatable({ }, {
__index = function(self, name)
local val = scope[name]
if val ~= nil then
2011-10-27 07:03:38 +00:00
return val
else
return old_env[name]
end
end
})
setfenv(fn, env)
return fn(...)
2011-10-27 07:03:38 +00:00
end
bind_methods = function(obj)
2011-10-27 07:03:38 +00:00
return setmetatable({ }, {
__index = function(self, name)
local val = obj[name]
if val and type(val) == "function" then
local bound
bound = function(...)
return val(obj, ...)
end
self[name] = bound
return bound
else
return val
end
end
})
end
defaultbl = function(t, fn)
if not fn then
fn = t
t = { }
end
return setmetatable(t, {
__index = function(self, name)
local val = fn(self, name)
rawset(self, name, val)
return val
end
})
end
extend = function(...)
local tbls = {
...
}
if #tbls < 2 then
return
end
for i = 1, #tbls - 1 do
local a = tbls[i]
local b = tbls[i + 1]
setmetatable(a, {
__index = b
})
end
return tbls[1]
end
copy = function(self)
local t = { }
for key, val in pairs(self) do
t[key] = val
end
return t
end
mixin = function(self, cls, ...)
local meta = getmetatable(cls)
for key, val in pairs(meta.__index) do
if not key:match("^__") then
self[key] = val
end
end
return cls.__init(self, ...)
end
mixin_object = function(self, object, methods)
local _list_0 = methods
for _index_0 = 1, #_list_0 do
local name = _list_0[_index_0]
self[name] = function(parent, ...)
return object[name](object, ...)
end
end
end
mixin_table = function(self, tbl, keys)
if keys then
local _list_0 = keys
for _index_0 = 1, #_list_0 do
local key = _list_0[_index_0]
self[key] = tbl[key]
end
else
for key, val in pairs(tbl) do
self[key] = val
end
end
end