local st = require "util.stanza";

local xmlns_hats = "xmpp:prosody.im/protocol/hats:1";

module:hook("muc-broadcast-presence", function (event)
	-- Strip any hats claimed by the client (to prevent spoofing)
	event.stanza:remove_children("hats", xmlns_hats);

	local aff_data = event.room:get_affiliation_data(event.occupant.bare_jid);
	local hats = aff_data and aff_data.hats;
	if not hats then return; end
	local hats_el;
	for hat_id, hat_data in pairs(hats) do
		if hat_data.active then
			if not hats_el then
				hats_el = st.stanza("hats", { xmlns = xmlns_hats });
			end
			hats_el:tag("hat", { uri = hat_id, title = hat_data.title }):up();
		end
	end
	if not hats_el then return; end
	event.stanza:add_direct_child(hats_el);
end);