88 lines
2.6 KiB
C#
88 lines
2.6 KiB
C#
using System.ClientModel;
|
|
using System.Net.WebSockets;
|
|
using System.Runtime.CompilerServices;
|
|
|
|
namespace BotSharp.Core.Session;
|
|
|
|
public class BotSharpRealtimeSession : IDisposable
|
|
{
|
|
private readonly IServiceProvider _services;
|
|
private readonly WebSocket _websocket;
|
|
private readonly ChatSessionOptions? _sessionOptions;
|
|
private readonly object _singleReceiveLock = new();
|
|
private AsyncWebsocketDataCollectionResult _receivedCollectionResult;
|
|
private bool _disposed = false;
|
|
|
|
public BotSharpRealtimeSession(
|
|
IServiceProvider services,
|
|
WebSocket websocket,
|
|
ChatSessionOptions? sessionOptions)
|
|
{
|
|
_services = services;
|
|
_websocket = websocket;
|
|
_sessionOptions = sessionOptions;
|
|
}
|
|
|
|
public async IAsyncEnumerable<ChatSessionUpdate> ReceiveUpdatesAsync([EnumeratorCancellation] CancellationToken cancellationToken = default)
|
|
{
|
|
await foreach (ClientResult result in ReceiveInnerUpdatesAsync(cancellationToken))
|
|
{
|
|
var update = HandleSessionResult(result);
|
|
yield return update;
|
|
}
|
|
}
|
|
|
|
private async IAsyncEnumerable<ClientResult> ReceiveInnerUpdatesAsync([EnumeratorCancellation] CancellationToken cancellationToken = default)
|
|
{
|
|
lock (_singleReceiveLock)
|
|
{
|
|
_receivedCollectionResult ??= new(_websocket, _sessionOptions, cancellationToken);
|
|
}
|
|
|
|
await foreach (var result in _receivedCollectionResult)
|
|
{
|
|
yield return result;
|
|
}
|
|
}
|
|
|
|
private ChatSessionUpdate HandleSessionResult(ClientResult result)
|
|
{
|
|
using var response = result.GetRawResponse();
|
|
var bytes = response.Content.ToArray();
|
|
var text = Encoding.UTF8.GetString(bytes, 0, bytes.Length);
|
|
return new ChatSessionUpdate
|
|
{
|
|
RawResponse = text
|
|
};
|
|
}
|
|
|
|
public async Task SendEventAsync(string message)
|
|
{
|
|
if (_disposed || _websocket.State != WebSocketState.Open)
|
|
{
|
|
return;
|
|
}
|
|
|
|
var buffer = Encoding.UTF8.GetBytes(message);
|
|
await _websocket.SendAsync(new ArraySegment<byte>(buffer), WebSocketMessageType.Text, true, CancellationToken.None);
|
|
}
|
|
|
|
public async Task DisconnectAsync()
|
|
{
|
|
if (_disposed || _websocket.State != WebSocketState.Open)
|
|
{
|
|
return;
|
|
}
|
|
|
|
await _websocket.CloseAsync(WebSocketCloseStatus.Empty, null, CancellationToken.None);
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
if (_disposed) return;
|
|
|
|
_disposed = true;
|
|
_websocket.Dispose();
|
|
}
|
|
}
|