Generic reports, including a dumb example.

2 files changed, 170 insertions, 0 deletions

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