#!/usr/bin/env python3
import json
import sys
import re

CATEGORIZE = {
    r".el7.x86_64.rpm$": "centos7_rpm",
    r".centos7.rpm$": "centos7_rpm",
    r".el8.x86_64.rpm$": "centos8_rpm",
    r".centos8.rpm$": "centos8_rpm",
    r".fc31.x86_64.rpm$": "fedora31_rpm",
    r".fedora31.rpm$": "fedora31_rpm",
    r".fc32.x86_64.rpm$": "fedora32_rpm",
    r".fedora32.rpm$": "fedora32_rpm",
    r"Debian9.12.deb$": "debian9_deb",
    r"Debian10.deb$": "debian10_deb",
    r"Ubuntu16.04.AppImage$": "ubuntu16_AppImage",
    r"^WezTerm-.*.x86_64.AppImage$": "ubuntu16_AppImage",
    r"Ubuntu16.04.deb$": "ubuntu16_deb",
    r"^wezterm-\d+-\d+-[a-f0-9]+.deb$": "ubuntu16_deb",
    r"Ubuntu18.04.deb$": "ubuntu18_deb",
    r"Ubuntu19.10.deb$": "ubuntu19_deb",
    r"Ubuntu20.04.deb$": "ubuntu20_deb",
    r"Ubuntu16.04.tar.xz$": "linux_raw_bin",
    r"^wezterm-\d+-\d+-[a-f0-9]+.tar.xz$": "linux_raw_bin",
    r"src.tar.gz$": "src",
    r"^WezTerm-macos-.*.zip$": "macos_zip",
    r"^WezTerm-windows-.*.zip$": "windows_zip",
}

def categorize(rel):
    downloads = {}

    tag_name = "wezterm-%s" % rel["tag_name"]
    for asset in rel["assets"]:
        url = asset["browser_download_url"]
        name = asset["name"]

        for k, v in CATEGORIZE.items():
            if re.search(k, name):
                downloads[v] = (url, name, tag_name)

    return downloads

def pretty(o):
    return json.dumps(o, indent=4, sort_keys=True, separators=(',', ':'))

def build_subst(subst, stable, categorized):
    for (kind, info) in categorized.items():
        if info is None:
            continue
        url, name, dir = info
        kind = f"{kind}_{stable}"
        subst["{{ %s }}" % kind] = url
        subst["{{ %s_asset }}" % kind] = name
        subst["{{ %s_dir }}" % kind] = dir

def load_release_info():
    with open("/tmp/wezterm.releases.json") as f:
        release_info = json.load(f)

    latest = release_info[0]
    nightly = None
    for rel in release_info:
        if rel["tag_name"] == "nightly":
            nightly = rel
            break

    latest = categorize(latest)
    nightly = categorize(nightly)

    print('latest: ', pretty(latest))
    print('nightly: ', pretty(nightly))

    subst = {}
    build_subst(subst, "stable", latest)
    build_subst(subst, "nightly", nightly)
    print(pretty(subst))

    with open("docs/installation.markdown", "r") as input:
        with open("docs/installation.md", "w") as output:
            for line in input:
                for (search, replace) in subst.items():
                    line = line.replace(search, replace)
                output.write(line)

def main():
    load_release_info()

main()