HTTPResponse.cpp

#include "HTTPResponse.h"
 
using namespace std;
 
namespace framework
{
    string HTTPResponse::getFullDate()
    {
        return format("{:%a, %d %b %Y %H:%M:%OS} GMT", chrono::system_clock::now());
    }
 
    HTTPResponse::HTTPResponse()
    {
        this->setDefault();
    }
 
    HTTPResponse& HTTPResponse::operator = (const web::HTTPBuilder& builder)
    {
        this->builder = builder;
 
        return *this;
    }
 
    void HTTPResponse::setHTTPVersion(const string& version)
    {
        builder.HTTPVersion(version);
    }
 
    void HTTPResponse::setResponseCode(web::responseCodes code)
    {
        builder.responseCode(code);
    }
 
    void HTTPResponse::addHeader(const string& name, const string& value)
    {
        builder.headers
        (
            name, value
        );
    }
 
    void HTTPResponse::addBody(const string& body)
    {
        this->body = body;
    }
 
    void HTTPResponse::addBody(const json::JSONBuilder& json)
    {
        builder.headers
        (
            "Content-Type", "application/json"
        );
 
        this->body = json.build();
    }
 
    void HTTPResponse::addBody(string&& body) noexcept
    {
        this->body = move(body);
    }
 
    HTTPResponse& HTTPResponse::appendBody(const std::string& body)
    {
        this->body += body;
 
        return *this;
    }
 
    void HTTPResponse::addCookie(const std::string& name, const std::string& value)
    {
        builder.headers
        (
            "Set-Cookie", name + "=" + value
        );
    }
 
    void HTTPResponse::setDefault()
    {
        builder.clear();
 
        builder.responseCode(web::responseCodes::ok);
 
        body.clear();
 
        this->setIsValid(true);
    }
 
    void HTTPResponse::setIsValid(bool isValid)
    {
        this->isValid = isValid;
    }
 
    HTTPResponse::operator bool() const
    {
        return isValid;
    }
 
    streams::IOSocketStream& operator << (streams::IOSocketStream& stream, HTTPResponse& response)
    {
        string result;
 
        response.builder.headers
        (
            "Date", HTTPResponse::getFullDate(),
            "Server", "WebFramework-Server"
        );
 
        if (response.body.size())
        {
            result = response.builder.build(response.body);
        }
        else
        {
            result = response.builder.build();
        }
 
        stream << result;
 
        return stream;
    }
}