#pragma once

#include "coro/io_scheduler.hpp"
#include "coro/net/ip_address.hpp"
#include "coro/net/socket.hpp"
#include "coro/net/tcp_client.hpp"
#include "coro/task.hpp"

#include <fcntl.h>
#include <functional>
#include <sys/socket.h>

namespace coro::net
{
class tcp_server : public io_scheduler
{
public:
    struct options
    {
        /// The ip address for the tcp server to bind and listen on.
        net::ip_address address{net::ip_address::from_string("0.0.0.0")};
        /// The port for the tcp server to bind and listen on.
        uint16_t port{8080};
        /// The kernel backlog of connections to buffer.
        int32_t backlog{128};
    };

    tcp_server(
        io_scheduler& scheduler,
        options       opts = options{.address = net::ip_address::from_string("0.0.0.0"), .port = 8080, .backlog = 128});

    tcp_server(const tcp_server&) = delete;
    tcp_server(tcp_server&&)      = delete;
    auto operator=(const tcp_server&) -> tcp_server& = delete;
    auto operator=(tcp_server&&) -> tcp_server& = delete;
    ~tcp_server()                               = default;

    /**
     * Polls for new incoming tcp connections.
     * @param timeout How long to wait for a new connection before timing out, zero waits indefinitely.
     * @return The result of the poll, 'event' means the poll was successful and there is at least 1
     *         connection ready to be accepted.
     */
    auto poll(std::chrono::milliseconds timeout = std::chrono::milliseconds{0}) -> coro::task<coro::poll_status>;

    /**
     * Accepts an incoming tcp client connection.  On failure the tcp clients socket will be set to
     * and invalid state, use the socket.is_value() to verify the client was correctly accepted.
     * @return The newly connected tcp client connection.
     */
    auto accept() -> coro::net::tcp_client;

private:
    /// The io scheduler for awaiting new connections.
    io_scheduler& m_io_scheduler;
    /// The bind and listen options for this server.
    options m_options;
    /// The socket for accepting new tcp connections on.
    net::socket m_accept_socket{-1};
};

} // namespace coro::net