local st = require "util.stanza";

local jid = require "util.jid";

local nodeprep = require "util.encodings".stringprep.nodeprep;

local rooms = module:shared "muc/rooms";

if not rooms then

        module:log("error", "This module only works on MUC components!");

        return;

end

local restrict_patterns = module:get_option("muc_restrict_matching", {});

local restrict_excepts = module:get_option_set("muc_restrict_exceptions", {});

local restrict_allow_admins = module:get_option_boolean("muc_restrict_allow_admins", false);

local function is_restricted(room, who)

	-- If admins can join prohibited rooms, we allow them to

	if restrict_allow_admins and usermanager.is_admin(who, module.host) then

		module:log("debug", "Admins are allowed to enter restricted rooms (%s on %s)", who, room)

		return nil;

	end

	-- Don't evaluate exceptions

	if restrict_excepts:contains(room) then

		module:log("debug", "Room %s is amongst restriction exceptions", room())

		return nil;

	end

	-- Evaluate regexps of restricted patterns

        for pattern,reason in pairs(restrict_patterns) do

                if room:match(pattern) then

			module:log("debug", "Room %s is restricted by pattern %s, user %s is not allowed to join (%s)", room, pattern, who, reason)

                        return reason;

                end

        end

	return nil

end

module:hook("presence/full", function(event)

        local stanza = event.stanza;

        if stanza.name == "presence" and stanza.attr.type == "unavailable" then   -- Leaving events get discarded

                return;

        end

	-- Get the room

	local room = jid.split(stanza.attr.to);

        if not room then return; end

	-- Get who has tried to join it

	local who = jid.bare(stanza.attr.from)

	-- Checking whether room is restricted

	local check_restricted = is_restricted(room, who)

        if check_restricted ~= nil then

                event.allowed = false;

                event.stanza.attr.type = 'error';

	        return event.origin.send(st.error_reply(event.stanza, "cancel", "forbidden", "You're not allowed to enter this room: " .. check_restricted));

        end

end, 10);