--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_muc_archive/mod_muc_archive.lua Fri Mar 27 15:51:57 2020 +0100
@@ -0,0 +1,191 @@
+-- The MIT License (MIT)
+--
+-- Copyright (C) 2009 Thilo Cestonaro
+-- Copyright (C) 2009 Waqas Hussain
+-- Copyright (C) 2009-2013 Matthew Wild
+-- Copyright (C) 2013 Kim Alvefur
+-- Copyright (C) 2013 Marco Cirillo
+-- Copyright (c) 2020 JC Brand
+--
+-- Permission is hereby granted, free of charge, to any person obtaining a copy of
+-- this software and associated documentation files (the "Software"), to deal in
+-- the Software without restriction, including without limitation the rights to
+-- use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+-- the Software, and to permit persons to whom the Software is furnished to do so,
+-- subject to the following conditions:
+--
+-- The above copyright notice and this permission notice shall be included in all
+-- copies or substantial portions of the Software.
+--
+-- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+-- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+-- FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+-- COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+-- IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+-- CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+--
+local hosts = prosody.hosts;
+local tostring = tostring;
+local st = require "util.stanza";
+local split_jid = require "util.jid".split;
+local jid_bare = require "util.jid".bare;
+local time_now = os.time;
+
+local muc_form_config_option = "muc#roomconfig_enablelogging"
+
+local log_all_rooms = module:get_option_boolean("muc_log_all_rooms", false);
+local log_by_default = module:get_option_boolean("muc_log_by_default", false);
+local log_presences = module:get_option_boolean("muc_log_presences", true);
+
+local archive_store = "muc_logging_archive";
+local archive = module:open_store(archive_store, "archive");
+
+local xmlns_muc_user = "http://jabber.org/protocol/muc#user";
+
+if archive.name == "null" or not archive.find then
+ if not archive.find then
+ module:log("error", "Attempt to open archive storage returned a driver without archive API support");
+ module:log("error", "mod_%s does not support archiving",
+ archive._provided_by or archive.name and "storage_"..archive.name.."(?)" or "<unknown>");
+ else
+ module:log("error", "Attempt to open archive storage returned null driver");
+ end
+ module:log("info", "See https://prosody.im/doc/storage and https://prosody.im/doc/archiving for more information");
+ return false;
+end
+
+
+-- Module Definitions
+
+local function get_room_from_jid(jid)
+ local node, host = split_jid(jid);
+ local component = hosts[host];
+ if component then
+ local muc = component.modules.muc
+ if muc and rawget(muc,"rooms") then
+ -- We're running 0.9.x or 0.10 (old MUC API)
+ return muc.rooms[jid];
+ elseif muc and rawget(muc,"get_room_from_jid") then
+ -- We're running >0.10 (new MUC API)
+ return muc.get_room_from_jid(jid);
+ else
+ return
+ end
+ end
+end
+
+local function logging_enabled(room)
+ if log_all_rooms then
+ return true;
+ end
+ if room._data.hidden then -- do not log data of private rooms
+ return false;
+ end
+ local enabled = room._data.logging;
+ if enabled == nil then
+ return log_by_default;
+ end
+ return enabled;
+end
+
+
+local function log_if_needed(event)
+ local stanza = event.stanza;
+
+ if (stanza.name == "presence") or
+ (stanza.name == "iq") or
+ (stanza.name == "message" and tostring(stanza.attr.type) == "groupchat")
+ then
+ local node, host = split_jid(stanza.attr.to);
+ if node and host then
+ local room = get_room_from_jid(node.."@"..host)
+ if not room then return end
+ -- Policy check
+ if not logging_enabled(room) then return end
+
+ local muc_to = nil
+ local muc_from = nil;
+
+ if stanza.name == "presence" and stanza.attr.type == nil then
+ muc_from = stanza.attr.to;
+ elseif stanza.name == "iq" and stanza.attr.type == "set" then
+ -- kick, to is the room, from is the admin, nick who is kicked is attr of iq->query->item
+ if stanza.tags[1] and stanza.tags[1].name == "query" then
+ local tmp = stanza.tags[1];
+ if tmp.tags[1] ~= nil and tmp.tags[1].name == "item" and tmp.tags[1].attr.nick then
+ tmp = tmp.tags[1];
+ for jid, nick in pairs(room._jid_nick) do
+ if nick == stanza.attr.to .. "/" .. tmp.attr.nick then
+ muc_to = nick;
+ break;
+ end
+ end
+ end
+ end
+ else
+ for jid, nick in pairs(room._jid_nick) do
+ if jid == stanza.attr.from then
+ muc_from = nick;
+ break;
+ end
+ end
+ end
+
+ if (muc_from or muc_to) then
+ local stored_stanza = st.clone(stanza);
+ stored_stanza.attr.from = muc_from;
+ stored_stanza.attr.to = muc_to;
+
+ if stanza.name == "message" then
+ local actor = jid_bare(room._occupants[muc_from].jid);
+ local affiliation = room:get_affiliation(actor) or "none";
+ local role = room:get_role(actor) or room:get_default_role(affiliation);
+ stored_stanza:add_direct_child(st.stanza("x", { xmlns = xmlns_muc_user })
+ :tag("item", { affiliation = affiliation; role = role; jid = actor }));
+ end
+
+ local with = stanza.name
+ if stanza.attr.type then
+ with = with .. "<" .. stanza.attr.type
+ end
+ archive:append(node, nil, stored_stanza, time_now(), with);
+ end
+ end
+ end
+end
+
+if not log_all_rooms then
+ module:hook("muc-config-form", function(event)
+ local room, form = event.room, event.form;
+ table.insert(form,
+ {
+ name = muc_form_config_option,
+ type = "boolean",
+ label = "Enable Logging?",
+ value = logging_enabled(room),
+ }
+ );
+ end);
+
+ module:hook("muc-config-submitted", function(event)
+ local room, fields, changed = event.room, event.fields, event.changed;
+ local new = fields[muc_form_config_option];
+ if new ~= room._data.logging then
+ room._data.logging = new;
+ if type(changed) == "table" then
+ changed[muc_form_config_option] = true;
+ else
+ event.changed = true;
+ end
+ end
+ end);
+end
+
+module:hook("message/bare", log_if_needed, 1);
+
+if log_presences then
+ module:hook("iq/bare", log_if_needed, 1);
+ module:hook("presence/full", log_if_needed, 1);
+end
+
+module:log("debug", "module mod_muc_archive loaded!");