-- (C) 2010-2011 Marco Cirillo (LW.Org)
-- (C) 2011 Kim Alvefur
--
-- 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", nil)
	local inst_text = module:get_option_string("registration_text", nil)
	local oob = module:get_option_boolean("registration_oob", 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_l = 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().."'s 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().."'s 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().."'s 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)