# Copyright: Ankitects Pty Ltd and contributors
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
# Based off Kieran Clancy's initial implementation.

import re
from typing import Any, Callable

from anki.hooks import addHook

r = r" ?([^ >]+?)\[(.+?)\]"
ruby = r"<ruby><rb>\1</rb><rt>\2</rt></ruby>"


def noSound(repl) -> Callable[[Any], Any]:
    def func(match):
        if match.group(2).startswith("sound:"):
            # return without modification
            return match.group(0)
        else:
            return re.sub(r, repl, match.group(0))

    return func


def _munge(s) -> Any:
    return s.replace("&nbsp;", " ")


def kanji(txt, *args) -> str:
    return re.sub(r, noSound(r"\1"), _munge(txt))


def kana(txt, *args) -> str:
    return re.sub(r, noSound(r"\2"), _munge(txt))


def furigana(txt, *args) -> str:
    return re.sub(r, noSound(ruby), _munge(txt))


def install() -> None:
    addHook("fmod_kanji", kanji)
    addHook("fmod_kana", kana)
    addHook("fmod_furigana", furigana)