artemis/titles/ongeki/const.py

from typing import Final, Dict
from enum import Enum


class OngekiConstants:
    GAME_CODE = "SDDT"

    CONFIG_NAME = "ongeki.yaml"

    VER_ONGEKI = 0
    VER_ONGEKI_PLUS = 1
    VER_ONGEKI_SUMMER = 2
    VER_ONGEKI_SUMMER_PLUS = 3
    VER_ONGEKI_RED = 4
    VER_ONGEKI_RED_PLUS = 5
    VER_ONGEKI_BRIGHT = 6
    VER_ONGEKI_BRIGHT_MEMORY = 7

    EVT_TYPES: Enum = Enum(
        "EVT_TYPES",
        [
            "None",
            "Announcement",
            "Movie",
            "AddMyList",
            "UnlockChapter",
            "JewelEvent",
            "RankingEvent",
            "AcceptRankingEvent",
            "UnlockMusic",
            "UnlockCard",
            "UnlockTrophy",
            "UnlockNamePlate",
            "UnlockLimitBreakItem",
            "MissionEvent",
            "DailyBonus",
            "UnlockBossLockEarly",
            "UnlockPurchaseItem",
            "TechChallengeEvent",
            "AcceptTechChallengeEvent",
            "SilverJewelEvent",
        ],
    )

    class CM_GACHA_KINDS(Enum):
        Normal = 0
        Pickup = 1
        BonusRestored = 2
        Free = 3
        PickupBonusRestored = 4

    class RARITY_TYPES(Enum):
        N = 0
        R = 1
        SR = 2
        SSR = 3
        SRPlus = 12

    # The game expects the server to give Lunatic an ID of 10, while the game uses 4 internally... except in Music.xml
    class DIFF_NAME(Enum):
        Basic = 0
        Advanced = 1
        Expert = 2
        Master = 3
        Lunatic = 10

    VERSION_NAMES = (
        "ONGEKI",
        "ONGEKI+",
        "ONGEKI Summer",
        "ONGEKI Summer+",
        "ONGEKI Red",
        "ONGEKI Red+",
        "ONGEKI Bright",
        "ONGEKI Bright Memory",
    )

    @classmethod
    def game_ver_to_string(cls, ver: int):
        return cls.VERSION_NAMES[ver]