rcpp_framework/libs/brynet/net/http/HttpFormat.hpp

#pragma once

#include <array>
#include <cassert>
#include <map>
#include <string>

namespace brynet { namespace net { namespace http {

class HttpQueryParameter final
{
public:
    void add(const std::string& k, const std::string& v)
    {
        if (!mParameter.empty())
        {
            mParameter += "&";
        }

        mParameter += k;
        mParameter += "=";
        mParameter += v;
    }

    const std::string& getResult() const
    {
        return mParameter;
    }

private:
    std::string mParameter;
};

class HttpRequest final
{
public:
    enum class HTTP_METHOD
    {
        HTTP_METHOD_HEAD,
        HTTP_METHOD_GET,
        HTTP_METHOD_POST,
        HTTP_METHOD_PUT,
        HTTP_METHOD_DELETE,
        HTTP_METHOD_MAX
    };

    HttpRequest()
    {
        setMethod(HTTP_METHOD::HTTP_METHOD_GET);
    }

    void setMethod(HTTP_METHOD protocol)
    {
        mMethod = protocol;
        assert(mMethod > HTTP_METHOD::HTTP_METHOD_HEAD &&
               mMethod < HTTP_METHOD::HTTP_METHOD_MAX);
    }

    void setHost(const std::string& host)
    {
        addHeadValue("Host", host);
    }

    void setUrl(const std::string& url)
    {
        mUrl = url;
    }

    void setCookie(const std::string& v)
    {
        addHeadValue("Cookie", v);
    }

    void setContentType(const std::string& v)
    {
        addHeadValue("Content-Type", v);
    }

    void setQuery(const std::string& query)
    {
        mQuery = query;
    }

    void setBody(const std::string& body)
    {
        addHeadValue("Content-Length", std::to_string(body.size()));
        mBody = body;
    }

    void setBody(std::string&& body)
    {
        addHeadValue("Content-Length", std::to_string(body.size()));
        mBody = std::move(body);
    }

    void addHeadValue(const std::string& field,
                      const std::string& value)
    {
        mHeadField[field] = value;
    }

    std::string getResult() const
    {
        const auto MethodMax = static_cast<size_t>(HTTP_METHOD::HTTP_METHOD_MAX);
        const static std::array<std::string, MethodMax> HttpMethodString =
                {"HEAD", "GET", "POST", "PUT", "DELETE"};

        std::string ret;
        if (mMethod >= HTTP_METHOD::HTTP_METHOD_HEAD &&
            mMethod < HTTP_METHOD::HTTP_METHOD_MAX)
        {
            ret += HttpMethodString[static_cast<size_t>(mMethod)];
        }

        ret += " ";
        ret += mUrl;
        if (!mQuery.empty())
        {
            ret += "?";
            ret += mQuery;
        }

        ret += " HTTP/1.1\r\n";

        for (auto& v : mHeadField)
        {
            ret += v.first;
            ret += ": ";
            ret += v.second;
            ret += "\r\n";
        }

        ret += "\r\n";

        if (!mBody.empty())
        {
            ret += mBody;
        }

        return ret;
    }

private:
    std::string mUrl;
    std::string mQuery;
    std::string mBody;
    HTTP_METHOD mMethod;
    std::map<std::string, std::string> mHeadField;
};

class HttpResponse final
{
public:
    enum class HTTP_RESPONSE_STATUS
    {
        NONE,
        OK = 200,
    };

    HttpResponse()
        : mStatus(HTTP_RESPONSE_STATUS::OK)
    {
    }

    void setStatus(HTTP_RESPONSE_STATUS status)
    {
        mStatus = status;
    }

    void setContentType(const std::string& v)
    {
        addHeadValue("Content-Type", v);
    }

    void addHeadValue(const std::string& field,
                      const std::string& value)
    {
        mHeadField[field] = value;
    }

    void setBody(const std::string& body)
    {
        addHeadValue("Content-Length", std::to_string(body.size()));
        mBody = body;
    }

    void setBody(std::string&& body)
    {
        addHeadValue("Content-Length", std::to_string(body.size()));
        mBody = std::move(body);
    }

    std::string getResult() const
    {
        std::string ret = "HTTP/1.1 ";

        ret += std::to_string(static_cast<int>(mStatus));
        switch (mStatus)
        {
            case HTTP_RESPONSE_STATUS::OK:
                ret += " OK";
                break;
            default:
                ret += "UNKNOWN";
                break;
        }

        ret += "\r\n";

        for (auto& v : mHeadField)
        {
            ret += v.first;
            ret += ": ";
            ret += v.second;
            ret += "\r\n";
        }

        ret += "\r\n";

        if (!mBody.empty())
        {
            ret += mBody;
        }

        return ret;
    }

private:
    HTTP_RESPONSE_STATUS mStatus;
    std::map<std::string, std::string> mHeadField;
    std::string mBody;
};

}}}// namespace brynet::net::http