local st = require "util.stanza";

local whitelist = module:get_option_inherited_set("s2s_whitelist", {});

module:hook("route/remote", function (event)

	if not whitelist:contains(event.to_host) then

		module:send(st.error_reply(event.stanza, "cancel", "not-allowed", "Communication with this domain is restricted"));

		return true;

	end

end, 100);

module:hook("s2s-stream-features", function (event)

	if not whitelist:contains(event.origin.from_host) then

		event.origin:close({

			condition = "policy-violation";

			text = "Communication with this domain is restricted";

		});

	end

end, 1000);