#include "src/sdk.h"
//
#include <boost/asio/signal_set.hpp>
#include <iostream>
#include <mutex>
#include <thread>
#include <vector>

#include "src/http_server.h"

namespace
{
namespace net = boost::asio;
using namespace std::literals;
namespace sys = boost::system;
namespace http = boost::beast::http;
using namespace std;
using net::ip::tcp;

// Запрос, тело которого представлено в виде строки
using StringRequest = http::request<http::string_body>;
// Ответ, тело которого представлено в виде строки
using StringResponse = http::response<http::string_body>;

struct ContentType
{
  ContentType() = delete;
  constexpr static std::string_view
  TEXT_HTML = "text/html"sv;
  // При необходимости внутрь ContentType можно добавить и другие типы контента
};

// Создаёт StringResponse с заданными параметрами
StringResponse MakeStringResponse(http::status status,
                                  std::string_view body,
                                  unsigned http_version,
                                  bool keep_alive,
                                  std::string_view content_type = ContentType::TEXT_HTML)
{
  StringResponse response(status, http_version);
  response.set(http::field::content_type, content_type);
  response.body() = body;
  response.content_length(body.size());
  response.keep_alive(keep_alive);
  return response;
}

StringResponse HandleRequest(StringRequest&& req)
{
  // Подставьте сюда код из синхронной версии HTTP-сервера
  return MakeStringResponse(http::status::ok,
                            "OK"sv,
                            req.version(),
                            req.keep_alive());
}

// Запускает функцию fn на n потоках, включая текущий
template<typename Fn>
void RunWorkers(unsigned n, const Fn& fn)
{
  n = std::max(1u, n);
  std::vector<std::jthread> workers;
  workers.reserve(n - 1);
  // Запускаем n-1 рабочих потоков, выполняющих функцию fn
  while (--n)
  {
    workers.emplace_back(fn);
  }
  fn();
}

}  // namespace

int main()
{
  const unsigned num_threads = std::thread::hardware_concurrency();

  net::io_context ioc(num_threads);
  tcp::acceptor   acceptor(net::make_strand(ioc));

  // Подписываемся на сигналы и при их получении завершаем работу сервера
  net::signal_set signals(ioc, SIGINT, SIGTERM);
  signals.async_wait([&ioc](const sys::error_code& ec,
                            [[maybe_unused]] int signal_number)
                     {
                       if (!ec)
                       {
                         cout << "Signal "sv << signal_number << " received"sv
                              << endl;
                         ioc.stop();
                       }
                     });

  const auto address = net::ip::make_address("0.0.0.0");
  constexpr
  net::ip::port_type port = 8080;
  http_server::ServeHttp(ioc, {address, port}, [](auto&& req, auto&& sender)
  {
    sender(HandleRequest(std::forward<decltype(req)>(req)));
  });

  net::steady_timer t {ioc, 30s};

  t.async_wait([](sys::error_code ec)
               {
                 cout << "Timer expired"sv << endl;
               });

  // Эта надпись сообщает тестам о том, что сервер запущен и готов обрабатывать запросы
  std::cout << "Server has started..."sv << std::endl;

  RunWorkers(num_threads, [&ioc]
  {
    ioc.run();
  });
}