59 lines
1.7 KiB
Lua
59 lines
1.7 KiB
Lua
local util = require("util")
|
|
|
|
function log(msg)
|
|
ngx.log(ngx.STDERR, tostring(msg))
|
|
end
|
|
|
|
local ctx = {}
|
|
|
|
function ctx:setWantedScripts(graph)
|
|
self._wanted_scripts = graph
|
|
end
|
|
|
|
function ctx:loadFromConfig(conf)
|
|
ctx:setWantedScripts(conf.wantedScripts)
|
|
ctx:loadChain()
|
|
end
|
|
|
|
|
|
function ctx:loadChain()
|
|
self.compiled_chain = {}
|
|
for module_name, module_config in pairs(self._wanted_scripts) do
|
|
local module = require('scripts.' .. module_name)
|
|
local module_config_readonly = table.readonly(module_config)
|
|
local module_state = module.init(module_config_readonly)
|
|
table.insert(self.compiled_chain, {module, module_config_readonly, module_state})
|
|
end
|
|
end
|
|
|
|
function ctx:onRequest()
|
|
local request_uri = ngx.var.uri
|
|
|
|
-- find out which modules to call based on their regexes
|
|
local callbacks_to_call = {}
|
|
for _, filter in ipairs(self.compiled_chain) do
|
|
local module, module_config, state = unpack(filter)
|
|
|
|
for callback_regex, callback_function in pairs(module.callbacks) do
|
|
local match = ngx.re.match(request_uri, callback_regex)
|
|
if match then
|
|
table.insert(callbacks_to_call, {module, callback_function, module_config, state})
|
|
end
|
|
end
|
|
end
|
|
|
|
for _, tuple in ipairs(callbacks_to_call) do
|
|
local module, callback_function, config, state = unpack(tuple)
|
|
local status_code, body = callback_function(config, state)
|
|
if status_code ~= nil then
|
|
log('filtered by module ' .. module.name)
|
|
ngx.status = status_code
|
|
ngx.say(body or "request denied")
|
|
ngx.exit(status_code)
|
|
return
|
|
end
|
|
end
|
|
end
|
|
|
|
return ctx
|