136 lines
5.9 KiB
Python
136 lines
5.9 KiB
Python
import base64
|
|
import os
|
|
from typing import Dict
|
|
|
|
from core.config import CoreConfig
|
|
from titles.mai2.buddiesplus import Mai2BuddiesPlus
|
|
from titles.mai2.const import Mai2Constants
|
|
from titles.mai2.config import Mai2Config
|
|
|
|
|
|
class Mai2Prism(Mai2BuddiesPlus):
|
|
def __init__(self, cfg: CoreConfig, game_cfg: Mai2Config) -> None:
|
|
super().__init__(cfg, game_cfg)
|
|
self.version = Mai2Constants.VER_MAIMAI_DX_PRISM
|
|
|
|
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)
|
|
|
|
# hardcode lastDataVersion for CardMaker
|
|
user_data["lastDataVersion"] = "1.50.00"
|
|
return user_data
|
|
|
|
|
|
async def handle_get_user_new_item_list_api_request(self, data: Dict) -> Dict:
|
|
return {
|
|
"user_id": data["userId"],
|
|
"userItemList": []
|
|
}
|
|
|
|
#used for downloading music scores online
|
|
async def handle_get_game_music_score_api_request(self, data: Dict) -> Dict:
|
|
if not self.game_config.charts.enable or not self.game_config.charts.chart_folder:
|
|
return {"gameMusicScore": {"musicId": data["musicId"], "level": data["level"], "type": data["type"], "scoreData": ""}}
|
|
|
|
padded_music_id = str(data["musicId"]).zfill(6)
|
|
padded_level_id = str(data["level"]).zfill(2)
|
|
music_folder = f"music{padded_music_id}"
|
|
|
|
if data["type"] == 0:
|
|
target_filename = f"{padded_music_id}_{padded_level_id}.ma2"
|
|
elif data["type"] == 1:
|
|
target_filename = f"{padded_music_id}_{padded_level_id}_L.ma2"
|
|
elif data["type"] == 2:
|
|
target_filename = f"{padded_music_id}_{padded_level_id}_R.ma2"
|
|
else:
|
|
self.logger.error("Invalid MusicScore type!")
|
|
return {"gameMusicScore": {"musicId": data["musicId"], "level": data["level"], "type": data["type"], "scoreData": ""}}
|
|
|
|
|
|
chart_path = os.path.join(self.game_config.charts.chart_folder, str(self.version), music_folder, target_filename)
|
|
if os.path.isfile(chart_path):
|
|
with open(chart_path, 'rb') as file:
|
|
file_content = file.read()
|
|
base64_content = base64.b64encode(file_content).decode('ascii')
|
|
return {
|
|
"gameMusicScore": {
|
|
"musicId": data["musicId"],
|
|
"level": data["level"],
|
|
"type": data["type"],
|
|
"scoreData": base64_content
|
|
}
|
|
}
|
|
else:
|
|
self.logger.warning(f"Version {self.version} Chart {target_filename} not found!")
|
|
return {"gameMusicScore": {"musicId": data["musicId"], "level": data["level"], "type": data["type"], "scoreData": ""}}
|
|
|
|
|
|
async def handle_get_game_kaleidx_scope_api_request(self, data: Dict) -> Dict:
|
|
return {
|
|
"gameKaleidxScopeList": [
|
|
{"gateId": 1, "phaseId": 6},
|
|
{"gateId": 2, "phaseId": 6},
|
|
{"gateId": 3, "phaseId": 6},
|
|
{"gateId": 4, "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:
|
|
return {"userId": data["userId"], "userKaleidxScopeList":[]}
|
|
|
|
kaleidxscope_list = []
|
|
for kaleidxscope_data in kaleidxscope:
|
|
tmp = kaleidxscope_data._asdict()
|
|
tmp.pop("user")
|
|
tmp.pop("id")
|
|
kaleidxscope_list.append(tmp)
|
|
return {
|
|
"userId": data["userId"],
|
|
"userKaleidxScopeList": kaleidxscope_list
|
|
} |