import asyncio
import sys
import threading
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
from pathlib import Path

EXAMPLES_DIR = Path(__file__).resolve().parent
sys.path = [entry for entry in sys.path if Path(entry or ".").resolve() != EXAMPLES_DIR]

import aiohttp
import httpx
import requests
import urllib3
import wreq


class Handler(BaseHTTPRequestHandler):
    def do_GET(self):
        routes = {
            "/204": (204, b"", None),
            "/empty": (200, b"", "application/json"),
            "/space": (200, b"   \n\t", "application/json"),
            "/null": (200, b"null", "application/json"),
            "/json": (200, b'{"ok": true}', "application/json"),
        }

        status, body, content_type = routes.get(self.path, (404, b"not found", "text/plain"))
        self.send_response(status)
        if content_type is not None:
            self.send_header("Content-Type", content_type)
        self.send_header("Content-Length", str(len(body)))
        self.end_headers()
        if body:
            self.wfile.write(body)

    def log_message(self, format, *args):
        return


def format_result(result):
    if isinstance(result, Exception):
        return f"{type(result).__name__}: {result}"
    return repr(result)


def run_requests(url):
    try:
        return requests.get(url, timeout=5).json()
    except Exception as exc:
        return exc


def run_httpx(url):
    try:
        return httpx.get(url, timeout=5).json()
    except Exception as exc:
        return exc


def run_urllib3(http, url):
    try:
        return http.request("GET", url, timeout=5.0).json()
    except Exception as exc:
        return exc


async def run_aiohttp(url):
    try:
        async with aiohttp.ClientSession() as session:
            async with session.get(url) as resp:
                return await resp.json()
    except Exception as exc:
        return exc


async def run_wreq(url):
    try:
        resp = await wreq.get(url)
        async with resp:
            return await resp.json()
    except Exception as exc:
        return exc


async def main():
    server = ThreadingHTTPServer(("127.0.0.1", 0), Handler)
    thread = threading.Thread(target=server.serve_forever, daemon=True)
    thread.start()

    base_url = f"http://127.0.0.1:{server.server_port}"
    paths = ["/204", "/empty", "/space", "/null", "/json"]
    http = urllib3.PoolManager()

    try:
        for path in paths:
            url = f"{base_url}{path}"
            print(f"\n=== {path} ===")
            print("requests:", format_result(run_requests(url)))
            print("httpx:", format_result(run_httpx(url)))
            print("urllib3:", format_result(run_urllib3(http, url)))
            print("aiohttp:", format_result(await run_aiohttp(url)))
            print("wreq:", format_result(await run_wreq(url)))
    finally:
        http.clear()
        server.shutdown()
        server.server_close()
        thread.join(timeout=1)


if __name__ == "__main__":
    asyncio.run(main())