if not moon or not moon.inject module "moon", package.seeall util = require "moonscript.util" lua = { :debug, :type } export * dump = util.dump p = (...) -> print dump ... is_object = (value) -> -- is a moonscript object lua.type(value) == "table" and value.__class type = (value) -> -- class aware type base_type = lua.type value if base_type == "table" cls = value.__class return cls if cls base_type debug = setmetatable { upvalue: (fn, k, v) -> upvalues = {} i = 1 while true name = lua.debug.getupvalue(fn, i) break if name == nil upvalues[name] = i i += 1 if not upvalues[k] error "Failed to find upvalue: " .. tostring k if not v _, value = lua.debug.getupvalue fn, upvalues[k] value else lua.debug.setupvalue fn, upvalues[k], v }, __index: lua.debug -- run a function with scope injected before its function environment run_with_scope = (fn, scope, ...) -> old_env = getfenv fn env = setmetatable {}, { __index: (name) => val = scope[name] if val != nil val else old_env[name] } setfenv fn, env fn ... -- wrap obj such that calls to methods do not need a reference to self bind_methods = (obj) -> setmetatable {}, { __index: (name) => val = obj[name] if val and lua.type(val) == "function" bound = (...) -> val obj, ... self[name] = bound bound else val } -- use a function to provide default values to table -- optionally specify a starting table -- fibanocci table: -- t = defaultbl {[0]: 0, [1]: 1}, (i) -> self[i - 1] + self[i - 2] defaultbl = (t, fn) -> if not fn fn = t t = {} setmetatable t, { __index: (name) => val = fn self, name rawset self, name, val val } -- chain together tables by __index metatables extend = (...) -> tbls = {...} return if #tbls < 2 for i = 1, #tbls - 1 a = tbls[i] b = tbls[i + 1] setmetatable a, __index: b tbls[1] -- shallow copy copy = => {key,val for key,val in pairs self} -- mixin class properties into self, call new mixin = (cls, ...) => meta = getmetatable cls for key, val in pairs meta.__index self[key] = val if not key\match"^__" cls.__init self, ... -- mixin methods from an object into self mixin_object = (object, methods) => for name in *methods self[name] = (parent, ...) -> object[name](object, ...) -- mixin table values into self mixin_table = (tbl, keys) => if keys for key in *keys self[key] = tbl[key] else for key, val in pairs tbl self[key] = val fold = (items, fn)-> len = #items if len > 1 accum = fn items[1], items[2] for i=3,len accum = fn acum, items[i] accum else items[1] nil