local adns = require "net.adns";

local rbl = module:get_option_string("registration_rbl");

local rbl_message = module:get_option_string("registration_rbl_message");

local st = require "util.stanza";

local function cleanup_ip(ip)

	if ip:sub(1,7):lower() == "::ffff:" then

		return ip:sub(8);

	end

	return ip;

end

local function reverse(ip, suffix)

	local a,b,c,d = ip:match("^(%d+).(%d+).(%d+).(%d+)$");

	if not a then return end

	return ("%d.%d.%d.%d.%s"):format(d,c,b,a, suffix);

end

module:hook("user-registered", function (event)

	local session = event.session;

	local ip = session and session.ip and cleanup_ip(session.ip);

	local rbl_ip = ip and reverse(ip, rbl);

	if rbl_ip then

		local registration_time = os.time();

		local log = session.log;

		adns.lookup(function (reply)

			if reply and reply[1] then

				log("warn", "Account %s@%s registered from IP %s found in RBL (%s)", event.username, event.host or module.host, ip, reply[1].a);

				local user = prosody.bare_sessions[event.username .. "@" .. module.host];

				if user and user.firewall_marks then

					user.firewall_marks.dnsbl_hit = registration_time;

				else

					module:open_store("firewall_marks", "map"):set(event.username, "dnsbl_hit", registration_time);

				end

				if rbl_message then

					module:log("debug", "Warning RBL registered user %s@%s", event.username, event.host);

					event.ip = ip;

					local rbl_stanza =

						st.message({ to = event.username.."@"..event.host, from = event.host },

							rbl_message:gsub("$(%w+)", event));

					module:send(rbl_stanza);

				end

			end

		end, rbl_ip);

	end

end);