#include "httpd.h"

#include <boost/bind.hpp>

HTTP::Server::Server(boost::asio::io_service& io_service, const tcp::endpoint& endpoint) : acceptor_(io_service, endpoint) {
	start_accept();
}

void HTTP::Server::add_handler(const std::string& name, Handler handler) {
	handlers[name] = handler;
}

void HTTP::Server::start_accept() {
	Connection::p new_connection = Connection::p(new Connection(acceptor_.io_service()));
	acceptor_.async_accept(new_connection->socket, boost::bind(&Server::handle_accept, this, new_connection, boost::asio::placeholders::error));
}

void HTTP::Server::handle_accept(Connection::p new_connection, const boost::system::error_code& error) {
	if(!error) {
		new_connection->read_request(boost::bind(&Server::handle_request, this, _1));
		start_accept();
	}
}

void HTTP::Server::handle_request(Connection::p connection) {
	std::string handler;
	
	if(connection->path.size()) {
		// Pop first element of path.
		handler = connection->path.front();
		connection->path.erase(connection->path.begin());
	} else {
		handler = "index";
	}
	
	if(handlers.count(handler)) {
		// Call handler.
		handlers[handler](connection);
	} else {
		// Error 404.
	}
}