#!/usr/bin/lua

--[[

Luci statistics - collectd configuration generator
(c) 2008 Freifunk Leipzig / Jo-Philipp Wich <xm@leipzig.freifunk.net>

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

$Id$

]]--


require("luci.model.uci")
require("luci.sys.iptparser")
require("luci.util")

local ipt = luci.sys.iptparser.IptParser()
local uci = luci.model.uci.cursor()
local sections = uci:get_all( "luci_statistics" )


function print(...)
	nixio.stdout:write(...)
	nixio.stdout:write("\n")
end

function section( plugin )

	local config = sections[ "collectd_" .. plugin ] or sections["collectd"]

	if type(config) == "table" and ( plugin == "collectd" or config.enable == "1" ) then

		local params = ""

		if type( plugins[plugin] ) == "function" then
			params = plugins[plugin]( config )
		else
			params = config_generic( config, plugins[plugin][1], plugins[plugin][2], plugins[plugin][3], plugin == "collectd" )
		end


		if plugin ~= "collectd" then
			print( "LoadPlugin " .. plugin )

			if params:len() > 0 then
				print( "<Plugin " .. plugin .. ">\n" .. params .. "</Plugin>\n" )
			else
				print( "" )
			end
		else
			print( params .. "\n" )
		end
	end
end

function config_generic( c, singles, bools, lists, nopad )
	local str = ""

	if type(c) == "table" then

		if type(singles) == "table" then
			for i, key in ipairs( singles ) do
				if preprocess[key] then
					c[key] = preprocess[key](c[key])
				end

				str = str .. _string( c[key], key, nopad )
			end
		end

		if type(bools) == "table" then
			for i, key in ipairs( bools ) do
				if preprocess[key] then
					c[key] = preprocess[key](c[key])
				end

				str = str .. _bool( c[key], key, nopad )
			end
		end

		if type(lists) == "table" then
			str = str .. _list_expand( c, lists, nopad )
		end
	end

	return str
end

function config_exec( c )
	local str = ""

	for s in pairs(sections) do
		for key, type in pairs({ Exec="collectd_exec_input", NotificationExec="collectd_exec_notify" }) do
			if sections[s][".type"] == type then

				cmd = sections[s].cmdline

				if cmd then
					cmd   = cmd:gsub("^%s+", ""):gsub("%s+$", "")
					user  = sections[s].cmduser  or "nobody"
					group = sections[s].cmdgroup

					str = str .. "\t" .. key .. ' "' ..
						user .. ( group and ":" .. group or "" ) .. '" "' ..
						cmd:gsub('%s+', '" "') .. '"\n'
				end
			end
		end
	end

	return str
end

function config_iptables( c )
	local str = ""

	for s in pairs(sections) do
		if sections[s][".type"] == "collectd_iptables_match" then

			search = { }

			for i, k in ipairs( {
				"table", "chain", "target", "protocol", "source", "destination",
				"inputif", "outputif", "options"
			} ) do
				v = sections[s][k]

				if type(v) == "string" then
					if k == "options" then v = luci.util.split( v, "%s+", nil, true ) end
					search[k] = v
				end
			end

			for i, rule in ipairs( ipt:find( search ) ) do

				name = sections[s].name:gsub( "%s+", "_" )
				if i > 1 then name = name .. "_(" .. i .. ")" end

				str = str .. "\tChain " .. rule.table .. " " .. rule.chain .. " " .. rule.index .. ' "' .. name .. "\"\n"
			end
		end
	end

	return str
end

function config_network( c )
	local str = ""

	for s in pairs(sections) do
		for key, type in pairs({ Listen="collectd_network_listen", Server="collectd_network_server" }) do
			if sections[s][".type"] == type then

				host = sections[s].host
				port = sections[s].port

				if host then
					if port then
						str = str .. "\t" .. key .. " \"" .. host .. "\" \"" .. port .. "\"\n"
					else
						str = str .. "\t" .. key .. " \"" .. host .. "\"\n"
					end
				end
						end
				end
		end

	return str .. _string( c["TimeToLive"], "TimeToLive" )
		   .. _string( c["CacheFlush"], "CacheFlush" )
		   .. _bool(   c["Forward"],    "Forward"    )
end


function _list_expand( c, l, nopad )
	local str = ""

	for i, n in ipairs(l) do
		if c[n] then
			if preprocess[n] then
				c[n] = preprocess[n](c[n])
			end

			if n:find("(%w+)ses") then
				k = n:gsub("(%w+)ses", "%1s")
			else
				k = n:gsub("(%w+)s", "%1")
			end

			str = str .. _expand( c[n], k, nopad )
		end
	end

	return str
end

function _expand( s, n, nopad )
	if type(s) == "string" then
		local str = ""

		for i, v in ipairs( luci.util.split( s, "%s+", nil, true ) ) do
			str = str .. _string( v, n, nopad )
		end

		return str
	end
end

function _bool( s, n, nopad )

	local str = ""
	local pad = ""
	if not nopad then pad = "\t" end

	if s and s == "1" then
		str = pad .. n .. " true"
	else
		str = pad .. n .. " false"
	end

	return str .. "\n"
end

function _string( s, n, nopad )

	local str = ""
	local pad = ""
	if not nopad then pad = "\t" end

	if s then
		if s:find("[^%d]") then
			if not s:find("[^%w]") then
				str = pad .. n .. " " .. luci.util.trim(s)
			else
				str = pad .. n .. ' "' .. luci.util.trim(s) .. '"'
			end
		else
			str = pad .. n .. " " .. luci.util.trim(s)
		end

		str = str .. "\n"
	end

	return str
end


plugins = {
	collectd = {
		{ "BaseDir", "Include", "PIDFile", "PluginDir", "TypesDB", "Interval", "ReadThreads", "Hostname" },
		{ },
		{ }
	},

	cpu	= {
		{ },
		{ },
		{ }
	},

	csv	= {
		{ "DataDir" },
		{ "StoreRates" },
		{ }
	},

	df	= {
		{ },
		{ "IgnoreSelected" },
		{ "Devices", "MountPoints", "FSTypes" }
	},

	disk	= {
		{ },
		{ "IgnoreSelected" },
		{ "Disks" }
	},

	dns	= {
		{ },
		{ },
		{ "Interfaces", "IgnoreSources" }
	},

	email	= {
		{ "SocketFile", "SocketGroup", "SocketPerms", "MaxConns" },
		{ },
		{ }
	},

	exec	= config_exec,

	interface = {
		{ },
		{ "IgnoreSelected" },
		{ "Interfaces" }
	},

	iptables = config_iptables,

	irq	= {
		{ },
		{ "IgnoreSelected" },
		{ "Irqs" }
	},

	load	= {
		{ },
		{ },
		{ }
	},

	logfile	= {
		{ "LogLevel", "File" },
		{ "Timestamp" },
		{ }
	},

	netlink	= {
		{ },
		{ "IgnoreSelected" },
		{ "Interfaces", "VerboseInterfaces", "QDiscs", "Classes", "Filters" }
	},

	network	= config_network,

	ping	= {
		{ "TTL", "Interval" },
		{ },
		{ "Hosts" }
	},

	processes = {
		{ },
		{ },
		{ "Processes" }
	},

	rrdtool	= {
		{ "DataDir", "StepSize", "HeartBeat", "RRARows", "XFF", "CacheFlush", "CacheTimeout" },
		{ "RRASingle" },
		{ "RRATimespans" }
	},

	tcpconns = {
		{ },
		{ "ListeningPorts" },
		{ "LocalPorts", "RemotePorts" }
	},

	unixsock = {
		{ "SocketFile", "SocketGroup", "SocketPerms" },
		{ },
		{ }
	},

	wireless = {
		{ },
		{ },
		{ }
	},
}

preprocess = {
	RRATimespans = function(val)
		local rv = { }
		for time in val:gmatch("[^%s]+") do
			table.insert( rv, luci.util.parse_units(time) )
		end
		return table.concat(rv, " ")
	end
}


section("collectd")

for plugin in pairs(plugins) do
	if plugin ~= "collectd" then
		section( plugin )
	end
end