feat: add rest client to get gateway url

src/StevesBot.Worker/Discord/DiscordGatewayClient.cs

@@ -5,44 +5,73 @@ internal class DiscordGatewayClient : IDiscordGatewayClient
   private readonly DiscordGatewayClientOptions _options;
   private readonly IWebSocketFactory _webSocketFactory;
   private readonly ILogger<DiscordGatewayClient> _logger;
-  private readonly JsonSerializerOptions _jsonSerializerOptions = new()
-  {
-    PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
-    ReferenceHandler = ReferenceHandler.IgnoreCycles,
-    Converters =
-    {
-      new DiscordEventConverter(),
-    },
-  };
+  private readonly IDiscordRestClient _discordRestClient;
+  // private readonly JsonSerializerOptions _jsonSerializerOptions = new()
+  // {
+  //   PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
+  //   ReferenceHandler = ReferenceHandler.IgnoreCycles,
+  //   Converters =
+  //   {
+  //     new DiscordEventConverter(),
+  //   },
+  // };
   private readonly AsyncLock _lock = new();
 
   private string _gatewayUrl = string.Empty;
   private IWebSocket? _webSocket;
   private Task? _receiveTask;
-  private DateTime _timeLastHeartbeatSent = DateTime.MinValue;
-  private DateTime _timeLastHeartbeatAcknowledged = DateTime.MinValue;
-  private CancellationTokenSource? _heartbeatCts;
-  private Task? _heartbeatTask;
-  private string _sessionId = string.Empty;
-  private string _resumeGatewayUrl = string.Empty;
+  // private DateTime _timeLastHeartbeatSent = DateTime.MinValue;
+  // private DateTime _timeLastHeartbeatAcknowledged = DateTime.MinValue;
+  // private CancellationTokenSource? _heartbeatCts;
+  // private Task? _heartbeatTask;
+  // private string _sessionId = string.Empty;
+  // private string _resumeGatewayUrl = string.Empty;
 
   public DiscordGatewayClient(
     DiscordGatewayClientOptions options,
     IWebSocketFactory webSocketFactory,
-    ILogger<DiscordGatewayClient> logger
+    ILogger<DiscordGatewayClient> logger,
+    IDiscordRestClient discordRestClient
   )
   {
     _options = options ?? throw new ArgumentNullException(nameof(options));
     _webSocketFactory = webSocketFactory ?? throw new ArgumentNullException(nameof(webSocketFactory));
     _logger = logger ?? throw new ArgumentNullException(nameof(logger));
+    _discordRestClient = discordRestClient ?? throw new ArgumentNullException(nameof(discordRestClient));
+  }
+
+  public async Task ConnectAsync(CancellationToken cancellationToken)
+  {
+    using (await _lock.LockAsync(cancellationToken))
+    {
+      if (string.IsNullOrEmpty(_gatewayUrl))
+      {
+        _gatewayUrl = await _discordRestClient.GetGatewayUrlAsync(cancellationToken);
+      }
+
+      _webSocket = _webSocketFactory.Create();
+
+      var uri = new Uri(_gatewayUrl);
+      await _webSocket.ConnectAsync(uri, cancellationToken);
+
+      _logger.LogInformation("Connected to Discord Gateway at {GatewayUrl}", _gatewayUrl);
+
+      _receiveTask = ReceiveMessagesAsync(cancellationToken);
+    }
+  }
+
+  private static Task ReceiveMessagesAsync(CancellationToken cancellationToken)
+  {
+    return Task.CompletedTask;
   }
 
   public void Dispose()
   {
-    _heartbeatCts?.Cancel();
-    _heartbeatCts?.Dispose();
-    _heartbeatTask?.Dispose();
+    // _heartbeatCts?.Cancel();
+    // _heartbeatCts?.Dispose();
+    // _heartbeatTask?.Dispose();
     _receiveTask?.Dispose();
     _webSocket?.Dispose();
+    _lock.Dispose();
   }
 }