#pragma once

#include "ASIO_Utils.h"

#include <deque>
#include <memory>
#include <vector>

namespace Psc::asio_socket {

class UDP_Server {
public:
    void set_bind_address(const Sockaddr_In& addr) { bind_address = addr; }
    void set_bind_address(const std::string& ip, std::uint32_t port) { bind_address = Sockaddr_In(ip, port); }
    [[nodiscard]] std::string to_string() const { return bind_address.to_string(); }

    void create();
    bool bind();
    void tick();
    void close();
    void reply_last_peer(const std::string& data);
    void send_to(const std::string& ip, uint16_t port, const std::string& data);
    void write_to_all_clients(const std::string& msg);

    struct Read_Info {
        Sockaddr_In peer;
        std::string data;
    };

    std::vector<Read_Info> read();

    enum State {
        Not_Created,
        Not_Set_Field,
        Not_Bind,
        Working,
    };

    State state = Not_Created;
    Sockaddr_In bind_address;
    std::vector<Sockaddr_In> clients;
    Socket_FD socket_fd = static_cast<Socket_FD>(-1);
    bool bound = false;
    int read_chunk_size = 1024 * 1024;
    Sockaddr_In last_peer{};
    bool has_last_peer = false;
    size_t buffer_size = 4096 * 10;

private:
    asio::io_context io_context;
    std::unique_ptr<asio::ip::udp::socket> socket;
    asio::ip::udp::endpoint recv_endpoint;
    ASIO_PacketRingBuffer recv_buffer;
    ASIO_PacketRingBuffer send_buffer;
    std::vector<Sockaddr_In> recv_peers;
    std::vector<asio::ip::udp::endpoint> send_endpoints;
    std::vector<char> recv_storage;
    std::vector<char> send_storage;
    bool read_pending = false;
    bool write_pending = false;

    void start_read();
    void start_write();
};

} // namespace Psc::asio_socket