feedback

Author: pavelsavara

src/Microsoft.Diagnostics.WebSocketServer/WebSocketServerImpl.cs

@@ -40,9 +40,18 @@ public async Task StartServer(string endpoint, CancellationToken cancellationTok
 
         ParseWebSocketURL(endpoint, out Uri uri);
 
+        string scheme = uri.Scheme switch
+        {
+            "ws" => "http",
+            "http" => "http",
+            "wss" => "https",
+            "https" => "https",
+            _ => throw new ArgumentException(string.Format("Unsupported Uri schema, \"{0}\"", uri.Scheme))
+        };
+
         EmbeddedWebSocketServer.Options options = new()
         {
-            Scheme = uri.Scheme,
+            Scheme = scheme,
             Host = uri.Host,
             Port = uri.Port.ToString(),
             Path = uri.PathAndQuery,
@@ -172,24 +181,24 @@ private static void ParseWebSocketURL(string endPoint, out Uri uri)
             uriToParse = endPoint;
         }
 
+        string[] supportedSchemes = new string[] { "ws", "wss", "http", "https" };
 
         if (!string.IsNullOrEmpty(uriToParse) && Uri.TryCreate(uriToParse, UriKind.Absolute, out uri))
         {
-            if (uri.Scheme == "ws")
-            {
-                uri = new UriBuilder(uri) { Scheme = "http" }.Uri;
-                return;
-            }
-            if (uri.Scheme == "wss")
+            bool supported = false;
+            foreach (string scheme in supportedSchemes)
             {
-                uri = new UriBuilder(uri) { Scheme = "https" }.Uri;
-                return;
+                if (string.Equals(uri.Scheme, scheme, StringComparison.InvariantCultureIgnoreCase))
+                {
+                    supported = true;
+                    break;
+                }
             }
-            if (uri.Scheme == "http" || uri.Scheme == "https")
+            if (!supported)
             {
-                return;
+                throw new ArgumentException(string.Format("Unsupported Uri schema, \"{0}\"", uri.Scheme));
             }
-            throw new ArgumentException(string.Format("Unsupported Uri schema, \"{0}\"", uri.Scheme));
+            return;
         }
         else
         {