"""Baseline network round-trip probe for hosted lanes (stdlib only).

For each host: n fresh TCP connects (pure network RTT) and n small HTTPS GETs

on one kept-alive TLS connection (RTT + server/edge time for a trivial

request). Separates network RTT from per-request server time.

"""

from __future__ import annotations

import http.client

import socket

import statistics

import time

def tcp_rtt_ms(host: str, port: int = 443, n: int = 20) -> list:

    out = []

    for _ in range(n):

        t = time.perf_counter()

        s = socket.create_connection((host, port), timeout=10)

        out.append((time.perf_counter() - t) * 1000)

        s.close()

    return out

def warm_get_ms(host: str, path: str, n: int = 20) -> list:

    c = http.client.HTTPSConnection(host, timeout=20)

    c.request("GET", path, headers={"User-Agent": "oakbench-rtt"})

    c.getresponse().read()

    out = []

    for _ in range(n):

        t = time.perf_counter()

        c.request("GET", path, headers={"User-Agent": "oakbench-rtt"})

        r = c.getresponse()

        r.read()

        out.append((time.perf_counter() - t) * 1000)

    c.close()

    return out

def summarize(vals: list) -> dict:

    v = sorted(vals)

    return {"n": len(v), "median_ms": statistics.median(v), "min_ms": v[0], "max_ms": v[-1]}

if __name__ == "__main__":

    import json

    import sys

    targets = [("oak.space", "/robots.txt"), ("cdn.oak.space", "/"), ("github.com", "/robots.txt")]

    res = {}

    for host, path in targets:

        res[host] = {"tcp_connect": summarize(tcp_rtt_ms(host)), "warm_https_get": summarize(warm_get_ms(host, path))}

    json.dump(res, sys.stdout, indent=1)