Add new KaleidxScope Condition handle method

titles/mai2/const.py

@@ -87,7 +87,16 @@ class Mai2Constants:
         "maimai DX BUDDiES PLUS",
         "maimai DX PRiSM"
     )
-
+    KALEIDXSCOPE_KEY_CONDITION={
+        1: [11009, 11008, 11100, 11097, 11098, 11099, 11163, 11162, 11161, 11228, 11229, 11231, 11463, 11464, 11465, 11538, 11539, 11541, 11620, 11622, 11623, 11737, 11738, 11164, 11230, 11466, 11540, 11621, 11739],
+        #青の扉: Played 29 songs
+        2: [11102, 11234, 11300, 11529, 11542, 11612],
+        #白の扉: set Frame as "Latent Kingdom" (459504), play 3 or 4 songs by the composer 大国奏音 in 1 pc
+        3: [],
+        #紫の扉: need to enter redeem code 51090942171709440000
+        4: [11023, 11106, 11221, 11222, 11300, 11374, 11458, 11523, 11619, 11663, 11746],
+        #青の扉: Played 11 songs
+    }
     MAI_VERSION_LUT = {
         "100": VER_MAIMAI,
         "110": VER_MAIMAI_PLUS,

titles/mai2/prism.py

@@ -43,12 +43,53 @@ class Mai2Prism(Mai2BuddiesPlus):
                 {"gateId": 2, "phaseId": 6},
                 {"gateId": 3, "phaseId": 6},
                 {"gateId": 4, "phaseId": 6},
-                {"gateId": 5, "phaseId": 6},
-                {"gateId": 6, "phaseId": 6}
             ]
         }
 
     async def handle_get_user_kaleidx_scope_api_request(self, data: Dict) -> Dict:
+        # kaleidxscope keyget condition judgement
+        # player may get key before GateFound
+        for gate in range(1,5):
+            if gate == 1 or gate == 4:
+                condition_satisfy = 0
+                for condition in Mai2Constants.KALEIDXSCOPE_KEY_CONDITION[gate]:
+                    score_list = await self.data.score.get_best_scores(user_id=data["userId"], song_id=condition)
+                    if score_list:
+                        condition_satisfy = condition_satisfy + 1
+                if len(Mai2Constants.KALEIDXSCOPE_KEY_CONDITION[gate]) == condition_satisfy:
+                    new_kaleidxscope = {'gateId': gate, "isKeyFound": True}
+                    await self.data.score.put_user_kaleidxscope(data["userId"], new_kaleidxscope)
+
+            elif gate == 2:
+                user_profile = await self.data.profile.get_profile_detail(user_id=data["userId"], version=self.version)
+                user_frame = user_profile["frameId"]
+                if user_frame == 459504:
+                    playlogs = await self.data.score.get_playlogs(user_id=data["userId"], idx=0, limit=0)
+
+                    playlog_dict = {}
+                    for playlog in playlogs:
+                        playlog_id = playlog["playlogId"]
+                        if playlog_id not in playlog_dict:
+                            playlog_dict[playlog_id] = []
+                        playlog_dict[playlog_id].append(playlog["musicId"])
+                    valid_playlogs = []
+                    allowed_music = set(Mai2Constants.KALEIDXSCOPE_KEY_CONDITION[2])
+                    for playlog_id, music_ids in playlog_dict.items():
+
+                        if len(music_ids) != len(set(music_ids)):
+                            continue
+                        all_valid = True
+                        for mid in music_ids:
+                            if mid not in allowed_music:
+                                all_valid = False
+                                break
+                        if all_valid:
+                            valid_playlogs.append(playlog_id)
+
+                    if valid_playlogs:
+                        new_kaleidxscope = {'gateId': 2, "isKeyFound": True}
+                        await self.data.score.put_user_kaleidxscope(data["userId"], new_kaleidxscope)
+
         kaleidxscope = await self.data.score.get_user_kaleidxscope_list(data["userId"])
 
         if kaleidxscope is None: