Response.hpp

#pragma once
 
#include "web/ng/Request.hpp"
 
#include <boost/asio/buffer.hpp>
#include <boost/beast/http/message.hpp>
#include <boost/beast/http/status.hpp>
#include <boost/beast/http/string_body.hpp>
#include <boost/json/object.hpp>
 
#include <string>
#include <variant>
 
namespace web::ng {
 
class Connection;

class Response {
public:
    std::variant<boost::beast::http::response<boost::beast::http::string_body>, std::string> data;
 
public:
    Response(boost::beast::http::status status, std::string message, Request const& request);

    Response(
        boost::beast::http::status status,
        boost::json::object const& message,
        Request const& request
    );

    Response(
        boost::beast::http::status status,
        boost::json::object const& message,
        Connection const& connection
    );

    Response(boost::beast::http::status status, std::string message, Connection const& connection);

    Response(
        boost::beast::http::response<boost::beast::http::string_body> response,
        Request const& request
    );

    [[nodiscard]] std::string const&
    message() const;

    void
    setMessage(std::string newMessage);

    void
    setMessage(boost::json::object const& newMessage);

    boost::beast::http::response<boost::beast::http::string_body>
    intoHttpResponse() &&;

    [[nodiscard]] boost::asio::const_buffer
    asWsResponse() const&;
};
 
}  // namespace web::ng