util.startup: Check root after detecting platform and reading config (thanks SigmaTel71)
Ensures that startup.detect_platform() runs so know whether to use the
POSIX method of checking the current user or something else. Also after
reading the config so we know whether the root override setting is set.
local names = { "Romeo", "Juliet", "Mercutio", "Tybalt", "Benvolio" };
local devices = { "", "phone", "laptop", "tablet", "toaster", "fridge", "shoe" };
local users = {};
local filters = require "prosody.util.filters";
local id = require "prosody.util.id";
local dt = require "prosody.util.datetime";
local dm = require "prosody.util.datamanager";
local st = require "prosody.util.stanza";
local record_id = id.short():lower();
local record_date = os.date("%Y%b%d"):lower();
local header_file = dm.getpath(record_id, "scansion", record_date, "scs", true);
local record_file = dm.getpath(record_id, "scansion", record_date, "log", true);
local head = io.open(header_file, "w");
local scan = io.open(record_file, "w+");
local function record(string)
scan:write(string);
scan:flush();
end
local function record_header(string)
head:write(string);
head:flush();
end
local function record_object(class, name, props)
head:write(("[%s] %s\n"):format(class, name));
for k,v in pairs(props) do
head:write(("\t%s: %s\n"):format(k, v));
end
head:write("\n");
head:flush();
end
local function record_event(session, event)
record(session.scansion_id.." "..event.."\n\n");
end
local function record_stanza(stanza, session, verb)
local flattened = tostring(stanza:indent(2, "\t"));
record(session.scansion_id.." "..verb..":\n\t"..flattened.."\n\n");
end
local function record_stanza_in(stanza, session)
if stanza.attr.xmlns == nil then
local copy = st.clone(stanza);
copy.attr.from = nil;
record_stanza(copy, session, "sends")
end
return stanza;
end
local function record_stanza_out(stanza, session)
if stanza.attr.xmlns == nil then
if not (stanza.name == "iq" and stanza:get_child("bind", "urn:ietf:params:xml:ns:xmpp-bind")) then
local copy = st.clone(stanza);
if copy.attr.to == session.full_jid then
copy.attr.to = nil;
end
record_stanza(copy, session, "receives");
end
end
return stanza;
end
module:hook("resource-bind", function (event)
local session = event.session;
if not users[session.username] then
users[session.username] = {
character = table.remove(names, 1) or id.short();
devices = {};
n_devices = 0;
};
end
local user = users[session.username];
local device = user.devices[session.resource];
if not device then
user.n_devices = user.n_devices + 1;
device = devices[user.n_devices] or ("device"..id.short());
user.devices[session.resource] = device;
end
session.scansion_character = user.character;
session.scansion_device = device;
session.scansion_id = user.character..(device ~= "" and "'s "..device or device);
record_object("Client", session.scansion_id, {
jid = session.full_jid,
password = "password",
});
module:log("info", "Connected: %s", session.scansion_id);
record_event(session, "connects");
filters.add_filter(session, "stanzas/in", record_stanza_in);
filters.add_filter(session, "stanzas/out", record_stanza_out);
end);
module:hook("resource-unbind", function (event)
local session = event.session;
if session.scansion_id then
record_event(session, "disconnects");
end
end)
record_header("# mod_scansion_record on host '"..module.host.."' recording started "..dt.datetime().."\n\n");
record[[
-----
]]
module:hook_global("server-stopping", function ()
record("# recording ended on "..dt.datetime().."\n");
module:log("info", "Scansion recording available in %s", header_file);
end);
prosody.events.add_handler("server-cleanup", function ()
scan:seek("set", 0);
for line in scan:lines() do
head:write(line, "\n");
end
scan:close();
os.remove(record_file);
head:close()
end);