TCPServer.h

// Copyright QED Software 2023.
 
#ifndef GRAIL_TCPSERVER_H
#define GRAIL_TCPSERVER_H
 
#include "Crc16.h"
#include "DataType.h"
#include "../libs/poco/Net/include/Poco/Net/SocketAddress.h"
#include "../libs/poco/Net/include/Poco/Net/SocketStream.h"
#include "../libs/poco/Net/include/Poco/Net/StreamSocket.h"
#include "../libs/poco/Net/include/Poco/Net/TCPServer.h"
 
#include <functional>
#include <memory>
#include <mutex>
#include <thread>
 
namespace grail
{
namespace live
{
    class TCPServer
    {
    public:
        TCPServer(Poco::UInt16 port);
        ~TCPServer();
 
        void Start();
        void Stop();
        void SendToAll(const std::string& data);
        void SendToAll(const uint8_t* data, std::size_t dataSize, DataType dataType);
 
        void AddEventOnConnection(std::function<void()> event);
 
        bool IsAnyConnected() const;
 
    private:
        void AcceptConnections();
        void CheckDisconnects();
 
        constexpr static const Poco::Int32 MAGIC = 0x22669977;
        const std::string STREAM_MAGIC;
        const Crc16 crc;
 
        std::atomic<bool> isWorking{false};
        Poco::UInt16 port;
        std::unique_ptr<Poco::Net::ServerSocket> server;
        std::unique_ptr<std::thread> connectionAcceptorThread = nullptr;
        std::unique_ptr<std::thread> disconnectCheckerThread = nullptr;
        std::vector<Poco::Net::StreamSocket> clients{};
        std::vector<std::function<void()>> onClientConnected{};
 
        mutable std::mutex clientsMutex{};
        std::mutex eventMutex{};
        std::mutex stateChangeMutex{};
    };
}
}
 
#endif // GRAIL_TCPSERVER_H