-- Prosody IM

-- Copyright (C) 2008-2017 Matthew Wild

-- Copyright (C) 2008-2017 Waqas Hussain

-- Copyright (C) 2011-2017 Kim Alvefur

-- Copyright (C) 2018 Emmanuel Gil Peyrot

--

-- This project is MIT/X11 licensed. Please see the

-- COPYING file in the source package for more information.

--

-- XEP-0313: Message Archive Management for Prosody

--

local st = require"util.stanza";

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

local xmlns_mam = "urn:xmpp:mam:2";

local default_attrs = {

	always = true, [true] = "always",

	never = false, [false] = "never",

	roster = "roster",

}

local function tostanza(prefs)

	local default = prefs[false];

	default = default_attrs[default];

	local prefstanza = st.stanza("prefs", { xmlns = xmlns_mam, default = default });

	local always = st.stanza("always");

	local never = st.stanza("never");

	for jid, choice in pairs(prefs) do

		if jid then

			(choice and always or never):tag("jid"):text(jid):up();

		end

	end

	prefstanza:add_child(always):add_child(never);

	return prefstanza;

end

local function fromstanza(prefstanza)

	local prefs = {};

	local default = prefstanza.attr.default;

	if default then

		prefs[false] = default_attrs[default];

	end

	local always = prefstanza:get_child("always");

	if always then

		for rule in always:childtags("jid") do

			local jid = jid_prep(rule:get_text());

			if jid then

				prefs[jid] = true;

			end

		end

	end

	local never = prefstanza:get_child("never");

	if never then

		for rule in never:childtags("jid") do

			local jid = jid_prep(rule:get_text());

			if jid then

				prefs[jid] = false;

			end

		end

	end

	return prefs;

end

return {

	tostanza = tostanza;

	fromstanza = fromstanza;

}