Fix proxy with TLS scenario with SocketConnectionFactory

src/NATS.Client.Core/NatsConnection.cs

@@ -335,7 +335,12 @@ private async ValueTask InitialConnectAsync()
                 }
 
                 _logger.LogInformation(NatsLogEvents.Connection, "Try to connect NATS {0}", uri);
-                if (uri.IsWebSocket)
+                if (Opts.SocketConnectionFactory != null)
+                {
+                    _logger.LogInformation(NatsLogEvents.Connection, "Try to connect using SocketConnectionFactory {0}", uri);
+                    _socket = await Opts.SocketConnectionFactory(uri, Opts, _disposedCancellationTokenSource.Token).ConfigureAwait(false);
+                }
+                else if (uri.IsWebSocket)
                 {
                     var conn = new WebSocketConnection();
                     await conn.ConnectAsync(uri, Opts).ConfigureAwait(false);
@@ -641,7 +646,13 @@ private async void ReconnectLoop()
                     }
 
                     _logger.LogInformation(NatsLogEvents.Connection, "Tried to connect NATS {Url} [{ReconnectCount}]", url, reconnectCount);
-                    if (url.IsWebSocket)
+
+                    if (Opts.SocketConnectionFactory != null)
+                    {
+                        _logger.LogInformation(NatsLogEvents.Connection, "Try to connect using SocketConnectionFactory {Url} [{ReconnectCount}]", url, reconnectCount);
+                        _socket = await Opts.SocketConnectionFactory(url, Opts, _disposedCancellationTokenSource.Token).ConfigureAwait(false);
+                    }
+                    else if (url.IsWebSocket)
                     {
                         _logger.LogDebug(NatsLogEvents.Connection, "Trying to reconnect using WebSocket {Url} [{ReconnectCount}]", url, reconnectCount);
                         var conn = new WebSocketConnection();

src/NATS.Client.Core/NatsOpts.cs

@@ -1,9 +1,7 @@
-using System.Net.WebSockets;
 using System.Text;
 using System.Threading.Channels;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Logging.Abstractions;
-using NATS.Client.Core.Internal;
 #if NETSTANDARD
 using Random = NATS.Client.Core.Internal.NetStandardExtensions.Random;
 #endif
@@ -142,6 +140,8 @@ public sealed record NatsOpts
     /// </remarks>
     public BoundedChannelFullMode SubPendingChannelFullMode { get; init; } = BoundedChannelFullMode.DropNewest;
 
+    public Func<NatsUri, NatsOpts, CancellationToken, ValueTask<ISocketConnection>>? SocketConnectionFactory { get; init; } = null;
+
     internal NatsUri[] GetSeedUris(bool suppressRandomization = false)
     {
         var urls = Url.Split(',');

src/NATS.Client.Core/Internal/NatsUri.cs → src/NATS.Client.Core/NatsUri.cs

@@ -1,6 +1,6 @@
-namespace NATS.Client.Core.Internal;
+namespace NATS.Client.Core;
 
-internal sealed class NatsUri : IEquatable<NatsUri>
+public sealed class NatsUri : IEquatable<NatsUri>
 {
     public const string DefaultScheme = "nats";