download_heroicons.py

#!/usr/bin/env python
"""
Download the latest heroicons zip file and select only the optimized icons.
"""
from __future__ import annotations

import argparse
import os
import subprocess
from io import BytesIO
from zipfile import ZIP_DEFLATED
from zipfile import ZipFile


def main(argv: list[str] | None = None) -> int:
    parser = argparse.ArgumentParser()
    parser.add_argument("version", help="dotted version number")
    args = parser.parse_args(argv)
    version: str = args.version

    proc = subprocess.run(
        [
            "curl",
            "--fail",
            "--location",
            f"https://github.com/tailwindlabs/heroicons/archive/v{version}.zip",
        ],
        stdout=subprocess.PIPE,
    )
    if proc.returncode != 0:
        raise SystemExit(1)

    input_zip = ZipFile(BytesIO(proc.stdout))
    input_prefix = f"heroicons-{version}/optimized/"

    output_path = "src/heroicons/heroicons.zip"

    try:
        os.remove(output_path)
    except FileNotFoundError:
        pass
    with ZipFile(
        output_path, "w", compression=ZIP_DEFLATED, compresslevel=9
    ) as output_zip:
        for name in sorted(input_zip.namelist()):
            if name.startswith(input_prefix) and name.endswith(".svg"):
                info = input_zip.getinfo(name)
                data = input_zip.read(name).replace(b' data-slot="icon"', b"")

                new_name = rename_file(name[len(input_prefix) :])

                info.filename = new_name
                output_zip.writestr(info, data)
                print(new_name)

    print("\n✅ Written!")

    return 0


def rename_file(filename: str) -> str:
    if filename.startswith("24/solid"):
        return filename[len("24/") :]
    elif filename.startswith("24/outline"):
        return filename[len("24/") :]
    elif filename.startswith("20/solid"):
        return "mini" + filename[len("20/solid") :]
    elif filename.startswith("16/solid"):
        return "micro" + filename[len("16/solid") :]
    else:
        raise ValueError(f"Unknown filename {filename!r}")


if __name__ == "__main__":
    raise SystemExit(main())