"""Draw the study's charts as SVG from results/portals.csv.

    python3 make_charts.py results/portals.csv out_dir/

Standard library only. Every chart states its own basis in the image, because
an embedded chart travels without the page's caption.
"""
import csv, os, sys
from xml.sax.saxutils import escape

NAMES = {"armor": "Armor Games", "coolmath": "Coolmath Games", "abcya": "ABCya", "kizi": "Kizi",
         "y8": "Y8", "itch": "itch.io", "friv": "Friv", "manxlab": "MANXLAB (ours)"}
BG, CARD, TEXT, MUTED, RULE = "#08120f", "#0d1a16", "#f0ede4", "#8fa89c", "#1d2f29"
HOT, MID, COOL = "#e8735f", "#e0b35a", "#3bd48f"
FONT = "Inter, -apple-system, 'Segoe UI', Roboto, Helvetica, Arial, sans-serif"
BASIS = ("Median of 10 games per portal (8 for MANXLAB), each the median of 3 loads. "
         "Chrome, US connection, 28 Sep 2026, nothing clicked.")
CREDIT = "manxlab.com/research/free-game-trackers"


def colour(v):
    return HOT if v >= 100 else MID if v >= 5 else COOL


def fmt(v):
    v = float(v)
    return f"{v:,.0f}" if v == int(v) else f"{v:,.1f}"


def wrap(text, n):
    words, lines, cur = text.split(), [], ""
    for w in words:
        if len(cur) + len(w) + 1 > n:
            lines.append(cur)
            cur = w
        else:
            cur = (cur + " " + w).strip()
    return lines + [cur]


def bars(path, title, subtitle, rows, value_key, range_keys=None, unit=""):
    W, L, R, top, band = 720, 170, 170, 110, 40
    rows = sorted(rows, key=lambda r: -float(r[value_key]))
    vmax = max(float(r[value_key]) for r in rows) or 1
    H = top + band * len(rows) + 110
    s = [f'<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 {W} {H}" width="{W}" height="{H}" '
         f'role="img" aria-label="{escape(title)}. {escape(subtitle)}" font-family="{FONT}">',
         f'<title>{escape(title)}</title><desc>{escape(subtitle + " " + BASIS)}</desc>',
         f'<rect width="{W}" height="{H}" fill="{BG}"/>',
         f'<text x="24" y="40" font-size="21" font-weight="700" fill="{TEXT}">{escape(title)}</text>']
    for k, line in enumerate(wrap(subtitle, 78)):
        s.append(f'<text x="24" y="{66 + k * 19}" font-size="14" fill="{MUTED}">{escape(line)}</text>')
    span = W - L - R
    for i, r in enumerate(rows):
        y = top + i * band
        v = float(r[value_key])
        w = max(2, span * v / vmax)
        s.append(f'<text x="{L - 12}" y="{y + 21}" font-size="14" text-anchor="end" fill="{TEXT}">'
                 f'{escape(NAMES.get(r["portal"], r["portal"]))}</text>')
        if v > 0:
            s.append(f'<rect x="{L}" y="{y + 6}" width="{w:.1f}" height="22" rx="4" fill="{colour(v)}"/>')
        label = fmt(v) + unit
        if range_keys:
            lo, hi = r[range_keys[0]], r[range_keys[1]]
            if lo != hi:
                label += f"  ({fmt(lo)}–{fmt(hi)})"
        s.append(f'<text x="{L + (w if v > 0 else 0) + 8:.1f}" y="{y + 22}" font-size="14" font-weight="600" fill="{TEXT}">'
                 f'{escape(label)}</text>')
    y0 = top + band * len(rows) + 22
    for k, line in enumerate(wrap(BASIS + (" Brackets: lowest–highest game." if range_keys else ""), 100)):
        s.append(f'<text x="24" y="{y0 + k * 17}" font-size="12" fill="{MUTED}">{escape(line)}</text>')
    s.append(f'<text x="{W - 24}" y="{H - 18}" font-size="12" font-weight="600" text-anchor="end" '
             f'fill="{COOL}">{CREDIT}</text></svg>')
    open(path, "w").write("\n".join(s))
    print("wrote", path, f"{W}x{H}")


def timeline(path, rows):
    W, H, L, R, top, bottom = 720, 470, 70, 210, 110, 370
    title = "The auction starts before the game does"
    sub = "Advertising and tracking companies contacted, counted at 5, 10, 30 and 60 seconds after the page opened."
    xs = [5, 10, 30, 60]
    keep = [r for r in rows if r["portal"] in ("armor", "coolmath", "abcya", "manxlab")]
    vmax = max(float(r["ad_tracking_companies_60s"]) for r in keep)
    px = lambda i: L + (W - L - R) * i / (len(xs) - 1)
    py = lambda v: bottom - (bottom - top) * v / vmax
    s = [f'<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 {W} {H}" width="{W}" height="{H}" '
         f'role="img" aria-label="{escape(title)}. {escape(sub)}" font-family="{FONT}">',
         f'<title>{escape(title)}</title><desc>{escape(sub + " " + BASIS)}</desc>',
         f'<rect width="{W}" height="{H}" fill="{BG}"/>',
         f'<text x="24" y="40" font-size="21" font-weight="700" fill="{TEXT}">{escape(title)}</text>']
    for k, line in enumerate(wrap(sub, 78)):
        s.append(f'<text x="24" y="{66 + k * 19}" font-size="14" fill="{MUTED}">{escape(line)}</text>')
    for i, x in enumerate(xs):
        s.append(f'<line x1="{px(i):.1f}" y1="{top}" x2="{px(i):.1f}" y2="{bottom}" stroke="{RULE}"/>')
        s.append(f'<text x="{px(i):.1f}" y="{bottom + 22}" font-size="13" text-anchor="middle" fill="{MUTED}">{x} s</text>')
    for v in range(0, int(vmax) + 1, 50):
        s.append(f'<line x1="{L}" y1="{py(v):.1f}" x2="{px(3):.1f}" y2="{py(v):.1f}" stroke="{RULE}"/>')
        s.append(f'<text x="{L - 10}" y="{py(v) + 4:.1f}" font-size="12" text-anchor="end" fill="{MUTED}">{v}</text>')
    for r in sorted(keep, key=lambda r: -float(r["ad_tracking_companies_60s"])):
        vals = [float(r[f"ad_tracking_companies_{x}s"]) for x in xs]
        c = colour(vals[-1])
        dash = ' stroke-dasharray="7 5"' if r["portal"] == "coolmath" else ""
        pts = " ".join(f"{px(i):.1f},{py(v):.1f}" for i, v in enumerate(vals))
        s.append(f'<polyline points="{pts}" fill="none" stroke="{c}" stroke-width="3" stroke-linejoin="round"{dash}/>')
        for i, v in enumerate(vals):
            s.append(f'<circle cx="{px(i):.1f}" cy="{py(v):.1f}" r="4" fill="{c}"/>')
        s.append(f'<text x="{px(3) + 12:.1f}" y="{py(vals[-1]) + 5:.1f}" font-size="14" font-weight="600" fill="{TEXT}">'
                 f'{escape(NAMES[r["portal"]])} {fmt(vals[-1])}</text>')
    for k, line in enumerate(wrap(BASIS, 100)):
        s.append(f'<text x="24" y="{bottom + 52 + k * 17}" font-size="12" fill="{MUTED}">{escape(line)}</text>')
    s.append(f'<text x="{W - 24}" y="{H - 18}" font-size="12" font-weight="600" text-anchor="end" '
             f'fill="{COOL}">{CREDIT}</text></svg>')
    open(path, "w").write("\n".join(s))
    print("wrote", path, f"{W}x{H}")


def main(portals_csv, out_dir):
    rows = list(csv.DictReader(open(portals_csv)))
    os.makedirs(out_dir, exist_ok=True)
    bars(os.path.join(out_dir, "companies.svg"),
         "Ad and tracking companies, first minute",
         "Advertising and tracking companies a free game page contacted in its first 60 seconds, before anything was clicked.",
         rows, "ad_tracking_companies_60s", ("ad_tracking_companies_60s_min", "ad_tracking_companies_60s_max"))
    bars(os.path.join(out_dir, "cookies.svg"),
         "Outside domains setting cookies",
         "Domains other than the portal's own that had set a cookie after 60 seconds, before anything was clicked.",
         rows, "third_party_cookie_domains_60s", ("third_party_cookie_domains_60s_min", "third_party_cookie_domains_60s_max"))
    bars(os.path.join(out_dir, "requests.svg"),
         "Network requests, first minute",
         "Requests a free game page made in its first 60 seconds, before anything was clicked.",
         rows, "requests_60s", ("requests_60s_min", "requests_60s_max"))
    timeline(os.path.join(out_dir, "timeline.svg"), rows)


if __name__ == "__main__":
    main(*sys.argv[1:])