From 82c2af5878bb4c695a48c2d0707dffdb9356fce3 Mon Sep 17 00:00:00 2001
From: cflip <cflip@cflip.net>
Date: Thu, 7 Jul 2022 12:06:33 -0600
Subject: Add some more abstractions to HttpResponse

---
 src/HttpResponse.h | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

(limited to 'src/HttpResponse.h')

diff --git a/src/HttpResponse.h b/src/HttpResponse.h
index 33148da..8a1d2d4 100644
--- a/src/HttpResponse.h
+++ b/src/HttpResponse.h
@@ -1,18 +1,22 @@
 #pragma once
 
-#include <sstream>
+#include <map>
 #include <string>
 
-enum class ResponseCode {
+enum class HttpStatusCode {
 	OK = 200
 };
 
 class HttpResponse {
 public:
-	HttpResponse(ResponseCode);
+	HttpResponse(HttpStatusCode status_code);
+
+	void add_header(const std::string& header, const std::string& value);
+	void set_content(const std::string& content) { m_content = content; }
 
-	void add_header(std::string header);
 	std::string to_string() const;
 private:
-	std::stringstream m_string_stream;
+	HttpStatusCode m_status_code;
+	std::map<std::string, std::string> m_headers;
+	std::string m_content;
 };
-- 
cgit v1.2.3