import json
import os
import urllib.error
import urllib.request

body = json.dumps({"query": "react admin dashboard with auth and data tables", "limit": 3}).encode("utf-8")
request = urllib.request.Request(
    "https://api.reposeek.ai/v1/search",
    data=body,
    headers={
        "Authorization": f"Bearer {os.environ['REPOSEEK_API_KEY']}",
        "Content-Type": "application/json",
    },
    method="POST",
)

try:
    with urllib.request.urlopen(request) as response:
        payload = json.loads(response.read().decode("utf-8"))
        request_id = response.headers.get("X-Request-Id")
        server_timing = response.headers.get("Server-Timing")
        print(json.dumps(payload, indent=2))
        if request_id:
            print(f"X-Request-Id: {request_id}")
        if server_timing:
            print(f"Server-Timing: {server_timing}")
except urllib.error.HTTPError as exc:
    payload = json.loads(exc.read().decode("utf-8"))
    print(json.dumps(payload, indent=2))
    raise SystemExit(exc.code)