| | | 1 | | using Microsoft.Extensions.Logging; |
| | | 2 | | using NostrSure.Infrastructure.Client.Abstractions; |
| | | 3 | | using System.Net.WebSockets; |
| | | 4 | | |
| | | 5 | | namespace NostrSure.Infrastructure.Client.Implementation; |
| | | 6 | | |
| | | 7 | | /// <summary> |
| | | 8 | | /// Refactored WebSocket connection implementation using the façade pattern with SOLID principles |
| | | 9 | | /// </summary> |
| | | 10 | | public sealed class WebSocketConnection : IWebSocketConnection |
| | | 11 | | { |
| | | 12 | | private readonly IConnectionManager _connectionManager; |
| | | 13 | | private readonly IMessageReceiver _messageReceiver; |
| | | 14 | | private readonly IMessageSender _messageSender; |
| | | 15 | | private readonly IConnectionErrorHandler _errorHandler; |
| | | 16 | | private readonly IConnectionStateManager _stateManager; |
| | | 17 | | private readonly ILogger<WebSocketConnection>? _logger; |
| | | 18 | | private bool _disposed; |
| | | 19 | | |
| | 2 | 20 | | public WebSocketConnection( |
| | 2 | 21 | | IConnectionManager connectionManager, |
| | 2 | 22 | | IMessageReceiver messageReceiver, |
| | 2 | 23 | | IMessageSender messageSender, |
| | 2 | 24 | | IConnectionErrorHandler errorHandler, |
| | 2 | 25 | | IConnectionStateManager stateManager, |
| | 2 | 26 | | ILogger<WebSocketConnection>? logger = null) |
| | 2 | 27 | | { |
| | 2 | 28 | | _connectionManager = connectionManager ?? throw new ArgumentNullException(nameof(connectionManager)); |
| | 2 | 29 | | _messageReceiver = messageReceiver ?? throw new ArgumentNullException(nameof(messageReceiver)); |
| | 2 | 30 | | _messageSender = messageSender ?? throw new ArgumentNullException(nameof(messageSender)); |
| | 2 | 31 | | _errorHandler = errorHandler ?? throw new ArgumentNullException(nameof(errorHandler)); |
| | 2 | 32 | | _stateManager = stateManager ?? throw new ArgumentNullException(nameof(stateManager)); |
| | 2 | 33 | | _logger = logger; |
| | 2 | 34 | | } |
| | | 35 | | |
| | 2 | 36 | | public WebSocketState State => _stateManager.CurrentState; |
| | | 37 | | |
| | | 38 | | // Delegate events to component implementations |
| | | 39 | | public event EventHandler<string>? MessageReceived |
| | | 40 | | { |
| | 1 | 41 | | add => _messageReceiver.MessageReceived += value; |
| | 0 | 42 | | remove => _messageReceiver.MessageReceived -= value; |
| | | 43 | | } |
| | | 44 | | |
| | | 45 | | public event EventHandler<Exception>? ErrorOccurred |
| | | 46 | | { |
| | 1 | 47 | | add => _errorHandler.ErrorOccurred += value; |
| | 0 | 48 | | remove => _errorHandler.ErrorOccurred -= value; |
| | | 49 | | } |
| | | 50 | | |
| | | 51 | | public event EventHandler? Disconnected |
| | | 52 | | { |
| | 1 | 53 | | add => _connectionManager.Disconnected += value; |
| | 0 | 54 | | remove => _connectionManager.Disconnected -= value; |
| | | 55 | | } |
| | | 56 | | |
| | | 57 | | public async Task ConnectAsync(Uri uri, CancellationToken cancellationToken = default) |
| | 1 | 58 | | { |
| | 1 | 59 | | _logger?.LogInformation("Connecting to {Uri}", uri); |
| | | 60 | | |
| | 1 | 61 | | await _connectionManager.ConnectAsync(uri, cancellationToken); |
| | 1 | 62 | | await _messageReceiver.StartReceivingAsync(cancellationToken); |
| | | 63 | | |
| | 1 | 64 | | _logger?.LogInformation("Successfully connected and started receiving messages"); |
| | 1 | 65 | | } |
| | | 66 | | |
| | | 67 | | public Task SendAsync(string message, CancellationToken cancellationToken = default) |
| | 1 | 68 | | { |
| | 1 | 69 | | return _messageSender.SendAsync(message, cancellationToken); |
| | 1 | 70 | | } |
| | | 71 | | |
| | | 72 | | public Task<string> ReceiveAsync(CancellationToken cancellationToken = default) |
| | 1 | 73 | | { |
| | 1 | 74 | | return _messageReceiver.ReceiveAsync(cancellationToken); |
| | 1 | 75 | | } |
| | | 76 | | |
| | | 77 | | public async Task CloseAsync(WebSocketCloseStatus closeStatus = WebSocketCloseStatus.NormalClosure, |
| | | 78 | | string? statusDescription = null, |
| | | 79 | | CancellationToken cancellationToken = default) |
| | 1 | 80 | | { |
| | 1 | 81 | | _logger?.LogInformation("Closing WebSocket connection"); |
| | | 82 | | |
| | 1 | 83 | | await _messageReceiver.StopReceivingAsync(); |
| | 1 | 84 | | await _connectionManager.CloseAsync(closeStatus, statusDescription, cancellationToken); |
| | | 85 | | |
| | 1 | 86 | | _logger?.LogInformation("WebSocket connection closed"); |
| | 1 | 87 | | } |
| | | 88 | | |
| | | 89 | | public void Dispose() |
| | 1 | 90 | | { |
| | 1 | 91 | | if (!_disposed) |
| | 1 | 92 | | { |
| | 1 | 93 | | _logger?.LogDebug("Disposing RefactoredWebSocketConnection"); |
| | | 94 | | |
| | 1 | 95 | | _messageReceiver?.Dispose(); |
| | 1 | 96 | | _connectionManager?.Dispose(); |
| | | 97 | | |
| | 1 | 98 | | _disposed = true; |
| | 1 | 99 | | } |
| | 1 | 100 | | } |
| | | 101 | | } |