generate-html.lua

#!/usr/bin/lua5.3

local json = require("cjson")
local lustache = require("lustache")
local yaml = require("lyaml")

local utils = require("utils")
local conf = require("config")


function get_branches(indexes)
	local res = {}
	for k,v in ipairs(indexes.master.branch) do
		table.insert(res, v.name)
	end
	return res
end

----
-- need to create one flipped table with both mirros and master
function flip_branches(branches)
	local res = {}
	for _,b in ipairs(branches) do
		for _,r in ipairs(b.repo) do
			for _,a in ipairs(r.arch) do
				if type(res[b.name]) == "nil" then res[b.name] = {} end
				if type(res[b.name][r.name]) == "nil" then
					res[b.name][r.name] = {}
				end
				res[b.name][r.name][a.name] = a
			end
		end
	end
	return res
end

----
-- convert table to array (values to keys)
function get_repo_arch(indexes)
	local t = {}
	for _,b in ipairs(indexes.master.branch) do
		for _,r in ipairs(b.repo) do
			for _,a in ipairs(r.arch) do
				if type(t[r.name]) == "nil" then t[r.name] = {} end
				t[r.name][a.name] = 1
			end
		end
	end
	local res = {}
	for repo in utils.kpairs(t, utils.sort_repo) do
		for arch in utils.kpairs(t[repo], utils.sort_arch) do
			table.insert(res, repo.."/"..arch)
		end
	end
	return res
end

----
-- convert table to array (values to keys) for each mirror
function flip_mirrors(mirrors)
	local res = {}
	for _,m in ipairs(mirrors) do
		res[m.url] = flip_branches(m.branch)
	end
	return res
end

----
-- format timestamp difference
function format_age(ts)
	local res = {}
	if ts < 3600 then
		res.text = "OK"
		res.class = "status-ok"
	elseif ts < 86400 then
		res.text = ("%dh"):format(math.ceil(ts/3600))
		res.class = "status-warn"
	elseif ts > 86400 then
		res.text = ("%dd"):format(math.ceil(ts/86400))
		res.class = "status-error"
	end
	return res
end

----
-- format status based on http status message
function format_status(status, age)
	local res = {}
	if status ~= "failed" then
		if status == "200" then
			res = format_age(age)
		elseif status == "404" then
			res.class = "status-na"
			res.text = "N/A"
		else
			res.class = "status-unk"
			res.text = status
		end
	else
		res.class = "status-na"
		res.text = "N/A"
	end
	return res
end

----
-- get status and format it based on http status and modified time
function get_status(fm, fb, date, mirror, branch, repo, arch)
	local res = { text = "N/A", class = "status-na" }
	if type(fm[mirror]) == "table" and
		type(fm[mirror][branch]) == "table" and
		type(fm[mirror][branch][repo]) == "table" and
		type(fm[mirror][branch][repo][arch]) == "table" then
		if type(fm[mirror][branch][repo][arch]) == "table" then
			local mirror = fm[mirror][branch][repo][arch]
			local master = fb[branch][repo][arch]
			local age = 0
			if type(mirror.modified) == "number" and
				type(master.modified) == "number" and
				master.modified ~= mirror.modified then
				age = master.modified - mirror.modified
			end
			res = format_status(mirror.status, age)
		end
	end
	return res
end


function get_mirrors()
	local mirrors = yaml.load(utils.read_file(conf.mirrors_yaml))
	for a,mirror in pairs(mirrors) do
		mirrors[a].location = mirror.location or "Unknown"
		mirrors[a].bandwidth = mirror.bandwidth or "Unknown"
		mirrors[a].num = a
		if mirror.name:len() > 32 then
			mirrors[a].name = ("%s..."):format(mirror.name:sub(1,32))
		end
		for b,url in pairs(mirror.urls) do
			local scheme = url:match("(.*)://*.")
			mirrors[a].urls[b] = {url = url, scheme = scheme}
		end
	end
	return mirrors
end

----
-- build the html table
function build_status_tables(indexes)
	local res = {}
	local fm = flip_mirrors(indexes.mirrors)
	local fb = flip_branches(indexes.master.branch)
	local thead = get_branches(indexes)
	table.insert(thead, 1, "branch/release")
	for idx,mirror in ipairs(indexes.mirrors) do
		local rows = {}
		for _,ra in ipairs(get_repo_arch(indexes)) do
			local repo, arch = ra:match("(.*)/(.*)")
			local row = {}
			table.insert(row, {text = ra})
			for _,branch in ipairs(get_branches(indexes)) do
				local status = get_status(fm, fb, indexes.date, mirror.url, branch, repo, arch)
				table.insert(row, status)
			end
			table.insert(rows, { row = row })
		end
		res[idx] = { 
			url = mirror.url, thead = thead, tbody = rows, num = idx,
			duration = mirror.duration, count = mirror.count
		}
	end
	return res
end

local out_json = ("%s/%s"):format(conf.outdir, conf.status_json)
local indexes = json.decode(utils.read_file(out_json))
local status = build_status_tables(indexes)
local mirrors = get_mirrors()
local last_update = os.date("%c", indexes.date)
local view = { last_update = last_update, mirrors = mirrors, status = status }
local tpl = utils.read_file("index.tpl")
local out_html = ("%s/%s"):format(conf.outdir, conf.mirrors_html)
utils.write_file(out_html, lustache:render(tpl, view))