File

plugins/carbons.lua @ 263:598e9f93de78

plugins.pubsub: Fix missing <item/> when publishing.
author Kim Alvefur <zash@zash.se>
date Sat, 10 Dec 2011 15:21:09 +0100
parent 250:a5ac643a7fd6
child 274:a228f3c7808a
line wrap: on
line source

local verse = require "verse";

local xmlns_carbons = "urn:xmpp:carbons:1";
local xmlns_forward = "urn:xmpp:forward:0";
local os_date = os.date;
local datetime = function(t) return os_date("!%Y-%m-%dT%H:%M:%SZ", t); end

-- TODO Check disco for support

function verse.plugins.carbons(stream)
	local carbons = {};
	carbons.enabled = false;
	stream.carbons = carbons;

	function carbons:enable(callback)
		stream:send_iq(verse.iq{type="set"}
		:tag("enable", { xmlns = xmlns_carbons })
		, function(result)
			local success = result.attr.type == "result";
			if success then
				carbons.enabled = true;
			end
			if callback then
				callback(success);
			end
		end or nil);
	end

	function carbons:disable(callback)
		stream:send_iq(verse.iq{type="set"}
		:tag("disable", { xmlns = xmlns_carbons })
		, function(result)
			local success = result.attr.type == "result";
			if success then
				carbons.enabled = false;
			end
			if callback then
				callback(success);
			end
		end or nil);
	end

	stream:hook("message", function(stanza)
		stream:debug(stanza);
		local fwd = stanza:get_child("forwarded", xmlns_forward);
		if fwd then
			local carbon_dir = fwd:get_child(nil, xmlns_carbons);
			carbon_dir = carbon_dir and carbon_dir.name;
			if carbon_dir then
				local fwd_stanza = fwd:get_child("message", "jabber:client");
				assert(fwd_stanza, "No stanza included.\n"..tostring(stanza).."\n--\n"..tostring(fwd_stanza));
				return stream:event("carbon", {
					dir = carbon_dir,
					stanza = fwd_stanza,
					timestamp = nil or datetime(), -- TODO check for delay tag
				});
			end
		end
	end, 1);
end