server.py

#!/usr/bin/env python
import sys

import mimetypes

mimetypes.init()

if ".wasm" not in mimetypes.types_map:
    print(
        "WARNING: wasm mimetype unsupported on that system, trying to correct",
        file=sys.stderr,
    )
    mimetypes.types_map[".wasm"] = "application/wasm"

import argparse
from http import server, HTTPStatus

import io
import os

import urllib


from pathlib import Path

CWD = os.getcwd()


parser = argparse.ArgumentParser(
    description="Start a local webserver with a Python terminal."
)
parser.add_argument(
    "--port", type=int, default=8000, help="port for the http server to listen on"
)
args = parser.parse_args()


class MyHTTPRequestHandler(server.SimpleHTTPRequestHandler):
    def end_headers(self):
        self.send_header("Access-Control-Allow-Origin", "*")
        self.send_header("Cross-Origin-Opener-Policy", "unsafe-none")
        self.send_header("Cross-Origin-Embedder-Policy", "unsafe-none")
        super().end_headers()

    def do_GET(self):
        f = self.send_head()
        if f:
            try:
                self.copyfile(f, self.wfile)
            finally:
                f.close()

    def do_HEAD(self):
        f = self.send_head()
        if f:
            f.close()

    def send_head(self):
        path = self.translate_path(self.path)
        print(path)

        if os.path.isdir(path):
            parts = urllib.parse.urlsplit(self.path)
            if not parts.path.endswith("/"):
                # redirect browser - doing basically what apache does
                self.send_response(HTTPStatus.MOVED_PERMANENTLY)
                new_parts = (parts[0], parts[1], parts[2] + "/", parts[3], parts[4])
                new_url = urllib.parse.urlunsplit(new_parts)
                self.send_header("Location", new_url)
                self.end_headers()
                return None
            for index in "index.html", "index.htm":
                index = os.path.join(path, index)
                if os.path.exists(index):
                    path = index
                    break
            else:
                return self.list_directory(path)

        ctype = self.guess_type(path)

        if not os.path.isfile(path):
            self.send_response(404)
            self.send_header("content-length", 0)
            self.end_headers()
            return b""

        self.send_response(HTTPStatus.OK)

        with open(path, "rb") as f:
            content = f.read()
            if path.endswith(".html"):
                content = content.replace(
                    b"https://pygame-web.github.io/", b"http://localhost:8000/"
                )

        self.send_header("content-type", ctype)
        self.send_header("content-length", str(len(content)))
        self.end_headers()

        return io.BytesIO(content)


server.test(HandlerClass=MyHTTPRequestHandler, protocol="HTTP/1.1", port=args.port)