Mercurial > prosody-modules
changeset 3957:7e96b95924bd
mod_muc_archive: Add fork of mod_muc_log that uses newer storage API
author | JC Brand <jc@opkode.com> |
---|---|
date | Fri, 27 Mar 2020 15:51:57 +0100 |
parents | ebc1f1d962c5 |
children | cacd753848b2 |
files | mod_muc_archive/README.markdown mod_muc_archive/mod_muc_archive.lua |
diffstat | 2 files changed, 234 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_muc_archive/README.markdown Fri Mar 27 15:51:57 2020 +0100 @@ -0,0 +1,43 @@ +--- +labels: +- 'Stage-Beta' +summary: Log MUC messages to disk +... + +# Introduction + +This module logs the conversations of chatrooms running on the server to Prosody's data store. + +This is a fork of [mod_muc_log](https://modules.prosody.im/mod_muc_log.html) which uses the newer storage API. +This allows you to also log messages to a SQL backend. + +## Changes between mod_muc_archive and mod_muc_log: + +- Use newer module storage API so that you can also store in SQL +- Adhere to config option `muc_log_all_rooms` (also used by mod_muc_mam) +- Add affiliation information in the logged stanza +- Remove code that set (and then removed) an "alreadyJoined" dummy element + +NOTE: The changes are unlikely to be entirely backwards compatible because the stanza +being logged is no longer wrapped with `<stanza time=...>`. + +Details +======= + +mod\_muc\_archive must be loaded individually for the components that need it. + +Assuming you have a MUC component already running on +conference.example.org then you can add muc\_archive to it like so: + + Component "conference.example.org" "muc" + modules_enabled = { + "muc_archive"; + } + + +Compatibility +============= + + ------ ----- + 0.11 Works + ------ -----
--- /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!");