-- /usr/lib/lua/luci/controller/zzz.lua
module("luci.controller.zzz", package.seeall)

function index()
	if not nixio.fs.access("/etc/config/zzz") then
		return
	end

	-- Menu
	entry({ "admin", "network", "zzz" }, cbi("zzz"), "ZZZ", 60).dependent = false

	-- Settings
	entry({ "admin", "network", "zzz", "service_control" }, call("service_control")).leaf = true

	-- Status API
	entry({ "admin", "network", "zzz", "get_status" }, call("act_status")).leaf = true
end

function service_control()
	local sys = require("luci.sys")
	local util = require("luci.util")
	local action = luci.http.formvalue("action")
	local result = { success = false, message = "" }

	local valid_actions = { start = true, stop = true, restart = true }

	if action and valid_actions[action] then
		local cmd = ""
		if action == "start" then
			cmd = "/etc/rc.d/S99zzz start"
		elseif action == "stop" then
			cmd = "/etc/rc.d/S99zzz stop"
		elseif action == "restart" then
			cmd = "/etc/rc.d/S99zzz stop && sleep 2 && /etc/rc.d/S99zzz start"
		end

		if cmd ~= "" then
			local ret = sys.call(cmd)
			if ret == 0 then
				result.success = true
				result.message = util.pcdata(action .. " 成功")
			else
				result.success = false
				result.message = util.pcdata(action .. " 失败")
			end
		end
	else
		result.message = "无效的操作"
	end

	luci.http.prepare_content("application/json")
	luci.http.write_json(result)
end

function act_status()
	local sys = require("luci.sys")
	local util = require("luci.util")
	local status = {}

	status.running = (sys.call("pgrep -f zzz >/dev/null") == 0)

	if status.running then
		status.process_info = util.trim(sys.exec("ps | grep -v grep | grep zzz"))
	end

	local log_file = "/tmp/zzz.log"
	if nixio.fs.access(log_file) then
		status.log = util.trim(sys.exec("tail -20 " .. log_file))
	else
		status.log = util.trim(sys.exec("logread | grep zzz | tail -10"))
	end

	if status.log then
		status.log = util.pcdata(status.log)
	end

	luci.http.prepare_content("application/json")
	luci.http.write_json(status)
end