blob: 68e3f2d228eb3271c9738efa7198b4b68df74cb3 [file] [log] [blame]
#pragma once
#include "http_request.h"
#include "logging.h"
#include "nlohmann/json.hpp"
#include <boost/beast/http.hpp>
#include <string>
namespace crow
{
template <typename Adaptor, typename Handler>
class Connection;
struct Response
{
template <typename Adaptor, typename Handler>
friend class crow::Connection;
using response_type =
boost::beast::http::response<boost::beast::http::string_body>;
std::optional<response_type> stringResponse;
nlohmann::json jsonValue;
void addHeader(const std::string_view key, const std::string_view value)
{
stringResponse->set(key, value);
}
void addHeader(boost::beast::http::field key, std::string_view value)
{
stringResponse->set(key, value);
}
Response() : stringResponse(response_type{})
{}
Response(Response&& r)
{
BMCWEB_LOG_DEBUG << "Moving response containers";
*this = std::move(r);
}
~Response()
{
BMCWEB_LOG_DEBUG << this << " Destroying response";
}
Response& operator=(const Response& r) = delete;
Response& operator=(Response&& r) noexcept
{
BMCWEB_LOG_DEBUG << "Moving response containers";
stringResponse = std::move(r.stringResponse);
r.stringResponse.emplace(response_type{});
jsonValue = std::move(r.jsonValue);
completed = r.completed;
return *this;
}
void result(boost::beast::http::status v)
{
stringResponse->result(v);
}
boost::beast::http::status result()
{
return stringResponse->result();
}
unsigned resultInt()
{
return stringResponse->result_int();
}
std::string_view reason()
{
return stringResponse->reason();
}
bool isCompleted() const noexcept
{
return completed;
}
std::string& body()
{
return stringResponse->body();
}
void keepAlive(bool k)
{
stringResponse->keep_alive(k);
}
bool keepAlive()
{
return stringResponse->keep_alive();
}
void preparePayload()
{
stringResponse->prepare_payload();
}
void clear()
{
BMCWEB_LOG_DEBUG << this << " Clearing response containers";
stringResponse.emplace(response_type{});
jsonValue.clear();
completed = false;
}
void write(std::string_view body_part)
{
stringResponse->body() += std::string(body_part);
}
void end()
{
if (completed)
{
BMCWEB_LOG_ERROR << "Response was ended twice";
return;
}
completed = true;
BMCWEB_LOG_DEBUG << "calling completion handler";
if (completeRequestHandler)
{
BMCWEB_LOG_DEBUG << "completion handler was valid";
completeRequestHandler();
}
}
void end(std::string_view body_part)
{
write(body_part);
end();
}
bool isAlive()
{
return isAliveHelper && isAliveHelper();
}
private:
bool completed{};
std::function<void()> completeRequestHandler;
std::function<bool()> isAliveHelper;
// In case of a JSON object, set the Content-Type header
void jsonMode()
{
addHeader("Content-Type", "application/json");
}
};
} // namespace crow