feat : add cookies to the client response

Ziko909 · May 23, 2023 · 1c7daaf · 1c7daaf
1 parent 32168ba
commit 1c7daaf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/include/Response.hpp b/include/Response.hpp
@@ -21,6 +21,7 @@ class Response
 		std::string _contentType;
 		size_t _contentLength;
 		std::string _location;
+		std::vector<std::string> _cookies;
 
 		void error(int status);
 		void error(int status, const Location &location, const Server &server);

diff --git a/src/Response.cpp b/src/Response.cpp
@@ -215,6 +215,7 @@ void Response::generateResponse(Request &req, const Server &server, const client
 				_status = cgi.getStatus();
 				_contentLength = cgi.getContentLength();
 				_contentType = cgi.getContentType();
+				_cookies = cgi.getCookies();
 				_body = cgi.getBody();
 				return ;
 			} catch (int &status) {
@@ -251,6 +252,11 @@ std::string Response::toString() const
 	{
 		stream << "Location: " << _location << "\r\n";
 	}
+	if (not _cookies.empty())
+	{
+		for (std::vector<std::string>::const_iterator it = _cookies.begin(); it != _cookies.end(); it++)
+			stream << "Set-Cookie: " << *it << "\r\n";
+	}
 	stream << "\r\n";
 	if (_status != NO_CONTENT and _status != CREATED)
 		stream << _body;
@@ -351,6 +357,7 @@ void Response::GET(Request &req, const Server &server, const client_info &client
 			cgi.run(req, server, client, path);
 			_contentLength = cgi.getContentLength();
 			_contentType = cgi.getContentType();
+			_cookies = cgi.getCookies();
 			_status = cgi.getStatus();
 			_body = cgi.getBody();
 			return ;