mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-05 06:07:37 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

mod_mam: Allow querying by set of IDs

Browse source
This commit is contained in:
Kim Alvefur 2021-01-12 19:00:44 +01:00
parent 3c919b0ad4
commit 2d03793897
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
plugins/mod_mam/mod_mam.lua
View file

@ -88,6 +88,11 @@ if archive.caps and archive.caps.full_id_range then
table.insert(query_form, { name = "after-id"; type = "text-single"; }); table.insert(query_form, { name = "after-id"; type = "text-single"; });
end end
if archive.caps and archive.caps.ids then
table.insert(query_form, { name = "ids"; type = "list-multi"; });
end
-- Serve form -- Serve form
module:hook("iq-get/self/"..xmlns_mam..":query", function(event) module:hook("iq-get/self/"..xmlns_mam..":query", function(event)
local origin, stanza = event.origin, event.stanza; local origin, stanza = event.origin, event.stanza;
@ -107,7 +112,7 @@ module:hook("iq-set/self/"..xmlns_mam..":query", function(event)
get_prefs(origin.username, true); get_prefs(origin.username, true);
-- Search query parameters -- Search query parameters
local qwith, qstart, qend, qbefore, qafter; local qwith, qstart, qend, qbefore, qafter, qids;
local form = query:get_child("x", "jabber:x:data"); local form = query:get_child("x", "jabber:x:data");
if form then if form then
local form_type, err = get_form_type(form); local form_type, err = get_form_type(form);
@ -125,6 +130,7 @@ module:hook("iq-set/self/"..xmlns_mam..":query", function(event)
end end
qwith, qstart, qend = form["with"], form["start"], form["end"]; qwith, qstart, qend = form["with"], form["start"], form["end"];
qbefore, qafter = form["before-id"], form["after-id"]; qbefore, qafter = form["before-id"], form["after-id"];
qids = form["ids"];
qwith = qwith and jid_bare(qwith); -- dataforms does jidprep qwith = qwith and jid_bare(qwith); -- dataforms does jidprep
end end
@ -164,6 +170,7 @@ module:hook("iq-set/self/"..xmlns_mam..":query", function(event)
with = qwith; with = qwith;
limit = qmax == 0 and 0 or qmax + 1; limit = qmax == 0 and 0 or qmax + 1;
before = before; after = after; before = before; after = after;
ids = qids;
reverse = reverse; reverse = reverse;
total = use_total or qmax == 0; total = use_total or qmax == 0;
}); });
@ -561,7 +568,6 @@ module:hook("message/bare", message_handler, 0);
module:hook("message/full", message_handler, 0); module:hook("message/full", message_handler, 0);
local advertise_extended = module:get_option_boolean("mam_advertise_extend", false); local advertise_extended = module:get_option_boolean("mam_advertise_extend", false);
-- TODO ids
-- TODO delete feature flag option -- TODO delete feature flag option
module:hook("account-disco-info", function(event) module:hook("account-disco-info", function(event)