#!/usr/bin/env python

from pip._vendor import requests
from pip._vendor import tomli
import json
import glob
import os

mods = dict()
count = dict()
count["server"] = 0
count["client"] = 0
count["both"] = 0

cache = dict()

if os.path.isfile("licenses.json"):
  with open("licenses.json") as f:
    cache = json.load(f)

for mod in glob.glob("pack/mods/*.toml"):
  with open(mod, "r") as f:
    data = tomli.load(f)
    moddata = dict()
    moddata["name"] = data["name"]
    moddata["url"] = ""
    moddata["side"] = data["side"]

    if "modrinth" in data["update"]:
      id = data["update"]["modrinth"]["mod-id"]
      moddata["id"] = id
      moddata["url"] = "https://modrinth.com/mod/" + id
      moddata["site"] = "Modrinth"

      if moddata["name"] in cache:
        data = cache[id]
      else:
        data = requests.get("https://api.modrinth.com/v2/project/" + id).json()["license"]
        cache[id] = data
    elif "curseforge" in data["update"]:
      moddata["url"] = "https://legacy.curseforge.com/projects/" + str(data["update"]["curseforge"]["project-id"])
      moddata["site"] = "CurseForge"

    count[moddata["side"]] += 1
    mods[moddata["name"]] = moddata

with open("wiki/Modlist.md", "w") as f:
  f.write("## Total Count\r\n")
  f.write("<table>")
  f.write("\r\n<tr>")
  f.write("<th>Side</th>")
  f.write("<th>Count</th>")
  f.write("</tr>\r\n")

  f.write("<tr>")
  f.write("<td>Total</td>")
  f.write("<td>" + str(count["server"] + count["client"] + count["both"]) + "</td>")
  f.write("</tr>\r\n")

  f.write("<tr>")
  f.write("<td>Shared</td>")
  f.write("<td>" + str(count["both"]) + "</td>")
  f.write("</tr>\r\n")

  f.write("<tr>")
  f.write("<td>Client</td>")
  f.write("<td>" + str(count["client"]) + "</td>")
  f.write("</tr>\r\n")

  f.write("<tr>")
  f.write("<td>Server</td>")
  f.write("<td>" + str(count["server"]) + "</td>")
  f.write("</tr>\r\n")

  f.write("</table>\r\n\r\n")

  f.write("## Individual Mods\r\n")
  f.write("<table>")
  f.write("\r\n<tr>")
  f.write("<th>Name</th>")
  f.write("<th>Side</th>")
  f.write("<th>Link</th>")
  f.write("</tr>\r\n")

  for mod in mods:
    data = mods[mod]
    f.write("\r\n<tr>")
    f.write("<td>" + mod + "</td>")
    f.write("<td>" + data["side"] + "</td>")
    if data["url"] != "":
      f.write("<td><a href=\"" + data["url"] + "\">" + data["site"] + "</a></td>")
    else:
      f.write("<td>N/A</td>")
    f.write("\r\n</tr>")
  f.write("\r\n</table>")

with open("wiki/Licenses.md", "w") as f:
  for mod in mods:
    data = mods[mod]
    f.write("## " + mod + "\r\n")
    f.write("<b>")
    if "license" in data:
      if data["license"]["url"] != None:
        f.write("<a href=\"" + data["license"]["url"] + "\">" + data["license"] + "")
      else:
        f.write(data["license"]["name"])
    else:
      f.write("Unknown")
    f.write("</b>\r\n")

with open("licenses.json") as f:
  json.dump(cache, f)