#pragma once

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

#include "protocol.hpp"

using boost::asio::ip::udp;

class server
{
public:
  server(boost::asio::io_context& io_context, short port)
    : socket_(io_context, udp::endpoint(udp::v4(), port))
  {
    do_receive();
  }

  void do_receive()
  {
    std::cout << "do_receive\n";
    socket_.async_receive_from(
        boost::asio::buffer(data_, max_length), sender_endpoint_,
        [this](boost::system::error_code ec, std::size_t bytes_recvd)
        {
          if (!ec && bytes_recvd > 0)
          {
            std::cout << data_ << std::endl;

            auto* msg_obj = reinterpret_cast<commons::protocol::generic_message_base*>(data_);

            std::cout << "Received Type: " << static_cast<int>(msg_obj->get_type()) << "\n";
            std::cout << "Received Size: " << msg_obj->get_size() << "\n";

            auto* casted_obj = reinterpret_cast<commons::protocol::draw_rectangle*>(msg_obj);

            std::cout << "Positon: X = " << casted_obj->position.x
                              << " Y = " << casted_obj->position.y << std::endl;

            do_send(bytes_recvd);
          }
          else
          {
            do_receive();
          }
        });
  }

  void do_send(std::size_t length)
  {
    socket_.async_send_to(
        boost::asio::buffer(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 };
  char data_[max_length];
};