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

#pragma once

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

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;
};