#pragma once

#include <boost/asio.hpp>
#include <cstdlib>
#include <future>
#include <iostream>
#include <sstream>

#include "protocol.hpp"
#include "spdlog/spdlog.h"

namespace commons {

using boost::asio::ip::udp;
using namespace protocol;

class fake_thread_pool {
 public:
  template <typename Task>
  static void add_task(Task&& task) {
    std::thread t{std::forward<Task&&>(task)};
    t.detach();
  }
};

class server {
 public:
  server(boost::asio::io_context& io_context, short port)
      : socket_(io_context, udp::endpoint(udp::v4(), port)),
        renderer(std::make_unique<cli_renderer>(commons::screen_size)) {
    do_receive();
  }

  void do_receive() {
    socket_.async_receive_from(
        boost::asio::buffer(data_.data(), max_length), sender_endpoint_,
        [this](boost::system::error_code ec, std::size_t bytes_recvd) {
          if (!ec && bytes_recvd > 0) {
            std::shared_ptr<protocol::generic_message_base> msg_obj =
                deserialize(data_.data(), bytes_recvd);

            fake_thread_pool::add_task([this, msg_obj = std::move(msg_obj)]() {
              msg_obj->handle_message(renderer.get());
            });

            // do_send(bytes_recvd);
          } else {
            spdlog::error("error during do receive");
          }

          do_receive();
        });
  }

  void do_send(std::size_t length) {
    socket_.async_send_to(boost::asio::buffer(data_.data(), length),
                          sender_endpoint_,
                          [this](boost::system::error_code /*ec*/,
                                 std::size_t /*bytes_sent*/) { do_receive(); });
  }

 private:
  udp::socket socket_;
  udp::endpoint sender_endpoint_;
  enum { max_length = 1024 };
  std::array<char, max_length> data_;

  std::unique_ptr<render_interface> renderer;
};

}  // namespace commons