server.py

import http.server
import json
import os
import uuid
from datetime import datetime as dt
from pathlib import Path
from typing import Callable
import time

BUILD_DIR = Path("_build")

server_id = uuid.uuid4()


def runserver(host: str, port: int, build: Callable[[Path], None]):
    build(BUILD_DIR)

    with http.server.ThreadingHTTPServer((host, port), RequestHandler) as httpd:  # noqa
        httpd.timeout = None
        changes = os.getenv("WATCHFILES_CHANGES")
        changes = json.loads(changes)
        if changes:
            print(f"\n{', '.join(v[1] for v in changes)} changed, reloading.\n")

        print(dt.now().strftime("%B %d, %Y %H:%M"))
        print("Sitegen version 0.1.0")
        print(f"Starting development server at http://{host}:{port}")
        print("Quit the server with CONTROL-C")
        try:
            httpd.serve_forever()
        except KeyboardInterrupt:
            print("Server shutting down ...")


class RequestHandler(http.server.SimpleHTTPRequestHandler):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, directory=str(BUILD_DIR), **kwargs)

    def end_headers(self):
        if not self.path.startswith("/__reload__"):
            self.send_header("Connection", "close")
            self.send_header("Cache-Control", "no-cache, no-store, must-revalidate")
            self.send_header("Pragma", "no-cache")
            self.send_header("Expires", "0")
        super().end_headers()

    def do_GET(self):  # noqa
        if self.path.startswith("/__reload__"):
            return self.handle_reload()
        super().do_GET()

    def handle_reload(self):
        self.send_header("Content-Type", "text/event-stream")
        self.send_header("Cache-Control", "no-cache")
        self.send_header("Access-Control-Allow-Origin", "*")
        self.end_headers()
        while True:
            self.wfile.write("data: reloading.\n\n".encode())
            self.wfile.flush()
            time.sleep(1)
        # while True:
        #     changes = os.getenv("WATCHFILES_CHANGES")
        #     changes = json.loads(changes)
        #     print(changes)
        #     self.wfile.write(b"data: reloading\n\n")
        #     time.sleep(1)
        

    def handle_one_request(self):
        try:
            super().handle_one_request()
        except ConnectionResetError:
            pass  # Ignore client disconnects