Request.hpp

#pragma once
 
#include <boost/beast/http/field.hpp>
#include <boost/beast/http/message.hpp>
#include <boost/beast/http/string_body.hpp>
 
#include <functional>
#include <optional>
#include <string>
#include <string_view>
#include <variant>
 
namespace web::ng {

class Request {
public:
    using HttpHeaders = boost::beast::http::request<boost::beast::http::string_body>::header_type;
 
private:
    struct WsData {
        std::string request;
        std::reference_wrapper<HttpHeaders const> headers;
    };
 
    using HttpRequest = boost::beast::http::request<boost::beast::http::string_body>;
    std::variant<HttpRequest, WsData> data_;
 
public:
    explicit Request(boost::beast::http::request<boost::beast::http::string_body> request);

    Request(std::string request, HttpHeaders const& headers);

    enum class Method { Get, Post, Websocket, Unsupported };

    [[nodiscard]] Method
    method() const;

    [[nodiscard]] bool
    isHttp() const;

    [[nodiscard]] std::optional<
        std::reference_wrapper<boost::beast::http::request<boost::beast::http::string_body> const>>
    asHttpRequest() const;

    [[nodiscard]] std::string_view
    message() const;

    [[nodiscard]] std::optional<std::string_view>
    target() const;

    [[nodiscard]] HttpHeaders const&
    httpHeaders() const;

    [[nodiscard]] std::optional<std::string_view>
    headerValue(boost::beast::http::field headerName) const;

    [[nodiscard]] std::optional<std::string_view>
    headerValue(std::string const& headerName) const;
 
private:
    [[nodiscard]] HttpRequest const&
    httpRequest() const;
};
 
}  // namespace web::ng