artemis/titles/chuni/config.py

from core.config import CoreConfig
from typing import Dict


class ChuniServerConfig:
    def __init__(self, parent_config: "ChuniConfig") -> None:
        self.__config = parent_config

    @property
    def enable(self) -> bool:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "server", "enable", default=True
        )

    @property
    def loglevel(self) -> int:
        return CoreConfig.str_to_loglevel(
            CoreConfig.get_config_field(
                self.__config, "chuni", "server", "loglevel", default="info"
            )
        )
        
    @property
    def news_msg(self) -> str:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "server", "news_msg", default=""
        )


class ChuniTeamConfig:
    def __init__(self, parent_config: "ChuniConfig") -> None:
        self.__config = parent_config

    @property
    def team_name(self) -> str:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "team", "name", default=""
        )
    @property
    def rank_scale(self) -> str:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "team", "rank_scale", default="False"
        )


class ChuniModsConfig:
    def __init__(self, parent_config: "ChuniConfig") -> None:
        self.__config = parent_config

    @property
    def use_login_bonus(self) -> bool:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "mods", "use_login_bonus", default=True
        )

    @property
    def stock_tickets(self) -> str:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "mods", "stock_tickets", default=None
        )

    @property
    def stock_count(self) -> int:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "mods", "stock_count", default=99
        )

    def forced_item_unlocks(self, item: str) -> bool:
        forced_item_unlocks = CoreConfig.get_config_field(
            self.__config, "chuni", "mods", "forced_item_unlocks", default={}
        )

        if item not in forced_item_unlocks.keys():
            # default to no forced unlocks
            return False
 
        return forced_item_unlocks[item]


class ChuniVersionConfig:
    def __init__(self, parent_config: "ChuniConfig") -> None:
        self.__config = parent_config

    def version(self, version: int) -> Dict:
        """
        in the form of:
        11: {"rom": 2.00.00, "data": 2.00.00}
        """
        versions = CoreConfig.get_config_field(
            self.__config, "chuni", "version", default={}
        )

        if version not in versions.keys():
            return None

        return versions[version]


class ChuniCryptoConfig:
    def __init__(self, parent_config: "ChuniConfig") -> None:
        self.__config = parent_config

    @property
    def keys(self) -> Dict:
        """
        in the form of:
        internal_version: [key, iv]
        all values are hex strings
        """
        return CoreConfig.get_config_field(
            self.__config, "chuni", "crypto", "keys", default={}
        )

    @property
    def encrypted_only(self) -> bool:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "crypto", "encrypted_only", default=False
        )

class ChuniMatchingConfig:
    def __init__(self, parent_config: "ChuniConfig") -> None:
        self.__config = parent_config
    
    @property
    def enable(self) -> bool:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "matching", "enable", default=False
        )
    
    @property
    def match_time_limit(self) -> int:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "matching", "match_time_limit", default=60
        )
    
    @property
    def match_error_limit(self) -> int:
        return CoreConfig.get_config_field(
            self.__config, "chuni", "matching", "match_error_limit", default=9999
        )

class ChuniConfig(dict):
    def __init__(self) -> None:
        self.server = ChuniServerConfig(self)
        self.team = ChuniTeamConfig(self)
        self.mods = ChuniModsConfig(self)
        self.version = ChuniVersionConfig(self)
        self.crypto = ChuniCryptoConfig(self)
        self.matching = ChuniMatchingConfig(self)