#include <boost/asio/io_context.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <boost/asio/spawn.hpp>
#include <boost/asio/write.hpp>
#include <iostream>
#include <memory>
#include <libs/fiber/examples/asio/yield.hpp>
#include <libs/fiber/examples/asio/detail/yield.hpp>
#include <libs/fiber/examples/asio/round_robin.hpp>

using boost::asio::ip::tcp;

void session(tcp::socket&& socket_)
{
  using boost::fibers::asio::yield;

  {
    try {
      boost::system::error_code ec;
      char                      data[16];
      for (;;) {
        std::size_t n = socket_.async_read_some(boost::asio::buffer(data), yield);

        if (!n) {
          break;
        }
        boost::asio::async_write(socket_, boost::asio::buffer(data, n), yield);
      }
    } catch (std::exception& e) {
      socket_.close();
    }
  }
}
void server(std::shared_ptr<boost::asio::io_context> io_ctx, int port);

int main(int argc, char* argv[])
{
  try {
    int port = 15999;
    if (argc == 2) {
      port = std::atoi(argv[1]);
    }

    std::shared_ptr<boost::asio::io_context> io_ctx = std::make_shared<boost::asio::io_context>();
                   boost::fibers::use_scheduling_algorithm<boost::fibers::asio::round_robin>(io_ctx);

    using boost::fibers::asio::yield;
    boost::fibers::fiber(server, io_ctx, port).detach();
    io_ctx->run();

  } catch (std::exception& e) {
    std::cerr << "Exception: " << e.what() << "\n";
  }
  return 0;
}

void server(std::shared_ptr<boost::asio::io_context> io_ctx, int port)
{
  tcp::acceptor acceptor(*io_ctx, tcp::endpoint(tcp::v4(), port));
  using boost::fibers::asio::yield;

  for (;;) {
    boost::system::error_code ec;
    tcp::socket               socket(*io_ctx);
    acceptor.async_accept(socket, yield[ec]);
    if (!ec) {
      boost::fibers::fiber(session, std::move(socket)).detach();
    }
  }
}