asioServer.cpp

Go to the documentation of this file.

///
/// @file AsioServer.cpp
/// @brief Asio-based implementation of INetworkServer interface
/// @namespace srv
///
 
#include "AsioServer/AsioServer.hpp"
#include "Interfaces/Protocol/Protocol.hpp"
#include "Utils/Event.hpp"
#include "Utils/EventBus.hpp"
#include "Utils/Logger.hpp"
 
#include <algorithm>
#include <chrono>
#include <random>
#include <ranges>
 
namespace srv
{
 
    AsioServer::AsioServer()
        : m_ioContext(std::make_unique<asio::io_context>()),
          m_socket(std::make_unique<asio::ip::udp::socket>(*m_ioContext)), m_port(0), m_tickRate(60), m_serverCaps(0),
          m_running(false), m_started(false), m_nextSessionId(1), m_nextLobbyId(1),
          m_packetHandler(std::make_unique<rnp::HandlerPacket>()), m_pingInterval(std::chrono::seconds(5)),
          m_clientTimeout(std::chrono::seconds(30)), m_componentId(utl::NETWORK_SERVER),
          m_eventBus(utl::EventBus::getInstance())
    {
        // utl::Logger::log("AsioServer: Constructor called", utl::LogLevel::INFO);
        // utl::Logger::log("AsioServer: Creating I/O context and socket", utl::LogLevel::INFO);
        setupPacketHandlers();
        // utl::Logger::log("AsioServer: Packet handlers setup complete", utl::LogLevel::INFO);
 
        // EventBus integration
        m_eventBus.registerComponent(m_componentId, "Asio Server");
        m_eventBus.subscribe(m_componentId, utl::EventType::SEND_TO_CLIENT);
        m_eventBus.subscribe(m_componentId, utl::EventType::BROADCAST_WORLD_STATE);
        m_eventBus.subscribe(m_componentId, utl::EventType::SEND_ENTITY_EVENTS);
        m_eventBus.subscribe(m_componentId, utl::EventType::GAME_OVER);
 
        // utl::Logger::log("AsioServer: Initialized with EventBus integration", utl::LogLevel::INFO);
    }
 
    AsioServer::~AsioServer()
    {
        // utl::Logger::log("AsioServer: Destructor called", utl::LogLevel::INFO);
        AsioServer::stop();
        // utl::Logger::log("AsioServer: Destroyed", utl::LogLevel::INFO);
    }
 
    void AsioServer::init(const std::string &host, std::uint16_t port)
    {
        // utl::Logger::log("AsioServer: init() called with host=" + host + ", port=" + std::to_string(port),
        //                  utl::LogLevel::INFO);
        m_host = host;
        m_port = port;
        // utl::Logger::log("AsioServer: Configuration stored - ready to start", utl::LogLevel::INFO);
    }
 
    void AsioServer::start()
    {
        // utl::Logger::log("AsioServer: Starting server...", utl::LogLevel::INFO);
 
        if (m_started.load())
        {
            // utl::Logger::log("AsioServer: Already started", utl::LogLevel::WARNING);
            return;
        }
 
        try
        {
            // utl::Logger::log("AsioServer: Creating endpoint for " + m_host + ":" + std::to_string(m_port),
            //                  utl::LogLevel::INFO);
 
            // Create endpoint
            asio::ip::udp::endpoint endpoint(asio::ip::make_address(m_host), m_port);
            // utl::Logger::log("AsioServer: Endpoint created successfully", utl::LogLevel::INFO);
 
            // Bind socket
            // utl::Logger::log("AsioServer: Opening socket...", utl::LogLevel::INFO);
            m_socket->open(endpoint.protocol());
            // utl::Logger::log("AsioServer: Setting socket options...", utl::LogLevel::INFO);
            m_socket->set_option(asio::ip::udp::socket::reuse_address(true));
            // utl::Logger::log("AsioServer: Binding socket to endpoint...", utl::LogLevel::INFO);
            m_socket->bind(endpoint);
            // utl::Logger::log("AsioServer: Socket bound successfully", utl::LogLevel::INFO);
 
            m_running.store(true);
            m_started.store(true);
 
            // Start network thread
            // utl::Logger::log("AsioServer: Starting network thread...", utl::LogLevel::INFO);
            m_networkThread = std::make_unique<std::thread>(&AsioServer::networkThreadLoop, this);
 
            // Start receiving
            // utl::Logger::log("AsioServer: Starting packet reception...", utl::LogLevel::INFO);
            startReceive();
 
            // utl::Logger::log("AsioServer: Started on " + m_host + ":" + std::to_string(m_port), utl::LogLevel::INFO);
        }
        catch (const std::exception &e)
        {
            m_running.store(false);
            m_started.store(false);
            // utl::Logger::log("AsioServer: Failed to start - " + std::string(e.what()), utl::LogLevel::WARNING);
            throw;
        }
    }
 
    void AsioServer::stop()
    {
        if (!m_started.load())
        {
            return;
        }
 
        m_running.store(false);
        m_started.store(false);
 
        // Send disconnect to all clients
        {
            std::lock_guard<std::mutex> lock(m_clientsMutex);
            for (auto &[sessionId, client] : m_clients)
            {
                if (client.isConnected)
                {
                    rnp::Serializer serializer;
                    rnp::PacketHeader header{};
                    header.type = static_cast<std::uint8_t>(rnp::PacketType::DISCONNECT);
                    header.length = sizeof(rnp::PacketDisconnect);
                    header.sessionId = sessionId;
 
                    rnp::PacketDisconnect disconnect{};
                    disconnect.reasonCode = static_cast<std::uint16_t>(rnp::DisconnectReason::SERVER_SHUTDOWN);
 
                    serializer.serializeHeader(header);
                    serializer.serializeDisconnect(disconnect);
 
                    sendPacketImmediate(serializer.getData(), client.endpoint);
                }
            }
        }
 
        // Stop io_context safely
        if (m_ioContext)
        {
            m_ioContext->stop();
        }
 
        // Wait for network thread with timeout
        if (m_networkThread && m_networkThread->joinable())
        {
            auto joinStart = std::chrono::steady_clock::now();
            bool joined = false;
 
            while (
                !joined &&
                std::chrono::duration_cast<std::chrono::seconds>(std::chrono::steady_clock::now() - joinStart).count() <
                    5)
            {
                if (m_networkThread->joinable())
                {
                    m_networkThread->join();
                    joined = true;
                }
                else
                {
                    std::this_thread::sleep_for(std::chrono::milliseconds(10));
                }
            }
 
            if (!joined)
            {
                // utl::Logger::log("AsioServer: Network thread join timeout, detaching", utl::LogLevel::WARNING);
                m_networkThread->detach();
            }
        }
 
        // Close socket safely
        {
            std::lock_guard<std::mutex> socketLock(m_socketMutex);
            if (m_socket && m_socket->is_open())
            {
                try
                {
                    m_socket->close();
                }
                catch (const std::exception &e)
                {
                    // utl::Logger::log("AsioServer: Exception closing socket - " + std::string(e.what()),
                    //                  utl::LogLevel::WARNING);
                }
            }
        }
 
        // Clear clients
        {
            std::lock_guard<std::mutex> lock(m_clientsMutex);
            m_clients.clear();
            m_endpointToSession.clear();
        }
 
        // utl::Logger::log("AsioServer: Stopped", utl::LogLevel::INFO);
    }
 
    void AsioServer::update()
    {
        processEventBusEvents();
 
        if (!m_running.load())
        {
            return;
        }
 
        // Process send queue
        processSendQueue();
 
        // Update client management (timeouts, pings)
        updateClientManagement();
    }
 
    void AsioServer::sendToClient(std::uint32_t sessionId, const std::vector<std::uint8_t> &data, bool reliable)
    {
        std::lock_guard<std::mutex> clientLock(m_clientsMutex);
        auto it = m_clients.find(sessionId);
        if (it == m_clients.end() || !it->second.isConnected)
        {
            // utl::Logger::log("AsioServer: Attempted to send to invalid session " + std::to_string(sessionId),
            //                  utl::LogLevel::WARNING);
            return;
        }
 
        std::lock_guard<std::mutex> queueLock(m_sendQueueMutex);
        m_sendQueue.emplace(data, it->second.endpoint, reliable);
    }
 
    void AsioServer::sendToAllClients(const std::vector<std::uint8_t> &data, bool reliable)
    {
        std::lock_guard<std::mutex> clientLock(m_clientsMutex);
        std::lock_guard<std::mutex> queueLock(m_sendQueueMutex);
 
        for (const auto &client : m_clients | std::views::values)
        {
            if (client.isConnected)
            {
                m_sendQueue.emplace(data, client.endpoint, reliable);
            }
        }
    }
 
    void AsioServer::disconnectClient(std::uint32_t sessionId)
    {
        std::uint32_t lobbyId = 0;
 
        // Scope for clientsLock to release it before broadcastLobbyUpdate
        {
            std::lock_guard<std::mutex> lock(m_clientsMutex);
            auto it = m_clients.find(sessionId);
            if (it == m_clients.end())
            {
                return;
            }
 
            utl::Logger::log("AsioServer: Disconnecting client " + std::to_string(sessionId), utl::LogLevel::INFO);
 
            rnp::PacketHeader header;
            header.type = static_cast<std::uint8_t>(rnp::PacketType::DISCONNECT);
            header.length = static_cast<std::uint16_t>(sizeof(rnp::PacketDisconnect));
            header.sessionId = sessionId;
 
            std::vector<std::uint8_t> packet;
            rnp::Serializer serializer(packet);
            rnp::PacketDisconnect disconnect{};
            disconnect.reasonCode = static_cast<std::uint16_t>(rnp::DisconnectReason::CLIENT_REQUEST);
 
            serializer.serializeHeader(header);
            serializer.serializeDisconnect(disconnect);
 
            sendPacketImmediate(serializer.getData(), it->second.endpoint);
 
            // Remove from lobby if in one
            if (it->second.currentLobbyId != 0)
            {
                lobbyId = it->second.currentLobbyId;
            }
 
            // Remove from endpoint mapping
            const std::string endpointStr = endpointToString(it->second.endpoint);
            m_endpointToSession.erase(endpointStr);
 
            // Remove client
            m_clients.erase(it);
 
            m_packetHandler->clearSession(sessionId);
 
            // utl::Logger::log("AsioServer: Disconnected client " + std::to_string(sessionId), utl::LogLevel::INFO);
        } // clientsLock released here
 
        // Leave lobby AFTER releasing clientsLock to avoid nested mutex acquisitions
        if (lobbyId != 0)
        {
            leaveLobby(sessionId);
            // Broadcast lobby update to remaining players
            broadcastLobbyUpdate(lobbyId);
        }
    }
 
    std::size_t AsioServer::getClientCount() const
    {
        std::lock_guard clientLock(m_clientsMutex);
        return std::ranges::count_if(m_clients, [](const auto &pair) { return pair.second.isConnected; });
    }
 
    std::vector<std::uint32_t> AsioServer::getConnectedSessions() const
    {
        std::lock_guard clientLock(m_clientsMutex);
        std::vector<std::uint32_t> sessions;
        for (const auto &[sessionId, client] : m_clients)
        {
            if (client.isConnected)
            {
                sessions.push_back(sessionId);
            }
        }
        return sessions;
    }
 
    bool AsioServer::isRunning() const { return m_running.load(); }
 
    void AsioServer::setTickRate(std::uint16_t tickRate)
    {
        m_tickRate = tickRate;
        // utl::Logger::log("AsioServer: Tick rate set to " + std::to_string(tickRate), utl::LogLevel::INFO);
    }
 
    void AsioServer::setServerCapabilities(std::uint32_t caps)
    {
        m_serverCaps = caps;
        // utl::Logger::log("AsioServer: Server capabilities set to " + std::to_string(caps), utl::LogLevel::INFO);
    }
 
    void AsioServer::setupPacketHandlers()
    {
        // CONNECT handler
        m_packetHandler->onConnect([this](const rnp::PacketConnect &packet, const rnp::PacketContext &context)
                                   { return handleConnect(packet, context); });
 
        // DISCONNECT handler
        m_packetHandler->onDisconnect([this](const rnp::PacketDisconnect &packet, const rnp::PacketContext &context)
                                      { return handleDisconnect(packet, context); });
 
        // PING handler
        m_packetHandler->onPing([this](const rnp::PacketPingPong &packet, const rnp::PacketContext &context)
                                { return handlePing(packet, context); });
 
        // PONG handler
        m_packetHandler->onPong([this](const rnp::PacketPingPong &packet, const rnp::PacketContext &context)
                                { return handlePong(packet, context); });
 
        // ENTITY_EVENT handler (for player inputs)
        m_packetHandler->onEntityEvent(
            [this](const std::vector<rnp::EventRecord> &events, const rnp::PacketContext &context)
            { return handleEntityEvent(events, context); });
 
        // LOBBY_LIST_REQUEST handler
        m_packetHandler->onLobbyListRequest([this](const rnp::PacketContext &context)
                                            { return handleLobbyListRequest(context); });
 
        // LOBBY_CREATE handler
        m_packetHandler->onLobbyCreate([this](const rnp::PacketLobbyCreate &packet, const rnp::PacketContext &context)
                                       { return handleLobbyCreate(packet, context); });
 
        // LOBBY_JOIN handler
        m_packetHandler->onLobbyJoin([this](const rnp::PacketLobbyJoin &packet, const rnp::PacketContext &context)
                                     { return handleLobbyJoin(packet, context); });
 
        // LOBBY_LEAVE handler
        m_packetHandler->onLobbyLeave([this](const rnp::PacketContext &context) { return handleLobbyLeave(context); });
 
        // START_GAME_REQUEST handler
        m_packetHandler->onStartGameRequest(
            [this](const rnp::PacketStartGameRequest &packet, const rnp::PacketContext &context)
            { return handleStartGameRequest(packet, context); });
 
        // utl::Logger::log("AsioServer: Packet handlers initialized", utl::LogLevel::INFO);
    }
 
    void AsioServer::startReceive()
    {
        // Check if we should continue receiving
        if (!m_running.load() || !m_socket || !m_socket->is_open())
        {
            utl::Logger::log("AsioServer: Cannot start receive - server not running or socket closed",
                             utl::LogLevel::WARNING);
            return;
        }
 
        m_socket->async_receive_from(
            asio::buffer(m_recvBuffer), m_senderEndpoint,
            [this](const std::error_code ec, const std::size_t bytesReceived)
            {
                if (!ec && m_running.load())
                {
                    utl::Logger::log("AsioServer: Received " + std::to_string(bytesReceived) + " bytes from " +
                                         m_senderEndpoint.address().to_string() + ":" +
                                         std::to_string(m_senderEndpoint.port()),
                                     utl::LogLevel::INFO);
                    handleReceive(bytesReceived);
                    startReceive(); // Continue receiving
                }
                else if (m_running.load() && ec != asio::error::operation_aborted)
                {
                    utl::Logger::log("AsioServer: Receive error - " + ec.message(), utl::LogLevel::WARNING);
                    std::this_thread::sleep_for(std::chrono::milliseconds(100));
                    startReceive(); // Try to continue
                }
                else
                {
                    utl::Logger::log("AsioServer: Server stopped or operation aborted, ending receive loop",
                                     utl::LogLevel::INFO);
                }
            });
    }
 
    void AsioServer::handleReceive(std::size_t bytesReceived)
    {
        utl::Logger::log("AsioServer: Processing received packet of " + std::to_string(bytesReceived) + " bytes",
                         utl::LogLevel::INFO);
 
        // PacketHeader serialized size: 1 (type) + 2 (length) + 4 (sessionId) = 7 bytes
        constexpr std::size_t PACKET_HEADER_SIZE = 7;
        if (bytesReceived < PACKET_HEADER_SIZE)
        {
            utl::Logger::log("AsioServer: Received packet too small (need " + std::to_string(PACKET_HEADER_SIZE) +
                                 " bytes, got " + std::to_string(bytesReceived) + ")",
                             utl::LogLevel::WARNING);
            return;
        }
 
        // Create packet context
        rnp::PacketContext context;
        context.receiveTime = std::chrono::steady_clock::now();
        context.senderAddress = m_senderEndpoint.address().to_string();
        context.senderPort = m_senderEndpoint.port();
 
        utl::Logger::log("AsioServer: Packet from " + context.senderAddress + ":" + std::to_string(context.senderPort),
                         utl::LogLevel::INFO);
 
        // Extract session ID from header for context
        std::vector<std::uint8_t> data(m_recvBuffer.begin(), m_recvBuffer.begin() + bytesReceived);
        if (data.size() >= PACKET_HEADER_SIZE)
        {
            rnp::Serializer headerSerializer(data);
            rnp::PacketHeader header = headerSerializer.deserializeHeader();
            context.sessionId = header.sessionId;
            utl::Logger::log("AsioServer: Packet type: " + std::to_string(static_cast<int>(header.type)) +
                                 ", Session ID: " + std::to_string(header.sessionId),
                             utl::LogLevel::INFO);
        }
 
        // Process packet
        utl::Logger::log("AsioServer: Processing packet...", utl::LogLevel::INFO);
        rnp::HandlerResult result = m_packetHandler->processPacket(data, context);
        if (result != rnp::HandlerResult::SUCCESS)
        {
            utl::Logger::log("AsioServer: Packet processing failed with result " +
                                 std::to_string(static_cast<int>(result)),
                             utl::LogLevel::WARNING);
        }
        else
        {
            utl::Logger::log("AsioServer: Packet processed successfully", utl::LogLevel::INFO);
        }
    }
 
    void AsioServer::networkThreadLoop() const
    {
        // utl::Logger::log("AsioServer: Network thread started", utl::LogLevel::INFO);
 
        while (m_running.load())
        {
            try
            {
                // Run I/O context and check if we should continue
                if (!m_running.load())
                {
                    break;
                }
 
                size_t handlersRun = m_ioContext->run_one();
                if (handlersRun == 0)
                {
                    // No handlers to run, sleep briefly to avoid busy waiting
                    std::this_thread::sleep_for(std::chrono::milliseconds(1));
                }
            }
            catch (const std::exception &e)
            {
                if (m_running.load())
                {
                    // utl::Logger::log("AsioServer: Network thread exception - " + std::string(e.what()),
                    //                  utl::LogLevel::WARNING);
                    std::this_thread::sleep_for(std::chrono::milliseconds(100));
                }
            }
        }
 
        // utl::Logger::log("AsioServer: Network thread stopped", utl::LogLevel::INFO);
    }
 
    std::uint32_t AsioServer::generateSessionId() const
    {
        std::random_device rd;
        std::mt19937 gen(rd());
        std::uniform_int_distribution<std::uint32_t> dis(1, UINT32_MAX);
 
        std::uint32_t sessionId = 0;
        while (sessionId == 0 || m_clients.contains(sessionId))
        {
            sessionId = dis(gen);
        }
 
        return sessionId;
    }
 
    std::string AsioServer::endpointToString(const asio::ip::udp::endpoint &endpoint)
    {
        return endpoint.address().to_string() + ":" + std::to_string(endpoint.port());
    }
 
    void AsioServer::sendPacketImmediate(const std::vector<std::uint8_t> &data,
                                         const asio::ip::udp::endpoint &destination)
    {
        utl::Logger::log("AsioServer: Sending " + std::to_string(data.size()) + " bytes to " +
                             destination.address().to_string() + ":" + std::to_string(destination.port()),
                         utl::LogLevel::INFO);
 
        // Protect socket access with mutex
        std::lock_guard<std::mutex> socketLock(m_socketMutex);
 
        if (!m_socket || !m_socket->is_open() || !m_running.load())
        {
            utl::Logger::log("AsioServer: Cannot send - socket not open or server not running", utl::LogLevel::WARNING);
            return;
        }
 
        try
        {
            size_t bytesSent = m_socket->send_to(asio::buffer(data), destination);
            utl::Logger::log("AsioServer: Successfully sent " + std::to_string(bytesSent) + " bytes",
                             utl::LogLevel::INFO);
        }
        catch (const std::exception &e)
        {
            if (m_running.load())
            {
                utl::Logger::log("AsioServer: Failed to send packet - " + std::string(e.what()),
                                 utl::LogLevel::WARNING);
            }
        }
    }
 
    rnp::HandlerResult AsioServer::handleConnect(const rnp::PacketConnect &packet, const rnp::PacketContext &context)
    {
        // utl::Logger::log("AsioServer: Handling CONNECT packet from " + context.senderAddress + ":" +
        //                      std::to_string(context.senderPort),
        //                  utl::LogLevel::INFO);
 
        std::lock_guard<std::mutex> lock(m_clientsMutex);
        // Check if server is full (avoid getClientCount() to prevent deadlock)
        const size_t currentClientCount =
            std::ranges::count_if(m_clients, [](const auto &pair) { return pair.second.isConnected; });
        // utl::Logger::log("AsioServer: Current client count: " + std::to_string(currentClientCount) + "/" +
        //                      std::to_string(MAX_CLIENTS),
        //                  utl::LogLevel::INFO);
 
        if (currentClientCount >= MAX_CLIENTS)
        {
            // utl::Logger::log("AsioServer: Server full, rejecting connection", utl::LogLevel::WARNING);
            sendError(rnp::ErrorCode::RATE_LIMITED, "Server full", m_senderEndpoint, 0);
            return rnp::HandlerResult::SUCCESS;
        }
 
        // Check if client already exists
        const std::string endpointStr = endpointToString(m_senderEndpoint);
        const auto existingIt = m_endpointToSession.find(endpointStr);
        if (existingIt != m_endpointToSession.end())
        {
            // utl::Logger::log("AsioServer: Client already connected, resending accept (Session ID: " +
            //                      std::to_string(existingIt->second) + ")",
            //                  utl::LogLevel::INFO);
            // Client already connected, resend accept
            sendConnectAccept(existingIt->second, m_senderEndpoint);
            return rnp::HandlerResult::SUCCESS;
        }
 
        // Create new session
        std::uint32_t sessionId = generateSessionId();
        // utl::Logger::log("AsioServer: Creating new session with ID: " + std::to_string(sessionId),
        // utl::LogLevel::INFO);
 
        ClientSession &client = m_clients[sessionId];
        client.sessionId = sessionId;
        client.endpoint = m_senderEndpoint;
        client.lastSeen = context.receiveTime;
        client.playerName = std::string(packet.playerName.data(), packet.nameLen);
        client.clientCaps = packet.clientCaps;
        client.isConnected = true;
 
        m_endpointToSession[endpointStr] = sessionId;
 
        // utl::Logger::log("AsioServer: Player name: '" + client.playerName +
        //                      "', Caps: " + std::to_string(client.clientCaps),
        //                  utl::LogLevel::INFO);
 
        // Send accept response
        // utl::Logger::log("AsioServer: Sending CONNECT_ACCEPT to session " + std::to_string(sessionId),
        //                  utl::LogLevel::INFO);
        sendConnectAccept(sessionId, m_senderEndpoint);
 
        // utl::Logger::log("AsioServer: Client connected - " + client.playerName + " (ID: " + std::to_string(sessionId)
        // +
        //                      ")",
        //                  utl::LogLevel::INFO);
 
        // Publier vers GameServer
        m_eventBus.publish(utl::EventType::PLAYER_CONNECTED, sessionId, m_componentId,
                           utl::GAME_LOGIC); // GameServer ID
 
        // utl::Logger::log(
        //     "AsioServer: Nouvelle connexion publiée vers GameServer (sessionId: " + std::to_string(sessionId) + ")",
        //     utl::LogLevel::INFO);
 
        return rnp::HandlerResult::SUCCESS;
    }
 
    rnp::HandlerResult AsioServer::handleDisconnect(const rnp::PacketDisconnect &packet,
                                                    const rnp::PacketContext &context)
    {
        // utl::Logger::log("AsioServer: Handling DISCONNECT packet from session " + std::to_string(context.sessionId) +
        //                      " from " + context.senderAddress + ":" + std::to_string(context.senderPort),
        //                  utl::LogLevel::INFO);
 
        std::lock_guard<std::mutex> lock(m_clientsMutex);
 
        auto it = m_clients.find(context.sessionId);
        if (it != m_clients.end())
        {
            // utl::Logger::log("AsioServer: Client disconnected - " + it->second.playerName +
            //                      " (ID: " + std::to_string(context.sessionId) + ")" +
            //                      ", Reason: " + std::to_string(static_cast<int>(packet.reasonCode)),
            //                  utl::LogLevel::INFO);
 
            // Remove from endpoint mapping
            const std::string endpointStr = endpointToString(it->second.endpoint);
            // utl::Logger::log("AsioServer: Removing endpoint mapping: " + endpointStr, utl::LogLevel::INFO);
            m_endpointToSession.erase(endpointStr);
 
            // Remove client
            // utl::Logger::log("AsioServer: Removing client from session list", utl::LogLevel::INFO);
            m_clients.erase(it);
        }
        else
        {
            // utl::Logger::log("AsioServer: DISCONNECT from unknown session " + std::to_string(context.sessionId),
            //                  utl::LogLevel::WARNING);
        }
 
        // Publier vers GameServer
        m_eventBus.publish(utl::EventType::PLAYER_DISCONNECTED, context.sessionId, m_componentId,
                           utl::GAME_LOGIC); // GameServer ID
 
        // utl::Logger::log(
        //     "AsioServer: Déconnexion publiée vers GameServer (sessionId: " + std::to_string(context.sessionId) + ")",
        //     utl::LogLevel::INFO);
 
        return rnp::HandlerResult::SUCCESS;
    }
 
    rnp::HandlerResult AsioServer::handlePing(const rnp::PacketPingPong &packet, const rnp::PacketContext &context)
    {
        // Update client last seen time
        {
            std::lock_guard<std::mutex> lock(m_clientsMutex);
            const auto it = m_clients.find(context.sessionId);
            if (it != m_clients.end())
            {
                it->second.lastSeen = context.receiveTime;
            }
        }
 
        // Send pong response
        sendPong(packet.nonce, m_senderEndpoint, context.sessionId);
        return rnp::HandlerResult::SUCCESS;
    }
 
    rnp::HandlerResult AsioServer::handlePong(const rnp::PacketPingPong &packet, const rnp::PacketContext &context)
    {
        // Update client latency and last seen time
        {
            std::lock_guard<std::mutex> lock(m_clientsMutex);
            auto it = m_clients.find(context.sessionId);
            if (it != m_clients.end())
            {
                it->second.lastSeen = context.receiveTime;
                // Calculate latency based on ping time
                auto now = std::chrono::steady_clock::now();
                auto pingTime = std::chrono::milliseconds(packet.sendTimeMs);
                auto currentTime = std::chrono::duration_cast<std::chrono::milliseconds>(now.time_since_epoch());
                it->second.latency = static_cast<std::uint32_t>((currentTime - pingTime).count());
            }
        }
 
        return rnp::HandlerResult::SUCCESS;
    }
 
    void AsioServer::sendPong(std::uint32_t nonce, const asio::ip::udp::endpoint &destination, std::uint32_t sessionId)
    {
        rnp::Serializer serializer;
        rnp::PacketHeader header{};
        header.type = static_cast<std::uint8_t>(rnp::PacketType::PONG);
        header.length = sizeof(rnp::PacketPingPong);
        header.sessionId = sessionId;
 
        rnp::PacketPingPong pong{};
        pong.nonce = nonce;
        pong.sendTimeMs = static_cast<std::uint32_t>(
            std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::steady_clock::now().time_since_epoch())
                .count());
 
        serializer.serializeHeader(header);
        serializer.serializePingPong(pong);
 
        sendPacketImmediate(serializer.getData(), destination);
    }
 
    void AsioServer::sendConnectAccept(std::uint32_t sessionId, const asio::ip::udp::endpoint &destination)
    {
        // utl::Logger::log("AsioServer: Preparing CONNECT_ACCEPT packet for session " + std::to_string(sessionId) +
        //                      " to " + destination.address().to_string() + ":" + std::to_string(destination.port()),
        //                  utl::LogLevel::INFO);
 
        rnp::Serializer serializer;
        rnp::PacketHeader header{};
        header.type = static_cast<std::uint8_t>(rnp::PacketType::CONNECT_ACCEPT);
        header.length = sizeof(rnp::PacketConnectAccept);
        header.sessionId = sessionId;
 
        rnp::PacketConnectAccept accept{};
        accept.sessionId = sessionId;
        accept.tickRateHz = m_tickRate;
        accept.mtuPayloadBytes = rnp::MAX_PAYLOAD;
        accept.serverCaps = m_serverCaps;
 
        serializer.serializeHeader(header);
        serializer.serializeConnectAccept(accept);
 
        // utl::Logger::log("AsioServer: Sending CONNECT_ACCEPT packet (" + std::to_string(serializer.getData().size())
        // +
        //                      " bytes)",
        //                  utl::LogLevel::INFO);
        sendPacketImmediate(serializer.getData(), destination);
    }
 
    void AsioServer::sendError(rnp::ErrorCode errorCode, const std::string &description,
                               const asio::ip::udp::endpoint &destination, std::uint32_t sessionId)
    {
        rnp::Serializer serializer;
        rnp::PacketHeader header{};
        header.type = static_cast<std::uint8_t>(rnp::PacketType::PACKET_ERROR);
        header.length = static_cast<std::uint16_t>(sizeof(rnp::PacketError) + description.length());
        header.sessionId = sessionId;
 
        rnp::PacketError error{};
        error.errorCode = static_cast<std::uint16_t>(errorCode);
        error.msgLen = static_cast<std::uint16_t>(description.length());
        error.description = description;
 
        serializer.serializeHeader(header);
        serializer.serializeError(error);
 
        sendPacketImmediate(serializer.getData(), destination);
    }
 
    void AsioServer::updateClientManagement()
    {
        const auto now = std::chrono::steady_clock::now();
 
        // Send pings periodically
        if (now - m_lastPingTime > m_pingInterval)
        {
            std::lock_guard<std::mutex> lock(m_clientsMutex);
            for (auto &[sessionId, client] : m_clients)
            {
                if (client.isConnected)
                {
                    // Send ping
                    rnp::Serializer serializer;
                    rnp::PacketHeader header{};
                    header.type = static_cast<std::uint8_t>(rnp::PacketType::PING);
                    header.length = sizeof(rnp::PacketPingPong);
                    header.sessionId = sessionId;
 
                    rnp::PacketPingPong ping{};
                    ping.nonce = ++client.lastPingSent;
                    ping.sendTimeMs = static_cast<std::uint32_t>(
                        std::chrono::duration_cast<std::chrono::milliseconds>(now.time_since_epoch()).count());
 
                    serializer.serializeHeader(header);
                    serializer.serializePingPong(ping);
 
                    std::lock_guard<std::mutex> queueLock(m_sendQueueMutex);
                    m_sendQueue.emplace(serializer.getData(), client.endpoint, false);
                }
            }
            m_lastPingTime = now;
        }
 
        // Check for timeouts
        std::vector<std::uint32_t> timedOutClients;
        {
            std::lock_guard<std::mutex> lock(m_clientsMutex);
            for (const auto &[sessionId, client] : m_clients)
            {
                if (client.isConnected && (now - client.lastSeen) > m_clientTimeout)
                {
                    timedOutClients.push_back(sessionId);
                }
            }
        }
 
        // Disconnect timed out clients
        for (std::uint32_t sessionId : timedOutClients)
        {
            // utl::Logger::log("AsioServer: Client " + std::to_string(sessionId) + " timed out", utl::LogLevel::INFO);
            disconnectClient(sessionId);
        }
    }
 
    void AsioServer::processSendQueue()
    {
        std::lock_guard<std::mutex> lock(m_sendQueueMutex);
        while (!m_sendQueue.empty())
        {
            const QueuedPacket &packet = m_sendQueue.front();
            sendPacketImmediate(packet.data, packet.destination);
            m_sendQueue.pop();
        }
    }
 
    void AsioServer::processEventBusEvents()
    {
        // Consommer les événements du GameServer
        const auto events = m_eventBus.consumeForTarget(m_componentId);
 
        for (const auto &event : events)
        {
            switch (event.type)
            {
                case utl::EventType::SEND_TO_CLIENT:
                    handleSendToClientEvent(event);
                    break;
                case utl::EventType::GAME_OVER:
                    handleGameOverEvent(event);
                    break;
 
                case utl::EventType::BROADCAST_WORLD_STATE:
                    handleBroadcastEvent(event);
                    break;
 
                case utl::EventType::SEND_ENTITY_EVENTS:
                    handleSendEntityEventToClients(event);
                    break;
 
                default:
                    // utl::Logger::log("AsioServer: Événement non géré: " +
                    //                      std::to_string(static_cast<uint32_t>(event.type)),
                    //                  utl::LogLevel::WARNING);
                    break;
            }
        }
    }
 
    void AsioServer::handleSendToClientEvent(const utl::Event &event)
    {
        // Extraire l'ID du client cible et les données
        if (event.data.size() >= sizeof(uint32_t))
        {
            uint32_t sessionId = 0;
            std::memcpy(&sessionId, event.data.data(), sizeof(uint32_t));
 
            std::vector<uint8_t> messageData(event.data.begin() + sizeof(uint32_t), event.data.end());
 
            // Utiliser la méthode existante
            sendToClient(sessionId, messageData);
 
            // utl::Logger::log("AsioServer: Message envoye au client " + std::to_string(sessionId) +
            //                      " (taille: " + std::to_string(messageData.size()) + " bytes)",
            //                  utl::LogLevel::INFO);
        }
    }
 
    void AsioServer::handleBroadcastEvent(const utl::Event &event)
    {
        // Utiliser la méthode existante pour broadcaster
        sendToAllClients(event.data);
 
        // utl::Logger::log(
        //     "AsioServer: Message diffuse a tous les clients (taille: " + std::to_string(event.data.size()) + "
        //     bytes)", utl::LogLevel::INFO);
    }
 
    void AsioServer::handleSendEntityEventToClients(const utl::Event &event)
    {
        // Forward entity events to all clients
        sendToAllClients(event.data);
 
        // utl::Logger::log("AsioServer: Entity events diffuses a tous les clients (taille: " +
        //                      std::to_string(event.data.size()) + " bytes)",
        //                  utl::LogLevel::INFO);
    }
 
    rnp::HandlerResult AsioServer::handleEntityEvent(const std::vector<rnp::EventRecord> &events,
                                                     const rnp::PacketContext &context) const
    {
        utl::Logger::log("AsioServer: Received " + std::to_string(events.size()) + " entity events from session " +
                             std::to_string(context.sessionId),
                         utl::LogLevel::INFO);
 
        // Filtrer les événements d'input et les publier vers GameServer
        for (const auto &eventRecord : events)
        {
            if (eventRecord.type == rnp::EventType::INPUT)
            {
                utl::Logger::log("AsioServer: Input event received, data size: " +
                                     std::to_string(eventRecord.data.size()),
                                 utl::LogLevel::INFO);
 
                // Create event with sessionId as sourceId so RTypeServer knows which player
                utl::Event event(utl::EventType::PLAYER_INPUT_RECEIVED, context.sessionId, utl::GAME_LOGIC);
                event.data = eventRecord.data;
                m_eventBus.publish(event);
 
                // utl::Logger::log("AsioServer: Input joueur publie vers GameServer (sessionId: " +
                //                      std::to_string(context.sessionId) + ")",
                //                  utl::LogLevel::INFO);
            }
            else
            {
                m_eventBus.publish(utl::EventType::ENTITY_EVENT_RECEIVED, eventRecord.data, m_componentId,
                                   utl::GAME_LOGIC); // GameServer ID
 
                // utl::Logger::log("AsioServer: Evenement entite publie vers GameServer (type: " +
                //                      std::to_string(static_cast<uint8_t>(eventRecord.type)) + ")",
                //                  utl::LogLevel::INFO);
            }
        }
 
        return rnp::HandlerResult::SUCCESS;
    }
 
    // Lobby System Implementation
 
    rnp::HandlerResult AsioServer::handleLobbyListRequest(const rnp::PacketContext &context)
    {
        utl::Logger::log("AsioServer: Handling LOBBY_LIST_REQUEST from session " + std::to_string(context.sessionId),
                         utl::LogLevel::INFO);
 
        // Verify session exists
        {
            std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
            auto clientIt = m_clients.find(context.sessionId);
            if (clientIt == m_clients.end() || !clientIt->second.isConnected)
            {
                sendError(rnp::ErrorCode::UNAUTHORIZED_SESSION, "Invalid session", m_senderEndpoint, context.sessionId);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
        } // Release the lock before calling sendLobbyList
 
        sendLobbyList(context.sessionId);
        return rnp::HandlerResult::SUCCESS;
    }
 
    rnp::HandlerResult AsioServer::handleLobbyCreate(const rnp::PacketLobbyCreate &packet,
                                                     const rnp::PacketContext &context)
    {
        utl::Logger::log("AsioServer: Handling LOBBY_CREATE from session " + std::to_string(context.sessionId),
                         utl::LogLevel::INFO);
 
        std::uint32_t lobbyId = 0;
        bool success = false;
 
        // Scope for clientsLock to release it before broadcastLobbyUpdate
        {
            std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
            auto clientIt = m_clients.find(context.sessionId);
            if (clientIt == m_clients.end() || !clientIt->second.isConnected)
            {
                sendLobbyCreateResponse(context.sessionId, 0, false, rnp::ErrorCode::UNAUTHORIZED_SESSION);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
 
            utl::Logger::log("AsioServer: LOBBY_JOIN - Client current lobby ID: " +
                                 std::to_string(clientIt->second.currentLobbyId),
                             utl::LogLevel::INFO);
 
            if (clientIt->second.currentLobbyId != 0)
            {
                sendLobbyCreateResponse(context.sessionId, 0, false, rnp::ErrorCode::ALREADY_IN_LOBBY);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
 
            // Extract lobby name
            std::string lobbyName;
            if (packet.nameLen > 0 && packet.nameLen <= 31)
            {
                lobbyName = std::string(packet.lobbyName.data(), packet.nameLen);
            }
            else
            {
                lobbyName = "Lobby " + std::to_string(m_nextLobbyId);
            }
 
            // Create lobby
            lobbyId = createLobby(lobbyName, context.sessionId, packet.maxPlayers, packet.gameMode);
            if (lobbyId > 0)
            {
                clientIt->second.currentLobbyId = lobbyId;
                sendLobbyCreateResponse(context.sessionId, lobbyId, true);
                success = true;
            }
            else
            {
                sendLobbyCreateResponse(context.sessionId, 0, false, rnp::ErrorCode::INTERNAL_ERROR);
            }
        } // clientsLock released here
 
        // Broadcast lobby update AFTER releasing clientsLock to avoid deadlock
        if (success && lobbyId > 0)
        {
            broadcastLobbyUpdate(lobbyId);
        }
 
        return rnp::HandlerResult::SUCCESS;
    }
 
    rnp::HandlerResult AsioServer::handleLobbyJoin(const rnp::PacketLobbyJoin &packet,
                                                   const rnp::PacketContext &context)
    {
        utl::Logger::log("AsioServer: Handling LOBBY_JOIN from session " + std::to_string(context.sessionId) +
                             " for lobby " + std::to_string(packet.lobbyId),
                         utl::LogLevel::INFO);
 
        bool joinSuccess = false;
 
        // Scope for clientsLock to release it before lobbyToLobbyInfo and broadcastLobbyUpdate
        {
            std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
            utl::Logger::log("AsioServer: LOBBY_JOIN - Checking client session " + std::to_string(context.sessionId),
                             utl::LogLevel::INFO);
            auto clientIt = m_clients.find(context.sessionId);
            if (clientIt == m_clients.end() || !clientIt->second.isConnected)
            {
                utl::Logger::log("AsioServer: LOBBY_JOIN failed - unauthorized session " +
                                     std::to_string(context.sessionId),
                                 utl::LogLevel::WARNING);
                sendLobbyJoinResponse(context.sessionId, packet.lobbyId, false, rnp::ErrorCode::UNAUTHORIZED_SESSION);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
 
            if (clientIt->second.currentLobbyId != 0)
            {
                utl::Logger::log("AsioServer: LOBBY_JOIN failed - session " + std::to_string(context.sessionId) +
                                     " already in lobby " + std::to_string(clientIt->second.currentLobbyId),
                                 utl::LogLevel::WARNING);
                sendLobbyJoinResponse(context.sessionId, packet.lobbyId, false, rnp::ErrorCode::ALREADY_IN_LOBBY);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
 
            // Try to join lobby
            LobbyStatus joinStatus = joinLobby(packet.lobbyId, context.sessionId);
            utl::Logger::log("AsioServer: joinLobby returned status: " + std::to_string(static_cast<int>(joinStatus)),
                             utl::LogLevel::INFO);
 
            if (joinStatus == LobbyStatus::SUCCESS)
            {
                clientIt->second.currentLobbyId = packet.lobbyId;
                utl::Logger::log("AsioServer: Session " + std::to_string(context.sessionId) +
                                     " successfully joined lobby " + std::to_string(packet.lobbyId),
                                 utl::LogLevel::INFO);
                joinSuccess = true;
            }
            else
            {
                utl::Logger::log("AsioServer: Session " + std::to_string(context.sessionId) + " failed to join lobby " +
                                     std::to_string(packet.lobbyId) +
                                     " - status: " + std::to_string(static_cast<int>(joinStatus)),
                                 utl::LogLevel::WARNING);
                sendLobbyJoinResponse(context.sessionId, packet.lobbyId, false, rnp::ErrorCode::LOBBY_NOT_FOUND,
                                      nullptr);
            }
        } // clientsLock released here
 
        // Get lobby info AFTER releasing clientsLock to avoid deadlock in lobbyToLobbyInfo
        if (joinSuccess)
        {
            rnp::LobbyInfo lobbyInfo;
            {
                std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
                auto lobbyIt = m_lobbies.find(packet.lobbyId);
                if (lobbyIt != m_lobbies.end())
                {
                    lobbyInfo = lobbyToLobbyInfo(lobbyIt->second);
                    utl::Logger::log("AsioServer: Lobby info - currentPlayers: " +
                                         std::to_string(static_cast<int>(lobbyInfo.currentPlayers)) + "/" +
                                         std::to_string(static_cast<int>(lobbyInfo.maxPlayers)),
                                     utl::LogLevel::INFO);
                }
            }
 
            sendLobbyJoinResponse(context.sessionId, packet.lobbyId, true, rnp::ErrorCode::NONE, &lobbyInfo);
        }
 
        // Broadcast lobby update AFTER releasing clientsLock to avoid deadlock
        if (joinSuccess)
        {
            broadcastLobbyUpdate(packet.lobbyId);
        }
 
        return rnp::HandlerResult::SUCCESS;
    }
 
    rnp::HandlerResult AsioServer::handleLobbyLeave(const rnp::PacketContext &context)
    {
        utl::Logger::log("AsioServer: Handling LOBBY_LEAVE from session " + std::to_string(context.sessionId),
                         utl::LogLevel::INFO);
 
        std::uint32_t lobbyId = 0;
 
        // Scope for clientsLock to release it before broadcastLobbyUpdate
        {
            std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
            auto clientIt = m_clients.find(context.sessionId);
            if (clientIt == m_clients.end() || !clientIt->second.isConnected)
            {
                sendError(rnp::ErrorCode::UNAUTHORIZED_SESSION, "Invalid session", m_senderEndpoint, context.sessionId);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
 
            // Check if client is in a lobby
            if (clientIt->second.currentLobbyId == 0)
            {
                sendError(rnp::ErrorCode::NOT_IN_LOBBY, "Not in lobby", m_senderEndpoint, context.sessionId);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
 
            lobbyId = clientIt->second.currentLobbyId;
            clientIt->second.currentLobbyId = 0; // Clear lobby association before leaving
        } // clientsLock released here
 
        // Leave lobby after releasing clientsLock to avoid nested mutex acquisitions
        leaveLobby(context.sessionId);
 
        // Broadcast update to remaining players AFTER releasing clientsLock to avoid deadlock
        if (lobbyId != 0)
        {
            broadcastLobbyUpdate(lobbyId);
        }
 
        return rnp::HandlerResult::SUCCESS;
    }
 
    void AsioServer::sendLobbyList(const std::uint32_t sessionId)
    {
        utl::Logger::log("AsioServer: Sending lobby list to session " + std::to_string(sessionId), utl::LogLevel::INFO);
 
        rnp::PacketLobbyListResponse response;
        {
            std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
            response.lobbyCount = static_cast<std::uint16_t>(m_lobbies.size());
            response.lobbies.reserve(m_lobbies.size());
 
            for (const auto &lobby : m_lobbies | std::views::values)
            {
                rnp::LobbyInfo lobbyInfo = lobbyToLobbyInfo(lobby);
                utl::Logger::log("AsioServer: Adding lobby " + std::to_string(lobbyInfo.lobbyId) + " with " +
                                     std::to_string(static_cast<int>(lobbyInfo.currentPlayers)) + "/" +
                                     std::to_string(static_cast<int>(lobbyInfo.maxPlayers)) + " players",
                                 utl::LogLevel::INFO);
                response.lobbies.push_back(lobbyInfo);
            }
        }
 
        // Serialize header and response data
        std::vector<std::uint8_t> packet;
        rnp::Serializer packetSerializer(packet);
 
        rnp::PacketHeader header;
        header.type = static_cast<std::uint8_t>(rnp::PacketType::LOBBY_LIST_RESPONSE);
        header.length = static_cast<std::uint16_t>(sizeof(rnp::PacketLobbyListResponse));
        header.sessionId = sessionId;
 
        packetSerializer.serializeHeader(header);
        packetSerializer.serializeLobbyListResponse(response);
 
        utl::Logger::log("AsioServer: Sending lobby list with " + std::to_string(response.lobbyCount) +
                             " lobbies, packet size: " + std::to_string(packetSerializer.getData().size()),
                         utl::LogLevel::INFO);
 
        sendPacketImmediate(packetSerializer.getData(), m_senderEndpoint);
    }
 
    std::uint32_t AsioServer::createLobby(const std::string &name, std::uint32_t hostSession, std::uint8_t maxPlayers,
                                          std::uint8_t gameMode)
    {
        std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
 
        std::uint32_t lobbyId = m_nextLobbyId++;
        Lobby lobby(lobbyId, name, hostSession, maxPlayers, gameMode);
        lobby.playerSessions.push_back(hostSession);
 
        m_lobbies[lobbyId] = std::move(lobby);
 
        utl::Logger::log("AsioServer: Created lobby " + std::to_string(lobbyId) + " '" + name + "' hosted by " +
                             std::to_string(hostSession),
                         utl::LogLevel::INFO);
 
        return lobbyId;
    }
 
    LobbyStatus AsioServer::joinLobby(std::uint32_t lobbyId, std::uint32_t sessionId)
    {
        std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
 
        utl::Logger::log("AsioServer: joinLobby - Looking for lobby " + std::to_string(lobbyId), utl::LogLevel::INFO);
 
        auto lobbyIt = m_lobbies.find(lobbyId);
        if (lobbyIt == m_lobbies.end())
        {
            utl::Logger::log("AsioServer: joinLobby - Lobby " + std::to_string(lobbyId) + " not found",
                             utl::LogLevel::WARNING);
            return LobbyStatus::NOT_FOUND;
        }
 
        Lobby &lobby = lobbyIt->second;
        utl::Logger::log(
            "AsioServer: joinLobby - Lobby found, current players: " + std::to_string(lobby.playerSessions.size()) +
                "/" + std::to_string(lobby.maxPlayers) + ", status: " + std::to_string(static_cast<int>(lobby.status)),
            utl::LogLevel::INFO);
 
        if (lobby.playerSessions.size() >= lobby.maxPlayers)
        {
            utl::Logger::log("AsioServer: joinLobby - Lobby is full", utl::LogLevel::WARNING);
            return LobbyStatus::FULL;
        }
 
        if (lobby.status != rnp::LobbyStatus::WAITING)
        {
            utl::Logger::log("AsioServer: joinLobby - Game already started", utl::LogLevel::WARNING);
            return LobbyStatus::IN_GAME;
        }
 
        if (auto playerIt = std::ranges::find(lobby.playerSessions, sessionId); playerIt != lobby.playerSessions.end())
        {
            return LobbyStatus::ALREADY_IN_LOBBY;
        }
 
        lobby.playerSessions.push_back(sessionId);
 
        utl::Logger::log("AsioServer: Player " + std::to_string(sessionId) + " joined lobby " +
                             std::to_string(lobbyId) + " (now " + std::to_string(lobby.playerSessions.size()) + "/" +
                             std::to_string(static_cast<int>(lobby.maxPlayers)) + " players)",
                         utl::LogLevel::INFO);
        return LobbyStatus::SUCCESS;
    }
 
    void AsioServer::leaveLobby(std::uint32_t sessionId)
    {
        std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
 
        for (auto &[lobbyId, lobby] : m_lobbies)
        {
            if (auto playerIt = std::ranges::find(lobby.playerSessions, sessionId);
                playerIt != lobby.playerSessions.end())
            {
                lobby.playerSessions.erase(playerIt);
 
                utl::Logger::log("AsioServer: Player " + std::to_string(sessionId) + " left lobby " +
                                     std::to_string(lobbyId),
                                 utl::LogLevel::INFO);
 
                // Clear the client's current lobby association
                {
                    std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
                    auto clientIt = m_clients.find(sessionId);
                    if (clientIt != m_clients.end())
                    {
                        clientIt->second.currentLobbyId = 0;
                        utl::Logger::log("AsioServer: Cleared lobby association for session " +
                                             std::to_string(sessionId),
                                         utl::LogLevel::INFO);
                    }
                }
 
                // If host left, assign new host or delete lobby
                if (lobby.hostSessionId == sessionId)
                {
                    if (!lobby.playerSessions.empty())
                    {
                        lobby.hostSessionId = lobby.playerSessions[0];
                        utl::Logger::log("AsioServer: New host for lobby " + std::to_string(lobbyId) + " is " +
                                             std::to_string(lobby.hostSessionId),
                                         utl::LogLevel::INFO);
                    }
                    else
                    {
                        // No players left, lobby will be cleaned up
                        utl::Logger::log("AsioServer: Lobby " + std::to_string(lobbyId) + " is now empty",
                                         utl::LogLevel::INFO);
                    }
                }
                break;
            }
        }
 
        cleanupEmptyLobbies();
    }
 
    void AsioServer::handleGameOverEvent(const utl::Event &event)
    {
        utl::Logger::log("AsioServer: Received GAME_OVER event - closing all active lobbies", utl::LogLevel::INFO);
 
        // Broadcast GAME_OVER to all clients in lobbies
        std::vector<std::uint32_t> lobbyIds;
        {
            std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
            for (const auto &[lobbyId, lobby] : m_lobbies)
            {
                lobbyIds.push_back(lobbyId);
            }
        }
 
        // Send GAME_OVER packet to all clients in each lobby
        for (std::uint32_t lobbyId : lobbyIds)
        {
            broadcastGameOverToLobby(lobbyId, event.data);
        }
 
        // Close all lobbies after game over
        {
            std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
            std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
 
            for (auto &[sessionId, client] : m_clients)
            {
                if (client.currentLobbyId != 0)
                {
                    utl::Logger::log("AsioServer: Clearing lobby association for session " + std::to_string(sessionId),
                                     utl::LogLevel::INFO);
                    client.currentLobbyId = 0;
                }
            }
 
            utl::Logger::log("AsioServer: Clearing all lobbies (" + std::to_string(m_lobbies.size()) + " lobbies)",
                             utl::LogLevel::INFO);
            m_lobbies.clear();
        }
    }
 
    void AsioServer::broadcastGameOverToLobby(std::uint32_t lobbyId, const std::vector<std::uint8_t> &gameOverData)
    {
        std::vector<std::uint32_t> playerSessions;
 
        {
            std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
            auto lobbyIt = m_lobbies.find(lobbyId);
            if (lobbyIt == m_lobbies.end())
            {
                return;
            }
            playerSessions = lobbyIt->second.playerSessions;
        }
 
        rnp::Serializer serializer;
        rnp::PacketHeader header;
        header.type = static_cast<std::uint8_t>(rnp::PacketType::GAME_OVER);
        header.length = static_cast<std::uint16_t>(gameOverData.size());
        header.sessionId = 0; // Broadcast to all
 
        serializer.serializeHeader(header);
        // Append game over data to serializer
        std::vector<std::uint8_t> packetData = serializer.getData();
        packetData.insert(packetData.end(), gameOverData.begin(), gameOverData.end());
 
        // Send to all players in lobby
        std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
        for (std::uint32_t sessionId : playerSessions)
        {
            auto clientIt = m_clients.find(sessionId);
            if (clientIt != m_clients.end() && clientIt->second.isConnected)
            {
                utl::Logger::log("AsioServer: Sending GAME_OVER to session " + std::to_string(sessionId),
                                 utl::LogLevel::INFO);
                sendPacketImmediate(packetData, clientIt->second.endpoint);
            }
        }
    }
 
    void AsioServer::broadcastLobbyUpdate(std::uint32_t lobbyId)
    {
        std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
        auto lobbyIt = m_lobbies.find(lobbyId);
        if (lobbyIt == m_lobbies.end())
        {
            return;
        }
 
        rnp::PacketLobbyUpdate update;
        update.lobbyInfo = lobbyToLobbyInfo(lobbyIt->second);
 
        // Serialize packet data
        std::vector<std::uint8_t> data;
        rnp::Serializer serializer(data);
        serializer.serializeLobbyUpdate(update);
 
        // Get the serialized data
        const std::vector<std::uint8_t> &lobbyUpdateData = serializer.getData();
 
        // Send to all players in lobby
        for (std::uint32_t sessionId : lobbyIt->second.playerSessions)
        {
            rnp::PacketHeader header;
            header.type = static_cast<std::uint8_t>(rnp::PacketType::LOBBY_UPDATE);
            header.length = static_cast<std::uint16_t>(lobbyUpdateData.size());
            header.sessionId = sessionId;
 
            std::vector<std::uint8_t> playerPacket;
            rnp::Serializer playerPacketSerializer(playerPacket);
            playerPacketSerializer.serializeHeader(header);
            playerPacketSerializer.serializeLobbyUpdate(update);
 
            // Lock clients mutex to safely access client data
            std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
            auto clientIt = m_clients.find(sessionId);
            if (clientIt != m_clients.end())
            {
                sendPacketImmediate(playerPacketSerializer.getData(), clientIt->second.endpoint);
            }
        }
 
        utl::Logger::log("AsioServer: Broadcasted lobby update for lobby " + std::to_string(lobbyId),
                         utl::LogLevel::INFO);
    }
 
    rnp::LobbyInfo AsioServer::lobbyToLobbyInfo(const Lobby &lobby)
    {
        rnp::LobbyInfo info;
        info.lobbyId = lobby.lobbyId;
        info.currentPlayers = static_cast<std::uint8_t>(lobby.playerSessions.size());
        info.maxPlayers = lobby.maxPlayers;
        info.gameMode = lobby.gameMode;
        info.status = static_cast<std::uint8_t>(lobby.status);
        info.hostSessionId = lobby.hostSessionId;
 
        // Copy lobby name
        std::size_t nameLen = std::min(lobby.lobbyName.size(), static_cast<std::size_t>(31));
        std::memcpy(info.lobbyName.data(), lobby.lobbyName.c_str(), nameLen);
        info.lobbyName[nameLen] = '\0';
 
        // Copy player names
        std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
        for (std::size_t i = 0; i < lobby.playerSessions.size() && i < 8; ++i)
        {
            auto clientIt = m_clients.find(lobby.playerSessions[i]);
            if (clientIt != m_clients.end())
            {
                std::size_t playerNameLen = std::min(clientIt->second.playerName.size(), static_cast<std::size_t>(31));
                std::memcpy(info.playerNames[i].data(), clientIt->second.playerName.c_str(), playerNameLen);
                info.playerNames[i][playerNameLen] = '\0';
            }
        }
 
        return info;
    }
 
    void AsioServer::sendLobbyCreateResponse(std::uint32_t sessionId, std::uint32_t lobbyId, bool success,
                                             rnp::ErrorCode errorCode)
    {
        rnp::PacketLobbyCreateResponse response;
        response.lobbyId = lobbyId;
        response.success = success ? 1 : 0;
        response.errorCode = static_cast<std::uint16_t>(errorCode);
 
        rnp::PacketHeader header;
        header.type = static_cast<std::uint8_t>(rnp::PacketType::LOBBY_CREATE_RESPONSE);
        header.length = static_cast<std::uint16_t>(sizeof(rnp::PacketLobbyCreateResponse));
        header.sessionId = sessionId;
 
        std::vector<std::uint8_t> packet;
        rnp::Serializer packetSerializer(packet);
        packetSerializer.serializeHeader(header);
        packetSerializer.serializeLobbyCreateResponse(response);
 
        sendPacketImmediate(packetSerializer.getData(), m_senderEndpoint);
 
        utl::Logger::log("AsioServer: Sent lobby create response to session " + std::to_string(sessionId) +
                             " (success: " + (success ? "true" : "false") + ")",
                         utl::LogLevel::INFO);
    }
 
    void AsioServer::sendLobbyJoinResponse(std::uint32_t sessionId, std::uint32_t lobbyId, bool success,
                                           rnp::ErrorCode errorCode, const rnp::LobbyInfo *lobbyInfo)
    {
        rnp::PacketLobbyJoinResponse response;
        response.lobbyId = lobbyId;
        response.success = success ? 1 : 0;
        response.errorCode = static_cast<std::uint16_t>(errorCode);
        if (success && lobbyInfo)
        {
            response.lobbyInfo = *lobbyInfo;
        }
 
        rnp::PacketHeader header;
        header.type = static_cast<std::uint8_t>(rnp::PacketType::LOBBY_JOIN_RESPONSE);
        header.length = static_cast<std::uint16_t>(sizeof(rnp::PacketLobbyJoinResponse));
        header.sessionId = sessionId;
 
        std::vector<std::uint8_t> packet;
        rnp::Serializer packetSerializer(packet);
        packetSerializer.serializeHeader(header);
        packetSerializer.serializeLobbyJoinResponse(response);
 
        sendPacketImmediate(packetSerializer.getData(), m_senderEndpoint);
 
        utl::Logger::log("AsioServer: Sent lobby join response to session " + std::to_string(sessionId) +
                             " (success: " + (success ? "true" : "false") + ")",
                         utl::LogLevel::INFO);
    }
 
    void AsioServer::cleanupEmptyLobbies()
    {
        // NOTE: Assumes m_lobbiesMutex is already held by caller
        auto it = m_lobbies.begin();
        while (it != m_lobbies.end())
        {
            if (it->second.playerSessions.empty())
            {
                utl::Logger::log("AsioServer: Cleaning up empty lobby " + std::to_string(it->first),
                                 utl::LogLevel::INFO);
                it = m_lobbies.erase(it);
            }
            else
            {
                ++it;
            }
        }
    }
 
    rnp::HandlerResult AsioServer::handleStartGameRequest(const rnp::PacketStartGameRequest &packet,
                                                          const rnp::PacketContext &context)
    {
        utl::Logger::log("AsioServer: Handling START_GAME_REQUEST from session " + std::to_string(context.sessionId) +
                             " for lobby " + std::to_string(packet.lobbyId),
                         utl::LogLevel::INFO);
 
        // Verify session exists and is in the lobby
        {
            std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
            auto clientIt = m_clients.find(context.sessionId);
            if (clientIt == m_clients.end() || !clientIt->second.isConnected)
            {
                sendError(rnp::ErrorCode::UNAUTHORIZED_SESSION, "Invalid session", m_senderEndpoint, context.sessionId);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
 
            if (clientIt->second.currentLobbyId != packet.lobbyId)
            {
                sendError(rnp::ErrorCode::NOT_IN_LOBBY, "Not in this lobby", m_senderEndpoint, context.sessionId);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
        }
 
        // Verify the session is the host and lobby exists
        {
            std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
            auto lobbyIt = m_lobbies.find(packet.lobbyId);
            if (lobbyIt == m_lobbies.end())
            {
                sendError(rnp::ErrorCode::LOBBY_NOT_FOUND, "Lobby not found", m_senderEndpoint, context.sessionId);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
 
            // Only host can start the game
            if (lobbyIt->second.hostSessionId != context.sessionId)
            {
                sendError(rnp::ErrorCode::UNAUTHORIZED_SESSION, "Only host can start game", m_senderEndpoint,
                          context.sessionId);
                return rnp::HandlerResult::PROCESSING_ERROR;
            }
 
            // Update lobby status to IN_GAME
            lobbyIt->second.status = rnp::LobbyStatus::IN_GAME;
        }
 
        utl::Logger::log("AsioServer: Starting game for lobby " + std::to_string(packet.lobbyId), utl::LogLevel::INFO);
 
        // Get sessionIds of players in lobby
        std::vector<std::uint32_t> playerSessions;
        {
            std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
            auto lobbyIt = m_lobbies.find(packet.lobbyId);
            if (lobbyIt != m_lobbies.end())
            {
                playerSessions = lobbyIt->second.playerSessions;
                utl::Logger::log("AsioServer: Player sessions in lobby: " + std::to_string(playerSessions.size()),
                                 utl::LogLevel::INFO);
            }
        }
 
        // Notify RTypeServer about game start with player sessions
        utl::Event startEvent(utl::EventType::SERVER_START, m_componentId, utl::GAME_LOGIC);
        std::vector<std::uint8_t> sessionsData;
        sessionsData.resize(sizeof(std::uint32_t) * playerSessions.size());
        for (size_t i = 0; i < playerSessions.size(); ++i)
        {
            std::memcpy(sessionsData.data() + i * sizeof(std::uint32_t), &playerSessions[i], sizeof(std::uint32_t));
        }
        startEvent.data = sessionsData;
        m_eventBus.publish(startEvent);
 
        utl::Logger::log("AsioServer: Sent SERVER_START to RTypeServer with " + std::to_string(playerSessions.size()) +
                             " players",
                         utl::LogLevel::INFO);
 
        // Broadcast game start to all players
        broadcastGameStart(packet.lobbyId);
 
        return rnp::HandlerResult::SUCCESS;
    }
 
    void AsioServer::broadcastGameStart(std::uint32_t lobbyId)
    {
        std::lock_guard<std::mutex> lobbiesLock(m_lobbiesMutex);
        auto lobbyIt = m_lobbies.find(lobbyId);
        if (lobbyIt == m_lobbies.end())
        {
            return;
        }
 
        rnp::PacketGameStart gameStart;
        gameStart.lobbyId = lobbyId;
 
        // Serialize packet
        std::vector<std::uint8_t> data;
        rnp::Serializer serializer(data);
        serializer.serializeGameStart(gameStart);
 
        const std::vector<std::uint8_t> &gameStartData = serializer.getData();
 
        // Send to all players in lobby
        for (std::uint32_t sessionId : lobbyIt->second.playerSessions)
        {
            rnp::PacketHeader header;
            header.type = static_cast<std::uint8_t>(rnp::PacketType::GAME_START);
            header.length = static_cast<std::uint16_t>(gameStartData.size());
            header.sessionId = sessionId;
 
            std::vector<std::uint8_t> playerPacket;
            rnp::Serializer playerPacketSerializer(playerPacket);
            playerPacketSerializer.serializeHeader(header);
            playerPacketSerializer.serializeGameStart(gameStart);
 
            // Lock clients mutex to safely access client data
            std::lock_guard<std::mutex> clientsLock(m_clientsMutex);
            auto clientIt = m_clients.find(sessionId);
            if (clientIt != m_clients.end())
            {
                sendPacketImmediate(playerPacketSerializer.getData(), clientIt->second.endpoint);
            }
        }
 
        utl::Logger::log("AsioServer: Broadcasted game start for lobby " + std::to_string(lobbyId),
                         utl::LogLevel::INFO);
 
        // Publish GAME_START event to game logic to start waves
        m_eventBus.publish(utl::EventType::GAME_START, gameStartData, utl::NETWORK_SERVER, utl::GAME_LOGIC);
        utl::Logger::log("AsioServer: Published GAME_START event to game logic", utl::LogLevel::INFO);
    }
 
} // namespace srv