local t_unpack = table.unpack or unpack; -- luacheck: ignore 113

local time_now = os.time;

local jid_prep = require "util.jid".prep;

local set = require "util.set";

local st = require "util.stanza";

local it = require "util.iterators";

local uuid_generate = require "util.uuid".generate;

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

local xmlns_pubsub = "http://jabber.org/protocol/pubsub";

local xmlns_pubsub_errors = "http://jabber.org/protocol/pubsub#errors";

local xmlns_pubsub_owner = "http://jabber.org/protocol/pubsub#owner";

local _M = {};

local handlers = {};

_M.handlers = handlers;

local pubsub_errors = {

	["conflict"] = { "cancel", "conflict" };

	["invalid-jid"] = { "modify", "bad-request", nil, "invalid-jid" };

	["jid-required"] = { "modify", "bad-request", nil, "jid-required" };

	["nodeid-required"] = { "modify", "bad-request", nil, "nodeid-required" };

	["item-not-found"] = { "cancel", "item-not-found" };

	["not-subscribed"] = { "modify", "unexpected-request", nil, "not-subscribed" };

	["invalid-options"] = { "modify", "bad-request", nil, "invalid-options" };

	["forbidden"] = { "auth", "forbidden" };

	["not-allowed"] = { "cancel", "not-allowed" };

	["not-acceptable"] = { "modify", "not-acceptable" };

	["internal-server-error"] = { "wait", "internal-server-error" };

};

local function pubsub_error_reply(stanza, error)

	local e = pubsub_errors[error];

	local reply = st.error_reply(stanza, t_unpack(e, 1, 3));

	if e[4] then

		reply:tag(e[4], { xmlns = xmlns_pubsub_errors }):up();

	end

	return reply;

end

_M.pubsub_error_reply = pubsub_error_reply;

-- util.pubsub is meant to be agnostic to XEP-0060

local function config_to_xep0060(node_config)

	return {

		["pubsub#title"] = node_config["title"];

		["pubsub#description"] = node_config["description"];

		["pubsub#max_items"] = tostring(node_config["max_items"]);

		["pubsub#persist_items"] = node_config["persist_items"];

		["pubsub#notification_type"] = node_config["notification_type"];

		["pubsub#access_model"] = node_config["access_model"];

	}

end

local function config_from_xep0060(config)

	return {

		["title"] = config["pubsub#title"];

		["description"] = config["pubsub#description"];

		["max_items"] = tonumber(config["pubsub#max_items"]);

		["persist_items"] = config["pubsub#persist_items"];

		["notification_type"] = config["pubsub#notification_type"];

		["access_model"] = config["pubsub#access_model"];

	}

end

local node_config_form = dataform {

	{

		type = "hidden";

		name = "FORM_TYPE";

		value = "http://jabber.org/protocol/pubsub#node_config";

	};

	{

		type = "text-single";

		name = "pubsub#title";

		label = "Title";

	};

	{

		type = "text-single";

		name = "pubsub#description";

		label = "Description";

	};

	{

		type = "text-single";

		name = "pubsub#max_items";

		label = "Max # of items to persist";

	};

	{

		type = "boolean";

		name = "pubsub#persist_items";

		label = "Persist items to storage";

	};

	{

		type = "list-single";

		name = "pubsub#access_model";

		label = "Specify the subscriber model";

		options = {

			{ value = "authorize" },

			{ value = "open" },

			{ value = "presence" },

			{ value = "roster" },

			{ value = "whitelist" },

		};

	};

	{

		type = "list-single";

		name = "pubsub#notification_type";

		label = "Specify the delivery style for notifications";

		options = {

			{ label = "Messages of type normal", value = "normal" },

			{ label = "Messages of type headline", value = "headline", default = true },

		};

	};

};

local options_form = dataform {

	{

		type = "hidden";

		name = "FORM_TYPE";

		value = "http://jabber.org/protocol/pubsub#subscribe_options";

	};

	{

		type = "boolean";

		name = "pubsub#include_body";

		label = "Receive message body in addition to payload?";

	};

};

local node_metadata_form = dataform {

	{

		type = "hidden";

		name = "FORM_TYPE";

		value = "http://jabber.org/protocol/pubsub#meta-data";

	};

	{

		type = "text-single";

		name = "pubsub#title";

	};

	{

		type = "text-single";

		name = "pubsub#description";

	};

};

local service_method_feature_map = {

	add_subscription = { "subscribe" };

	create = { "create-nodes", "instant-nodes", "item-ids", "create-and-configure" };

	delete = { "delete-nodes" };

	get_items = { "retrieve-items" };

	get_subscriptions = { "retrieve-subscriptions" };

	node_defaults = { "retrieve-default" };

	publish = { "publish" };

	purge = { "purge-nodes" };

	retract = { "delete-items", "retract-items" };

	set_node_config = { "config-node" };

	set_affiliation = { "modify-affiliations" };

};

local service_config_feature_map = {

	autocreate_on_publish = { "auto-create" };

};

function _M.get_feature_set(service)

	local supported_features = set.new();

	for method, features in pairs(service_method_feature_map) do

		if service[method] then

			for _, feature in ipairs(features) do

				if feature then

					supported_features:add(feature);

				end

			end

		end

	end

	for option, features in pairs(service_config_feature_map) do

		if service.config[option] then

			for _, feature in ipairs(features) do

				if feature then

					supported_features:add(feature);

				end

			end

		end

	end

	for affiliation in pairs(service.config.capabilities) do

		if affiliation ~= "none" and affiliation ~= "owner" then

			supported_features:add(affiliation.."-affiliation");

		end

	end

	if service.node_defaults.access_model then

		supported_features:add("access-"..service.node_defaults.access_model);

	end

	if rawget(service.config, "itemstore") and rawget(service.config, "nodestore") then

		supported_features:add("persistent-items");

	end

	return supported_features;

end

function _M.handle_disco_info_node(event, service)

	local stanza, reply, node = event.stanza, event.reply, event.node;

	local ok, ret = service:get_nodes(stanza.attr.from);

	local node_obj = ret[node];

	if not ok or not node_obj then

		return;

	end

	event.exists = true;

	reply:tag("identity", { category = "pubsub", type = "leaf" }):up();

	if node_obj.config then

		reply:add_child(node_metadata_form:form({

			["pubsub#title"] = node_obj.config.title;

			["pubsub#description"] = node_obj.config.description;

		}, "result"));

	end

end

function _M.handle_disco_items_node(event, service)

	local stanza, reply, node = event.stanza, event.reply, event.node;

	local ok, ret = service:get_items(node, stanza.attr.from);

	if not ok then

		return;

	end

	for _, id in ipairs(ret) do

		reply:tag("item", { jid = module.host, name = id }):up();

	end

	event.exists = true;

end

function _M.handle_pubsub_iq(event, service)

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

	local pubsub_tag = stanza.tags[1];

	local action = pubsub_tag.tags[1];

	if not action then

		return origin.send(st.error_reply(stanza, "cancel", "bad-request"));

	end

	local prefix = "";

	if pubsub_tag.attr.xmlns == xmlns_pubsub_owner then

		prefix = "owner_";

	end

	local handler = handlers[prefix..stanza.attr.type.."_"..action.name];

	if handler then

		handler(origin, stanza, action, service);

		return true;

	end

end

function handlers.get_items(origin, stanza, items, service)

	local node = items.attr.node;

	local item = items:get_child("item");

	local item_id = item and item.attr.id;

	if not node then

		origin.send(pubsub_error_reply(stanza, "nodeid-required"));

		return true;

	end

	local ok, results = service:get_items(node, stanza.attr.from, item_id);

	if not ok then

		origin.send(pubsub_error_reply(stanza, results));

		return true;

	end

	local data = st.stanza("items", { node = node });

	for _, id in ipairs(results) do

		data:add_child(results[id]);

	end

	local reply;

	if data then

		reply = st.reply(stanza)

			:tag("pubsub", { xmlns = xmlns_pubsub })

				:add_child(data);

	else

		reply = pubsub_error_reply(stanza, "item-not-found");

	end

	origin.send(reply);

	return true;

end

function handlers.get_subscriptions(origin, stanza, subscriptions, service)

	local node = subscriptions.attr.node;

	local ok, ret = service:get_subscriptions(node, stanza.attr.from, stanza.attr.from);

	if not ok then

		origin.send(pubsub_error_reply(stanza, ret));

		return true;

	end

	local reply = st.reply(stanza)

		:tag("pubsub", { xmlns = xmlns_pubsub })

			:tag("subscriptions");

	for _, sub in ipairs(ret) do

		reply:tag("subscription", { node = sub.node, jid = sub.jid, subscription = 'subscribed' }):up();

	end

	origin.send(reply);

	return true;

end

function handlers.owner_get_subscriptions(origin, stanza, subscriptions, service)

	local node = subscriptions.attr.node;

	local ok, ret = service:get_subscriptions(node, stanza.attr.from);

	if not ok then

		origin.send(pubsub_error_reply(stanza, ret));

		return true;

	end

	local reply = st.reply(stanza)

		:tag("pubsub", { xmlns = xmlns_pubsub_owner })

			:tag("subscriptions");

	for _, sub in ipairs(ret) do

		reply:tag("subscription", { node = sub.node, jid = sub.jid, subscription = 'subscribed' }):up();

	end

	origin.send(reply);

	return true;

end

function handlers.owner_set_subscriptions(origin, stanza, subscriptions, service)

	local node = subscriptions.attr.node;

	if not node then

		origin.send(pubsub_error_reply(stanza, "nodeid-required"));

		return true;

	end

	if not service:may(node, stanza.attr.from, "subscribe_other") then

		origin.send(pubsub_error_reply(stanza, "forbidden"));

		return true;

	end

	local node_obj = service.nodes[node];

	if not node_obj then

		origin.send(pubsub_error_reply(stanza, "item-not-found"));

		return true;

	end

	for subscription_tag in subscriptions:childtags("subscription") do

		if subscription_tag.attr.subscription == 'subscribed' then

			local ok, err = service:add_subscription(node, stanza.attr.from, subscription_tag.attr.jid);

			if not ok then

				origin.send(pubsub_error_reply(stanza, err));

				return true;

			end

		elseif subscription_tag.attr.subscription == 'none' then

			local ok, err = service:remove_subscription(node, stanza.attr.from, subscription_tag.attr.jid);

			if not ok then

				origin.send(pubsub_error_reply(stanza, err));

				return true;

			end

		end

	end

	local reply = st.reply(stanza);

	origin.send(reply);

	return true;

end

function handlers.set_create(origin, stanza, create, service)

	local node = create.attr.node;

	local ok, ret, reply;

	local config;

	local configure = stanza.tags[1]:get_child("configure");

	if configure then

		local config_form = configure:get_child("x", "jabber:x:data");

		if not config_form then

			origin.send(st.error_reply(stanza, "modify", "bad-request", "Missing dataform"));

			return true;

		end

		local form_data, err = node_config_form:data(config_form);

		if not form_data then

			origin.send(st.error_reply(stanza, "modify", "bad-request", err));

			return true;

		end

		config = config_from_xep0060(form_data);