CrystalConnector/CrystalConnector/Connector/WebSocketConnectionManager.cs

using System.Net.WebSockets;
using CrystalConnector.Connector.Packet.S2C;
using CrystalConnector.Utilities;

namespace CrystalConnector.Connector;

public class WebSocketConnectionManager
{
    internal static Dictionary<WebSocket, WebSocketConnectionInfo> Info { get; } = new();

    public static void DisconnectAll()
    {
        foreach (var (webSocket, info) in Info)
        {
            webSocket.Disconnect();
        }
    }

    public static async Task Broadcast(string origin, string channelId, byte[] payload, bool allowUnauthenticated = false)
    {
        var packet = new S2CForwardMessagePacket(origin, channelId, payload);
        
        foreach (var (webSocket, info) in Info)
        {
            if (info.RegisteredChannels.TryGetValue(channelId, out var value) 
                && value.Direction.HasFlag(MessageDirection.Incoming))
            {
                await webSocket.Send(packet);
            }
        }
    }
}