diff --git a/mesecons/util.lua b/mesecons/util.lua index d5d42ef..5f957b8 100644 --- a/mesecons/util.lua +++ b/mesecons/util.lua @@ -201,9 +201,8 @@ function mesecon.tablecopy_change_env(t, env, seen) seen[t] = n for k, v in pairs(t) do if type(v) == "function" then - local newfunc = v - setfenv(newfunc, env) - n[(type(k) == "table" and (seen[k] or mesecon.tablecopy_change_env(k, env, seen))) or k] = newfunc + setfenv(v, env) + n[(type(k) == "table" and (seen[k] or mesecon.tablecopy_change_env(k, env, seen))) or k] = v else n[(type(k) == "table" and (seen[k] or mesecon.tablecopy_change_env(k, env, seen))) or k] = (type(v) == "table" and (seen[v] or mesecon.tablecopy_change_env(v, env, seen))) or v diff --git a/mesecons_luacontroller/init.lua b/mesecons_luacontroller/init.lua index 44880c9..d9bafb5 100644 --- a/mesecons_luacontroller/init.lua +++ b/mesecons_luacontroller/init.lua @@ -459,12 +459,20 @@ local function get_digiline_send(pos, itbl, send_warning) end end +-- Mods can place their own "libraries" in here to be loaded via require() from in a Luacontroller. +-- These can take two different forms: +-- Function (recommended for libraries adding new functionality): A function that, when called, returns something that will be passed to the LuaC code. +-- Function signature is getlibrary(env, pos) where 'env' is the environment that the Luacontroller code is running in, and 'pos' is the position of the controller. +-- Table (recommended for libraries containing mostly lookup tables): A table that will be copied, and the copy returned to the LuaC code. +-- When using the table format, any functions in the table will have their environment changed to that of the Luacontroller. mesecon.luacontroller_libraries = {} -local function get_require(env) +local function get_require(pos, env) return function(name) - if mesecon.luacontroller_libraries[name] then - return mesecon.tablecopy_change_env(mesecon.luacontroller_libraries[name],env) + if type(mesecon.luacontroller_libraries[name]) == "function" then + return mesecon.luacontroller_libraries[name](env, pos) + elseif type(mesecon.luacontroller_libraries[name]) == "table" then + return mesecon.tablecopy_change_env(mesecon.luacontroller_libraries[name], env) end end end @@ -557,7 +565,7 @@ local function create_environment(pos, mem, event, itbl, send_warning) env[name] = _G[name] end - env.require = get_require(env) + env.require = get_require(pos, env) return env end