File

+
−
+
−local st = require "util.stanza";
+
−local jid_bare = require "util.jid".bare;
+
−local rsm = require "util.rsm";
+
−local dataform = require "util.dataforms".new;
+
−local datetime = require "util.datetime".datetime;
+
−
+
−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.counts) do
+
−		reply:tag("item", { jid = jid });
+
−		if type(count) == "number" then
+
−			reply:text_tag("count", ("%d"):format(count));
+
−		end
+
−		if summary.earliest and summary.earliest[jid] then
+
−			reply:text_tag("start", datetime(summary.earliest[jid]));
+
−		end
+
−		if summary.latest and summary.latest[jid] then
+
−			reply:text_tag("end", datetime(summary.latest[jid]));
+
−		end
+
−		if summary.body and summary.body[jid] then
+
−			reply:text_tag("body", summary.body[jid]);
+
−		end
+
−		reply:up();
+
−	end
+
−
+
−	module:send(reply);
+
−	return true;
+
−end);