module:depends("adhoc");
local adhoc_new = module:require("adhoc").new;
local hats_api = module:depends("muc_hats_api");

local dataforms_new = require "util.dataforms".new;
local adhoc_simple = require "util.adhoc".new_simple_form;

local function generate_error_message(errors)
	local errmsg = {};
	for name, err in pairs(errors) do
		errmsg[#errmsg + 1] = name .. ": " .. err;
	end
	return { status = "completed", error = { message = table.concat(errmsg, "\n") } };
end

local add_hat_form_layout = dataforms_new {
	title = "Add a hat";
	instructions = "Assign a hat to a room member";

	{ name = "user",  type = "jid-single",  label = "User JID", required = true };
	{ name = "room",  type = "jid-single",  label = "Room JID", required = true };
	{ name = "title", type = "text-single", label = "Hat title" };
	{ name = "uri",   type = "text-single", label = "Hat URI",  required = true };
};

local add_hat_handler = adhoc_simple(add_hat_form_layout, function (fields, errs)
	if errs then
		return generate_error_message(errs);
	end

	local ok, err_cond, err_text = hats_api.add_user_hat(fields.user, fields.room, fields.uri, {
		active = true;
		required = true;
		title = fields.title;
	});

	return {
		status = "completed";
		info = ok and "The hat has been added successfully" or ("There was an error adding the hat: "..(err_text or err_cond));
	};

end);

local remove_hat_form_layout = dataforms_new {
	title = "Remove a hat";
	instructions = "Remove a hat from a room member";

	{ name = "user",  type = "jid-single",  label = "User JID", required = true };
	{ name = "room",  type = "jid-single",  label = "Room JID", required = true };
	{ name = "uri",   type = "text-single", label = "Hat URI",  required = true };
};

local remove_hat_handler = adhoc_simple(remove_hat_form_layout, function (fields, errs)
	if errs then
		return generate_error_message(errs);
	end

	local ok, err_cond, err_text = hats_api.remove_user_hat(fields.user, fields.room, fields.uri);

	return {
		status = "completed";
		info = ok and "The hat has been removed successfully" or ("There was an error removing the hat: "..(err_text or err_cond));
	};

end);

local add_hat_desc = adhoc_new("Add hat to a user", "http://prosody.im/protocol/hats#add", add_hat_handler, "admin");
local remove_hat_desc = adhoc_new("Remove hat from a user", "http://prosody.im/protocol/hats#remove", remove_hat_handler, "admin");

module:provides("adhoc", add_hat_desc);
module:provides("adhoc", remove_hat_desc);