from typing import Optional
from core.utils import floor_to_nearest_005

class Mai2Constants:
    GRADE = {
        "D": 0,
        "C": 1,
        "B": 2,
        "BB": 3,
        "BBB": 4,
        "A": 5,
        "AA": 6,
        "AAA": 7,
        "S": 8,
        "S+": 9,
        "SS": 10,
        "SS+": 11,
        "SSS": 12,
        "SSS+": 13,
    }
    FC = {"None": 0, "FC": 1, "FC+": 2, "AP": 3, "AP+": 4}
    SYNC = {"None": 0, "FS": 1, "FS+": 2, "FDX": 3, "FDX+": 4}

    DATE_TIME_FORMAT = "%Y-%m-%d %H:%M:%S"

    GAME_CODE = "SBXL"
    GAME_CODE_GREEN = "SBZF"
    GAME_CODE_ORANGE = "SDBM"
    GAME_CODE_PINK = "SDCQ"
    GAME_CODE_MURASAKI = "SDDK"
    GAME_CODE_MILK = "SDDZ"
    GAME_CODE_FINALE = "SDEY"
    GAME_CODE_DX = "SDEZ"
    GAME_CODE_DX_INT = "SDGA"

    CONFIG_NAME = "mai2.yaml"

    VER_MAIMAI = 0
    VER_MAIMAI_PLUS = 1
    VER_MAIMAI_GREEN = 2
    VER_MAIMAI_GREEN_PLUS = 3
    VER_MAIMAI_ORANGE = 4
    VER_MAIMAI_ORANGE_PLUS = 5
    VER_MAIMAI_PINK = 6
    VER_MAIMAI_PINK_PLUS = 7
    VER_MAIMAI_MURASAKI = 8
    VER_MAIMAI_MURASAKI_PLUS = 9
    VER_MAIMAI_MILK = 10
    VER_MAIMAI_MILK_PLUS = 11
    VER_MAIMAI_FINALE = 12

    VER_MAIMAI_DX = 13
    VER_MAIMAI_DX_PLUS = 14
    VER_MAIMAI_DX_SPLASH = 15
    VER_MAIMAI_DX_SPLASH_PLUS = 16
    VER_MAIMAI_DX_UNIVERSE = 17
    VER_MAIMAI_DX_UNIVERSE_PLUS = 18
    VER_MAIMAI_DX_FESTIVAL = 19
    VER_MAIMAI_DX_FESTIVAL_PLUS = 20
    VER_MAIMAI_DX_BUDDIES = 21
    VER_MAIMAI_DX_BUDDIES_PLUS = 22
    VER_MAIMAI_DX_PRISM = 23

    VERSION_STRING = (
        "maimai",
        "maimai PLUS",
        "maimai GreeN",
        "maimai GreeN PLUS",
        "maimai ORANGE",
        "maimai ORANGE PLUS",
        "maimai PiNK",
        "maimai PiNK PLUS",
        "maimai MURASAKi",
        "maimai MURASAKi PLUS",
        "maimai MiLK",
        "maimai MiLK PLUS",
        "maimai FiNALE",
        "maimai DX",
        "maimai DX PLUS",
        "maimai DX Splash",
        "maimai DX Splash PLUS",
        "maimai DX UNiVERSE",
        "maimai DX UNiVERSE PLUS",
        "maimai DX FESTiVAL",
        "maimai DX FESTiVAL PLUS",
        "maimai DX BUDDiES",
        "maimai DX BUDDiES PLUS",
        "maimai DX PRiSM"
    )

    MAI_VERSION_LUT = {
        "100": VER_MAIMAI,
        "110": VER_MAIMAI_PLUS,
        "120": VER_MAIMAI_GREEN,
        "130": VER_MAIMAI_GREEN_PLUS,
        "140": VER_MAIMAI_ORANGE,
        "150": VER_MAIMAI_ORANGE_PLUS,
        "160": VER_MAIMAI_PINK,
        "170": VER_MAIMAI_PINK_PLUS,
        "180": VER_MAIMAI_MURASAKI,
        "185": VER_MAIMAI_MURASAKI_PLUS,
        "190": VER_MAIMAI_MILK,
        "195": VER_MAIMAI_MILK_PLUS,
        "197": VER_MAIMAI_FINALE,
    }

    MAI2_VERSION_LUT = {
        "100": VER_MAIMAI_DX,
        "105": VER_MAIMAI_DX_PLUS,
        "110": VER_MAIMAI_DX_SPLASH,
        "115": VER_MAIMAI_DX_SPLASH_PLUS,
        "120": VER_MAIMAI_DX_UNIVERSE,
        "125": VER_MAIMAI_DX_UNIVERSE_PLUS,
        "130": VER_MAIMAI_DX_FESTIVAL,
        "135": VER_MAIMAI_DX_FESTIVAL_PLUS,
        "140": VER_MAIMAI_DX_BUDDIES,
        "145": VER_MAIMAI_DX_BUDDIES_PLUS,
    }

    @classmethod
    def game_ver_to_string(cls, ver: int):
        """ Takes an internal game version (ex 13 for maimai DX) and returns a the full name of the version """
        return cls.VERSION_STRING[ver]

    @classmethod
    def int_ver_to_game_ver(cls, ver: int, is_dx = True) -> Optional[int]:
        """ Takes an int ver (ex 100 for 1.00) and returns an internal game version """
        if is_dx:
            return cls.MAI2_VERSION_LUT.get(str(floor_to_nearest_005(ver)), None)
        else:
            if ver >= 197:
                return cls.VER_MAIMAI_FINALE
            return cls.MAI_VERSION_LUT.get(str(floor_to_nearest_005(ver)), None)