#!/usr/bin/env python

import glob, re, json, stringcase

files = (
    # glob.glob("../../pylib/**/*.py", recursive=True)
    glob.glob("../../qt/**/*.py", recursive=True)
    # glob.glob("../../qt/**/forms/*.ui", recursive=True)
)
string_re = re.compile(r'ngettext\(\s*"(.+?)",\s+".+?",\s+(.+?)\) % \2', re.DOTALL)

map = json.load(open("keys_by_text.json"))

# unused or missing strings
blacklist = {
    "Label1",
    "After pressing OK, you can choose which tags to include.",
    "Filter/Cram",
    "Show %s",
    "~",
    "about:blank",
    "%d card imported.",
    # need to update manually
    "Browse (%(cur)d card shown; %(sel)s)",
    # previewer.py needs updating to fix these
    "Shortcut key: R",
    "Shortcut key: B",
}

from html.entities import name2codepoint

reEnts = re.compile(r"&#?\w+;")


def decode_ents(html):
    def fixup(m):
        text = m.group(0)
        if text[:2] == "&#":
            # character reference
            try:
                if text[:3] == "&#x":
                    return chr(int(text[3:-1], 16))
                else:
                    return chr(int(text[2:-1]))
            except ValueError:
                pass
        else:
            # named entity
            try:
                text = chr(name2codepoint[text[1:-1]])
            except KeyError:
                pass
        return text  # leave as is

    return reEnts.sub(fixup, html)


def munge_key(key):
    if key == "browsing-note":
        return "browsing-note-count"
    if key == "card-templates-card":
        return "card-templates-card-count"
    return key


def repl(m):
    print(m.group(0))
    text = decode_ents(m.group(1))
    if text in blacklist:
        return m.group(0)

    (module, key) = map[text]
    key = munge_key(key)

    screaming = stringcase.constcase(key)

    ret = f"tr(TR.{screaming}, count={m.group(2)})"
    print(ret)
    return ret


for file in files:
    if file.endswith("stats.py"):
        continue
    buf = open(file).read()
    buf2 = string_re.sub(repl, buf)
    if buf != buf2:
        lines = buf2.split("\n")
        lines.insert(3, "from aqt.utils import tr, TR")
        buf2 = "\n".join(lines)
        open(file, "w").write(buf2)