mROA/mROA/Implementation/Backend/ConnectionHub.cs

using System;
using System.Collections.Generic;
using mROA.Abstract;

namespace mROA.Implementation.Backend
{
    public class ConnectionHub : IConnectionHub
    {
        private readonly Dictionary<int, IChannelInteractionModule> _connections = new();
        private IContextualSerializationToolKit _serializationToolkit;

        public ConnectionHub(IContextualSerializationToolKit serializationToolkit)
        {
            _serializationToolkit = serializationToolkit;
        }

        public void RegisterInteraction(IChannelInteractionModule interaction)
        {
            if (_serializationToolkit is null)
                throw new NullReferenceException("Serialization toolkit is null");

            _connections.Add(interaction.ConnectionId, interaction);
            var module = new RepresentationModule(interaction, _serializationToolkit);
            OnConnected?.Invoke(module);
        }

        public IChannelInteractionModule GetInteraction(int id)
        {
            return _connections!.GetValueOrDefault(id, null) ?? _connections!.GetValueOrDefault(-id, null) ??
                throw new Exception("No connection found");
        }

        public event ConnectionHandler? OnConnected;
        public event DisconnectionHandler? OnDisconnected;
    }
}