1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
--[[
Copyright (c) 2022, Vsevolod Stakhov <vsevolod@rspamd.com>
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
]]--
-- Controller endpoints
local local_conf = rspamd_paths['LOCAL_CONFDIR']
local local_rules = rspamd_paths['RULESDIR']
local rspamd_util = require "rspamd_util"
local lua_util = require "lua_util"
local rspamd_logger = require "rspamd_logger"
-- Define default controller paths, could be overridden in local.d/controller.lua
local controller_plugin_paths = {
maps = dofile(local_rules .. "/controller/maps.lua"),
neural = dofile(local_rules .. "/controller/neural.lua"),
selectors = dofile(local_rules .. "/controller/selectors.lua"),
fuzzy = dofile(local_rules .. "/controller/fuzzy.lua"),
}
if rspamd_util.file_exists(local_conf .. '/controller.lua') then
local controller_overrides = dofile(local_conf .. '/controller.lua')
if controller_overrides and type(controller_overrides) == 'table' then
controller_plugin_paths = lua_util.override_defaults(controller_plugin_paths, controller_overrides)
end
end
for plug, paths in pairs(controller_plugin_paths) do
if not rspamd_plugins[plug] then
rspamd_plugins[plug] = {}
end
if not rspamd_plugins[plug].webui then
rspamd_plugins[plug].webui = {}
end
local webui = rspamd_plugins[plug].webui
for path, attrs in pairs(paths) do
if type(attrs) == 'table' then
if type(attrs.handler) ~= 'function' then
rspamd_logger.infox(rspamd_config, 'controller plugin %s; webui path %s has invalid handler: %s; ignore it',
plug, path, type(attrs.handler))
else
webui[path] = lua_util.shallowcopy(attrs)
rspamd_logger.infox(rspamd_config, 'controller plugin %s; register webui path %s',
plug, path)
end
else
rspamd_logger.infox(rspamd_config, 'controller plugin %s; webui path %s has invalid type: %s; ignore it',
plug, path, type(attrs))
end
end
end
|
