prosody-modules: mod_cloud_notify/mod_cloud

comparison mod_cloud_notify/mod_cloud_notify.lua @ 2736:fff185e7ab73

mod_cloud_notify: Implement the "stripped stanzas" proposal. See https://mail.jabber.org/pipermail/standards/2017-July/033089.html

author	tmolitor <thilo@eightysoft.de>
date	Mon, 14 Aug 2017 23:31:05 +0200
parents	75b137cf869a
children	9756211fcbe3

comparison

equal deleted inserted replaced

-:b5fae17e4403
+:fff185e7ab73
 -- Copyright (C) 2015-2016 Kim Alvefur
 -- Copyright (C) 2017 Thilo Molitor
 --
 -- This file is MIT/X11 licensed.
+local t_insert = table.insert;
 local st = require"util.stanza";
 local jid = require"util.jid";
 local dataform = require"util.dataforms".new;
 local filters = require"util.filters";
 local hashes = require"util.hashes";
 	for push_identifier, _ in pairs(user_push_services) do
 		if hashes.sha256(push_identifier, true) == stanza.attr.id then
 			if user_push_services[push_identifier] and user_push_services[push_identifier].jid == from and push_errors[push_identifier] > 0 then
 				push_errors[push_identifier] = 0;
-				module:log("debug", "Push succeeded, error count for identifier '%s' is now at %s", push_identifier, tostring(push_errors[push_identifier]));
+				module:log("debug", "Push succeeded, error count for identifier '%s' is now at %s again", push_identifier, tostring(push_errors[push_identifier]));
 			end
 		end
 	end
 	return true;
 end
 	origin.log("debug", "Attempting to enable push notifications");
 	-- MUST contain a 'jid' attribute of the XMPP Push Service being enabled
 	local push_jid = enable.attr.jid;
 	-- SHOULD contain a 'node' attribute
 	local push_node = enable.attr.node;
+	-- CAN contain a 'include_payload' attribute
+	local include_payload = enable.attr.include_payload;
 	if not push_jid then
 		origin.log("debug", "Push notification enable request missing the 'jid' field");
 		origin.send(st.error_reply(stanza, "modify", "bad-request", "Missing jid"));
 		return true;
 	end
 	end
 	local push_identifier = push_jid .. "<" .. (push_node or "");
 	local push_service = {
 		jid = push_jid;
 		node = push_node;
+		include_payload = include_payload;
 		count = 0;
 		options = publish_options and st.preserialize(publish_options);
 	};
 	local ok = push_store:set_identifier(origin.username, push_identifier, push_service);
 	if not ok then
 		origin.send(st.reply(stanza));
 	end
 	return true;
 end
 module:hook("iq-set/self/"..xmlns_push..":disable", push_disable);
+-- clone a stanza and strip it
+local function strip_stanza(stanza)
+	local tags = {};
+	local new = { name = stanza.name, attr = { xmlns = stanza.attr.xmlns, type = stanza.attr.type }, tags = tags };
+	for i=1,#stanza do
+		local child = stanza[i];
+		if type(child) == "table" then		-- don't add raw text nodes
+			if child.name then
+				child = strip_stanza(child);
+				t_insert(tags, child);
+			end
+			t_insert(new, child);
+		end
+	end
+	return setmetatable(new, st.stanza_mt);
+end
 local push_form = dataform {
 	{ name = "FORM_TYPE"; type = "hidden"; value = "urn:xmpp:push:summary"; };
 	{ name = "message-count"; type = "text-single"; };
 	{ name = "pending-subscription-count"; type = "text-single"; };
 			end
 			if stanza and include_body then
 				form_data["last-message-body"] = stanza:get_child_text("body");
 			end
 			push_publish:add_child(push_form:form(form_data));
+			if stanza and push_info.include_payload == "stripped" then
+				push_publish:tag("payload", { type = "stripped" })
+					:add_child(strip_stanza(stanza));
+				push_publish:up(); -- / payload
+			end
+			if stanza and push_info.include_payload == "full" then
+				push_publish:tag("payload", { type = "full" })
+					:add_child(st.clone(stanza));
+				push_publish:up(); -- / payload
+			end
 			push_publish:up(); -- / notification
 			push_publish:up(); -- / publish
 			push_publish:up(); -- / pubsub
 			if push_info.options then
 				push_publish:tag("publish-options"):add_child(st.deserialize(push_info.options));
 			end
 			-- send out push
 			module:log("debug", "Sending push notification for %s@%s to %s (%s)", node, module.host, push_info.jid, tostring(push_info.node));
+			-- module:log("debug", "PUSH STANZA: %s", tostring(push_publish));
 			-- handle push errors for this node
 			if push_errors[push_identifier] == nil then
 				push_errors[push_identifier] = 0;
 				module:hook("iq-error/bare/"..stanza_id, handle_push_error);
 				module:hook("iq-result/bare/"..stanza_id, handle_push_success);
 end
 -- publish on offline message
 module:hook("message/offline/handle", function(event)
 	local node, user_push_services = get_push_settings(event.stanza, event.origin);
+	module:log("debug", "Invoking cloud handle_notify_request() for offline stanza");
 	handle_notify_request(event.stanza, node, user_push_services);
 end, 1);
 -- publish on unacked smacks message
 local function process_smacks_stanza(stanza, session)
 	local session = event.origin;
 	local queue = event.queue;
 	-- process unacked stanzas
 	process_smacks_queue(queue, session);
 	-- process future unacked (hibernated) stanzas
-	filters.add_filter(session, "stanzas/out", process_smacks_stanza);
+	filters.add_filter(session, "stanzas/out", process_smacks_stanza, -990);
 end
 -- smacks hibernation is ended
 local function restore_session(event)
 	local session = event.resumed;
 	handle_notify_request(nil, user, push_services);
 end
 -- can be used by other modules to ping one or more (or all) push endpoints
 module:hook("cloud-notify-ping", send_ping);
--- TODO: this has to be done on first connect not on offline broadcast, else the counter will be incorrect
--- TODO: it seems this is already done, so this could be safely removed, couldn't it?
--- module:hook("message/offline/broadcast", function(event)
--- 	local origin = event.origin;
--- 	local user_push_services = push_store:get(origin.username);
--- 	if not #user_push_services then return end
---
--- 	for _, push_info in pairs(user_push_services) do
--- 		if push_info then
--- 			push_info.count = 0;
--- 		end
--- 	end
--- 	push_store:set(origin.username, user_push_services);
--- end, 1);
 module:log("info", "Module loaded");
 function module.unload()
 	if module.unhook then
 		module:unhook("account-disco-info", account_dico_info);
 		module:unhook("iq-set/self/"..xmlns_push..":enable", push_enable);

Mercurial > prosody-modules

comparison mod_cloud_notify/mod_cloud_notify.lua @ 2736:fff185e7ab73