artemis/titles/cm/base.py

from datetime import date, datetime, timedelta
from typing import Any, Dict, List
import json
import logging
from enum import Enum

from core.config import CoreConfig
from core.data.cache import cached
from titles.cm.const import CardMakerConstants
from titles.cm.config import CardMakerConfig


class CardMakerBase:
    def __init__(self, core_cfg: CoreConfig, game_cfg: CardMakerConfig) -> None:
        self.core_cfg = core_cfg
        self.game_cfg = game_cfg
        self.date_time_format = "%Y-%m-%d %H:%M:%S"
        self.date_time_format_ext = (
            "%Y-%m-%d %H:%M:%S.%f"  # needs to be lopped off at [:-5]
        )
        self.date_time_format_short = "%Y-%m-%d"
        self.logger = logging.getLogger("cardmaker")
        self.game = CardMakerConstants.GAME_CODE
        self.version = CardMakerConstants.VER_CARD_MAKER

    def handle_get_game_connect_api_request(self, data: Dict) -> Dict:
        if self.core_cfg.server.is_develop:
            uri = f"http://{self.core_cfg.title.hostname}:{self.core_cfg.title.port}"
        else:
            uri = f"http://{self.core_cfg.title.hostname}"

        # CHUNITHM = 0, maimai = 1, ONGEKI = 2
        return {
            "length": 3,
            "gameConnectList": [
                {"modelKind": 0, "type": 1, "titleUri": f"{uri}/SDHD/200/"},
                {"modelKind": 1, "type": 1, "titleUri": f"{uri}/SDEZ/120/"},
                {"modelKind": 2, "type": 1, "titleUri": f"{uri}/SDDT/130/"},
            ],
        }

    def handle_get_game_setting_api_request(self, data: Dict) -> Dict:
        reboot_start = date.strftime(
            datetime.now() + timedelta(hours=3), self.date_time_format
        )
        reboot_end = date.strftime(
            datetime.now() + timedelta(hours=4), self.date_time_format
        )

        return {
            "gameSetting": {
                "dataVersion": "1.30.00",
                "ongekiCmVersion": "1.30.01",
                "chuniCmVersion": "2.00.00",
                "maimaiCmVersion": "1.20.00",
                "requestInterval": 10,
                "rebootStartTime": reboot_start,
                "rebootEndTime": reboot_end,
                "maxCountCharacter": 100,
                "maxCountItem": 100,
                "maxCountCard": 100,
                "watermark": False,
                "isMaintenance": False,
                "isBackgroundDistribute": False,
            },
            "isDumpUpload": False,
            "isAou": False,
        }

    def handle_get_client_bookkeeping_api_request(self, data: Dict) -> Dict:
        return {"placeId": data["placeId"], "length": 0, "clientBookkeepingList": []}

    def handle_upsert_client_setting_api_request(self, data: Dict) -> Dict:
        return {"returnCode": 1, "apiName": "UpsertClientSettingApi"}

    def handle_upsert_client_bookkeeping_api_request(self, data: Dict) -> Dict:
        return {"returnCode": 1, "apiName": "UpsertClientBookkeepingApi"}