#!/usr/bin/env python3

"""

Serve the build directory using builtin Python web server.

"""

import http.server

import pathlib

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

class Handler(http.server.SimpleHTTPRequestHandler):

def __init__(

self,

request: bytes,

client_address: tuple[str, int],

server: http.server.HTTPServer,

directory: str | None = ...

) -> None:

super().__init__(request, client_address, server, directory=str(BUILD_DIR))

def end_headers(self) -> None:

# custom headers needed for some web API features

self.send_header("Cross-Origin-Opener-Policy", "same-origin")

self.send_header("Cross-Origin-Embedder-Policy", "require-corp")

return super().end_headers()

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()