local connlisteners_start = require "net.connlisteners".start;

local serialize = require "util.serialization".serialize;

local sha1 = require "util.hashes".sha1;

local _config = {};

local transfers = {};

_config.port = config_get(_host, "core", "port");

_config.interface = config_get(_host, "core", "interface");

local function bin2hex(bin)

	return bin:gsub(".", function (c) return ("%02x"):format(c:byte()); end)

end

function new_session(conn)

	local w = function(s) conn.write(s:gsub("\n", "\r\n")); end;

	local session = { conn = conn;

			send = function (t) w(tostring(t)); end;

			print = function (t) w("| "..tostring(t).."\n"); end;

			disconnect = function () conn.close(); end;

			};

	return session;

end

function connlistener.listener(conn, data)

	module:log("debug", "listener called....")

	local session = sessions[conn];

	if data ~= nil then module:log("debug", bin2hex(data)); end

	if not session and data ~= nil and data:byte() == string.char(5):byte() and data:len() > 2 then

		local nmethods = data:sub(2):byte();

		local methods = data:sub(3);

		local supported = false;

		for i=1, nmethods, 1 do

			if(methods:sub(i):byte() == string.char(0):byte()) then

				supported = true;

				break;

			end

		end

		if(supported) then

			module:log("debug", "new session found ... ")

			session = new_session(conn);

			sessions[conn] = session;

			session.send(string.char(5, 0));

		end

	elseif data ~= nil and data:len() > 6 and 

		data:sub(1):byte() == string.char(5):byte() and -- SOCKS5 has 5 in first byte

		data:sub(2):byte() == string.char(1):byte() and -- CMD must be 1

		data:sub(3):byte() == string.char(0):byte() and -- RSV must be 0

		data:sub(4):byte() == string.char(3):byte() and -- ATYP must be 3		

		data:sub(-2):byte() == string.char(0):byte() and data:sub(-1):byte() == string.char(0):byte() -- PORT must be 0, size 2 byte		

	then

		local sha = data:sub(6, data:len() - 2);

		module:log("debug", "gotten sha: >%s<", sha);

		if transfers[sha] == nil then

			transfers[sha] = {};

			transfers[sha].target = conn;

			module:log("debug", "target connected ... ");

		elseif transfers[sha].target ~= nil then

			transfers[sha].initiator = conn;

			module:log("debug", "initiator connected ... ");

		end

		session.send(string.char(5, 0, 0, 3, 40) .. sha .. string.char(0, 0)); -- VER, REP, RSV, ATYP, BND.ADDR (sha), BND.PORT (2 Byte)

	end

end

function connlistener.disconnect(conn, err)

end

local function get_disco_info(stanza)

	local reply = replies_cache.disco_info;

	if reply == nil then

	 	reply = st.iq({type='result', from=_host}):query("http://jabber.org/protocol/disco#info")

			:tag("identity", {category='proxy', type='bytestreams', name=_name}):up()

			:tag("feature", {var="http://jabber.org/protocol/bytestreams"});

		replies_cache.disco_info = reply;

	end

	reply.attr.id = stanza.attr.id;

	reply.attr.to = stanza.attr.from;

	return reply;

end

local function get_disco_items(stanza)

	local reply = replies_cache.disco_items;

	if reply == nil then

	 	reply = st.iq({type='result', from=_host}):query("http://jabber.org/protocol/disco#items");

		replies_cache.disco_info = reply;

	end

	reply.attr.id = stanza.attr.id;

	reply.attr.to = stanza.attr.from;

	return reply;

end

local function get_stream_host(stanza)

	local reply = replies_cache.stream_host;

	local sid = stanza.tags[1].attr.sid;

	if reply == nil then

		reply = st.iq({type="result", from=_host})

			:query("http://jabber.org/protocol/bytestreams")

			:tag("streamhost", {jid=_host, host=_config.interface, port=_config.port}); -- TODO get the correct data

		replies_cache.stream_host = reply;

	end

	reply.attr.id = stanza.attr.id;

	reply.attr.to = stanza.attr.from;

	reply.tags[1].attr.sid = sid;

	return reply;

end

module.unload = function()

	component_deregister(_host);

	connlisteners_deregister("proxy65");

end

local function set_activation(stanza)

	local from = nil;

	local to = nil;

	local sid = nil;

	local reply = nil;

	if stanza.attr ~= nil then

		from = stanza.attr.from;

	end

	if stanza.tags[1] ~= nil and tostring(stanza.tags[1].name) == "query" then

		if stanza.tags[1].attr ~= nil then

			sid = stanza.tags[1].attr.sid;

		end

		if stanza.tags[1].tags[1] ~= nil and tostring(stanza.tags[1].tags[1].name) == "activate" then

			to = stanza.tags[1].tags[1][1];

		end

	end

	if from ~= nil and to ~= nil and sid ~= nil then

		reply = st.iq({type="result", from=_host});

		reply.attr.id = stanza.attr.id;

	end

	return reply, from, to, sid;

end

local function forward(initiator, target)

	module:log("debug", "forward it ....");

end

	connlistener.default_port = _config.port;

	connlistener.default_interface = "*"; 

	connlistener.default_mode = "*a";

		connlistener.handler = connlisteners_start('proxy65');

		module:log("debug", "Connection listener registered ... ")

					elseif xmlns == "http://jabber.org/protocol/bytestreams" then

				elseif stanza.name == "iq" and type == "set" then

					local reply, from, to, sid = set_activation(stanza);

					if reply ~= nil and from ~= nil and to ~= nil and sid ~= nil then

						module:log("debug", "need to build sha1 of data: from: %s, to: %s, sid: %s", from, to, sid);

						local sha = sha1(sid .. from .. to, true);

						module:log("debug", "generated sha: %s", sha);

						if(transfers[sha] ~= nil and transfers[sha].initiator ~= nil and transfers[sha].target ~= nil) then

							origin.send(reply);

							forward(transfers[sha].initiator, transfers[sha].target);

							transfers[sha] = nil;

						end

					end