modulemanager: Add module:hook([xmlns, ] name, handler) which uses new stanza events

1 files changed, 11 insertions, 0 deletions

diff --git a/core/modulemanager.lua b/core/modulemanager.lua
index 18b511e6..92a50abb 100644
--- a/core/modulemanager.lua
+++ b/core/modulemanager.lua
@@ -373,6 +373,17 @@ function api:hook(event, handler)
 	(hosts[self.host] or prosody).events.add_handler(event, handler);
 end
 
+function api:hook_stanza(xmlns, name, handler)
+	if not handler and type(name) == "function" then
+		-- If only 2 options then they specified no xmlns
+		xmlns, name, handler = nil, xmlns, name;	
+	elseif not (handler and name and xmlns) then
+		module:log("warn", "Error: Insufficient parameters to module:hook_stanza()");
+		return;
+	end
+	return api.hook(self, "stanza/"..(xmlns and (xmlns..":") or "")..name, function (data) return handler(data.origin, data.stanza, data); end);
+end
+
 --------------------------------------------------------------------
 
 local actions = {};