from http.server import BaseHTTPRequestHandler, HTTPServer
import json
import os


class HealthHandler(BaseHTTPRequestHandler):
    def do_GET(self) -> None:
        if self.path != "/healthz":
            self.send_response(404)
            self.end_headers()
            return
        body = json.dumps({"status": "ok", "agent": "nexafabric"}).encode()
        self.send_response(200)
        self.send_header("Content-Type", "application/json")
        self.send_header("Content-Length", str(len(body)))
        self.end_headers()
        self.wfile.write(body)


def main() -> None:
    host = os.getenv("NEXAFABRIC_AGENT_HOST", "0.0.0.0")
    port = int(os.getenv("NEXAFABRIC_AGENT_PORT", "9844"))
    HTTPServer((host, port), HealthHandler).serve_forever()


if __name__ == "__main__":
    main()