#!/usr/bin/env python3
"""Fetch Klaus Dormann's 6502/65C02 functional test binaries.

They are not committed: they are GPL-licensed and 64K each, and this repository
is BSD. tests/test_klaus.c reports them skipped when they are absent, so this is
optional for a local build and a single step in CI.

    python tests/fetch_klaus.py

The listings that come with them (bin_files/*.lst) map an address to the check
that stopped there, which is how you read a failure. They are large, so they are
not fetched here; look them up in the upstream repository when you need one.
"""

import sys
import urllib.request
from pathlib import Path

DEST = Path(__file__).resolve().parent / "klaus"
BASE = ("https://raw.githubusercontent.com/Klaus2m5/"
        "6502_65C02_functional_tests/master/bin_files")

BINARIES = (
    "6502_functional_test.bin",
    "65C02_extended_opcodes_test.bin",
)

# The listings map an address back to the check that stopped there, which is
# what turns "trapped at $0ECE" into a line of source. Large, but the difference
# between a usable failure and a scavenger hunt.
LISTINGS = (
    "6502_functional_test.lst",
    "65C02_extended_opcodes_test.lst",
)

EXPECTED_SIZE = 65536


def get(name, path, expect_size=None):
    if path.exists() and (expect_size is None or path.stat().st_size == expect_size):
        print(f"  have {name}")
        return True
    print(f"  get  {name}")
    try:
        with urllib.request.urlopen(f"{BASE}/{name}", timeout=180) as r:
            data = r.read()
    except Exception as exc:
        print(f"  FAIL {name}: {exc}")
        return False
    if expect_size is not None and len(data) != expect_size:
        print(f"  FAIL {name}: got {len(data)} bytes, expected {expect_size}")
        return False
    path.write_bytes(data)
    return True


def main():
    DEST.mkdir(parents=True, exist_ok=True)
    ok = True

    for name in BINARIES:
        ok &= get(name, DEST / name, EXPECTED_SIZE)
    for name in LISTINGS:
        ok &= get(name, DEST / name)

    print(f"\ntest images in {DEST}")
    return 0 if ok else 1


if __name__ == "__main__":
    sys.exit(main())