1 files changed, 10 insertions, 3 deletions
diff --git a/http.h b/http.h
index f7aec59..3ab66dc 100644
--- a/http.h
+++ b/http.h
@@ -1,6 +1,8 @@
 #ifndef HTTP_H
 #define HTTP_H
 
+#include <boost/asio.hpp>
+
 #include <iostream>
 #include <string>
 #include <map>
@@ -15,13 +17,18 @@ class HTTPRequest {
 };
 
 class HTTPResponse {
-	public:
+	private:
 		HTTPHeaders headers;
+		boost::asio::ip::tcp::socket& socket;
+		void write_headers();
+		bool headers_written;
+	public:
 		int code;
 		std::string httpver, status;
-		HTTPResponse();
+		HTTPResponse(boost::asio::ip::tcp::socket& socket_);
 		void add_header(std::string key, std::string value);
-		void write_headers(std::ostream& os);
+		void write(char *data, unsigned int len);
+		void write(std::string str);
 };
 
 #endif