#include "http.h"

#include <boost/format.hpp>
#include <boost/algorithm/string.hpp>
#include <boost/algorithm/string/regex.hpp>

#include <vector>

HTTPRequest::HTTPRequest(std::istream& is) {
	std::string firstline;
	std::getline(is, firstline);

	std::vector<std::string> splitvec;
	boost::algorithm::split(splitvec, firstline, boost::algorithm::is_space());

	type = splitvec[0];
	path = splitvec[1];
	httpver = splitvec[2];
	std::cout << boost::format("%s %s %s\n") % type % path % httpver;

	while(is.good()) {
		std::string line;
		std::getline(is, line);
		boost::trim(line);
		if(!line.size()) continue;
		std::vector<std::string> v;
		boost::algorithm::split_regex(v, line, boost::regex(": "));
		headers[v[0]] = v[1];
	}
}

HTTPResponse::HTTPResponse(boost::asio::ip::tcp::socket& socket_) : socket(socket_){
	httpver = "1.1";
	headers_written = false;
}

void HTTPResponse::add_header(std::string key, std::string value) {
	headers[key] = value;
}

void HTTPResponse::write_headers() {
	write(boost::str(boost::format("HTTP/%s %d %s\r\n") % httpver % code % status));
	for(HTTPHeaders::iterator it = headers.begin(); it != headers.end(); it++) {
		write(boost::str(boost::format("%s: %s\r\n") % it->first % it->second));
	}
	write("\r\n");
}

void HTTPResponse::write(char *data, unsigned int len) {
	write(std::string(data, len));
}

void HTTPResponse::write(std::string str) {
	if(!headers_written) {
		// make sure to set headers_written before calling write_headers
		headers_written = true;
		write_headers();
	}
	boost::asio::streambuf b;
	std::ostream os(&b);
	os << str;
	boost::asio::write(socket, b);
}