summaryrefslogtreecommitdiff
path: root/httpd.cpp
blob: 8eac976805c8843be32eb635591e9380f72f99f8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#include "httpd.h"
#include "music.h"
#include "http.h"

#include <boost/bind.hpp>
#include <boost/algorithm/string/split.hpp>
#include <boost/algorithm/string/classification.hpp>
#include <boost/algorithm/string/predicate.hpp>
#include <boost/format.hpp>

#include <iostream>

HTTPConnection::HTTPConnection(boost::asio::io_service& io_service) : socket(io_service) {
}

void HTTPConnection::handle_write(const boost::system::error_code& error, size_t bytes_transferred) {
}

void HTTPConnection::handle_read(const boost::system::error_code& error, size_t bytes_transferred) {
	std::istream is(&buf);

	HTTPRequest req(is);

	boost::asio::streambuf b;
	std::ostream os(&b);

	HTTPResponse res;

	MusicListing *ml = music::find(req.path);
	if(ml) {
		res.code = 200;
		res.status = "OK";
		res.add_header("content-type", "text/html");
		res.write_headers(os);

		ml->render(os);
	} else {
		res.code = 404;
		res.status = "Not Found";
		res.write_headers(os);
	}

	boost::asio::write(socket, b);
}

HTTPConnection::pointer HTTPConnection::create(boost::asio::io_service& io_service) {
	return pointer(new HTTPConnection(io_service));
}

void HTTPConnection::start() {
	boost::asio::async_read_until(socket, buf, "\r\n\r\n", boost::bind(&HTTPConnection::handle_read, shared_from_this(),
				boost::asio::placeholders::error, boost::asio::placeholders::bytes_transferred));
}

HTTPServer::HTTPServer(boost::asio::io_service& io_service) : acceptor_(io_service, tcp::endpoint(tcp::v4(), 8000)) {
	start_accept();
}

void HTTPServer::start_accept() {
	HTTPConnection::pointer new_connection = HTTPConnection::create(acceptor_.io_service());
	acceptor_.async_accept(new_connection->socket, boost::bind(&HTTPServer::handle_accept, this, new_connection, boost::asio::placeholders::error));
}

void HTTPServer::handle_accept(HTTPConnection::pointer new_connection, const boost::system::error_code& error) {
	if(!error) {
		new_connection->start();
		start_accept();
	}
}