diff mod_map/mod_map.lua @ 3496:262e68821f3f

mod_map: Experimental module exposing MAM summary
author Kim Alvefur <zash@zash.se>
date Mon, 25 Feb 2019 15:52:46 +0100
parents
children 30743ae1fe1a
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_map/mod_map.lua	Mon Feb 25 15:52:46 2019 +0100
@@ -0,0 +1,67 @@
+
+local st = require "util.stanza";
+local jid_bare = require "util.jid".bare;
+local rsm = require "util.rsm";
+local dataform = require "util.dataforms".new;
+
+local archive = module:open_store("archive", "archive");
+
+local query_form = dataform {
+	{ name = "with"; type = "jid-single"; };
+	{ name = "start"; type = "text-single" };
+	{ name = "end"; type = "text-single"; };
+};
+
+if not archive.summary then
+	module:log("error", "The archive:summary() API is not supported by %s", archive._provided_by);
+	return
+end
+
+module:hook("iq-get/self/xmpp:prosody.im/mod_map:summary", function(event)
+	local origin, stanza = event.origin, event.stanza;
+
+	local query = stanza.tags[1];
+
+	-- Search query parameters
+	local qwith, qstart, qend;
+	local form = query:get_child("x", "jabber:x:data");
+	if form then
+		local err;
+		form, err = query_form:data(form);
+		if err then
+			origin.send(st.error_reply(stanza, "modify", "bad-request", select(2, next(err))));
+			return true;
+		end
+		qwith, qstart, qend = form["with"], form["start"], form["end"];
+		qwith = qwith and jid_bare(qwith); -- dataforms does jidprep
+	end
+
+	local qset = rsm.get(query);
+	local qmax = qset and qset.max;
+	local before, after = qset and qset.before, qset and qset.after;
+	if type(before) ~= "string" then before = nil; end
+
+	local summary = archive:summary(origin.username, {
+		start = qstart; ["end"] = qend; -- Time range
+		with = qwith;
+		limit = qmax;
+		before = before; after = after;
+	});
+	if not summary then
+		module:send(st.error_reply(stanza, "wait", "internal-server-error"));
+		return true;
+	end
+
+	local reply = st.reply(stanza);
+	reply:tag("summary", { xmlns = "xmpp:prosody.im/mod_map" });
+	for jid, count in pairs(summary) do
+		reply:tag("item", { jid = jid });
+		if type(count) == "number" then
+			reply:text_tag("count", ("%d"):format(count));
+		end
+		reply:up();
+	end
+
+	module:send(reply);
+	return true;
+end);