Changeset

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_register_redirect/mod_register_redirect.lua	Thu Sep 22 15:03:51 2011 +0000
@@ -0,0 +1,93 @@
+-- Registration Redirect module for Prosody
+-- 
+-- Redirects IP addresses not in the whitelist to a web page or another method to complete the registration.
+
+local st = require "util.stanza";
+local cman = require "core.configmanager";
+
+function reg_redirect(event)
+	local stanza, origin = event.stanza, event.origin;
+	local ip_wl = module:get_option("registration_whitelist") or { "127.0.0.1" };
+	local url = module:get_option_string("registration_url" or nil);
+	local inst_text = module:get_option_string("registration_text" or nil);
+	local oob = module:get_option_boolean("registration_oob" or true);
+	local admins_g = cman.get("*", core, "admins");
+	local admins_l = cman.get(module:get_host(), core, "admins");
+	local no_wl = module:get_option_boolean("no_registration_whitelist", false);
+	local test_ip = false;
+
+	if type(admins_g) ~= "table" then admins_g = nil; end
+	if type(admins_l) ~= "table" then admins_g = nil; end
+
+	-- perform checks to set default responses and sanity checks.
+	if not inst_text then
+		if url and oob then
+			if url:match("^%w+[:].*$") then
+				if url:match("^(%w+)[:].*$") == "http" or url:match("^(%w+)[:].*$") == "https" then
+					inst_text = "Please visit "..url.." to register an account on this server."
+				elseif url:match("^(%w+)[:].*$") == "mailto" then
+					inst_text = "Please send an e-mail at "..url:match("^%w+[:](.*)$").." to register an account on this server."
+				elseif url:match("^(%w+)[:].*$") == "xmpp" then
+					inst_text = "Please contact "..module:get_host().." server administrator via xmpp to register an account on this server at: "..url:match("^%w�+[:](.*)$")
+				else
+					module:log("error", "This module supports only http/https, mailto or xmpp as URL formats.")
+					module:log("error", "If you want to use personalized instructions without an Out-Of-Band method,")
+					module:log("error", "specify: register_oob = false; -- in your configuration along your banner string (register_text).")
+					origin.send(st.error_reply(stanza, "wait", "internal-server-error")); return true -- bouncing request.
+				end
+			else
+				module:log("error", "Please check your configuration, the URL you specified is invalid")
+				origin.send(st.error_reply(stanza, "wait", "internal-server-error")); return true -- bouncing request.
+			end
+		else
+			if admins_l then
+				local ajid; for _,v in ipairs(admins_l) do ajid = v; break; end
+				inst_text = "Please contact "..module:get_host().." server administrator via xmpp to register an account on this server at: "..ajid
+			else
+				if admins_g then
+					local ajid; for _,v in ipairs(admins_g) do ajid = v; break; end
+					inst_text = "Please contact "..module:get_host().." server administrator via xmpp to register an account on this server at: "..ajid
+				else
+					module:log("error", "Please be sure to, _at the very least_, configure one server administrator either global or hostwise...")
+					module:log("error", "if you want to use this module, or read it's configuration wiki at: http://code.google.com/p/prosody-modules/wiki/mod_register_redirect")
+					origin.send(st.error_reply(stanza, "wait", "internal-server-error")) -- bouncing request.
+					return true;
+				end
+			end
+		end
+	elseif text and oob then
+		if not url:match("^%w+[:].*$") then
+			module:log("error", "Please check your configuration, the URL specified is not valid.")
+			origin.send(st.error_reply(stanza, "wait", "internal-server-error")) -- bouncing request.
+		end
+	end
+
+	if not no_wl then
+		for i,ip in ipairs(ip_wl) do
+			if origin.ip == ip then test_ip = true; end
+			break;
+		end
+	end
+
+	-- Prepare replies.
+	local reply = st.reply(event.stanza);
+	if oob then
+		reply:query("jabber:iq:register")
+			:tag("instructions"):text(inst_text):up()
+			:tag("x", {xmlns = "jabber:x:oob"})
+				:tag("url"):text(url);
+	else
+		reply:query("jabber:iq:register")
+			:tag("instructions"):text(inst_text):up()
+	end
+	
+	if stanza.attr.type == "get" then
+		origin.send(reply);
+		return true;
+	else
+		origin.send(st.error_reply(stanza, "cancel", "not-authorized"))
+		return true
+	end
+end
+
+module:hook("stanza/iq/jabber:iq:register:query", reg_redirect, 10);