#!/usr/bin/env python3
import os
import re
from pathlib import Path
from typing import List

BASE_URL = "https://huggingface.co/csukuangfj/sherpa-onnx-apk/resolve/main/"

from dataclasses import dataclass
@dataclass
class APK:
    major: int
    minor: int
    patch: int
    arch: str
    lang: str
    src: str # piper, coqui
    def __init__(self, s):
        s = str(s)
        split = s.split('-')
        self.major, self.minor, self.patch = list(map(int, split[2].split('.')))
        self.arch = split[3]
        self.lang = split[4]
        self.src = split[7]
        if 'arm' in s:
            self.arch += '-' + split[4]
            self.lang = split[5]
            self.src = split[8]

        if 'armeabi' in self.arch:
            self.arch = 'y' + self.arch

        if 'arm64' in self.arch:
            self.arch = 'z' + self.arch

def sort_by_apk(x):
    x = APK(x)
    return (x.major, x.minor, x.patch, x.arch, -ord(x.src[0]),-ord(x.lang[0]))

def generate_url(files: List[str]) -> List[str]:
    ans = []
    base = BASE_URL
    for f in files:
        ans.append(base + str(f))
    return ans


def get_all_files(d: str, suffix: str) -> List[str]:
    ans = sorted(Path(d).glob(suffix), key=sort_by_apk, reverse=True)
    return list(map(lambda x: BASE_URL + str(x), ans))


def to_file(filename: str, files: List[str]):
    content = r"""
<h1> APKs for text-to-speech </h1>
This page lists the <strong>text-to-speech</strong> APKs for <a href="http://github.com/k2-fsa/sherpa-onnx">sherpa-onnx</a>,
one of the deployment frameworks of <a href="https://github.com/k2-fsa">the Next-gen Kaldi project</a>.
<br/>
The name of an APK has the following rule:
<ul>
 <li> sherpa-onnx-{version}-{arch}-{lang}-tts-{model}.apk
</ul>
where
<ul>
 <li> version: It specifies the current version, e.g., 1.8.7
 <li> arch: The architecture targeted by this APK, e.g., arm64-v8a, armeabi-v7a, x86_64, x86
 <li> lang: The language supported by this APK, e.g., en for English, zh for Chinese, fr for French, de for German, es for Spanish
 <li> model: The name of the model used in the APK, e.g., vits-ljs, vits-piper-de_DE-thorsten-low, vits-piper-de_DE-thorsten-medium
</ul>

<span style="color:red;">Note:</span> Models from
<a href="https://github.com/rhasspy/piper">piper</a> have their names prefixed
with <strong>vits-piper-</strong>. For instance, for the model
<strong>vits-piper-en_US-lessac-medium.apk</strong>, its original name
in <a href="https://github.com/rhasspy/piper">piper</a> is
<strong>en_US-lessac-medium.apk</strong>, which is available at
<a href="https://huggingface.co/rhasspy/piper-voices/blob/main/en/en_US/lessac/medium/en_US-lessac-medium.onnx">
https://huggingface.co/rhasspy/piper-voices/blob/main/en/en_US/lessac/medium/en_US-lessac-medium.onnx
</a><br/><br/>

<span style="color:red;">Note:</span> Models from
<a href="https://github.com/coqui-ai/TTS">coqui-ai/TTS</a> have their names prefixed
with <strong>coqui-</strong>.
<br/><br/>

You can find many more models that have not been converted to <strong>sherpa-onnx</strong>
at
<a href="https://huggingface.co/rhasspy/piper-voices">https://huggingface.co/rhasspy/piper-voices</a>


<br/>
<br/>
<div/>
    """
    with open(filename, "w") as f:
        print(content, file=f)
        for x in files:
            name = x.rsplit("/", maxsplit=1)[-1]
            print(f'<a href="{x}" />{name}<br/>', file=f)


def main():
    apk = get_all_files("tts", suffix="*.apk")
    to_file("./apk.html", apk)


if __name__ == "__main__":
    main()