import json


# Utils
def slugify(name):
    return name.lower().replace(" ", "-").replace("(", "").replace(")", "")


def generate_table_of_contents():
    with open("core/data/static/categories.json", "r", encoding="utf-8") as f:
        data = json.load(f)
    categories = data.get("categories", [])
    subcategories = data.get("subcategories", [])

    # Build the alphabetical list (ignoring parent categories)
    subcat_names = [sub["name"] for sub in subcategories]
    subcat_names.sort(key=lambda x: x.lower())
    alphabetical_md = ""
    for name in subcat_names:
        alphabetical_md += f"- [{name}](#{slugify(name)})\n"

    # Build the categorized list
    parent_map = {cat["id"]: cat["name"] for cat in categories}

    grouped = {}
    for sub in subcategories:
        parent = sub.get("parent", "other")
        grouped.setdefault(parent, []).append(sub["name"])

    for key in grouped:
        grouped[key].sort(key=lambda x: x.lower())

    parents = [(pid, parent_map.get(pid, "Other")) for pid in grouped if pid != "other"]
    parents.sort(key=lambda x: x[1].lower())
    if "other" in grouped:
        parents.append(("other", "Other"))

    categorized_md_lines = []
    for pid, pname in parents:
        categorized_md_lines.append(f"- {pname}")
        for subname in grouped[pid]:
            categorized_md_lines.append(f"    - [{subname}](#{slugify(subname)})")

    # Append fixed sections at the end of the categorized TOC
    fixed_sections = [
        "Removed Projects",
        "FAQ",
        "Honorable Mentions of Closed-Source Software",
    ]
    for item in fixed_sections:
        categorized_md_lines.append(f"- [{item}](#{slugify(item)})")

    categorized_md = "\n".join(categorized_md_lines)

    toc = f"""## Table of Contents

<details>
  <summary><b>Alphabetical</b></summary> <br />

{alphabetical_md}
</details>

<details open>
  <summary><b>Categorized</b></summary> <br />

{categorized_md}
</details>
"""
    return toc


if __name__ == "__main__":
    # For testing the TOC generator
    print(generate_table_of_contents())