1 files changed, 216 insertions, 0 deletions
diff --git a/applications/luci-app-nlbwmon/luasrc/controller/nlbw.lua b/applications/luci-app-nlbwmon/luasrc/controller/nlbw.lua
new file mode 100644
index 0000000000..4f41bb8c31
--- /dev/null
+++ b/applications/luci-app-nlbwmon/luasrc/controller/nlbw.lua
@@ -0,0 +1,216 @@
+-- Copyright 2017 Jo-Philipp Wich <jo@mein.io>
+-- Licensed to the public under the Apache License 2.0.
+
+module("luci.controller.nlbw", package.seeall)
+
+function index()
+	entry({"admin", "nlbw"}, firstchild(), _("Bandwidth Monitor"), 80)
+	entry({"admin", "nlbw", "display"}, template("nlbw/display"), _("Display"), 1)
+	entry({"admin", "nlbw", "config"}, cbi("nlbw/config"), _("Configuration"), 2)
+	entry({"admin", "nlbw", "backup"}, template("nlbw/backup"), _("Backup"), 3)
+	entry({"admin", "nlbw", "data"}, call("action_data"), nil, 4)
+	entry({"admin", "nlbw", "list"}, call("action_list"), nil, 5)
+	entry({"admin", "nlbw", "ptr"}, call("action_ptr"), nil, 6).leaf = true
+	entry({"admin", "nlbw", "download"}, call("action_download"), nil, 7)
+	entry({"admin", "nlbw", "restore"}, post("action_restore"), nil, 8)
+end
+
+local function exec(cmd, args, writer)
+	local os = require "os"
+	local nixio = require "nixio"
+
+	local fdi, fdo = nixio.pipe()
+	local pid = nixio.fork()
+
+	if pid > 0 then
+		fdo:close()
+
+		while true do
+			local buffer = fdi:read(2048)
+			local wpid, stat, code = nixio.waitpid(pid, "nohang")
+
+			if writer and buffer and #buffer > 0 then
+				writer(buffer)
+			end
+
+			if wpid and stat == "exited" then
+				break
+			end
+		end
+	elseif pid == 0 then
+		nixio.dup(fdo, nixio.stdout)
+		fdi:close()
+		fdo:close()
+		nixio.exece(cmd, args, nil)
+		nixio.stdout:close()
+		os.exit(1)
+	end
+end
+
+function action_data()
+	local http = require "luci.http"
+
+	local types = {
+		csv = "text/csv",
+		json = "application/json"
+	}
+
+	local args = { }
+	local mtype = http.formvalue("type") or "json"
+	local delim = http.formvalue("delim") or ";"
+	local period = http.formvalue("period")
+	local group_by = http.formvalue("group_by")
+	local order_by = http.formvalue("order_by")
+
+	if types[mtype] then
+		args[#args+1] = "-c"
+		args[#args+1] = mtype
+	else
+		http.status(400, "Unsupported type")
+		return
+	end
+
+	if delim and #delim > 0 then
+		args[#args+1] = "-s%s" % delim
+	end
+
+	if period and #period > 0 then
+		args[#args+1] = "-t"
+		args[#args+1] = period
+	end
+
+	if group_by and #group_by > 0 then
+		args[#args+1] = "-g"
+		args[#args+1] = group_by
+	end
+
+	if order_by and #order_by > 0 then
+		args[#args+1] = "-o"
+		args[#args+1] = order_by
+	end
+
+	http.prepare_content(types[mtype])
+	exec("/usr/sbin/nlbw", args, http.write)
+end
+
+function action_list()
+	local http = require "luci.http"
+
+	local fd = io.popen("/usr/sbin/nlbw -c list")
+	local periods = { }
+
+	if fd then
+		while true do
+			local period = fd:read("*l")
+
+			if not period then
+				break
+			end
+
+			periods[#periods+1] = period
+		end
+
+		fd:close()
+	end
+
+	http.prepare_content("application/json")
+	http.write_json(periods)
+end
+
+function action_ptr(...)
+	local http = require "luci.http"
+	local util = require "luci.util"
+
+	http.prepare_content("application/json")
+	http.write_json(util.ubus("network.rrdns", "lookup", {
+		addrs = {...}, timeout = 3000
+	}))
+end
+
+function action_download()
+	local nixio = require "nixio"
+	local http = require "luci.http"
+	local sys = require "luci.sys"
+	local uci = require "luci.model.uci".cursor()
+
+	local dir = uci:get_first("nlbwmon", "nlbwmon", "database_directory")
+		or "/var/lib/nlbwmon"
+
+	if dir and nixio.fs.stat(dir, "type") == "dir" then
+		local n = "nlbwmon-backup-%s-%s.tar.gz"
+			%{ sys.hostname(), os.date("%Y-%m-%d") }
+
+		http.prepare_content("application/octet-stream")
+		http.header("Content-Disposition", "attachment; filename=\"%s\"" % n)
+		exec("/bin/tar", { "-C", dir, "-c", "-z", ".", "-f", "-" }, http.write)
+	else
+		http.status(500, "Unable to find database directory")
+	end
+end
+
+function action_restore()
+	local nixio = require "nixio"
+	local http = require "luci.http"
+	local i18n = require "luci.i18n"
+	local tpl = require "luci.template"
+	local uci = require "luci.model.uci".cursor()
+
+	local tmp = "/tmp/nlbw-restore.tar.gz"
+	local dir = uci:get_first("nlbwmon", "nlbwmon", "database_directory")
+		or "/var/lib/nlbwmon"
+
+	local fp
+	http.setfilehandler(
+		function(meta, chunk, eof)
+			if not fp and meta and meta.name == "archive" then
+				fp = io.open(tmp, "w")
+			end
+			if fp and chunk then
+				fp:write(chunk)
+			end
+			if fp and eof then
+				fp:close()
+			end
+		end)
+
+	local files = { }
+	local tar = io.popen("/bin/tar -tzf %s" % tmp, "r")
+	if tar then
+		while true do
+			local file = tar:read("*l")
+			if not file then
+				break
+			elseif file:match("^%d%d%d%d%d%d%d%d%.db%.gz$") or
+			       file:match("^%./%d%d%d%d%d%d%d%d%.db%.gz$") then
+				files[#files+1] = file
+			end
+		end
+		tar:close()
+	end
+
+	if #files == 0 then
+		http.status(500, "Internal Server Error")
+		tpl.render("nlbw/backup", {
+			message = i18n.translate("Invalid or empty backup archive")
+		})
+		return
+	end
+
+
+	local output = { }
+
+	exec("/etc/init.d/nlbwmon", { "stop" })
+	exec("/bin/mkdir", { "-p", dir })
+
+	exec("/bin/tar", { "-C", dir, "-vxzf", tmp, unpack(files) },
+		function(chunk) output[#output+1] = chunk:match("%S+") end)
+
+	exec("/bin/rm", { "-f", tmp })
+	exec("/etc/init.d/nlbwmon", { "start" })
+
+	tpl.render("nlbw/backup", {
+		message = i18n.translatef(
+			"The following database files have been restored: %s",
+			table.concat(output, ", "))
+	})
+end