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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
local exports = {}
local function rspamd_map_add(mname, optname, mtype, description)
local ret = {
get_key = function(t, k)
if t.__data then
return t.__data:get_key(k)
end
return nil
end
}
local ret_mt = {
__index = function(t, k)
if t.__data then
return t.get_key(k)
end
return nil
end
}
local opt = rspamd_config:get_module_opt(mname, optname)
if not opt then
return nil
end
if type(opt) == 'string' then
-- We have a single string, so we treat it as a map
local map = rspamd_config:add_map{
type = mtype,
description = description,
url = opt,
}
if map then
ret.__data = map
setmetatable(ret, ret_mt)
return ret
end
elseif type(opt) == 'table' then
-- it might be plain map or map of plain elements
if opt[1] then
if mtype == 'radix' then
if string.find(opt[1], '^%d') then
local map = rspamd_config:radix_from_config(mname, optname)
if map then
ret.__data = map
setmetatable(ret, ret_mt)
return ret
end
else
-- Plain table
local map = rspamd_config:add_map{
type = mtype,
description = description,
url = opt,
}
if map then
ret.__data = map
setmetatable(ret, ret_mt)
return ret
end
end
elseif mtype == 'regexp' then
-- Plain table
local map = rspamd_config:add_map{
type = mtype,
description = description,
url = opt,
}
if map then
ret.__data = map
setmetatable(ret, ret_mt)
return ret
end
else
if string.find(opt[1], '^/%a') or string.find(opt[1], '^http') then
-- Plain table
local map = rspamd_config:add_map{
type = mtype,
description = description,
url = opt,
}
if map then
ret.__data = map
setmetatable(ret, ret_mt)
return ret
end
else
local data = {}
local nelts = 0
for _,elt in ipairs(opt) do
if type(elt) == 'string' then
data[elt] = true
nelts = nelts + 1
end
end
if nelts > 0 then
ret.__data = data
ret.get_key = function(t, k)
if k ~= '__data' then
return t.__data[k]
end
return nil
end
return ret
end
end
end
else
local map = rspamd_config:add_map{
type = mtype,
description = description,
url = opt,
}
if map then
ret.__data = map
setmetatable(ret, ret_mt)
return ret
end
end
end
return nil
end
exports.rspamd_map_add = rspamd_map_add
return exports
|
