server.hxx

#pragma once
 
#include <StormByte/network/endpoint.hxx>
 
#include <atomic>
#include <mutex>
#include <thread>
#include <unordered_map>
 
namespace StormByte::Network {
    namespace Connection {
        class Client;                                        
    }
 
    namespace Socket {
        class Server;                                        
    }
 
    class STORMBYTE_NETWORK_PUBLIC Server: private Endpoint {
        public:
            Server(const DeserializePacketFunction& deserialize_packet_function, std::shared_ptr<Logger::Log> logger) noexcept;
 
            Server(const Server& other)                                                     = delete;
 
            Server(Server&& other) noexcept                                                 = default;
 
            virtual ~Server() noexcept;
 
            Server& operator=(const Server& other)                                          = delete;
 
            Server& operator=(Server&& other) noexcept                                      = default;
 
            bool                                                                            Connect(const Connection::Protocol& protocol, const std::string& address, const unsigned short& port) override;
 
            void                                                                            Disconnect() noexcept override;
 
            inline Connection::Status                                                       Status() const noexcept override {
                return m_status.load();
            }
 
        protected:
            void                                                                            DisconnectClient(const std::string& uuid) noexcept;
 
        private:
            std::unique_ptr<Socket::Server> m_socket_server;                                
            std::atomic<Connection::Status> m_status;                                       
            std::thread m_accept_thread;                                                    
            std::unordered_map<std::string, std::shared_ptr<Connection::Client>> m_clients; 
            std::unordered_map<std::string, std::thread> m_handle_msg_threads;              
            std::mutex m_mutex;                                                             
 
            void                                                                            AcceptClients() noexcept;
 
            void                                                                            HandleClientCommunication(const std::string& client_uuid) noexcept;
            
            virtual PacketPointer                                                           ProcessClientPacket(const std::string& client_uuid, PacketPointer packet) noexcept = 0;
    };
}