from __future__ import annotations

import json
from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
from pathlib import Path


RESULTS_DIR = Path(".cache/browser-results")


class BrowserHandler(SimpleHTTPRequestHandler):
    def do_POST(self) -> None:
        if self.path != "/benchmark-result":
            self.send_error(404)
            return

        length = int(self.headers.get("Content-Length", "0"))
        if length <= 0 or length > 1_000_000:
            self.send_error(400, "invalid result size")
            return

        result = json.loads(self.rfile.read(length))
        bundle = result.get("bundle")
        if bundle not in {
            "compact-qdq",
            "compact-fp16",
            "compact-unified-qdq",
            "compact-unified-gather",
            "balanced-unified-qdq",
            "balanced-unified-gather",
            "balanced-qdq",
            "lossless",
        }:
            self.send_error(400, "unknown bundle")
            return

        RESULTS_DIR.mkdir(parents=True, exist_ok=True)
        destination = RESULTS_DIR / f"hayai13-{bundle}.json"
        destination.write_text(json.dumps(result, ensure_ascii=False, indent=2) + "\n")
        self.send_response(204)
        self.end_headers()

    def end_headers(self) -> None:
        self.send_header("Cross-Origin-Opener-Policy", "same-origin")
        self.send_header("Cross-Origin-Embedder-Policy", "require-corp")
        self.send_header("Cross-Origin-Resource-Policy", "same-origin")
        super().end_headers()


if __name__ == "__main__":
    ThreadingHTTPServer(("127.0.0.1", 8765), BrowserHandler).serve_forever()