from typing import Optional, Dict, List, Tuple, Any from typing_extensions import Final from bemani.backend.reflec.base import ReflecBeatBase from bemani.backend.reflec.limelight import ReflecBeatLimelight from bemani.common import Profile, ValidatedDict, VersionConstants, ID, Time from bemani.data import UserID, Achievement from bemani.protocol import Node class ReflecBeatColette(ReflecBeatBase): name: str = "REFLEC BEAT colette" version: int = VersionConstants.REFLEC_BEAT_COLETTE # Clear types according to the game GAME_CLEAR_TYPE_NO_PLAY: Final[int] = 0 GAME_CLEAR_TYPE_FAILED: Final[int] = 1 GAME_CLEAR_TYPE_CLEARED: Final[int] = 2 GAME_CLEAR_TYPE_ALMOST_COMBO: Final[int] = 3 GAME_CLEAR_TYPE_FULL_COMBO: Final[int] = 4 def previous_version(self) -> Optional[ReflecBeatBase]: return ReflecBeatLimelight(self.data, self.config, self.model) @classmethod def get_settings(cls) -> Dict[str, Any]: """ Return all of our front-end modifiably settings. """ return { "bools": [ { "name": "Force Song Unlock", "tip": "Force unlock all songs.", "category": "game_config", "setting": "force_unlock_songs", }, ], "ints": [], } def __db_to_game_clear_type(self, db_clear_type: int, db_combo_type: int) -> int: if db_clear_type == self.CLEAR_TYPE_NO_PLAY: return self.GAME_CLEAR_TYPE_NO_PLAY if db_clear_type == self.CLEAR_TYPE_FAILED: return self.GAME_CLEAR_TYPE_FAILED if db_clear_type in [ self.CLEAR_TYPE_CLEARED, self.CLEAR_TYPE_HARD_CLEARED, self.CLEAR_TYPE_S_HARD_CLEARED, ]: if db_combo_type == self.COMBO_TYPE_NONE: return self.GAME_CLEAR_TYPE_CLEARED if db_combo_type == self.COMBO_TYPE_ALMOST_COMBO: return self.GAME_CLEAR_TYPE_ALMOST_COMBO if db_combo_type in [ self.COMBO_TYPE_FULL_COMBO, self.COMBO_TYPE_FULL_COMBO_ALL_JUST, ]: return self.GAME_CLEAR_TYPE_FULL_COMBO raise Exception(f"Invalid db_combo_type {db_combo_type}") raise Exception(f"Invalid db_clear_type {db_clear_type}") def __game_to_db_clear_type(self, game_clear_type: int) -> Tuple[int, int]: if game_clear_type == self.GAME_CLEAR_TYPE_NO_PLAY: return (self.CLEAR_TYPE_NO_PLAY, self.COMBO_TYPE_NONE) if game_clear_type == self.GAME_CLEAR_TYPE_FAILED: return (self.CLEAR_TYPE_FAILED, self.COMBO_TYPE_NONE) if game_clear_type == self.GAME_CLEAR_TYPE_CLEARED: return (self.CLEAR_TYPE_CLEARED, self.COMBO_TYPE_NONE) if game_clear_type == self.GAME_CLEAR_TYPE_ALMOST_COMBO: return (self.CLEAR_TYPE_CLEARED, self.COMBO_TYPE_ALMOST_COMBO) if game_clear_type == self.GAME_CLEAR_TYPE_FULL_COMBO: return (self.CLEAR_TYPE_CLEARED, self.COMBO_TYPE_FULL_COMBO) raise Exception(f"Invalid game_clear_type {game_clear_type}") def handle_pcb_error_request(self, request: Node) -> Node: return Node.void("pcb") def handle_pcb_uptime_update_request(self, request: Node) -> Node: return Node.void("pcb") def handle_pcb_boot_request(self, request: Node) -> Node: shop_id = ID.parse_machine_id(request.child_value("lid")) machine = self.get_machine_by_id(shop_id) if machine is not None: machine_name = machine.name close = machine.data.get_bool("close") hour = machine.data.get_int("hour") minute = machine.data.get_int("minute") else: machine_name = "" close = False hour = 0 minute = 0 root = Node.void("pcb") sinfo = Node.void("sinfo") root.add_child(sinfo) sinfo.add_child(Node.string("nm", machine_name)) sinfo.add_child(Node.bool("cl_enbl", close)) sinfo.add_child(Node.u8("cl_h", hour)) sinfo.add_child(Node.u8("cl_m", minute)) return root def handle_shop_setting_write_request(self, request: Node) -> Node: return Node.void("shop") def handle_shop_info_write_request(self, request: Node) -> Node: self.update_machine_name(request.child_value("sinfo/nm")) self.update_machine_data( { "close": request.child_value("sinfo/cl_enbl"), "hour": request.child_value("sinfo/cl_h"), "minute": request.child_value("sinfo/cl_m"), "pref": request.child_value("sinfo/prf"), } ) return Node.void("shop") def __add_event_info(self, root: Node) -> None: event_ctrl = Node.void("event_ctrl") root.add_child(event_ctrl) events: Dict[int, int] = { # Tricolette Park unlock event. 9: 0, } for (eventid, phase) in events.items(): data = Node.void("data") event_ctrl.add_child(data) data.add_child(Node.s32("type", eventid)) data.add_child(Node.s32("phase", phase)) item_lock_ctrl = Node.void("item_lock_ctrl") root.add_child(item_lock_ctrl) # Contains zero or more nodes like: # # any # any # 0-3 # def handle_info_common_request(self, request: Node) -> Node: root = Node.void("info") self.__add_event_info(root) return root def handle_info_ranking_request(self, request: Node) -> Node: version = request.child_value("ver") root = Node.void("info") root.add_child(Node.s32("ver", version)) ranking = Node.void("ranking") root.add_child(ranking) def add_hitchart( name: str, start: int, end: int, hitchart: List[Tuple[int, int]] ) -> None: base = Node.void(name) ranking.add_child(base) base.add_child(Node.s32("bt", start)) base.add_child(Node.s32("et", end)) new = Node.void("new") base.add_child(new) for (mid, plays) in hitchart: d = Node.void("d") new.add_child(d) d.add_child(Node.s16("mid", mid)) d.add_child(Node.s32("cnt", plays)) # Weekly hit chart add_hitchart( "weekly", Time.now() - Time.SECONDS_IN_WEEK, Time.now(), self.data.local.music.get_hit_chart(self.game, self.version, 1024, 7), ) # Monthly hit chart add_hitchart( "monthly", Time.now() - Time.SECONDS_IN_DAY * 30, Time.now(), self.data.local.music.get_hit_chart(self.game, self.version, 1024, 30), ) # All time hit chart add_hitchart( "total", Time.now() - Time.SECONDS_IN_DAY * 365, Time.now(), self.data.local.music.get_hit_chart(self.game, self.version, 1024, 365), ) return root def handle_info_pzlcmt_read_request(self, request: Node) -> Node: extid = request.child_value("uid") teamid = request.child_value("tid") limit = request.child_value("limit") userid = self.data.remote.user.from_extid(self.game, self.version, extid) comments = [ achievement for achievement in self.data.local.user.get_all_time_based_achievements( self.game, self.version ) if achievement[1].type == "puzzle_comment" ] comments.sort(key=lambda x: x[1].timestamp, reverse=True) favorites = [comment for comment in comments if comment[0] == userid] teamcomments = [ comment for comment in comments if comment[1].data.get_int("teamid") == teamid ] # Cap all comment blocks to the limit if limit >= 0: comments = comments[:limit] favorites = favorites[:limit] teamcomments = teamcomments[:limit] root = Node.void("info") comment = Node.void("comment") root.add_child(comment) comment.add_child(Node.s32("time", Time.now())) # Mapping of profiles to userIDs uid_mapping = { uid: prof for (uid, prof) in self.get_any_profiles([c[0] for c in comments]) } # Handle anonymous comments by returning a default profile uid_mapping[UserID(0)] = Profile( self.game, self.version, "", 0, {"name": "ＰＬＡＹＥＲ"}, ) def add_comments(name: str, selected: List[Tuple[UserID, Achievement]]) -> None: for (uid, ach) in selected: cmnt = Node.void(name) root.add_child(cmnt) cmnt.add_child(Node.s32("uid", uid_mapping[uid].extid)) cmnt.add_child(Node.string("name", uid_mapping[uid].get_str("name"))) cmnt.add_child(Node.s16("icon", ach.data.get_int("icon"))) cmnt.add_child(Node.s8("bln", ach.data.get_int("bln"))) cmnt.add_child(Node.s32("tid", ach.data.get_int("teamid"))) cmnt.add_child(Node.string("t_name", ach.data.get_str("teamname"))) cmnt.add_child(Node.s8("pref", ach.data.get_int("prefecture"))) cmnt.add_child(Node.s32("time", ach.timestamp)) cmnt.add_child(Node.string("comment", ach.data.get_str("comment"))) cmnt.add_child(Node.bool("is_tweet", ach.data.get_bool("tweet"))) # Add all comments add_comments("c", comments) # Add personal comments (favorites) add_comments("cf", favorites) # Add team comments add_comments("ct", teamcomments) return root def handle_info_pzlcmt_write_request(self, request: Node) -> Node: extid = request.child_value("uid") userid = self.data.remote.user.from_extid(self.game, self.version, extid) if userid is None: # Anonymous comment userid = UserID(0) icon = request.child_value("icon") bln = request.child_value("bln") teamid = request.child_value("tid") teamname = request.child_value("t_name") prefecture = request.child_value("pref") comment = request.child_value("comment") is_tweet = request.child_value("is_tweet") # Link comment to user's profile self.data.local.user.put_time_based_achievement( self.game, self.version, userid, 0, # We never have an ID for this, since comments are add-only "puzzle_comment", { "icon": icon, "bln": bln, "teamid": teamid, "teamname": teamname, "prefecture": prefecture, "comment": comment, "tweet": is_tweet, }, ) return Node.void("info") def handle_jbrbcollabo_save_request(self, request: Node) -> Node: jbrbcollabo = Node.void("jbrbcollabo") jbrbcollabo.add_child(Node.u16("marathontype", 0)) jbrbcollabo.add_child(Node.u32("smith_start", 0)) jbrbcollabo.add_child(Node.u32("pastel_start", 0)) jbrbcollabo.add_child(Node.u32("smith_run", 0)) jbrbcollabo.add_child(Node.u32("pastel_run", 0)) jbrbcollabo.add_child(Node.u16("smith_ouen", 0)) jbrbcollabo.add_child(Node.u16("pastel_ouen", 0)) jbrbcollabo.add_child(Node.u32("smith_water_run", 0)) jbrbcollabo.add_child(Node.u32("pastel_water_run", 0)) jbrbcollabo.add_child(Node.bool("getwater", False)) jbrbcollabo.add_child(Node.bool("smith_goal", False)) jbrbcollabo.add_child(Node.bool("pastel_goal", False)) jbrbcollabo.add_child(Node.u16("distancetype", 0)) jbrbcollabo.add_child(Node.bool("run1_1_j_flg", False)) jbrbcollabo.add_child(Node.bool("run1_2_j_flg", False)) jbrbcollabo.add_child(Node.bool("run1_3_j_flg", False)) jbrbcollabo.add_child(Node.bool("run1_4_flg", False)) jbrbcollabo.add_child(Node.bool("run1_1_r_flg", False)) jbrbcollabo.add_child(Node.bool("run1_2_r_flg", False)) jbrbcollabo.add_child(Node.bool("run1_3_r_flg", False)) jbrbcollabo.add_child(Node.bool("run1_4_flg", False)) jbrbcollabo.add_child(Node.bool("run2_1_j_flg", False)) jbrbcollabo.add_child(Node.bool("run2_2_j_flg", False)) jbrbcollabo.add_child(Node.bool("run2_3_j_flg", False)) jbrbcollabo.add_child(Node.bool("run2_4_flg", False)) jbrbcollabo.add_child(Node.bool("run2_1_r_flg", False)) jbrbcollabo.add_child(Node.bool("run2_2_r_flg", False)) jbrbcollabo.add_child(Node.bool("run2_3_r_flg", False)) jbrbcollabo.add_child(Node.bool("run2_4_flg", False)) jbrbcollabo.add_child(Node.bool("run3_1_j_flg", False)) jbrbcollabo.add_child(Node.bool("run3_2_j_flg", False)) jbrbcollabo.add_child(Node.bool("run3_3_j_flg", False)) jbrbcollabo.add_child(Node.bool("run3_4_flg", False)) jbrbcollabo.add_child(Node.bool("run3_1_r_flg", False)) jbrbcollabo.add_child(Node.bool("run3_2_r_flg", False)) jbrbcollabo.add_child(Node.bool("run3_3_r_flg", False)) jbrbcollabo.add_child(Node.bool("run3_4_flg", False)) jbrbcollabo.add_child(Node.bool("run4_1_j_flg", False)) jbrbcollabo.add_child(Node.bool("run4_1_r_flg", False)) jbrbcollabo.add_child(Node.bool("run4_2_flg", False)) jbrbcollabo.add_child(Node.bool("run4_2_flg", False)) jbrbcollabo.add_child(Node.bool("start_flg", False)) return jbrbcollabo def handle_lobby_entry_request(self, request: Node) -> Node: root = Node.void("lobby") root.add_child(Node.s32("interval", 120)) root.add_child(Node.s32("interval_p", 120)) # Create a lobby entry for this user extid = request.child_value("e/uid") userid = self.data.remote.user.from_extid(self.game, self.version, extid) if userid is not None: profile = self.get_profile(userid) self.data.local.lobby.put_lobby( self.game, self.version, userid, { "mid": request.child_value("e/mid"), "ng": request.child_value("e/ng"), "mopt": request.child_value("e/mopt"), "tid": request.child_value("e/tid"), "tn": request.child_value("e/tn"), "topt": request.child_value("e/topt"), "lid": request.child_value("e/lid"), "sn": request.child_value("e/sn"), "pref": request.child_value("e/pref"), "stg": request.child_value("e/stg"), "pside": request.child_value("e/pside"), "eatime": request.child_value("e/eatime"), "ga": request.child_value("e/ga"), "gp": request.child_value("e/gp"), "la": request.child_value("e/la"), "ver": request.child_value("e/ver"), }, ) lobby = self.data.local.lobby.get_lobby( self.game, self.version, userid, ) root.add_child(Node.s32("eid", lobby.get_int("id"))) e = Node.void("e") root.add_child(e) e.add_child(Node.s32("eid", lobby.get_int("id"))) e.add_child(Node.u16("mid", lobby.get_int("mid"))) e.add_child(Node.u8("ng", lobby.get_int("ng"))) e.add_child(Node.s32("uid", profile.extid)) e.add_child(Node.s32("uattr", profile.get_int("uattr"))) e.add_child(Node.string("pn", profile.get_str("name"))) e.add_child(Node.s16("mg", profile.get_int("mg"))) e.add_child(Node.s32("mopt", lobby.get_int("mopt"))) e.add_child(Node.s32("tid", lobby.get_int("tid"))) e.add_child(Node.string("tn", lobby.get_str("tn"))) e.add_child(Node.s32("topt", lobby.get_int("topt"))) e.add_child(Node.string("lid", lobby.get_str("lid"))) e.add_child(Node.string("sn", lobby.get_str("sn"))) e.add_child(Node.u8("pref", lobby.get_int("pref"))) e.add_child(Node.s8("stg", lobby.get_int("stg"))) e.add_child(Node.s8("pside", lobby.get_int("pside"))) e.add_child(Node.s16("eatime", lobby.get_int("eatime"))) e.add_child(Node.u8_array("ga", lobby.get_int_array("ga", 4))) e.add_child(Node.u16("gp", lobby.get_int("gp"))) e.add_child(Node.u8_array("la", lobby.get_int_array("la", 4))) e.add_child(Node.u8("ver", lobby.get_int("ver"))) return root def handle_lobby_read_request(self, request: Node) -> Node: root = Node.void("lobby") root.add_child(Node.s32("interval", 120)) root.add_child(Node.s32("interval_p", 120)) # Look up all lobbies matching the criteria specified ver = request.child_value("var") mg = request.child_value("m_grade") # noqa: F841 extid = request.child_value("uid") limit = request.child_value("max") userid = self.data.remote.user.from_extid(self.game, self.version, extid) if userid is not None: lobbies = self.data.local.lobby.get_all_lobbies(self.game, self.version) for (user, lobby) in lobbies: if limit <= 0: break if user == userid: # If we have our own lobby, don't return it continue if ver != lobby.get_int("ver"): # Don't return lobby data for different versions continue profile = self.get_profile(user) if profile is None: # No profile info, don't return this lobby continue e = Node.void("e") root.add_child(e) e.add_child(Node.s32("eid", lobby.get_int("id"))) e.add_child(Node.u16("mid", lobby.get_int("mid"))) e.add_child(Node.u8("ng", lobby.get_int("ng"))) e.add_child(Node.s32("uid", profile.extid)) e.add_child(Node.s32("uattr", profile.get_int("uattr"))) e.add_child(Node.string("pn", profile.get_str("name"))) e.add_child(Node.s16("mg", profile.get_int("mg"))) e.add_child(Node.s32("mopt", lobby.get_int("mopt"))) e.add_child(Node.s32("tid", lobby.get_int("tid"))) e.add_child(Node.string("tn", lobby.get_str("tn"))) e.add_child(Node.s32("topt", lobby.get_int("topt"))) e.add_child(Node.string("lid", lobby.get_str("lid"))) e.add_child(Node.string("sn", lobby.get_str("sn"))) e.add_child(Node.u8("pref", lobby.get_int("pref"))) e.add_child(Node.s8("stg", lobby.get_int("stg"))) e.add_child(Node.s8("pside", lobby.get_int("pside"))) e.add_child(Node.s16("eatime", lobby.get_int("eatime"))) e.add_child(Node.u8_array("ga", lobby.get_int_array("ga", 4))) e.add_child(Node.u16("gp", lobby.get_int("gp"))) e.add_child(Node.u8_array("la", lobby.get_int_array("la", 4))) e.add_child(Node.u8("ver", lobby.get_int("ver"))) limit = limit - 1 return root def handle_lobby_delete_request(self, request: Node) -> Node: eid = request.child_value("eid") self.data.local.lobby.destroy_lobby(eid) return Node.void("lobby") def handle_player_start_request(self, request: Node) -> Node: root = Node.void("player") # Create a new play session based on info from the request refid = request.child_value("rid") userid = self.data.remote.user.from_refid(self.game, self.version, refid) if userid is not None: self.data.local.lobby.put_play_session_info( self.game, self.version, userid, { "ga": request.child_value("ga"), "gp": request.child_value("gp"), "la": request.child_value("la"), }, ) info = self.data.local.lobby.get_play_session_info( self.game, self.version, userid, ) if info is not None: play_id = info.get_int("id") else: play_id = 0 else: play_id = 0 # Session stuff, and resend global defaults root.add_child(Node.s32("plyid", play_id)) root.add_child(Node.u64("start_time", Time.now() * 1000)) self.__add_event_info(root) # Event settings and such lincle_link_4 = Node.void("lincle_link_4") root.add_child(lincle_link_4) lincle_link_4.add_child(Node.u32("qpro", 0)) lincle_link_4.add_child(Node.u32("glass", 0)) lincle_link_4.add_child(Node.u32("treasure", 0)) lincle_link_4.add_child(Node.bool("for_iidx_0_0", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_1", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_2", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_3", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_4", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_5", False)) lincle_link_4.add_child(Node.bool("for_iidx_0_6", False)) lincle_link_4.add_child(Node.bool("for_iidx_0", False)) lincle_link_4.add_child(Node.bool("for_iidx_1", False)) lincle_link_4.add_child(Node.bool("for_iidx_2", False)) lincle_link_4.add_child(Node.bool("for_iidx_3", False)) lincle_link_4.add_child(Node.bool("for_iidx_4", False)) lincle_link_4.add_child(Node.bool("for_rb_0_0", False)) lincle_link_4.add_child(Node.bool("for_rb_0_1", False)) lincle_link_4.add_child(Node.bool("for_rb_0_2", False)) lincle_link_4.add_child(Node.bool("for_rb_0_3", False)) lincle_link_4.add_child(Node.bool("for_rb_0_4", False)) lincle_link_4.add_child(Node.bool("for_rb_0_5", False)) lincle_link_4.add_child(Node.bool("for_rb_0_6", False)) lincle_link_4.add_child(Node.bool("for_rb_0", False)) lincle_link_4.add_child(Node.bool("for_rb_1", False)) lincle_link_4.add_child(Node.bool("for_rb_2", False)) lincle_link_4.add_child(Node.bool("for_rb_3", False)) lincle_link_4.add_child(Node.bool("for_rb_4", False)) lincle_link_4.add_child(Node.bool("qproflg", False)) lincle_link_4.add_child(Node.bool("glassflg", False)) lincle_link_4.add_child(Node.bool("complete", False)) jbrbcollabo = Node.void("jbrbcollabo") root.add_child(jbrbcollabo) jbrbcollabo.add_child(Node.bool("run1_1_j_flg", False)) jbrbcollabo.add_child(Node.bool("run1_2_j_flg", False)) jbrbcollabo.add_child(Node.bool("run1_3_j_flg", False)) jbrbcollabo.add_child(Node.bool("run1_4_flg", False)) jbrbcollabo.add_child(Node.bool("run1_1_r_flg", False)) jbrbcollabo.add_child(Node.bool("run1_2_r_flg", False)) jbrbcollabo.add_child(Node.bool("run1_3_r_flg", False)) jbrbcollabo.add_child(Node.bool("run1_4_flg", False)) jbrbcollabo.add_child(Node.bool("run2_1_j_flg", False)) jbrbcollabo.add_child(Node.bool("run2_2_j_flg", False)) jbrbcollabo.add_child(Node.bool("run2_3_j_flg", False)) jbrbcollabo.add_child(Node.bool("run2_4_flg", False)) jbrbcollabo.add_child(Node.bool("run2_1_r_flg", False)) jbrbcollabo.add_child(Node.bool("run2_2_r_flg", False)) jbrbcollabo.add_child(Node.bool("run2_3_r_flg", False)) jbrbcollabo.add_child(Node.bool("run2_4_flg", False)) jbrbcollabo.add_child(Node.bool("run3_1_j_flg", False)) jbrbcollabo.add_child(Node.bool("run3_2_j_flg", False)) jbrbcollabo.add_child(Node.bool("run3_3_j_flg", False)) jbrbcollabo.add_child(Node.bool("run3_4_flg", False)) jbrbcollabo.add_child(Node.bool("run3_1_r_flg", False)) jbrbcollabo.add_child(Node.bool("run3_2_r_flg", False)) jbrbcollabo.add_child(Node.bool("run3_3_r_flg", False)) jbrbcollabo.add_child(Node.bool("run3_4_flg", False)) jbrbcollabo.add_child(Node.u16("marathontype", 0)) jbrbcollabo.add_child(Node.u32("smith_start", 0)) jbrbcollabo.add_child(Node.u32("pastel_start", 0)) jbrbcollabo.add_child(Node.u16("smith_ouen", 0)) jbrbcollabo.add_child(Node.u16("pastel_ouen", 0)) jbrbcollabo.add_child(Node.u16("distancetype", 0)) jbrbcollabo.add_child(Node.bool("smith_goal", False)) jbrbcollabo.add_child(Node.bool("pastel_goal", False)) jbrbcollabo.add_child(Node.bool("run4_1_j_flg", False)) jbrbcollabo.add_child(Node.bool("run4_1_r_flg", False)) jbrbcollabo.add_child(Node.bool("run4_2_flg", False)) jbrbcollabo.add_child(Node.bool("run4_2_flg", False)) jbrbcollabo.add_child(Node.bool("start_flg", False)) tricolettepark = Node.void("tricolettepark") root.add_child(tricolettepark) tricolettepark.add_child(Node.s32("open_music", -1)) tricolettepark.add_child(Node.s32("boss0_damage", -1)) tricolettepark.add_child(Node.s32("boss1_damage", -1)) tricolettepark.add_child(Node.s32("boss2_damage", -1)) tricolettepark.add_child(Node.s32("boss3_damage", -1)) tricolettepark.add_child(Node.s32("boss0_stun", -1)) tricolettepark.add_child(Node.s32("boss1_stun", -1)) tricolettepark.add_child(Node.s32("boss2_stun", -1)) tricolettepark.add_child(Node.s32("boss3_stun", -1)) tricolettepark.add_child(Node.s32("magic_gauge", -1)) tricolettepark.add_child(Node.s32("today_party", -1)) tricolettepark.add_child(Node.bool("union_magic", False)) tricolettepark.add_child(Node.bool("is_complete", False)) tricolettepark.add_child(Node.float("base_attack_rate", 1.0)) return root def handle_player_delete_request(self, request: Node) -> Node: return Node.void("player") def handle_player_end_request(self, request: Node) -> Node: # Destroy play session based on info from the request refid = request.child_value("rid") userid = self.data.remote.user.from_refid(self.game, self.version, refid) if userid is not None: # Kill any lingering lobbies by this user lobby = self.data.local.lobby.get_lobby( self.game, self.version, userid, ) if lobby is not None: self.data.local.lobby.destroy_lobby(lobby.get_int("id")) self.data.local.lobby.destroy_play_session_info( self.game, self.version, userid ) return Node.void("player") def handle_player_succeed_request(self, request: Node) -> Node: refid = request.child_value("rid") userid = self.data.remote.user.from_refid(self.game, self.version, refid) if userid is not None: previous_version = self.previous_version() profile = previous_version.get_profile(userid) achievements = self.data.local.user.get_achievements( previous_version.game, previous_version.version, userid ) scores = self.data.remote.music.get_scores( previous_version.game, previous_version.version, userid ) else: profile = None root = Node.void("player") if profile is None: root.add_child(Node.string("name", "")) root.add_child(Node.s16("lv", -1)) root.add_child(Node.s32("exp", -1)) root.add_child(Node.s32("grd", -1)) root.add_child(Node.s32("ap", -1)) root.add_child(Node.void("released")) root.add_child(Node.void("mrecord")) else: root.add_child(Node.string("name", profile.get_str("name"))) root.add_child(Node.s16("lv", profile.get_int("lvl"))) root.add_child(Node.s32("exp", profile.get_int("exp"))) root.add_child(Node.s32("grd", profile.get_int("mg"))) # This is a guess root.add_child(Node.s32("ap", profile.get_int("ap"))) released = Node.void("released") root.add_child(released) for item in achievements: if item.type != "item_0": continue released.add_child(Node.s16("i", item.id)) mrecord = Node.void("mrecord") root.add_child(mrecord) for score in scores: mrec = Node.void("mrec") mrecord.add_child(mrec) mrec.add_child(Node.s16("mid", score.id)) mrec.add_child(Node.s8("ntgrd", score.chart)) mrec.add_child(Node.s32("pc", score.plays)) mrec.add_child( Node.s8( "ct", self.__db_to_game_clear_type( score.data.get_int("clear_type"), score.data.get_int("combo_type"), ), ) ) mrec.add_child(Node.s16("ar", score.data.get_int("achievement_rate"))) mrec.add_child(Node.s16("scr", score.points)) mrec.add_child(Node.s16("cmb", score.data.get_int("combo"))) mrec.add_child(Node.s16("ms", score.data.get_int("miss_count"))) mrec.add_child(Node.u16("ver", 0)) return root def handle_player_read_request(self, request: Node) -> Node: refid = request.child_value("rid") profile = self.get_profile_by_refid(refid) if profile: return profile return Node.void("player") def handle_player_write_request(self, request: Node) -> Node: refid = request.child_value("pdata/account/rid") profile = self.put_profile_by_refid(refid, request) root = Node.void("player") if profile is None: root.add_child(Node.s32("uid", 0)) else: root.add_child(Node.s32("uid", profile.extid)) return root def format_profile(self, userid: UserID, profile: Profile) -> Node: statistics = self.get_play_statistics(userid) game_config = self.get_game_config() achievements = self.data.local.user.get_achievements( self.game, self.version, userid ) scores = self.data.remote.music.get_scores(self.game, self.version, userid) links = self.data.local.user.get_links(self.game, self.version, userid) root = Node.void("player") pdata = Node.void("pdata") root.add_child(pdata) account = Node.void("account") pdata.add_child(account) account.add_child(Node.s32("usrid", profile.extid)) account.add_child(Node.s32("tpc", statistics.total_plays)) account.add_child(Node.s32("dpc", statistics.today_plays)) account.add_child(Node.s32("crd", 1)) account.add_child(Node.s32("brd", 1)) account.add_child(Node.s32("tdc", statistics.total_days)) account.add_child(Node.s32("intrvld", 0)) account.add_child(Node.s16("ver", 5)) account.add_child(Node.u64("pst", 0)) account.add_child(Node.u64("st", Time.now() * 1000)) # Base account info base = Node.void("base") pdata.add_child(base) base.add_child(Node.string("name", profile.get_str("name"))) base.add_child(Node.s32("exp", profile.get_int("exp"))) base.add_child(Node.s32("lv", profile.get_int("lvl"))) base.add_child(Node.s32("mg", profile.get_int("mg"))) base.add_child(Node.s32("ap", profile.get_int("ap"))) base.add_child(Node.s32("tid", profile.get_int("team_id", -1))) base.add_child(Node.string("tname", profile.get_str("team_name", ""))) base.add_child(Node.string("cmnt", "")) base.add_child(Node.s32("uattr", profile.get_int("uattr"))) base.add_child( Node.s32_array("hidden_param", profile.get_int_array("hidden_param", 50)) ) base.add_child(Node.s32("tbs", -1)) base.add_child(Node.s32("tbs_r", -1)) # Rivals rival = Node.void("rival") pdata.add_child(rival) slotid = 0 for link in links: if link.type != "rival": continue rprofile = self.get_profile(link.other_userid) if rprofile is None: continue r = Node.void("r") rival.add_child(r) r.add_child(Node.s32("slot_id", slotid)) r.add_child(Node.s32("id", rprofile.extid)) r.add_child(Node.string("name", rprofile.get_str("name"))) r.add_child(Node.bool("friend", True)) r.add_child(Node.bool("locked", False)) r.add_child(Node.s32("rc", 0)) slotid = slotid + 1 # Player customizations custom = Node.void("custom") customdict = profile.get_dict("custom") pdata.add_child(custom) custom.add_child(Node.u8("st_shot", customdict.get_int("st_shot"))) custom.add_child(Node.u8("st_frame", customdict.get_int("st_frame"))) custom.add_child(Node.u8("st_expl", customdict.get_int("st_expl"))) custom.add_child(Node.u8("st_bg", customdict.get_int("st_bg"))) custom.add_child(Node.u8("st_shot_vol", customdict.get_int("st_shot_vol"))) custom.add_child(Node.u8("st_bg_bri", customdict.get_int("st_bg_bri"))) custom.add_child(Node.u8("st_obj_size", customdict.get_int("st_obj_size"))) custom.add_child(Node.u8("st_jr_gauge", customdict.get_int("st_jr_gauge"))) custom.add_child(Node.u8("st_clr_gauge", customdict.get_int("st_clr_gauge"))) custom.add_child(Node.u8("st_jdg_disp", customdict.get_int("st_jdg_disp"))) custom.add_child(Node.u8("st_tm_disp", customdict.get_int("st_tm_disp"))) custom.add_child(Node.u8("st_rnd", customdict.get_int("st_rnd"))) custom.add_child( Node.s16_array("schat_0", customdict.get_int_array("schat_0", 9)) ) custom.add_child( Node.s16_array("schat_1", customdict.get_int_array("schat_1", 9)) ) custom.add_child( Node.s16_array("ichat_0", customdict.get_int_array("ichat_0", 6)) ) custom.add_child( Node.s16_array("ichat_1", customdict.get_int_array("ichat_1", 6)) ) # Player external config config = Node.void("config") configdict = profile.get_dict("config") pdata.add_child(config) config.add_child(Node.u8("msel_bgm", configdict.get_int("msel_bgm"))) config.add_child( Node.u8("narrowdown_type", configdict.get_int("narrowdown_type")) ) config.add_child(Node.s16("icon_id", configdict.get_int("icon_id"))) config.add_child(Node.s16("byword_0", configdict.get_int("byword_0"))) config.add_child(Node.s16("byword_1", configdict.get_int("byword_1"))) config.add_child( Node.bool("is_auto_byword_0", configdict.get_bool("is_auto_byword_0")) ) config.add_child( Node.bool("is_auto_byword_1", configdict.get_bool("is_auto_byword_1")) ) config.add_child(Node.u8("mrec_type", configdict.get_int("mrec_type"))) config.add_child(Node.u8("tab_sel", configdict.get_int("tab_sel"))) config.add_child(Node.u8("card_disp", configdict.get_int("card_disp"))) config.add_child( Node.u8("score_tab_disp", configdict.get_int("score_tab_disp")) ) config.add_child( Node.s16("last_music_id", configdict.get_int("last_music_id", -1)) ) config.add_child( Node.u8("last_note_grade", configdict.get_int("last_note_grade")) ) config.add_child(Node.u8("sort_type", configdict.get_int("sort_type"))) config.add_child( Node.u8("rival_panel_type", configdict.get_int("rival_panel_type")) ) config.add_child( Node.u64("random_entry_work", configdict.get_int("random_entry_work")) ) config.add_child( Node.u8("folder_lamp_type", configdict.get_int("folder_lamp_type")) ) config.add_child(Node.bool("is_tweet", configdict.get_bool("is_tweet"))) config.add_child( Node.bool("is_link_twitter", configdict.get_bool("is_link_twitter")) ) # Stamps stamp = Node.void("stamp") stampdict = profile.get_dict("stamp") pdata.add_child(stamp) stamp.add_child( Node.s32_array("stmpcnt", stampdict.get_int_array("stmpcnt", 5)) ) stamp.add_child( Node.s32_array("tcktcnt", stampdict.get_int_array("tcktcnt", 5)) ) stamp.add_child(Node.s64("area", stampdict.get_int("area"))) stamp.add_child(Node.s64("prfvst", stampdict.get_int("prfvst"))) stamp.add_child(Node.s32("reserve", stampdict.get_int("reserve"))) # Unlocks released = Node.void("released") pdata.add_child(released) for item in achievements: if item.type[:5] != "item_": continue itemtype = int(item.type[5:]) if game_config.get_bool("force_unlock_songs") and itemtype == 0: # Don't echo unlocks when we're force unlocking, we'll do it later continue info = Node.void("info") released.add_child(info) info.add_child(Node.u8("type", itemtype)) info.add_child(Node.u16("id", item.id)) info.add_child(Node.u16("param", item.data.get_int("param"))) if game_config.get_bool("force_unlock_songs"): ids: Dict[int, int] = {} songs = self.data.local.music.get_all_songs(self.game, self.version) for song in songs: if song.id not in ids: ids[song.id] = 0 if song.data.get_int("difficulty") > 0: ids[song.id] = ids[song.id] | (1 << song.chart) for songid in ids: if ids[songid] == 0: continue info = Node.void("info") released.add_child(info) info.add_child(Node.u8("type", 0)) info.add_child(Node.u16("id", songid)) info.add_child(Node.u16("param", ids[songid])) # Favorite songs fav_music_slot = Node.void("fav_music_slot") pdata.add_child(fav_music_slot) for item in achievements: if item.type != "music": continue slot = Node.void("slot") fav_music_slot.add_child(slot) slot.add_child(Node.u8("slot_id", item.id)) slot.add_child(Node.s16("music_id", item.data.get_int("music_id"))) # Event stuff order = Node.void("order") pdata.add_child(order) order.add_child(Node.s32("exp", profile.get_int("order_exp"))) for item in achievements: if item.type != "order": continue data = Node.void("d") order.add_child(data) data.add_child(Node.s16("order", item.id)) data.add_child(Node.s16("slt", item.data.get_int("slt"))) data.add_child(Node.s32("ccnt", item.data.get_int("ccnt"))) data.add_child(Node.s32("fcnt", item.data.get_int("fcnt"))) data.add_child(Node.s32("fcnt1", item.data.get_int("fcnt1"))) data.add_child(Node.s32("prm", item.data.get_int("param"))) seedpod = Node.void("seedpod") pdata.add_child(seedpod) for item in achievements: if item.type != "seedpod": continue data = Node.void("data") seedpod.add_child(data) data.add_child(Node.s16("id", item.id)) data.add_child(Node.s16("pod", item.data.get_int("pod"))) eqpexp = Node.void("eqpexp") pdata.add_child(eqpexp) for item in achievements: if item.type[:7] != "eqpexp_": continue stype = int(item.type[7:]) data = Node.void("data") eqpexp.add_child(data) data.add_child(Node.s16("id", item.id)) data.add_child(Node.s32("exp", item.data.get_int("exp"))) data.add_child(Node.s16("stype", stype)) eventexp = Node.void("evntexp") pdata.add_child(eventexp) for item in achievements: if item.type != "eventexp": continue data = Node.void("data") eventexp.add_child(data) data.add_child(Node.s16("id", item.id)) data.add_child(Node.s32("exp", item.data.get_int("exp"))) # Scores record = Node.void("record") pdata.add_child(record) record_old = Node.void("record_old") pdata.add_child(record_old) for score in scores: rec = Node.void("rec") record.add_child(rec) rec.add_child(Node.s16("mid", score.id)) rec.add_child(Node.s8("ntgrd", score.chart)) rec.add_child(Node.s32("pc", score.plays)) rec.add_child( Node.s8( "ct", self.__db_to_game_clear_type( score.data.get_int("clear_type"), score.data.get_int("combo_type"), ), ) ) rec.add_child(Node.s16("ar", score.data.get_int("achievement_rate"))) rec.add_child(Node.s16("scr", score.points)) rec.add_child(Node.s16("cmb", score.data.get_int("combo"))) rec.add_child(Node.s16("ms", score.data.get_int("miss_count"))) rec.add_child(Node.s32("bscrt", score.timestamp)) rec.add_child( Node.s32("bart", score.data.get_int("best_achievement_rate_time")) ) rec.add_child(Node.s32("bctt", score.data.get_int("best_clear_type_time"))) rec.add_child(Node.s32("bmst", score.data.get_int("best_miss_count_time"))) rec.add_child(Node.s32("time", score.data.get_int("last_played_time"))) return root def unformat_profile( self, userid: UserID, request: Node, oldprofile: Profile ) -> Profile: game_config = self.get_game_config() newprofile = oldprofile.clone() newprofile.replace_int( "lid", ID.parse_machine_id(request.child_value("pdata/account/lid")) ) newprofile.replace_str("name", request.child_value("pdata/base/name")) newprofile.replace_int("exp", request.child_value("pdata/base/exp")) newprofile.replace_int("lvl", request.child_value("pdata/base/lvl")) newprofile.replace_int("mg", request.child_value("pdata/base/mg")) newprofile.replace_int("ap", request.child_value("pdata/base/ap")) newprofile.replace_int_array( "hidden_param", 50, request.child_value("pdata/base/hidden_param") ) configdict = newprofile.get_dict("config") config = request.child("pdata/config") if config: configdict.replace_int("msel_bgm", config.child_value("msel_bgm")) configdict.replace_int( "narrowdown_type", config.child_value("narrowdown_type") ) configdict.replace_int("icon_id", config.child_value("icon_id")) configdict.replace_int("byword_0", config.child_value("byword_0")) configdict.replace_int("byword_1", config.child_value("byword_1")) configdict.replace_bool( "is_auto_byword_0", config.child_value("is_auto_byword_0") ) configdict.replace_bool( "is_auto_byword_1", config.child_value("is_auto_byword_1") ) configdict.replace_int("mrec_type", config.child_value("mrec_type")) configdict.replace_int("tab_sel", config.child_value("tab_sel")) configdict.replace_int("card_disp", config.child_value("card_disp")) configdict.replace_int( "score_tab_disp", config.child_value("score_tab_disp") ) configdict.replace_int("last_music_id", config.child_value("last_music_id")) configdict.replace_int( "last_note_grade", config.child_value("last_note_grade") ) configdict.replace_int("sort_type", config.child_value("sort_type")) configdict.replace_int( "rival_panel_type", config.child_value("rival_panel_type") ) configdict.replace_int( "random_entry_work", config.child_value("random_entry_work") ) configdict.replace_int( "folder_lamp_type", config.child_value("folder_lamp_type") ) configdict.replace_bool("is_tweet", config.child_value("is_tweet")) configdict.replace_bool( "is_link_twitter", config.child_value("is_link_twitter") ) newprofile.replace_dict("config", configdict) customdict = newprofile.get_dict("custom") custom = request.child("pdata/custom") if custom: customdict.replace_int("st_shot", custom.child_value("st_shot")) customdict.replace_int("st_frame", custom.child_value("st_frame")) customdict.replace_int("st_expl", custom.child_value("st_expl")) customdict.replace_int("st_bg", custom.child_value("st_bg")) customdict.replace_int("st_shot_vol", custom.child_value("st_shot_vol")) customdict.replace_int("st_bg_bri", custom.child_value("st_bg_bri")) customdict.replace_int("st_obj_size", custom.child_value("st_obj_size")) customdict.replace_int("st_jr_gauge", custom.child_value("st_jr_gauge")) customdict.replace_int("st_clr_gauge", custom.child_value("st_clr_gauge")) customdict.replace_int("st_jdg_disp", custom.child_value("st_jdg_disp")) customdict.replace_int("st_tm_disp", custom.child_value("st_tm_disp")) customdict.replace_int("st_rnd", custom.child_value("st_rnd")) customdict.replace_int_array("schat_0", 9, custom.child_value("schat_0")) customdict.replace_int_array("schat_1", 9, custom.child_value("schat_1")) customdict.replace_int_array("ichat_0", 6, custom.child_value("ichat_0")) customdict.replace_int_array("ichat_1", 6, custom.child_value("ichat_1")) newprofile.replace_dict("custom", customdict) # Stamps stampdict = newprofile.get_dict("stamp") stamp = request.child("pdata/stamp") if stamp: stampdict.replace_int_array("stmpcnt", 5, stamp.child_value("stmpcnt")) stampdict.replace_int_array("tcktcnt", 5, stamp.child_value("tcktcnt")) stampdict.replace_int("area", stamp.child_value("area")) stampdict.replace_int("prfvst", stamp.child_value("prfvst")) stampdict.replace_int("reserve", stamp.child_value("reserve")) newprofile.replace_dict("stamp", stampdict) # Unlockable orders newprofile.replace_int("order_exp", request.child_value("pdata/order/exp")) order = request.child("pdata/order") if order: for child in order.children: if child.name != "d": continue orderid = child.child_value("order") slt = child.child_value("slt") ccnt = child.child_value("ccnt") fcnt = child.child_value("fcnt") fcnt1 = child.child_value("fcnt1") param = child.child_value("prm") if slt == -1: # The game doesn't return valid data for this selection # type, so be sure not to accidentally overwrite the # finished flags. continue self.data.local.user.put_achievement( self.game, self.version, userid, orderid, "order", { "slt": slt, "ccnt": ccnt, "fcnt": fcnt, "fcnt1": fcnt1, "param": param, }, ) # Music unlocks and other stuff released = request.child("pdata/released") if released: for child in released.children: if child.name != "info": continue item_id = child.child_value("id") item_type = child.child_value("type") param = child.child_value("param") if game_config.get_bool("force_unlock_songs") and item_type == 0: # Don't save unlocks when we're force unlocking continue self.data.local.user.put_achievement( self.game, self.version, userid, item_id, f"item_{item_type}", { "param": param, }, ) # Favorite music fav_music_slot = request.child("pdata/fav_music_slot") if fav_music_slot: for child in fav_music_slot.children: if child.name != "slot": continue slot_id = child.child_value("slot_id") music_id = child.child_value("music_id") if music_id == -1: # Delete this favorite self.data.local.user.destroy_achievement( self.game, self.version, userid, slot_id, "music", ) else: # Add/update this favorite self.data.local.user.put_achievement( self.game, self.version, userid, slot_id, "music", { "music_id": music_id, }, ) # Event stuff seedpod = request.child("pdata/seedpod") if seedpod: for child in seedpod.children: if child.name != "data": continue seedpod_id = child.child_value("id") seedpod_pod = child.child_value("pod") self.data.local.user.put_achievement( self.game, self.version, userid, seedpod_id, "seedpod", { "pod": seedpod_pod, }, ) eventexp = request.child("pdata/evntexp") if eventexp: for child in eventexp.children: if child.name != "data": continue eventexp_id = child.child_value("id") eventexp_exp = child.child_value("exp") # Experience is additive, so load it first and add the updated amount data = ( self.data.local.user.get_achievement( self.game, self.version, userid, eventexp_id, "eventexp", ) or ValidatedDict() ) self.data.local.user.put_achievement( self.game, self.version, userid, eventexp_id, "eventexp", { "exp": data.get_int("exp") + eventexp_exp, }, ) eqpexp = request.child("pdata/eqpexp") if eqpexp: for child in eqpexp.children: if child.name != "data": continue eqpexp_id = child.child_value("id") eqpexp_exp = child.child_value("exp") eqpexp_stype = child.child_value("stype") # Experience is additive, so load it first and add the updated amount data = ( self.data.local.user.get_achievement( self.game, self.version, userid, eqpexp_id, f"eqpexp_{eqpexp_stype}", ) or ValidatedDict() ) self.data.local.user.put_achievement( self.game, self.version, userid, eqpexp_id, f"eqpexp_{eqpexp_stype}", { "exp": data.get_int("exp") + eqpexp_exp, }, ) # Grab any new records set during this play session songplays = request.child("pdata/stglog") if songplays: for child in songplays.children: if child.name != "log": continue songid = child.child_value("mid") chart = child.child_value("ng") clear_type = child.child_value("ct") if songid == 0 and chart == 0 and clear_type == -1: # Dummy song save during profile create continue points = child.child_value("sc") achievement_rate = child.child_value("ar") clear_type, combo_type = self.__game_to_db_clear_type(clear_type) combo = child.child_value("cmb") miss_count = child.child_value("jt_ms") self.update_score( userid, songid, chart, points, achievement_rate, clear_type, combo_type, miss_count, combo=combo, ) # Keep track of play statistics self.update_play_statistics(userid) return newprofile