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
|
--[[
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.
]]--
local argparse = require "argparse"
-- Define command line options
local parser = argparse()
:name "rspamadm template"
:description "Apply jinja templates for strings/files"
:help_description_margin(30)
parser:argument "file"
:description "File to process"
:argname "<file>"
:args "*"
parser:flag "-n --no-vars"
:description "Don't add Rspamd internal variables"
parser:option "-e --env"
:description "Load additional environment vars from specific file (name=value)"
:argname "<filename>"
:count "*"
parser:option "-l --lua-env"
:description "Load additional environment vars from specific file (lua source)"
:argname "<filename>"
:count "*"
parser:mutex(
parser:option "-s --suffix"
:description "Store files with the new suffix"
:argname "<suffix>",
parser:flag "-i --inplace"
:description "Replace input file(s)"
)
local lua_util = require "lua_util"
local function set_env(opts, env)
if opts.env then
for _, fname in ipairs(opts.env) do
for kv in assert(io.open(fname)):lines() do
if not kv:match('%s*#.*') then
local k, v = kv:match('([^=%s]+)%s*=%s*(.+)')
if k and v then
env[k] = v
else
io.write(string.format('invalid env line in %s: %s\n', fname, kv))
end
end
end
end
end
if opts.lua_env then
for _, fname in ipairs(opts.env) do
local ret, res_or_err = pcall(loadfile(fname))
if not ret then
io.write(string.format('cannot load %s: %s\n', fname, res_or_err))
else
if type(res_or_err) == 'table' then
for k, v in pairs(res_or_err) do
env[k] = lua_util.deepcopy(v)
end
else
io.write(string.format('cannot load %s: not a table\n', fname))
end
end
end
end
end
local function read_file(file)
local content
if file == '-' then
content = io.read("*all")
else
local f = assert(io.open(file, "rb"))
content = f:read("*all")
f:close()
end
return content
end
local function handler(args)
local opts = parser:parse(args)
local env = {}
set_env(opts, env)
if not opts.file or #opts.file == 0 then
opts.file = { '-' }
end
for _, fname in ipairs(opts.file) do
local content = read_file(fname)
local res = lua_util.jinja_template(content, env, opts.no_vars)
if opts.inplace then
local nfile = string.format('%s.new', fname)
local out = assert(io.open(nfile, 'w'))
out:write(content)
out:close()
os.rename(nfile, fname)
elseif opts.suffix then
local nfile = string.format('%s.%s', opts.suffix)
local out = assert(io.open(nfile, 'w'))
out:write(content)
out:close()
else
io.write(res)
end
end
end
return {
handler = handler,
description = parser._description,
name = 'template'
}
|
