mirror of
https://github.com/leafo/moonscript.git
synced 2024-12-24 01:04:21 +00:00
165 lines
3.2 KiB
Lua
165 lines
3.2 KiB
Lua
if not moon or not moon.inject then
|
|
module("moon", package.seeall)
|
|
end
|
|
local util = require("moonscript.util")
|
|
local lua = {
|
|
debug = debug,
|
|
type = type
|
|
}
|
|
dump = util.dump
|
|
p = function(...)
|
|
return print(dump(...))
|
|
end
|
|
is_object = function(value)
|
|
return lua.type(value) == "table" and value.__class
|
|
end
|
|
type = function(value)
|
|
local base_type = lua.type(value)
|
|
if base_type == "table" then
|
|
local cls = value.__class
|
|
if cls then
|
|
return cls
|
|
end
|
|
end
|
|
return base_type
|
|
end
|
|
debug = setmetatable({
|
|
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
|
|
}, {
|
|
__index = lua.debug
|
|
})
|
|
run_with_scope = function(fn, scope, ...)
|
|
local old_env = getfenv(fn)
|
|
local env = setmetatable({ }, {
|
|
__index = function(self, name)
|
|
local val = scope[name]
|
|
if val ~= nil then
|
|
return val
|
|
else
|
|
return old_env[name]
|
|
end
|
|
end
|
|
})
|
|
setfenv(fn, env)
|
|
return fn(...)
|
|
end
|
|
bind_methods = function(obj)
|
|
return setmetatable({ }, {
|
|
__index = function(self, name)
|
|
local val = obj[name]
|
|
if val and lua.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)
|
|
return (function()
|
|
local _tbl_0 = { }
|
|
for key, val in pairs(self) do
|
|
_tbl_0[key] = val
|
|
end
|
|
return _tbl_0
|
|
end)()
|
|
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
|
|
fold = function(items, fn)
|
|
local len = #items
|
|
if len > 1 then
|
|
local accum = fn(items[1], items[2])
|
|
for i = 3, len do
|
|
accum = fn(accum, items[i])
|
|
end
|
|
return accum
|
|
else
|
|
return items[1]
|
|
end
|
|
end
|
|
return nil
|