Stupidly recursively walk narinfo's

walk-narinfo.py

import urllib3
import sys

CACHE = "https://cache.nixos.org"

visited: set[str] = set()

def get_narinfo(hash) -> dict[str, str | list[str]]:
    resp = urllib3.request("GET", f"{CACHE}/{hash}.narinfo")
    if resp.status != 200:
        raise ValueError(f"Status: {resp.status}")

    ret = { }
    for line in resp.data.decode().split("\n"):
        if not line:
            continue
        name, data = line.split(": ")
        if name == "References":
            data = data.split()
        ret[name] = data

    return ret


def recurse(hash: str):
    if hash in visited:
        return
    visited.add(hash)

    narinfo = get_narinfo(hash)
    print(narinfo["StorePath"].split("/")[-1])
    for reference in narinfo.get("References", []):
        hash, _ = reference.split("-", 1)
        recurse(hash)

def main() -> None:
    store_path = sys.argv[1]
    hash, _ = store_path.split("/")[-1].split("-", 1)
    recurse(hash)