RequestBuilder.hpp

#pragma once
 
#include "util/log/Logger.hpp"
#include "util/requests/Types.hpp"
 
#include <boost/asio/spawn.hpp>
#include <boost/asio/ssl/context.hpp>
#include <boost/beast.hpp>
#include <boost/beast/core/error.hpp>
#include <boost/beast/http/field.hpp>
#include <boost/beast/http/message.hpp>
#include <boost/beast/http/string_body.hpp>
#include <boost/beast/http/verb.hpp>
 
#include <chrono>
#include <expected>
#include <string>
#include <string_view>
#include <vector>
 
namespace util::requests {

class RequestBuilder {
    util::Logger log_{"RequestBuilder"};
    std::string host_;
    std::string port_;
    std::chrono::milliseconds timeout_{kDEFAULT_TIMEOUT};
    boost::beast::http::request<boost::beast::http::string_body> request_;
 
public:
    RequestBuilder(std::string host, std::string port);

    RequestBuilder&
    addHeader(HttpHeader const& header);

    RequestBuilder&
    addHeaders(std::vector<HttpHeader> const& headers);

    RequestBuilder&
    addData(std::string data);

    RequestBuilder&
    setTimeout(std::chrono::milliseconds timeout);

    RequestBuilder&
    setTarget(std::string_view target);

    std::expected<std::string, RequestError>
    getSsl(boost::asio::yield_context yield);

    std::expected<std::string, RequestError>
    getPlain(boost::asio::yield_context yield);

    std::expected<std::string, RequestError>
    get(boost::asio::yield_context yield);

    std::expected<std::string, RequestError>
    postSsl(boost::asio::yield_context yield);

    std::expected<std::string, RequestError>
    postPlain(boost::asio::yield_context yield);

    std::expected<std::string, RequestError>
    post(boost::asio::yield_context yield);
 
    static constexpr std::chrono::milliseconds kDEFAULT_TIMEOUT{
        30000
    }; 
 
private:
    std::expected<std::string, RequestError>
    doSslRequest(boost::asio::yield_context yield, boost::beast::http::verb method);
 
    std::expected<std::string, RequestError>
    doPlainRequest(boost::asio::yield_context yield, boost::beast::http::verb method);
 
    std::expected<std::string, RequestError>
    doRequest(boost::asio::yield_context yield, boost::beast::http::verb method);
 
    template <typename StreamDataType>
    std::expected<std::string, RequestError>
    doRequestImpl(
        StreamDataType&& streamData,
        boost::asio::yield_context yield,
        boost::beast::http::verb method
    );
};
 
}  // namespace util::requests