clix/export.lua

local verse = require "verse";

short_opts.i = "interactive";

local split_jid = require "prosody.util.jid".split;

local timer = require "prosody.util.timer";

local xmlns_pie = "urn:xmpp:pie:0";

local xmlns_nodes = {

	{ "query", "jabber:iq:roster" },

	{ "query", "jabber:iq:privacy" },

	{ "vCard", "vcard-temp" },

}

local xmlns_private = "jabber:iq:private";

local xmlns_private_nodes = {

	{ "storage", "storage:bookmarks" },

}

local function on_connect(conn)

	local export = verse.stanza("server-data", { xmlns = xmlns_pie })

		:tag("host", { jid = conn.host })

			:tag("user", { name = conn.username, password = conn.password });

	local out = 0;

	for i = 1,#xmlns_nodes do

		out = out + 1;

		local node = xmlns_nodes[i];

		conn:debug("requesting %s*%s", node[2], node[1]);

		conn:send_iq(verse.iq{type="get"}

			:tag(node[1], { xmlns = node[2] })

		, function(stanza)

			conn:debug("got %s*%s", node[2], node[1]);

			local query = stanza:get_child(node[1], node[2]);

			if query then

				export:add_child(query);

			end

			out = out - 1;

		end);

	end

	for i = 1,#xmlns_private_nodes do

		out = out + 1;

		local node = xmlns_private_nodes[i];

		conn:debug("requesting private node %s*%s", node[2], node[1]);

		conn:send_iq(verse.iq{type="get"}

			:tag("query", { xmlns = xmlns_private })

				:tag(node[1], { xmlns = node[2] })

		, function(stanza)

			conn:debug("got private %s*%s", node[2], node[1]);

			local query = stanza:get_child("query", xmlns_private);

			if query and #query > 0 then

				export:add_child(query);

			end

			out = out - 1;

		end);

	end

	timer.add_task(1, function ()

		if out > 0 then

			return 1

		end

		print("<?xml version='1.0' encoding='UTF-8'?>");

		print(export);

		conn:close();

	end)

end

return function (opts, arg)

	if opts.short_help then

		print("Export your account");

		return;

	end

	if opts.help then

		return 0;

	end

	clix_connect(opts, on_connect);

end