#pragma once

#include "Socket.hpp"
#include "DirectConnection.hpp"
#include "IpAddress.hpp"
#include "PubsubKey.hpp"
#include <vector>
#include <mutex>

namespace sibs
{
    class BootstrapException : public std::runtime_error
    {
    public:
        BootstrapException(const std::string &errMsg) : std::runtime_error(errMsg) {}
    };
    
    class BootstrapNode
    {
        DISABLE_COPY(BootstrapNode)
    public:
        // Throws SocketCreateException or BootstrapException on error
        BootstrapNode(const Ipv4 &address);
        ~BootstrapNode();
    private:
        void acceptConnections();
        void messageFromClient(std::shared_ptr<DirectConnectionPeer> peer, MessageType messageType, const void *data, const usize size);
    private:
        DirectConnections connections;
        std::unique_ptr<Socket> socket;
        std::thread acceptConnectionsThread;
        PubsubKeyMap<std::vector<std::shared_ptr<DirectConnectionPeer>>> subscribedPeers;
        std::mutex subscribedPeersMutex;
    };
}