local dm_load = require "util.datamanager".load;

local dm_store = require "util.datamanager".store;

local usermanager = require "core.usermanager";

local dataforms_new = require "util.dataforms".new;

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

local vcard = module:require "vcard";

local rawget, rawset = rawget, rawset;

local st = require "util.stanza";

local template = require "util.template";

local get_reply = template[[

<query xmlns="jabber:iq:search">

  <instructions>Fill in one or more fields to search for any matching Jabber users.</instructions>

  <first/>

  <last/>

  <nick/>

  <email/>

</query>

]].apply({});

local item_template = template[[

<item xmlns="jabber:iq:search" jid="{jid}">

  <first>{first}</first>

  <last>{last}</last>

  <nick>{nick}</nick>

  <email>{email}</email>

</item>

]];

module:add_feature("jabber:iq:search");

local opted_in;

function module.load()

	opted_in = dm_load(nil, module.host, "user_index") or {};

end

function module.unload()

	dm_store(nil, module.host, "user_index", opted_in);

end

local opt_in_layout = dataforms_new{

	title = "Search settings";

	instructions = "Do you want to appear in search results?";

	{

		name = "searchable",

		label = "Appear in search results?",

		type = "boolean",

	},

};

local vCard_mt = {

	__index = function(t, k)

		if type(k) ~= "string" then return nil end

		for i=1,#t do

			local t_i = rawget(t, i);

			if t_i and t_i.name == k then

				rawset(t, k, t_i);

				return t_i;

			end

		end

	end

};

local function get_user_vcard(user)

	local vCard = dm_load(user, module.host, "vcard");

	if vCard then

		vCard = st.deserialize(vCard);

		vCard = vcard.from_xep54(vCard);

		return setmetatable(vCard, vCard_mt);

	end

end

local at_host = "@"..module.host;

module:hook("iq/host/jabber:iq:search:query", function(event)

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

	if stanza.attr.type == "get" then

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

	else -- type == "set"

		local query = stanza.tags[1];

		local first, last, nick, email =

			(query:get_child_text"first" or false),

			(query:get_child_text"last" or false),

			(query:get_child_text"nick" or false),

			(query:get_child_text"email" or false);

		if not ( first or last or nick or email ) then

			origin.send(st.error_reply(stanza, "modify", "not-acceptable", "All fields were empty"));

			return true;

		end

		local reply = st.reply(stanza):query("jabber:iq:search");

		local username, hostname = jid_split(email);

		if hostname == module.host and username and usermanager.user_exists(username, hostname) then

			local vCard = get_user_vcard(username);

			if vCard then

				reply:add_child(item_template.apply{

					jid = username..at_host;

					first = vCard.N and vCard.N[2] or nil;

					last = vCard.N and vCard.N[1] or nil;

					nick = vCard.NICKNAME and vCard.NICKNAME[1] or username;

					email = vCard.EMAIL and vCard.EMAIL[1] or nil;

				});

			end

		else

			for username in pairs(opted_in) do

				local vCard = get_user_vcard(username);

				if vCard and (

				(vCard.N and vCard.N[2] == first) or

				(vCard.N and vCard.N[1] == last) or

				(vCard.NICKNAME and vCard.NICKNAME[1] == nick) or

				(vCard.EMAIL and vCard.EMAIL[1] == email)) then

					reply:add_child(item_template.apply{

						jid = username..at_host;

						first = vCard.N and vCard.N[2] or nil;

						last = vCard.N and vCard.N[1] or nil;

						nick = vCard.NICKNAME and vCard.NICKNAME[1] or username;

						email = vCard.EMAIL and vCard.EMAIL[1] or nil;

					});

				end

			end

		end

		origin.send(reply);

	end

	return true;

end);

local function opt_in_handler(self, data, state)

	local username, hostname = jid_split(data.from);

	if state then -- the second return value

		if data.action == "cancel" then

			return { status = "canceled" };

		end

		if not username or not hostname or hostname ~= module.host then

			return { status = "error", error = { type = "cancel",

				condition = "forbidden", message = "Invalid user or hostname." } };

		end

		local fields = opt_in_layout:data(data.form);

		opted_in[username] = fields.searchable or nil

		return { status = "completed" }

	else -- No state, send the form.

		return { status = "executing", actions  = { "complete" },

	end

end

local adhoc_new = module:require "adhoc".new;

local adhoc_vjudsetup = adhoc_new("Search settings", "vjudsetup", opt_in_handler);--, "self");-- and nil);

module:depends"adhoc";

module:provides("adhoc", adhoc_vjudsetup);