"""Convert Chinese text to pinyin character sequence."""

import re
from pypinyin import lazy_pinyin, Style
from .symbols import get_phoneme_ids

_whitespace_re = re.compile(r"\s+")


def text_to_sequence(text: str) -> list[int]:
    """Chinese text -> pinyin character IDs.

    Example: "你好" -> "ni3 hao3" -> [n, i, 3, <space>, h, a, o, 3]
    """
    text = re.sub(_whitespace_re, "", text)

    # Convert to pinyin with tone digits
    py_list = lazy_pinyin(text, style=Style.TONE3, neutral_tone_with_five=True)

    # Join with spaces between words
    pinyin_str = " ".join(py_list)

    return get_phoneme_ids(pinyin_str)


def pinyin_to_ids(pinyin_str: str) -> list[int]:
    return get_phoneme_ids(pinyin_str)