Merge pull request #10 from HoughIO/fix_websocket_over_https

update websocket uri to use wss if https is in the Server.public_url

Author: willmcgugan

src/textual_serve/server.py

@@ -93,6 +93,8 @@ def __init__(
         if public_url is None:
             if self.port == 80:
                 self.public_url = f"http://{self.host}"
+            elif self.port == 443:
+                self.public_url = f"https://{self.host}"
             else:
                 self.public_url = f"http://{self.host}:{self.port}"
         else:
@@ -251,7 +253,11 @@ def get_url(route: str, **args) -> str:
         def get_websocket_url(route: str, **args) -> str:
             """Get a URL with a websocket prefix."""
             url = get_url(route, **args)
-            return "ws:" + url.split(":", 1)[1]
+
+            if self.public_url.startswith("https"):
+                return "wss:" + url.split(":", 1)[1]
+            else:
+                return "ws:" + url.split(":", 1)[1]
 
         context = {
             "font_size": font_size,