From 84ef8a73ebd601f8348adb5f83b60fba3e77cf00 Mon Sep 17 00:00:00 2001 From: Kim Alvefur Date: Fri, 10 Dec 2021 22:37:32 +0100 Subject: mod_mam: Allow plugging into archive decision Similar procedure as mod_csi_simple --- plugins/mod_mam/mod_mam.lua | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) (limited to 'plugins') diff --git a/plugins/mod_mam/mod_mam.lua b/plugins/mod_mam/mod_mam.lua index d4b2b060..6b1a0973 100644 --- a/plugins/mod_mam/mod_mam.lua +++ b/plugins/mod_mam/mod_mam.lua @@ -311,7 +311,7 @@ local function strip_stanza_id(stanza, user) return stanza; end -local function should_store(stanza, c2s) --> boolean, reason: string +local function should_store(stanza) --> boolean, reason: string local st_type = stanza.attr.type or "normal"; -- FIXME pass direction of stanza and use that along with bare/full JID addressing -- for more accurate MUC / type=groupchat check @@ -320,7 +320,7 @@ local function should_store(stanza, c2s) --> boolean, reason: string -- Headline messages are ephemeral by definition return false, "headline"; end - if st_type == "error" and not c2s then + if st_type == "error" then -- Errors not sent sent from a local client -- Why would a client send an error anyway? if jid_resource(stanza.attr.to) then @@ -380,6 +380,12 @@ local function should_store(stanza, c2s) --> boolean, reason: string return false, "default"; end +module:hook("archive-should-store", function (event) + local should, why = should_store(event.stanza); + event.reason = why; + return should; +end, -1) + -- Handle messages local function message_handler(event, c2s) local origin, stanza = event.origin, event.stanza; @@ -396,9 +402,12 @@ local function message_handler(event, c2s) -- Filter out that claim to be from us event.stanza = strip_stanza_id(stanza, store_user); - local should, why = should_store(stanza, c2s); + local event_payload = { stanza = stanza; session = origin }; + local should = module:fire_event("archive-should-store", event_payload); + local why = event_payload.reason; + if not should then - log("debug", "Not archiving stanza: %s (%s)", stanza:top_tag(), why); + log("debug", "Not archiving stanza: %s (%s)", stanza:top_tag(), event_payload.reason); return; end -- cgit v1.2.3