8d43b8a0b1d0f7de87ea6d62039017057047b3d1
[project/luci.git] / applications / luci-statistics / root / usr / bin / stat-genconfig
1 #!/usr/bin/lua
2
3 --[[
4
5 Luci statistics - collectd configuration generator
6 (c) 2008 Freifunk Leipzig / Jo-Philipp Wich <xm@leipzig.freifunk.net>
7
8 Licensed under the Apache License, Version 2.0 (the "License");
9 you may not use this file except in compliance with the License.
10 You may obtain a copy of the License at
11
12 http://www.apache.org/licenses/LICENSE-2.0
13
14 $Id$
15
16 ]]--
17
18
19 require("luci.model.uci")
20 require("luci.sys.iptparser")
21 require("luci.util")
22
23 local ipt = luci.sys.iptparser.IptParser()
24 local uci = luci.model.uci.cursor()
25 local sections = uci:get_all( "luci_statistics" )
26
27
28 function print(...)
29 nixio.stdout:write(...)
30 nixio.stdout:write("\n")
31 end
32
33 function section( plugin )
34
35 local config = sections[ "collectd_" .. plugin ] or sections["collectd"]
36
37 if type(config) == "table" and ( plugin == "collectd" or config.enable == "1" ) then
38
39 local params = ""
40
41 if type( plugins[plugin] ) == "function" then
42 params = plugins[plugin]( config )
43 else
44 params = config_generic( config, plugins[plugin][1], plugins[plugin][2], plugins[plugin][3], plugin == "collectd" )
45 end
46
47
48 if plugin ~= "collectd" then
49 print( "LoadPlugin " .. plugin )
50
51 if params:len() > 0 then
52 print( "<Plugin " .. plugin .. ">\n" .. params .. "</Plugin>\n" )
53 else
54 print( "" )
55 end
56 else
57 print( params .. "\n" )
58 end
59 end
60 end
61
62 function config_generic( c, singles, bools, lists, nopad )
63 local str = ""
64
65 if type(c) == "table" then
66
67 if type(singles) == "table" then
68 for i, key in ipairs( singles ) do
69 if preprocess[key] then
70 c[key] = preprocess[key](c[key])
71 end
72
73 str = str .. _string( c[key], key, nopad )
74 end
75 end
76
77 if type(bools) == "table" then
78 for i, key in ipairs( bools ) do
79 if preprocess[key] then
80 c[key] = preprocess[key](c[key])
81 end
82
83 str = str .. _bool( c[key], key, nopad )
84 end
85 end
86
87 if type(lists) == "table" then
88 str = str .. _list_expand( c, lists, nopad )
89 end
90 end
91
92 return str
93 end
94
95 function config_exec( c )
96 local str = ""
97
98 for s in pairs(sections) do
99 for key, type in pairs({ Exec="collectd_exec_input", NotificationExec="collectd_exec_notify" }) do
100 if sections[s][".type"] == type then
101
102 cmd = sections[s].cmdline
103
104 if cmd then
105 cmd = cmd:gsub("^%s+", ""):gsub("%s+$", "")
106 user = sections[s].cmduser or "nobody"
107 group = sections[s].cmdgroup
108
109 str = str .. "\t" .. key .. ' "' ..
110 user .. ( group and ":" .. group or "" ) .. '" "' ..
111 cmd:gsub('%s+', '" "') .. '"\n'
112 end
113 end
114 end
115 end
116
117 return str
118 end
119
120 function config_iptables( c )
121 local str = ""
122
123 for s in pairs(sections) do
124 if sections[s][".type"] == "collectd_iptables_match" then
125
126 search = { }
127
128 for i, k in ipairs( {
129 "table", "chain", "target", "protocol", "source", "destination",
130 "inputif", "outputif", "options"
131 } ) do
132 v = sections[s][k]
133
134 if type(v) == "string" then
135 if k == "options" then v = luci.util.split( v, "%s+", nil, true ) end
136 search[k] = v
137 end
138 end
139
140 for i, rule in ipairs( ipt:find( search ) ) do
141
142 name = sections[s].name:gsub( "%s+", "_" )
143 if i > 1 then name = name .. "_(" .. i .. ")" end
144
145 str = str .. "\tChain " .. rule.table .. " " .. rule.chain .. " " .. rule.index .. ' "' .. name .. "\"\n"
146 end
147 end
148 end
149
150 return str
151 end
152
153 function config_network( c )
154 local str = ""
155
156 for s in pairs(sections) do
157 for key, type in pairs({ Listen="collectd_network_listen", Server="collectd_network_server" }) do
158 if sections[s][".type"] == type then
159
160 host = sections[s].host
161 port = sections[s].port
162
163 if host then
164 if port then
165 str = str .. "\t" .. key .. " \"" .. host .. "\" \"" .. port .. "\"\n"
166 else
167 str = str .. "\t" .. key .. " \"" .. host .. "\"\n"
168 end
169 end
170 end
171 end
172 end
173
174 return str .. _string( c["TimeToLive"], "TimeToLive" )
175 .. _string( c["CacheFlush"], "CacheFlush" )
176 .. _bool( c["Forward"], "Forward" )
177 end
178
179
180 function _list_expand( c, l, nopad )
181 local str = ""
182
183 for i, n in ipairs(l) do
184 if c[n] then
185 if preprocess[n] then
186 c[n] = preprocess[n](c[n])
187 end
188
189 if n:find("(%w+)ses") then
190 k = n:gsub("(%w+)ses", "%1s")
191 else
192 k = n:gsub("(%w+)s", "%1")
193 end
194
195 str = str .. _expand( c[n], k, nopad )
196 end
197 end
198
199 return str
200 end
201
202 function _expand( s, n, nopad )
203 if type(s) == "string" then
204 local str = ""
205
206 for i, v in ipairs( luci.util.split( s, "%s+", nil, true ) ) do
207 str = str .. _string( v, n, nopad )
208 end
209
210 return str
211 end
212 end
213
214 function _bool( s, n, nopad )
215
216 local str = ""
217 local pad = ""
218 if not nopad then pad = "\t" end
219
220 if s and s == "1" then
221 str = pad .. n .. " true"
222 else
223 str = pad .. n .. " false"
224 end
225
226 return str .. "\n"
227 end
228
229 function _string( s, n, nopad )
230
231 local str = ""
232 local pad = ""
233 if not nopad then pad = "\t" end
234
235 if s then
236 if s:find("[^%d]") or n == "Port" then
237 if not s:find("[^%w]") and n ~= "Port" then
238 str = pad .. n .. " " .. luci.util.trim(s)
239 else
240 str = pad .. n .. ' "' .. luci.util.trim(s) .. '"'
241 end
242 else
243 str = pad .. n .. " " .. luci.util.trim(s)
244 end
245
246 str = str .. "\n"
247 end
248
249 return str
250 end
251
252
253 plugins = {
254 collectd = {
255 { "BaseDir", "Include", "PIDFile", "PluginDir", "TypesDB", "Interval", "ReadThreads", "Hostname" },
256 { },
257 { }
258 },
259
260 conntrack = {
261 { },
262 { },
263 { }
264 },
265
266 cpu = {
267 { },
268 { },
269 { }
270 },
271
272 csv = {
273 { "DataDir" },
274 { "StoreRates" },
275 { }
276 },
277
278 df = {
279 { },
280 { "IgnoreSelected" },
281 { "Devices", "MountPoints", "FSTypes" }
282 },
283
284 disk = {
285 { },
286 { "IgnoreSelected" },
287 { "Disks" }
288 },
289
290 dns = {
291 { },
292 { },
293 { "Interfaces", "IgnoreSources" }
294 },
295
296 email = {
297 { "SocketFile", "SocketGroup", "SocketPerms", "MaxConns" },
298 { },
299 { }
300 },
301
302 exec = config_exec,
303
304 interface = {
305 { },
306 { "IgnoreSelected" },
307 { "Interfaces" }
308 },
309
310 iptables = config_iptables,
311
312 irq = {
313 { },
314 { "IgnoreSelected" },
315 { "Irqs" }
316 },
317
318 load = {
319 { },
320 { },
321 { }
322 },
323
324 logfile = {
325 { "LogLevel", "File" },
326 { "Timestamp" },
327 { }
328 },
329
330 madwifi = {
331 { "WatchSet" },
332 { },
333 { "Interfaces", "WatchAdds" }
334 },
335
336 memory = {
337 { },
338 { },
339 { }
340 },
341
342 netlink = {
343 { },
344 { "IgnoreSelected" },
345 { "Interfaces", "VerboseInterfaces", "QDiscs", "Classes", "Filters" }
346 },
347
348 network = config_network,
349
350 olsrd = {
351 { "Host", "Port", "CollectLinks","CollectRoutes","CollectTopology"},
352 { },
353 { }
354 },
355
356 ping = {
357 { "TTL", "Interval" },
358 { },
359 { "Hosts" }
360 },
361
362 processes = {
363 { },
364 { },
365 { "Processes" }
366 },
367
368 rrdtool = {
369 { "DataDir", "StepSize", "HeartBeat", "RRARows", "XFF", "CacheFlush", "CacheTimeout" },
370 { "RRASingle" },
371 { "RRATimespans" }
372 },
373
374 tcpconns = {
375 { },
376 { "ListeningPorts" },
377 { "LocalPorts", "RemotePorts" }
378 },
379
380 unixsock = {
381 { "SocketFile", "SocketGroup", "SocketPerms" },
382 { },
383 { }
384 },
385
386 wireless = {
387 { },
388 { },
389 { }
390 },
391 }
392
393 preprocess = {
394 RRATimespans = function(val)
395 local rv = { }
396 for time in val:gmatch("[^%s]+") do
397 table.insert( rv, luci.util.parse_units(time) )
398 end
399 return table.concat(rv, " ")
400 end
401 }
402
403
404 section("collectd")
405
406 for plugin in pairs(plugins) do
407 if plugin ~= "collectd" then
408 section( plugin )
409 end
410 end