local st = require "util.stanza"

local jid_split = require "util.jid".split;

local mod_pep = module:depends("pep");

local sha1 = require "util.hashes".sha1;

local base64_decode = require "util.encodings".base64.decode;

local vcards = module:open_store("vcard");

module:add_feature("vcard-temp");

-- Simple translations

-- <foo><text>hey</text></foo> -> <FOO>hey</FOO>

local simple_map = {

	nickname = "text";

	title = "text";

	role = "text";

	categories = "text";

	note = "text";

	url = "uri";

	bday = "date";

}

module:hook("iq-get/bare/vcard-temp:vCard", function (event)

	local origin, stanza = event.origin, event.stanza;

	local pep_service = mod_pep.get_pep_service(jid_split(stanza.attr.to) or origin.username);

	local ok, id, vcard4_item = pep_service:get_last_item("urn:xmpp:vcard4", stanza.attr.from);

	local vcard_temp = st.stanza("vCard", { xmlns = "vcard-temp" });

	if ok and vcard4_item then

		local vcard4 = vcard4_item.tags[1];

		local fn = vcard4:get_child("fn");

		vcard_temp:text_tag("FN", fn and fn:get_child_text("text"));

		local v4n = vcard4:get_child("n");

		vcard_temp:tag("N")

			:text_tag("FAMILY", v4n and v4n:get_child_text("surname"))

			:text_tag("GIVEN", v4n and v4n:get_child_text("given"))

			:text_tag("MIDDLE", v4n and v4n:get_child_text("additional"))

			:text_tag("PREFIX", v4n and v4n:get_child_text("prefix"))

			:text_tag("SUFFIX", v4n and v4n:get_child_text("suffix"))

			:up();

		for tag in vcard4:childtags() do

			local typ = simple_map[tag.name];

			if typ then

				local text = tag:get_child_text(typ);

				if text then

					vcard_temp:text_tag(tag.name:upper(), text);

				end

			elseif tag.name == "email" then

				local text = tag:get_child_text("text");

				if text then

					vcard_temp:tag("EMAIL")

						:text_tag("USERID", text)

						:tag("INTERNET"):up();

					if tag:find"parameters/type/text#" == "home" then

						vcard_temp:tag("HOME"):up();

					elseif tag:find"parameters/type/text#" == "work" then

						vcard_temp:tag("WORK"):up();

					end

					vcard_temp:up();

				end

			elseif tag.name == "tel" then

				local text = tag:get_child_text("uri");

				if text then

					if text:sub(1, 4) == "tel:" then

						text = text:sub(5)

					end

					vcard_temp:tag("TEL"):text_tag("NUMBER", text);

					if tag:find"parameters/type/text#" == "home" then

						vcard_temp:tag("HOME"):up();

					elseif tag:find"parameters/type/text#" == "work" then

						vcard_temp:tag("WORK"):up();

					end

					vcard_temp:up();

				end

			elseif tag.name == "adr" then

				vcard_temp:tag("ADR")

					:text_tag("POBOX", tag:get_child_text("pobox"))

					:text_tag("EXTADD", tag:get_child_text("ext"))

					:text_tag("STREET", tag:get_child_text("street"))

					:text_tag("LOCALITY", tag:get_child_text("locality"))

					:text_tag("REGION", tag:get_child_text("region"))

					:text_tag("PCODE", tag:get_child_text("code"))

					:text_tag("CTRY", tag:get_child_text("country"));

				if tag:find"parameters/type/text#" == "home" then

					vcard_temp:tag("HOME"):up();

				elseif tag:find"parameters/type/text#" == "work" then

					vcard_temp:tag("WORK"):up();

				end

				vcard_temp:up();

			end

		end

	else

		local legacy_vcard = st.deserialize(vcards:get(jid_split(stanza.attr.to) or origin.username));

		origin.send(st.reply(stanza):add_child(legacy_vcard or vcard_temp));

		return true;

	end

	local meta_ok, avatar_meta = pep_service:get_items("urn:xmpp:avatar:metadata", stanza.attr.from);

	local data_ok, avatar_data = pep_service:get_items("urn:xmpp:avatar:data", stanza.attr.from);

	if meta_ok and data_ok then

		for _, hash in ipairs(avatar_meta) do

			local meta = avatar_meta[hash];

			local data = avatar_data[hash];

			local info = meta and meta.tags[1]:get_child("info");

			vcard_temp:tag("PHOTO");

			if info and info.attr.type then

				vcard_temp:text_tag("TYPE", info.attr.type);

			end

			if data then

				vcard_temp:text_tag("BINVAL", data.tags[1]:get_text());

			elseif info and info.attr.url then

				vcard_temp:text_tag("EXTVAL", info.attr.uri);

			end

			vcard_temp:up();

		end

	end

	origin.send(st.reply(stanza):add_child(vcard_temp));

	return true;

end);

module:hook("iq-set/self/vcard-temp:vCard", function (event)

	local origin, stanza = event.origin, event.stanza;

	local pep_service = mod_pep.get_pep_service(origin.username);

	local vcard_temp = stanza.tags[1];

	local vcard4 = st.stanza("item", { xmlns = "http://jabber.org/protocol/pubsub", id = "current" })

		:tag("vcard", { xmlns = 'urn:ietf:params:xml:ns:vcard-4.0' });

	if pep_service:purge("urn:xmpp:avatar:metadata", origin.full_jid) then

		pep_service:publish("urn:xmpp:avatar:data", origin.full_jid);

	end

	vcard4:tag("fn"):text_tag("text", vcard_temp:get_child_text("FN")):up();

	local N = vcard_temp:get_child("N");

	vcard4:tag("n")

		:text_tag("surname", N and N:get_child_text("FAMILY"))

		:text_tag("given", N and N:get_child_text("GIVEN"))

		:text_tag("additional", N and N:get_child_text("MIDDLe"))

		:text_tag("prefix", N and N:get_child_text("PREFIX"))

		:text_tag("suffix", N and N:get_child_text("SUFFIX"))

	:up();

	for tag in vcard_temp:childtags() do

		local typ = simple_map[tag.name:lower()];

		if typ then

			local text = tag:get_text();

			if text then

				vcard4:tag(tag.name:lower()):text_tag(typ, text):up();

			end

		elseif tag.name == "EMAIL" then

			local text = tag:get_child_text("USERID");

			if text then

				vcard4:tag("email")

				vcard4:text_tag("text", text)

				vcard4:tag("parameters"):tag("type");

				if tag:get_child("HOME") then

					vcard4:text_tag("text", "home");

				elseif tag:get_child("WORK") then

					vcard4:text_tag("text", "work");

				end

				vcard4:up():up():up();

			end

		elseif tag.name == "TEL" then

			local text = tag:get_child_text("NUMBER");

			if text then

				vcard4:tag("tel"):text_tag("uri", "tel:"..text);

			end

			vcard4:tag("parameters"):tag("type");

			if tag:get_child("HOME") then

				vcard4:text_tag("text", "home");

			elseif tag:get_child("WORK") then

				vcard4:text_tag("text", "work");

			end

			vcard4:up():up():up();

		elseif tag.name == "ORG" then

			local text = tag:get_child_text("ORGNAME");

			if text then

				vcard4:tag("org"):text_tag("text", text):up();

			end

		elseif tag.name == "DESC" then

			local text = tag:get_text();

			if text then

				vcard4:tag("note"):text_tag("text", text):up();

			end

		elseif tag.name == "ADR" then

			vcard4:tag("adr")

				:text_tag("pobox", tag:get_child_text("POBOX"))

				:text_tag("ext", tag:get_child_text("EXTADD"))

				:text_tag("street", tag:get_child_text("STREET"))

				:text_tag("locality", tag:get_child_text("LOCALITY"))

				:text_tag("region", tag:get_child_text("REGION"))

				:text_tag("code", tag:get_child_text("PCODE"))

				:text_tag("country", tag:get_child_text("CTRY"));

			vcard4:tag("parameters"):tag("type");

			if tag:get_child("HOME") then

				vcard4:text_tag("text", "home");

			elseif tag:get_child("WORK") then

				vcard4:text_tag("text", "work");

			end

			vcard4:up():up():up();

		elseif tag.name == "PHOTO" then

			local avatar_type = tag:get_child_text("TYPE");

			local avatar_payload = tag:get_child_text("BINVAL");

			if avatar_payload then

				local avatar_raw = base64_decode(avatar_payload);

				local avatar_hash = sha1(avatar_raw, true);

				local avatar_meta = st.stanza("item", { id = avatar_hash, xmlns = "http://jabber.org/protocol/pubsub" })

					:tag("metadata", { xmlns="urn:xmpp:avatar:metadata" })

						:tag("info", {

							bytes = tostring(#avatar_raw),

							id = avatar_hash,

							type = avatar_type,

						});

				local avatar_data = st.stanza("item", { id = avatar_hash, xmlns = "http://jabber.org/protocol/pubsub" })

					:tag("data", { xmlns="urn:xmpp:avatar:data" })

						:text(avatar_payload);

				if pep_service:publish("urn:xmpp:avatar:data", origin.full_jid, avatar_hash, avatar_data) then

					pep_service:publish("urn:xmpp:avatar:metadata", origin.full_jid, avatar_hash, avatar_meta);

				end

			end

		end

	end

	local ok, err = pep_service:publish("urn:xmpp:vcard4", origin.full_jid, "current", vcard4);

	if ok then

		origin.send(st.reply(stanza));

	elseif err == "forbidden" then

		origin.send(st.error_reply(stanza, "auth", "forbidden"));

	elseif err == "internal-server-error" then

		origin.send(st.error_reply(stanza, "wait", "internal-server-error"));

	else

		origin.send(st.error_reply(stanza, "modify", "undefined-condition", err));

	end

	return true;

end);

local function inject_xep153(event)

	local origin, stanza = event.origin, event.stanza;

	local username = origin.username;

	if not username then return end

	local pep = mod_pep.get_pep_service(username);

	stanza:remove_children("x", "vcard-temp:x:update");

	local x_update = st.stanza("x", { xmlns = "vcard-temp:x:update" });

	local ok, avatar_hash = pep:get_last_item("urn:xmpp:avatar:metadata", true);

	if ok and avatar_hash then

		x_update:text_tag("photo", avatar_hash);

	end

	stanza:add_direct_child(x_update);

end

module:hook("pre-presence/full", inject_xep153, 1);

module:hook("pre-presence/bare", inject_xep153, 1);

module:hook("pre-presence/host", inject_xep153, 1);