#!/usr/bin/env python3
"""HTTP server that serves the POC with COOP/COEP headers so that
crossOriginIsolated is true and SharedArrayBuffer is available."""
import http.server
import socketserver
import sys

PORT = 8080
if len(sys.argv) > 1:
    PORT = int(sys.argv[1])


class Handler(http.server.SimpleHTTPRequestHandler):
    def end_headers(self):
        self.send_header('Cross-Origin-Opener-Policy', 'same-origin')
        self.send_header('Cross-Origin-Embedder-Policy', 'require-corp')
        self.send_header('Cache-Control', 'no-store')
        super().end_headers()

    def log_message(self, fmt, *args):
        sys.stderr.write("[server] " + (fmt % args) + "\n")


socketserver.TCPServer.allow_reuse_address = True
with socketserver.TCPServer(("127.0.0.1", PORT), Handler) as httpd:
    sys.stderr.write(f"[server] serving on http://127.0.0.1:{PORT}/\n")
    httpd.serve_forever()
