-- Prosody IM

-- Copyright (C) 2008-2010 Matthew Wild

-- Copyright (C) 2008-2010 Waqas Hussain

-- 

-- This project is MIT/X11 licensed. Please see the

-- COPYING file in the source package for more information.

--

local log = require "util.logger".init("httpclient_listener");

local t_concat, t_insert = table.concat, table.insert;

local connlisteners_register = require "net.connlisteners".register;

local requests = {}; -- Open requests

local buffers = {}; -- Buffers of partial lines

local httpclient = { default_port = 80, default_mode = "*a" };

function httpclient.onconnect(conn)

	local req = requests[conn];

	-- Send the request

	local request_line = { req.method or "GET", " ", req.path, " HTTP/1.1\r\n" };

	if req.query then

		t_insert(request_line, 4, "?"..req.query);

	end

	conn:write(t_concat(request_line));

	local t = { [2] = ": ", [4] = "\r\n" };

	for k, v in pairs(req.headers) do

		t[1], t[3] = k, v;

		conn:write(t_concat(t));

	end

	conn:write("\r\n");

	if req.body then

		conn:write(req.body);

	end

end

function httpclient.onincoming(conn, data)

	local request = requests[conn];

	if not request then

		log("warn", "Received response from connection %s with no request attached!", tostring(conn));

		return;

	end

	if data and request.reader then

		request:reader(data);

	end

end

function httpclient.ondisconnect(conn, err)

	local request = requests[conn];

	if request and request.conn then

		request:reader(nil);

	end

	requests[conn] = nil;

end

function httpclient.register_request(conn, req)

	log("debug", "Attaching request %s to connection %s", tostring(req.id or req), tostring(conn));

	requests[conn] = req;

end

connlisteners_register("httpclient", httpclient);