Mercurial > prosody-modules
diff mod_firewall/mod_firewall.lua @ 2366:14021c93a962
mod_firewall: Allow prefixing script paths with 'module:' to specify path relative to module file
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Tue, 15 Nov 2016 15:00:38 +0000 |
parents | 05dae9adf778 |
children | 3ebd3cb4d7d2 |
line wrap: on
line diff
--- a/mod_firewall/mod_firewall.lua Tue Nov 15 14:59:49 2016 +0000 +++ b/mod_firewall/mod_firewall.lua Tue Nov 15 15:00:38 2016 +0000 @@ -408,12 +408,21 @@ return chunk; end +local function resolve_script_path(script_path) + local relative_to = prosody.paths.config; + if script_path:match("^module:") then + relative_to = module.path:sub(1, -#("/mod_"..module.name..".lua")); + script_path = script_path:match("^module:(.+)$"); + end + return resolve_relative_path(relative_to, script_path); +end + function module.load() if not prosody.arg then return end -- Don't run in prosodyctl active_definitions = {}; local firewall_scripts = module:get_option_set("firewall_scripts", {}); for script in firewall_scripts do - script = resolve_relative_path(prosody.paths.config, script); + script = resolve_script_path(script); local chain_functions, err = compile_firewall_rules(script) if not chain_functions then @@ -459,6 +468,7 @@ end for _, filename in ipairs(arg) do + filename = resolve_script_path(filename); print("do -- File "..filename); local chain_functions = assert(compile_firewall_rules(filename)); if verbose then