artemis/titles/chuni/luminous.py

from datetime import timedelta
from typing import Dict

from core.config import CoreConfig
from titles.chuni.sunplus import ChuniSunPlus
from titles.chuni.const import ChuniConstants, MapAreaConditionLogicalOperator, MapAreaConditionType
from titles.chuni.config import ChuniConfig


class ChuniLuminous(ChuniSunPlus):
    def __init__(self, core_cfg: CoreConfig, game_cfg: ChuniConfig) -> None:
        super().__init__(core_cfg, game_cfg)
        self.version = ChuniConstants.VER_CHUNITHM_LUMINOUS

    async def handle_cm_get_user_preview_api_request(self, data: Dict) -> Dict:
        user_data = await super().handle_cm_get_user_preview_api_request(data)

        # Does CARD MAKER 1.35 work this far up?
        user_data["lastDataVersion"] = "2.20.00"
        return user_data
    
    async def handle_get_user_c_mission_api_request(self, data: Dict) -> Dict:
        user_id = data["userId"]
        mission_id = data["missionId"]

        progress_list = []
        point = 0

        mission_data = await self.data.item.get_cmission(user_id, mission_id)
        progress_data = await self.data.item.get_cmission_progress(user_id, mission_id)
        
        if mission_data and progress_data:
            point = mission_data["point"]

            for progress in progress_data:
                progress_list.append(
                    {
                        "order": progress["order"],
                        "stage": progress["stage"],
                        "progress": progress["progress"],
                    }
                )

        return {
            "userId": user_id,
            "missionId": mission_id,
            "point": point,
            "userCMissionProgressList": progress_list,
        }

    async def handle_get_user_net_battle_ranking_info_api_request(self, data: Dict) -> Dict:
        user_id = data["userId"]

        net_battle = {}
        net_battle_data = await self.data.profile.get_net_battle(user_id)
        
        if net_battle_data:
            net_battle = {
                "isRankUpChallengeFailed": net_battle_data["isRankUpChallengeFailed"],
                "highestBattleRankId": net_battle_data["highestBattleRankId"],
                "battleIconId": net_battle_data["battleIconId"],
                "battleIconNum": net_battle_data["battleIconNum"],
                "avatarEffectPoint": net_battle_data["avatarEffectPoint"],
            }

        return {
            "userId": user_id,
            "userNetBattleData": net_battle,
        }

    async def handle_get_game_map_area_condition_api_request(self, data: Dict) -> Dict:
        # There is no game data for this, everything is server side.
        # However, we can selectively show/hide events as data is imported into the server.
        events = await self.data.static.get_enabled_events(self.version)
        event_by_id = {evt["eventId"]: evt for evt in events}
        conditions = []

        # The Mystic Rainbow of LUMINOUS map unlocks when any mainline LUMINOUS area
        # (ep. I, ep. II, ep. III) are completed.
        mystic_area_1_conditions = {
            "mapAreaId": 3229301,  # Mystic Rainbow of LUMINOUS Area 1
            "length": 0,
            "mapAreaConditionList": [],
        }
        mystic_area_1_added = False

        # Secret AREA: MUSIC GAME
        if 14029 in event_by_id:
            start_date = event_by_id[14029]["startDate"].strftime(self.date_time_format)
            mission_in_progress_end_date = "2099-12-31 00:00:00.0"

            # The "MISSION in progress" trophy required to trigger the secret area
            # is only available in the first CHUNITHM mission. If the second mission
            # (event ID 14214) was imported into ARTEMiS, we disable the requirement
            # for this trophy.
            if 14214 in event_by_id:
                mission_in_progress_end_date = (event_by_id[14214]["startDate"] - timedelta(hours=2)).strftime(self.date_time_format)
            
            conditions.extend([
                {
                    "mapAreaId": 2206201,  # BlythE ULTIMA
                    "length": 1,
                    # Obtain the trophy "MISSION in progress".
                    "mapAreaConditionList": [
                        {
                            "type": MapAreaConditionType.TROPHY_OBTAINED.value,
                            "conditionId": 6832,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": mission_in_progress_end_date,
                        }
                    ],
                },
                {
                    "mapAreaId": 2206202,  # PRIVATE SERVICE ULTIMA
                    "length": 1,
                    # Obtain the trophy "MISSION in progress".
                    "mapAreaConditionList": [
                        {
                            "type": MapAreaConditionType.TROPHY_OBTAINED.value,
                            "conditionId": 6832,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": mission_in_progress_end_date,
                        }
                    ],
                },
                {
                    "mapAreaId": 2206203,  # New York Back Raise
                    "length": 1,
                    # SS NightTheater's EXPERT chart and get the title
                    # "今宵、劇場に映し出される景色とは――――。"
                    "mapAreaConditionList": [
                        {
                            "type": MapAreaConditionType.TROPHY_OBTAINED.value,
                            "conditionId": 6833,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": "2099-12-31 00:00:00.0",
                        },
                    ],
                },
                {
                    "mapAreaId": 2206204,  # Spasmodic
                    "length": 2,
                    # - Get 1 miss on Random (any difficulty) and get the title "当たり待ち"
                    # - Get 1 miss on 花たちに希望を (any difficulty) and get the title "花たちに希望を"
                    "mapAreaConditionList": [
                        {
                            "type": MapAreaConditionType.TROPHY_OBTAINED.value,
                            "conditionId": 6834,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": "2099-12-31 00:00:00.0",
                        },
                        {
                            "type": MapAreaConditionType.TROPHY_OBTAINED.value,
                            "conditionId": 6835,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": "2099-12-31 00:00:00.0",
                        },
                    ],
                },
                {
                    "mapAreaId": 2206205,  # ΩΩPARTS
                    "length": 2,
                    # - S Sage EXPERT to get the title "ﾏﾀｰﾘ進行ｷﾎﾞﾝﾇ"
                    # - Equip this title and play cab-to-cab with another person with this title
                    # to get "ﾏﾀｰﾘしようよ". Disabled because it is difficult to play cab2cab
                    # on data setups. A network operator may consider re-enabling it by uncommenting
                    # the second condition.
                    "mapAreaConditionList": [
                        {
                            "type": MapAreaConditionType.TROPHY_OBTAINED.value,
                            "conditionId": 6836,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": "2099-12-31 00:00:00.0",
                        },
                        # {
                        #     "type": MapAreaConditionType.TROPHY_OBTAINED.value,
                        #     "conditionId": 6837,
                        #     "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                        #     "startDate": start_date,
                        #     "endDate": "2099-12-31 00:00:00.0",
                        # },
                    ],
                },
                {
                    "mapAreaId": 2206206,  # Blow My Mind
                    "length": 1,
                    # SS on CHAOS EXPERT, Hydra EXPERT, Surive EXPERT and Jakarta PROGRESSION EXPERT
                    # to get the title "Can you hear me?"
                    "mapAreaConditionList": [
                        {
                            "type": MapAreaConditionType.TROPHY_OBTAINED.value,
                            "conditionId": 6838,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": "2099-12-31 00:00:00.0",
                        },
                    ],
                },
                {
                    "mapAreaId": 2206207,  # VALLIS-NERIA
                    "length": 6,
                    # Finish the 6 other areas
                    "mapAreaConditionList": [
                        {
                            "type": MapAreaConditionType.MAP_AREA_CLEARED.value,
                            "conditionId": x,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": "2099-12-31 00:00:00.0",
                        }
                        for x in range(2206201, 2206207)
                    ],
                },
            ])
        
        # LUMINOUS ep. I
        if 14005 in event_by_id:
            start_date = event_by_id[14005]["startDate"].strftime(self.date_time_format)

            if not mystic_area_1_added:
                conditions.append(mystic_area_1_conditions)
                mystic_area_1_added = True
            
            mystic_area_1_conditions["length"] += 1
            mystic_area_1_conditions["mapAreaConditionList"].append(
                {
                    "type": MapAreaConditionType.MAP_CLEARED.value,
                    "conditionId": 3020701,
                    "logicalOpe": MapAreaConditionLogicalOperator.OR.value,
                    "startDate": start_date,
                    "endDate": "2099-12-31 00:00:00.0",
                }
            )

            conditions.append(
                {
                    "mapAreaId": 3229302,  # Mystic Rainbow of LUMINOUS Area 2,
                    "length": 1,
                    # Unlocks when LUMINOUS ep. I is completed.
                    "mapAreaConditionList": [
                        {
                            "type": MapAreaConditionType.MAP_CLEARED.value,
                            "conditionId": 3020701,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": "2099-12-31 00:00:00.0",
                        },
                    ],
                }
            )
        
        # LUMINOUS ep. II
        if 14251 in event_by_id:
            start_date = event_by_id[14251]["startDate"].strftime(self.date_time_format)
            
            if not mystic_area_1_added:
                conditions.append(mystic_area_1_conditions)
                mystic_area_1_added = True
            
            mystic_area_1_conditions["length"] += 1
            mystic_area_1_conditions["mapAreaConditionList"].append(
                {
                    "type": MapAreaConditionType.MAP_CLEARED.value,
                    "conditionId": 3020702,
                    "logicalOpe": MapAreaConditionLogicalOperator.OR.value,
                    "startDate": start_date,
                    "endDate": "2099-12-31 00:00:00.0",
                }
            )

            conditions.append(
                {
                    "mapAreaId": 3229303,  # Mystic Rainbow of LUMINOUS Area 3,
                    "length": 1,
                    # Unlocks when LUMINOUS ep. II is completed.
                    "mapAreaConditionList": [
                        {
                            "type": MapAreaConditionType.MAP_CLEARED.value,
                            "conditionId": 3020702,
                            "logicalOpe": MapAreaConditionLogicalOperator.AND.value,
                            "startDate": start_date,
                            "endDate": "2099-12-31 00:00:00.0",
                        },
                    ],
                }
            )


        return {
            "length": len(conditions),
            "gameMapAreaConditionList": conditions,
        }