Mercurial > riddim / changeset

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/plugins/pubsub2room.lua	Fri Mar 11 22:37:06 2011 +0100
@@ -0,0 +1,54 @@
+--[[ pubsub2room
+relays pubsub broadcasts to rooms
+
+example conf:
+pubsub2room = {
+	["pubsub.prosody.im#commits"] = {
+		room = "prosody@conference.prosody.im";
+		template = "${author.name} committed: ${title}";
+	};
+};
+--]]
+
+-- FIXME Should this really be here?
+local extractor_mt = {
+	__index = function (t, k)
+		local n = t.stanza;
+		for x in k:gmatch("[^.]+") do
+			n = n:get_child(x);
+			if not n then return end
+		end
+		return n[1];
+	end
+};
+
+local function new_extractor(stanza, ...)
+	local st = stanza:get_child(...)
+	return st and setmetatable({ stanza = st }, extractor_mt) or nil;
+end
+
+function riddim.plugins.pubsub2room(bot)
+	local bare_jid = require "util.jid".bare;
+	bot:add_plugin("pubsub");
+
+	local config = bot.config.pubsub2room;
+	bot:hook("pubsub/event", function(event)
+		local conf = config[event.from .. "#" .. event.node];
+		local room = bot.rooms[conf.room];
+		local entry = event.item and new_extractor(event.item, "entry", "http://www.w3.org/2005/Atom")
+		-- FIXME or forever be limited to Atom!
+
+		if not conf or not entry or not room then return end
+		local message = conf.template:gsub("%${([^}]+)}", entry);
+		room:send_message(message);
+	end);
+
+	-- FIXME When to unsubscribe?
+	bot:hook("started", function()
+		local jid = bare_jid(bot.stream.jid);
+		for hostnode in pairs(config) do
+			local host, node = hostnode:match("^([^#]+)#(.*)");
+			bot.pubsub:subscribe(host, node, jid, print);
+		end
+	end);
+end