diacraft/make-list.py

import toml
import glob
from typing import Optional
from httpx import AsyncClient
import asyncio

try:
    with open("modinfo.toml", "r") as fp:
        modinfo = toml.load(fp)
except OSError:
    modinfo = { "mods": {}, "categories": {} }


mods = {}
name_list: list[str] = []
for filename in glob.glob("mods/*.pw.toml"):
    modname = filename.removesuffix(".pw.toml").split("/", 1)[1]
    with open(filename, "r") as fp:
        mod = mods[modname] = toml.load(fp)

    modinfo["mods"].setdefault(modname, {})

    link: Optional[str] = None
    if "modrinth" in mod["update"]:
        link = f"https://modrinth.com/mod/{modname}"

    for k, v in {
        "name": mod["name"],
        "description": "XXX: update",
        "link": link,
        "side": mod.get("side", "unknown"),
        "optional": mod.get("option", {}).get("optional", False)
    }.items():
        modinfo["mods"][modname].setdefault(k, v)

    if modinfo["mods"][modname]["description"] == "XXX: update" or modinfo["mods"][modname].get("icon") is None:
        name_list.append(modname)

async def fetch_description(modname: str) -> tuple[str, dict]:
    async with AsyncClient(headers={
        "User-Agent": "Packwiz-Description-Generator/1.0 (hatkidchan@gmail.com)"
    }) as c:
        print(f"[ + ] GET {modname}")
        try:
            d = await c.get(f"https://api.modrinth.com/v2/project/{modname}")
            data = d.json()
        except Exception:
            return modname, {}
        return modname, data

async def update_descriptions():
    result = await asyncio.gather(*[
        fetch_description(modname) for modname in name_list
    ])
    for modname, data in result:
        modinfo["mods"][modname]["description"] = data.get("description", "")
        modinfo["mods"][modname]["icon"] = data.get("icon_url", "")

asyncio.run(update_descriptions())

categories: dict[str, list] = {}

for modname in modinfo["mods"]:
    if modname not in mods:
        modinfo["mods"][modname]["removed"] = True
    info = modinfo["mods"][modname]
    if info.get("icon", "") == "":
        print(f"[W] No icon set for {modname}")
    category = info.get("category", "unset")
    categories.setdefault(category, [])
    categories[category].append(modname)

modinfo.setdefault("categories", {})
for catname, category in categories.items():
    modinfo["categories"].setdefault(catname, {})
    for k, v in {
        "title": catname,
        "description": ""
    }.items():
        modinfo["categories"][catname].setdefault(k, v)

with open("modinfo.toml", "w") as fp:
    toml.dump(modinfo, fp)


with open("README.md", "w") as fp:
    fp.write("# Mods list\n")
    for catname, modlist in categories.items():
        catinfo = modinfo["categories"][catname]
        fp.write(f"## {catinfo['title']}\n")
        fp.write(catinfo["description"] + "\n")

        for modname in sorted(modlist):
            info = modinfo["mods"][modname]
            if info.get("removed"):
                continue
            tags = []
            if info.get("optional", False):
                tags.append(("O", "optional"))
            side = info.get("side", "unknown")
            if side == "client":
                tags.append(("CO", "side:client"))
            elif side == "server":
                tags.append(("SO", "side:server"))
            elif side == "both":
                tags.append(("CS", "side:both"))

            try:
                fp.write(" * ")
                fp.write(f'<img width="32" height="32" src="{info['icon']}">')
                fp.write(str.join("", [
                    f"&#91;<abbr title=\"{tag[1]}\">{tag[0]}</abbr>&#93;"
                    for tag in tags
                ]))
                fp.write(f" **[{info['name']}]({info['link']})**")
                fp.write(f" - {info['description']}\n")
            except Exception as e:
                print(f"Err: {e!r} for {modname = }")
                raise