From 99a7271471c1fb01df5dc7d6bb8ddf1f42aefffd Mon Sep 17 00:00:00 2001
From: onefang
Date: Tue, 5 Nov 2019 17:30:08 +1000
Subject: Generic reports, including a dumb example.

---
 apt-panopticon-report-DNS.lua | 144 ++++++++++++++++++++++++++++++++++++++++++
 apt-panopticon.lua            |  26 ++++++++
 2 files changed, 170 insertions(+)
 create mode 100755 apt-panopticon-report-DNS.lua

diff --git a/apt-panopticon-report-DNS.lua b/apt-panopticon-report-DNS.lua
new file mode 100755
index 0000000..b26bde5
--- /dev/null
+++ b/apt-panopticon-report-DNS.lua
@@ -0,0 +1,144 @@
+#!/usr/bin/env luajit
+
+local args = {...}
+
+local logFile
+
+
+--[[ Ordered table iterator, allow to iterate on the natural order of the keys of a table.
+    From http://lua-users.org/wiki/SortedIteration
+  ]]
+function __genOrderedIndex( t )
+    local orderedIndex = {}
+    for key in pairs(t) do
+        table.insert( orderedIndex, key )
+    end
+    table.sort( orderedIndex )
+    return orderedIndex
+end
+function orderedNext(t, state)
+    -- Equivalent of the next function, but returns the keys in the alphabetic
+    -- order. We use a temporary ordered key table that is stored in the
+    -- table being iterated.
+
+    local key = nil
+    --print("orderedNext: state = "..tostring(state) )
+    if state == nil then
+        -- the first time, generate the index
+        t.__orderedIndex = __genOrderedIndex( t )
+        key = t.__orderedIndex[1]
+    else
+        -- fetch the next value
+        for i = 1,table.getn(t.__orderedIndex) do
+            if t.__orderedIndex[i] == state then
+                key = t.__orderedIndex[i+1]
+            end
+        end
+    end
+
+    if key then
+        return key, t[key]
+    end
+
+    -- no more value to return, cleanup
+    t.__orderedIndex = nil
+    return
+end
+function orderedPairs(t)
+    -- Equivalent of the pairs() function on tables. Allows to iterate
+    -- in order
+    return orderedNext, t, nil
+end
+
+
+-- Use this to dump a table to a string.
+dumpTable = function (table, space, name)
+    local r = ""
+    if "" == space then r = r .. space .. name .. " =\n" else r = r .. space .. "[" .. name .. "] =\n" end
+    r = r .. space .. "{\n"
+    r = r .. dumpTableSub(table, space .. "  ")
+    if "" == space then r = r .. space .. "}\n" else r = r .. space .. "},\n" end
+    return r
+end
+dumpTableSub = function (table, space)
+    local r = ""
+--    for k, v in pairs(table) do 
+    for k, v in orderedPairs(table) do 
+        if type(k) == "string" then k = '"' .. k .. '"' end
+	if type(v) == "table" then
+	    r = r .. dumpTable(v, space, k)
+	elseif type(v) == "string" then
+	    r = r .. space .. "[" .. k .. "] = '" .. v .. "';\n"
+	elseif type(v) == "function" then
+	    r = r .. space .. "[" .. k .. "] = function ();\n"
+	elseif type(v) == "userdata" then
+	    r = r .. space .. "userdata " .. "[" .. k .. "];\n"
+	elseif type(v) == "boolean" then
+	    if (v) then
+		r = r .. space .. "[" .. k .. "] = true;\n"
+	    else
+		r = r .. space .. "[" .. k .. "] = false;\n"
+	    end
+	else
+	    r = r .. space .. "[" .. k .. "] = " .. v .. ";\n"
+	end
+    end
+    return r
+end
+
+local results = {}
+
+local log = function(v, t, s, prot, test, host)
+    local x = ""
+    if nil == prot then prot = "" end
+    if nil ~= test then x = x .. test else test = "" end
+    if nil ~= host then
+	if #x > 0 then x = x .. " " end
+	x = x .. host
+    end
+    if #x > 0 then
+	t = t .. "(" .. x ..  ")"
+	if "" == test then
+	    if v == 0 then results[prot].errors = results[prot].errors + 1 end
+	    if v == 1 then results[prot].warnings = results[prot].warnings + 1 end
+	else
+	    if v == 0 then results[prot][test].errors = results[prot][test].errors + 1 end
+	    if v == 1 then results[prot][test].warnings = results[prot][test].warnings + 1 end
+	end
+    end
+    if v <= verbosity then
+	if 3 <= verbosity then t = os.date() .. " " .. t end
+	print(t .. ": " .. s)
+    end
+    if nil ~= logFile then
+	logFile:write(os.date() .. " " .. t .. ": " .. s .. "\n")
+	logFile:flush()
+    end
+end
+local D = function(s) log(3,  "DEBUG   ", s) end
+local I = function(s) log(2,  "INFO    ", s) end
+local W = function(s, p, t, h) log(1,  "WARNING ", s, p, t, h) end
+local E = function(s, p, t, h) log(0,  "ERROR   ", s, p, t, h) end
+local C = function(s) log(-1, "CRITICAL", s) end
+
+local mirrors = {}
+
+mirrors = loadfile("results/mirrors.lua")()
+
+for k, v in pairs(mirrors) do
+    mirrors[k].Protocols = nil
+    mirrors[k].FQDN = nil
+    mirrors[k].Active = nil
+    mirrors[k].Rate = nil
+    mirrors[k].BaseURL = nil
+    mirrors[k].Country = nil
+    mirrors[k].Bandwidth = nil
+end
+
+
+local file, e = io.open("results/REPORT-DNS.txt", "w+")
+if nil == file then C("opening mirrors file - " .. e) else
+    file:write(dumpTable(mirrors, "", "DNS report"))
+    file:close()
+end
+
diff --git a/apt-panopticon.lua b/apt-panopticon.lua
index 2a9631b..788d72c 100755
--- a/apt-panopticon.lua
+++ b/apt-panopticon.lua
@@ -48,6 +48,20 @@ options =
 --	        "Updated",
 	    },
 	},
+    reports =
+	{
+	    typ = "table",
+	    help = "",
+	    value =
+	    {
+		"DNS",
+		"email",
+--		"Nagios",
+		"Prometheus",
+--		"RRD",
+		"web",
+	    },
+	},
 }
 
 local defaultURL = {scheme = "http"}
@@ -630,5 +644,17 @@ else
 	while 0 < checkExes(downloadLock) do os.execute("sleep 10") end
     end
     os.execute("rm -f results/*.check")
+
+    -- Create the reports.
+    for n, r in pairs(options.reports.value) do
+	local report = "apt-panopticon-report-" .. r .. ".lua"
+	local rfile, e = io.open(report, "r")
+	if nil == rfile then C("opening " .. report .. " file - " .. e) else
+	    rfile:close()
+	    I("Creating " .. report .. " report.")
+	    execute("./" .. report .. " &")
+	end
+    end
+
     logFile:close()
 end
-- 
cgit v1.1