mod_pep: Handle service discovery queries for bare account JIDs (thanks darkrain).

1 files changed, 18 insertions, 0 deletions

diff --git a/plugins/mod_pep.lua b/plugins/mod_pep.lua
index ef03ab0f..b2f2f8f3 100644
--- a/plugins/mod_pep.lua
+++ b/plugins/mod_pep.lua
@@ -277,3 +277,21 @@ module:hook("iq/bare/disco", function(event)
 		end
 	end
 end);
+
+module:hook("account-disco-info", function(event)
+	local stanza = event.stanza;
+	stanza:tag('identity', {category='pubsub', type='pep'}):up();
+	stanza:tag('feature', {var='http://jabber.org/protocol/pubsub#publish'}):up();
+end);
+
+module:hook("account-disco-items", function(event)
+	local session, stanza = event.session, event.stanza;
+	local bare = session.username..'@'..session.host;
+	local user_data = data[bare];
+
+	if user_data then
+		for node, _ in pairs(user_data) do
+			stanza:tag('item', {jid=bare, node=node}):up();
+		end
+	end
+end);