artemis-bugfix/titles/wacca/handlers/user_status.py

from typing import List, Dict, Optional

from titles.wacca.handlers.base import BaseRequest, BaseResponse
from titles.wacca.handlers.helpers import *


# ---user/status/get----
class UserStatusGetRequest(BaseRequest):
    aimeId: int = 0

    def __init__(self, data: Dict) -> None:
        super().__init__(data)
        self.aimeId = int(data["params"][0])


class UserStatusGetV1Response(BaseResponse):
    def __init__(self) -> None:
        super().__init__()
        self.userStatus: UserStatusV1 = UserStatusV1()
        self.setTitleId: int = 0
        self.setIconId: int = 0
        self.profileStatus: ProfileStatus = ProfileStatus.ProfileGood
        self.versionStatus: PlayVersionStatus = PlayVersionStatus.VersionGood
        self.lastGameVersion: ShortVersion = ShortVersion()

    def make(self) -> Dict:
        self.params = [
            self.userStatus.make(),
            self.setTitleId,
            self.setIconId,
            self.profileStatus.value,
            [self.versionStatus.value, str(self.lastGameVersion)],
        ]

        return super().make()


class UserStatusGetV2Response(UserStatusGetV1Response):
    def __init__(self) -> None:
        super().__init__()
        self.userStatus: UserStatusV2 = UserStatusV2()
        self.options: List[UserOption] = []

    def make(self) -> Dict:
        super().make()
        opts = []

        for x in self.options:
            opts.append(x.make())

        self.params.append(opts)

        return super(UserStatusGetV1Response, self).make()


# ---user/status/getDetail----
class UserStatusGetDetailRequest(BaseRequest):
    userId: int = 0

    def __init__(self, data: Dict) -> None:
        super().__init__(data)
        self.userId = data["params"][0]


class UserStatusGetDetailResponseV1(BaseResponse):
    def __init__(self) -> None:
        super().__init__()
        self.userStatus: UserStatusV1 = UserStatusV1()
        self.options: List[UserOption] = []
        self.seasonalPlayModeCounts: List[PlayModeCounts] = []
        self.userItems: UserItemInfoV1 = UserItemInfoV1()
        self.scores: List[BestScoreDetailV1] = []
        self.songPlayStatus: List[int] = [0, 0]
        self.seasonInfo: SeasonalInfoV1 = SeasonalInfoV1()
        self.playAreaList: List = [
            [0],
            [0, 0, 0, 0, 0, 0],
            [0, 0, 0, 0, 0, 0, 0],
            [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
            [0, 0, 0, 0, 0, 0],
            [0, 0, 0, 0, 0],
            [0, 0, 0, 0],
            [0, 0, 0, 0, 0, 0, 0],
            [0],
        ]
        self.songUpdateTime: int = 0

    def make(self) -> Dict:
        opts = []
        play_modes = []
        scores = []

        for x in self.seasonalPlayModeCounts:
            play_modes.append(x.make())

        for x in self.scores:
            scores.append(x.make())

        for x in self.options:
            opts.append(x.make())

        self.params = [
            self.userStatus.make(),
            opts,
            play_modes,
            self.userItems.make(),
            scores,
            self.songPlayStatus,
            self.seasonInfo.make(),
            self.playAreaList,
            self.songUpdateTime,
        ]

        return super().make()

    def find_score_idx(
        self,
        song_id: int,
        difficulty: int = 1,
        start_idx: int = 0,
        stop_idx: Optional[int] = None,
    ) -> Optional[int]:
        if stop_idx is None or stop_idx > len(self.scores):
            stop_idx = len(self.scores)

        for x in range(start_idx, stop_idx):
            if (
                self.scores[x].songId == song_id
                and self.scores[x].difficulty == difficulty
            ):
                return x

        return None


class UserStatusGetDetailResponseV2(UserStatusGetDetailResponseV1):
    def __init__(self) -> None:
        super().__init__()
        self.userStatus: UserStatusV2 = UserStatusV2()
        self.seasonInfo: SeasonalInfoV2 = SeasonalInfoV2()
        self.userItems: UserItemInfoV2 = UserItemInfoV2()
        self.favorites: List[int] = []
        self.stoppedSongIds: List[int] = []
        self.eventInfo: List[UserEventInfo] = []
        self.gateInfo: List[GateDetailV1] = []
        self.lastSongInfo: LastSongDetail = LastSongDetail()
        self.gateTutorialFlags: List[GateTutorialFlag] = []
        self.gatchaInfo: List[GachaInfo] = []
        self.friendList: List[FriendDetail] = []

    def make(self) -> Dict:
        super().make()
        gates = []
        friends = []
        tut_flg = []
        evts = []
        gacha = []

        for x in self.gateInfo:
            gates.append(x.make())

        for x in self.friendList:
            friends.append(x.make())

        for x in self.gateTutorialFlags:
            tut_flg.append(x.make())

        while len(tut_flg) < 5:
            flag_id = len(tut_flg) + 1
            tut_flg.append([flag_id, 0])
        
        for x in self.eventInfo:
            evts.append(x.make())

        for x in self.gatchaInfo:
            gacha.append(x.make())

        self.params.append(self.favorites)
        self.params.append(self.stoppedSongIds)
        self.params.append(evts)
        self.params.append(gates)
        self.params.append(self.lastSongInfo.make())
        self.params.append(tut_flg)
        self.params.append(gacha)
        self.params.append(friends)

        return super(UserStatusGetDetailResponseV1, self).make()


class UserStatusGetDetailResponseV3(UserStatusGetDetailResponseV2):
    def __init__(self) -> None:
        super().__init__()
        self.gateInfo: List[GateDetailV2] = []


class UserStatusGetDetailResponseV4(UserStatusGetDetailResponseV3):
    def __init__(self) -> None:
        super().__init__()
        self.userItems: UserItemInfoV3 = UserItemInfoV3()
        self.bingoStatus: BingoDetail = BingoDetail(0)
        self.scores: List[BestScoreDetailV2] = []

    def make(self) -> Dict:
        super().make()
        self.params.append(self.bingoStatus.make())

        return super(UserStatusGetDetailResponseV1, self).make()


# ---user/status/login----
class UserStatusLoginRequest(BaseRequest):
    userId: int = 0

    def __init__(self, data: Dict) -> None:
        super().__init__(data)
        self.userId = data["params"][0]


class UserStatusLoginResponseV1(BaseResponse):
    def __init__(
        self, is_first_login_daily: bool = False, last_login_date: int = 0
    ) -> None:
        super().__init__()
        self.dailyBonus: List[LoginBonusInfo] = []
        self.consecBonus: List[LoginBonusInfo] = []
        self.otherBonus: List[LoginBonusInfo] = []
        self.firstLoginDaily = is_first_login_daily
        self.lastLoginDate = last_login_date

    def make(self) -> Dict:
        super().make()
        daily = []
        consec = []
        other = []

        for bonus in self.dailyBonus:
            daily.append(bonus.make())

        for bonus in self.consecBonus:
            consec.append(bonus.make())

        for bonus in self.otherBonus:
            other.append(bonus.make())

        self.params = [daily, consec, other, int(self.firstLoginDaily)]
        return super().make()


class UserStatusLoginResponseV2(UserStatusLoginResponseV1):
    def __init__(
        self, is_first_login_daily: bool = False, last_login_date: int = 0
    ) -> None:
        super().__init__(is_first_login_daily)
        self.lastLoginDate = last_login_date

        self.vipInfo = VipInfo()

    def make(self) -> Dict:
        super().make()
        self.params.append(self.vipInfo.make())
        self.params.append(self.lastLoginDate)
        return super(UserStatusLoginResponseV1, self).make()


class UserStatusLoginResponseV3(UserStatusLoginResponseV2):
    def __init__(
        self, is_first_login_daily: bool = False, last_login_date: int = 0
    ) -> None:
        super().__init__(is_first_login_daily, last_login_date)
        self.unk: List = [] # Ticket info, item info, message, title, voice name (not sure how they fit...)

    def make(self) -> Dict:
        super().make()
        self.params.append(self.unk)
        return super(UserStatusLoginResponseV1, self).make()


# ---user/status/create---
class UserStatusCreateRequest(BaseRequest):
    def __init__(self, data: Dict) -> None:
        super().__init__(data)
        self.aimeId = data["params"][0]
        self.username = data["params"][1]


class UserStatusCreateResponseV1(BaseResponse):
    def __init__(self, userId: int, username: str) -> None:
        super().__init__()
        self.userStatus = UserStatusV1()
        self.userStatus.userId = userId
        self.userStatus.username = username

    def make(self) -> Dict:
        self.params = [self.userStatus.make()]
        return super().make()


class UserStatusCreateResponseV2(UserStatusCreateResponseV1):
    def __init__(self, userId: int, username: str) -> None:
        super().__init__(userId, username)
        self.userStatus: UserStatusV2 = UserStatusV2()
        self.userStatus.userId = userId
        self.userStatus.username = username


# ---user/status/logout---
class UserStatusLogoutRequest(BaseRequest):
    userId: int

    def __init__(self, data: Dict) -> None:
        super().__init__(data)
        self.userId = data["params"][0]


# ---user/status/update---
class UserStatusUpdateRequestV1(BaseRequest):
    def __init__(self, data: Dict) -> None:
        super().__init__(data)
        self.profileId: int = data["params"][0]
        self.playType: PlayType = PlayType(data["params"][1])
        self.itemsRecieved: List[GenericItemRecv] = []

        for itm in data["params"][2]:
            self.itemsRecieved.append(GenericItemRecv(itm[0], itm[1], itm[2]))


class UserStatusUpdateRequestV2(UserStatusUpdateRequestV1):
    def __init__(self, data: Dict) -> None:
        super().__init__(data)
        self.isContinue = bool(data["params"][3])
        self.isFirstPlayFree = bool(data["params"][4])
        self.itemsUsed = data["params"][5]
        self.lastSongInfo = LastSongDetail(
            data["params"][6][0],
            data["params"][6][1],
            data["params"][6][2],
            data["params"][6][3],
            data["params"][6][4],
        )