scripts/serve: Optionally serve over https.

Allows local testing on mobile devices with Bluetooth, USB, and camera suppport.

Author: laurensvalk

scripts/serve.py

@@ -6,9 +6,11 @@
 
 import http.server
 import pathlib
-import socketserver
+import ssl
 
 BUILD_DIR = (pathlib.Path(__file__).parent.parent / "build").resolve()
+CERT_FILE = BUILD_DIR / "cert.pem"
+KEY_FILE = BUILD_DIR / "key.pem"
 PORT = 8000
 
 
@@ -17,10 +19,10 @@ def __init__(
         self,
         request: bytes,
         client_address: tuple[str, int],
-        server: socketserver.BaseServer,
+        server: http.server.HTTPServer,
         directory: str | None = ...
     ) -> None:
-        super().__init__(request, client_address, server, directory=BUILD_DIR)
+        super().__init__(request, client_address, server, directory=str(BUILD_DIR))
 
     def end_headers(self) -> None:
         # custom headers needed for some web API features
@@ -29,6 +31,14 @@ def end_headers(self) -> None:
         return super().end_headers()
 
 
-with socketserver.TCPServer(("", PORT), Handler) as httpd:
-    print(f"serving at http://0.0.0.0:{PORT}")
-    httpd.serve_forever()
+httpd = http.server.HTTPServer(("", PORT), Handler)
+
+if CERT_FILE.exists() and KEY_FILE.exists():
+    ctx = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
+    ctx.load_cert_chain(certfile=CERT_FILE, keyfile=KEY_FILE)
+    httpd.socket = ctx.wrap_socket(httpd.socket, server_side=True)
+    print(f"serving at https://0.0.0.0:{PORT}")
+else:
+    print(f"cert/key not found, serving without TLS at http://0.0.0.0:{PORT}")
+
+httpd.serve_forever()