ClientWebSocketExtensions.cs

namespace Texnomic.Types.Extensions;

public static class ClientWebSocketExtensions
{
    extension(ClientWebSocket ClientWebSocket)
    {
        public async ValueTask SendJsonAsync<TRequest>(TRequest Request, JsonSerializerOptions JsonSerializerOptions, CancellationToken CancellationToken = default)
        {
            var Bytes = JsonSerializer.SerializeToUtf8Bytes(Request, JsonSerializerOptions);

            await ClientWebSocket.SendAsync(Bytes, WebSocketMessageType.Text, WebSocketMessageFlags.EndOfMessage, CancellationToken);
        }

        public async ValueTask<TResponse> ReceiveJsonAsync<TResponse>(Memory<byte> Buffer, JsonSerializerOptions JsonSerializerOptions, CancellationToken CancellationToken = default)
        {
            var Pointer = 0;

            while (true)
            {
                var Result = await ClientWebSocket.ReceiveAsync(Buffer[Pointer..], CancellationToken);

                Pointer += Result.Count;

                if (Result.EndOfMessage) break;
            }

            var Response = JsonSerializer.Deserialize<TResponse>(Buffer.Span[..Pointer], JsonSerializerOptions);

            return Response;
        }
    }
}