response.h

#pragma once

#include <string>
#include <string_view>
#include <unordered_map>

struct Response {
  int status_code = 200;
  std::string_view status_text = "OK";
  std::unordered_map<std::string, std::string> headers;
  std::string body;

  std::string serialize() {
    if (!headers.contains("Content-Length"))
      headers["Content-Length"] = std::to_string(body.size());
    if (!headers.contains("Connection"))
      headers["Connection"] = "close";
    if (!headers.contains("Content-Type"))
      headers["Content-Type"] = "text/plain";

    std::string out;
    out.reserve(128 + body.size());

    out += "HTTP/1.1 ";
    out += std::to_string(status_code);
    out += " ";
    out += status_text;
    out += "\r\n";

    for (auto const &[k, v] : headers) {
      out += k + ": " + v + "\r\n";
    }

    out += "\r\n";
    out += body;
    return out;
  }
};

inline Response make_text_response(int code, std::string reason,
                                   std::string body) {
  Response res;
  res.status_code = code;
  res.status_text = reason;
  res.body = std::move(body);
  res.headers["Content-Type"] = "text/plain";
  return res;
}

inline Response bad_request() {
  return make_text_response(400, "Bad Request", "400 Bad Request\n");
}