-- This module allows you to probe the MUC presences for multiple occupants.

-- Copyright (C) 2020 JC Brand

local st = require "util.stanza";

local mod_muc = module:depends"muc";

local get_room_from_jid = rawget(mod_muc, "get_room_from_jid") or

	function (jid)

		local rooms = rawget(mod_muc, "rooms");

		return rooms[jid];

	end

module:log("debug", "Module loaded");

local function respondToBatchedProbe(event)

	local stanza = event.stanza;

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

		return;

	end

	local query = stanza:get_child("query", "http://jabber.org/protocol/muc#user");

	if not query then

		return;

	end;

	local origin = event.origin;

	local room = get_room_from_jid(stanza.attr.to);

	local probing_occupant = room:get_occupant_by_real_jid(stanza.attr.from);

	if probing_occupant == nil then

		origin.send(st.error_reply(stanza, "cancel", "not-acceptable", "You are not currently connected to this chat", room.jid));

		return true;

	end

	for item in query:children() do

		local probed_jid = item.attr.jid;

		local probed_occupant = room:get_occupant_by_nick(probed_jid);

		if probed_occupant == nil then

			local pr = room:build_unavailable_presence(probed_jid, stanza.attr.from);

			if pr then

				room:route_stanza(pr);

			end

		else

			local x = st.stanza("x", {xmlns = "http://jabber.org/protocol/muc#user"});

			room:publicise_occupant_status(probed_occupant, x, nil, nil, nil, nil, false, probing_occupant);

		end

	end

	origin.send(st.reply(stanza));

	return true;

end

module:hook("iq/bare", respondToBatchedProbe, 1);