"""
HPCI 4軸版 — 本人 alias 定義

S軸（Self-involvement）算出で使用される本人 alias の事前登録集合。
未登録 alias の自動マージは行わない（本体プロンプト §4）。

新規アーティストを追加する場合は、確認済みの表記揺れのみを追加すること。
"""

ARTIST_ALIASES: dict[str, list[str]] = {
    "Drake": [
        "Drake",
        "Aubrey Graham",
        "Aubrey Drake Graham",
    ],
    "Kendrick Lamar": [
        "Kendrick Lamar",
        "K. Duckworth",
        "Kendrick Duckworth",
    ],
    "Snoop Dogg": [
        "Snoop Dogg",
        "Calvin Broadus Jr.",
        "Calvin Cordozar Broadus Jr.",
        "Broadus",
    ],
}


def get_aliases(artist: str) -> list[str]:
    if artist not in ARTIST_ALIASES:
        raise KeyError(
            f"Artist '{artist}' not in ARTIST_ALIASES. "
            f"未登録 alias の自動マージは禁止されています。"
            f"確認済みの表記揺れのみ src/aliases.py に追加してください。"
        )
    return ARTIST_ALIASES[artist]