# vim: set fileencoding=utf-8 import random import struct from typing import Optional, Dict, Any, List, Tuple from typing_extensions import Final from bemani.backend.iidx.base import IIDXBase from bemani.backend.iidx.course import IIDXCourse from bemani.backend.iidx.pendual import IIDXPendual from bemani.common import ( Profile, ValidatedDict, VersionConstants, BroadcastConstants, Time, ID, ) from bemani.data import Data, UserID from bemani.protocol import Node class IIDXCopula(IIDXCourse, IIDXBase): name: str = "Beatmania IIDX copula" version: int = VersionConstants.IIDX_COPULA GAME_CLTYPE_SINGLE: Final[int] = 0 GAME_CLTYPE_DOUBLE: Final[int] = 1 DAN_STAGES: Final[int] = 4 GAME_CLEAR_STATUS_NO_PLAY: Final[int] = 0 GAME_CLEAR_STATUS_FAILED: Final[int] = 1 GAME_CLEAR_STATUS_ASSIST_CLEAR: Final[int] = 2 GAME_CLEAR_STATUS_EASY_CLEAR: Final[int] = 3 GAME_CLEAR_STATUS_CLEAR: Final[int] = 4 GAME_CLEAR_STATUS_HARD_CLEAR: Final[int] = 5 GAME_CLEAR_STATUS_EX_HARD_CLEAR: Final[int] = 6 GAME_CLEAR_STATUS_FULL_COMBO: Final[int] = 7 GAME_GHOST_TYPE_RIVAL: Final[int] = 1 GAME_GHOST_TYPE_GLOBAL_TOP: Final[int] = 2 GAME_GHOST_TYPE_GLOBAL_AVERAGE: Final[int] = 3 GAME_GHOST_TYPE_LOCAL_TOP: Final[int] = 4 GAME_GHOST_TYPE_LOCAL_AVERAGE: Final[int] = 5 GAME_GHOST_TYPE_DAN_TOP: Final[int] = 6 GAME_GHOST_TYPE_DAN_AVERAGE: Final[int] = 7 GAME_GHOST_TYPE_RIVAL_TOP: Final[int] = 8 GAME_GHOST_TYPE_RIVAL_AVERAGE: Final[int] = 9 GAME_GHOST_LENGTH: Final[int] = 64 GAME_SP_DAN_RANK_7_KYU: Final[int] = 0 GAME_SP_DAN_RANK_6_KYU: Final[int] = 1 GAME_SP_DAN_RANK_5_KYU: Final[int] = 2 GAME_SP_DAN_RANK_4_KYU: Final[int] = 3 GAME_SP_DAN_RANK_3_KYU: Final[int] = 4 GAME_SP_DAN_RANK_2_KYU: Final[int] = 5 GAME_SP_DAN_RANK_1_KYU: Final[int] = 6 GAME_SP_DAN_RANK_1_DAN: Final[int] = 7 GAME_SP_DAN_RANK_2_DAN: Final[int] = 8 GAME_SP_DAN_RANK_3_DAN: Final[int] = 9 GAME_SP_DAN_RANK_4_DAN: Final[int] = 10 GAME_SP_DAN_RANK_5_DAN: Final[int] = 11 GAME_SP_DAN_RANK_6_DAN: Final[int] = 12 GAME_SP_DAN_RANK_7_DAN: Final[int] = 13 GAME_SP_DAN_RANK_8_DAN: Final[int] = 14 GAME_SP_DAN_RANK_9_DAN: Final[int] = 15 GAME_SP_DAN_RANK_10_DAN: Final[int] = 16 GAME_SP_DAN_RANK_CHUDEN: Final[int] = 17 GAME_SP_DAN_RANK_KAIDEN: Final[int] = 18 GAME_DP_DAN_RANK_7_KYU: Final[int] = 0 GAME_DP_DAN_RANK_6_KYU: Final[int] = 1 GAME_DP_DAN_RANK_5_KYU: Final[int] = 2 GAME_DP_DAN_RANK_4_KYU: Final[int] = 3 GAME_DP_DAN_RANK_3_KYU: Final[int] = 4 GAME_DP_DAN_RANK_2_KYU: Final[int] = 5 GAME_DP_DAN_RANK_1_KYU: Final[int] = 6 GAME_DP_DAN_RANK_1_DAN: Final[int] = 7 GAME_DP_DAN_RANK_2_DAN: Final[int] = 8 GAME_DP_DAN_RANK_3_DAN: Final[int] = 9 GAME_DP_DAN_RANK_4_DAN: Final[int] = 10 GAME_DP_DAN_RANK_5_DAN: Final[int] = 11 GAME_DP_DAN_RANK_6_DAN: Final[int] = 12 GAME_DP_DAN_RANK_7_DAN: Final[int] = 13 GAME_DP_DAN_RANK_8_DAN: Final[int] = 14 GAME_DP_DAN_RANK_9_DAN: Final[int] = 15 GAME_DP_DAN_RANK_10_DAN: Final[int] = 16 GAME_DP_DAN_RANK_CHUDEN: Final[int] = 17 GAME_DP_DAN_RANK_KAIDEN: Final[int] = 18 FAVORITE_LIST_LENGTH: Final[int] = 20 GAME_CHART_TYPE_N7: Final[int] = 0 GAME_CHART_TYPE_H7: Final[int] = 1 GAME_CHART_TYPE_A7: Final[int] = 2 GAME_CHART_TYPE_N14: Final[int] = 3 GAME_CHART_TYPE_H14: Final[int] = 4 GAME_CHART_TYPE_A14: Final[int] = 5 GAME_CHART_TYPE_B7: Final[int] = 6 def previous_version(self) -> Optional[IIDXBase]: return IIDXPendual(self.data, self.config, self.model) @classmethod def run_scheduled_work( cls, data: Data, config: Dict[str, Any] ) -> List[Tuple[str, Dict[str, Any]]]: """ Insert dailies into the DB. """ events = [] if data.local.network.should_schedule( cls.game, cls.version, "daily_charts", "daily" ): # Generate a new list of three dailies. start_time, end_time = data.local.network.get_schedule_duration("daily") all_songs = list( set( [ song.id for song in data.local.music.get_all_songs( cls.game, cls.version ) ] ) ) if len(all_songs) >= 3: daily_songs = random.sample(all_songs, 3) data.local.game.put_time_sensitive_settings( cls.game, cls.version, "dailies", { "start_time": start_time, "end_time": end_time, "music": daily_songs, }, ) events.append( ( "iidx_daily_charts", { "version": cls.version, "music": daily_songs, }, ) ) # Mark that we did some actual work here. data.local.network.mark_scheduled( cls.game, cls.version, "daily_charts", "daily" ) return events @classmethod def get_settings(cls) -> Dict[str, Any]: """ Return all of our front-end modifiably settings. """ return { "bools": [ { "name": "Global Shop Ranking", "tip": "Return network-wide ranking instead of shop ranking on results screen.", "category": "game_config", "setting": "global_shop_ranking", }, { "name": "Events In Omnimix", "tip": "Allow events to be enabled at all for Omnimix.", "category": "game_config", "setting": "omnimix_events_enabled", }, ], "ints": [ { "name": "Event Phase", "tip": "Event phase for all players.", "category": "game_config", "setting": "event_phase", "values": { 0: "No Event", 1: "Tokotoko Line", 2: "Mystery Line Phase 1", 3: "Mystery Line Phase 2", 4: "Mystery Line Phase 3", }, }, ], } def db_to_game_status(self, db_status: int) -> int: return { self.CLEAR_STATUS_NO_PLAY: self.GAME_CLEAR_STATUS_NO_PLAY, self.CLEAR_STATUS_FAILED: self.GAME_CLEAR_STATUS_FAILED, self.CLEAR_STATUS_ASSIST_CLEAR: self.GAME_CLEAR_STATUS_ASSIST_CLEAR, self.CLEAR_STATUS_EASY_CLEAR: self.GAME_CLEAR_STATUS_EASY_CLEAR, self.CLEAR_STATUS_CLEAR: self.GAME_CLEAR_STATUS_CLEAR, self.CLEAR_STATUS_HARD_CLEAR: self.GAME_CLEAR_STATUS_HARD_CLEAR, self.CLEAR_STATUS_EX_HARD_CLEAR: self.GAME_CLEAR_STATUS_EX_HARD_CLEAR, self.CLEAR_STATUS_FULL_COMBO: self.GAME_CLEAR_STATUS_FULL_COMBO, }[db_status] def game_to_db_status(self, game_status: int) -> int: return { self.GAME_CLEAR_STATUS_NO_PLAY: self.CLEAR_STATUS_NO_PLAY, self.GAME_CLEAR_STATUS_FAILED: self.CLEAR_STATUS_FAILED, self.GAME_CLEAR_STATUS_ASSIST_CLEAR: self.CLEAR_STATUS_ASSIST_CLEAR, self.GAME_CLEAR_STATUS_EASY_CLEAR: self.CLEAR_STATUS_EASY_CLEAR, self.GAME_CLEAR_STATUS_CLEAR: self.CLEAR_STATUS_CLEAR, self.GAME_CLEAR_STATUS_HARD_CLEAR: self.CLEAR_STATUS_HARD_CLEAR, self.GAME_CLEAR_STATUS_EX_HARD_CLEAR: self.CLEAR_STATUS_EX_HARD_CLEAR, self.GAME_CLEAR_STATUS_FULL_COMBO: self.CLEAR_STATUS_FULL_COMBO, }[game_status] def db_to_game_rank(self, db_dan: int, cltype: int) -> int: # Special case for no DAN rank if db_dan == -1: return -1 if cltype == self.GAME_CLTYPE_SINGLE: return { self.DAN_RANK_7_KYU: self.GAME_SP_DAN_RANK_7_KYU, self.DAN_RANK_6_KYU: self.GAME_SP_DAN_RANK_6_KYU, self.DAN_RANK_5_KYU: self.GAME_SP_DAN_RANK_5_KYU, self.DAN_RANK_4_KYU: self.GAME_SP_DAN_RANK_4_KYU, self.DAN_RANK_3_KYU: self.GAME_SP_DAN_RANK_3_KYU, self.DAN_RANK_2_KYU: self.GAME_SP_DAN_RANK_2_KYU, self.DAN_RANK_1_KYU: self.GAME_SP_DAN_RANK_1_KYU, self.DAN_RANK_1_DAN: self.GAME_SP_DAN_RANK_1_DAN, self.DAN_RANK_2_DAN: self.GAME_SP_DAN_RANK_2_DAN, self.DAN_RANK_3_DAN: self.GAME_SP_DAN_RANK_3_DAN, self.DAN_RANK_4_DAN: self.GAME_SP_DAN_RANK_4_DAN, self.DAN_RANK_5_DAN: self.GAME_SP_DAN_RANK_5_DAN, self.DAN_RANK_6_DAN: self.GAME_SP_DAN_RANK_6_DAN, self.DAN_RANK_7_DAN: self.GAME_SP_DAN_RANK_7_DAN, self.DAN_RANK_8_DAN: self.GAME_SP_DAN_RANK_8_DAN, self.DAN_RANK_9_DAN: self.GAME_SP_DAN_RANK_9_DAN, self.DAN_RANK_10_DAN: self.GAME_SP_DAN_RANK_10_DAN, self.DAN_RANK_CHUDEN: self.GAME_SP_DAN_RANK_CHUDEN, self.DAN_RANK_KAIDEN: self.GAME_SP_DAN_RANK_KAIDEN, }[db_dan] elif cltype == self.GAME_CLTYPE_DOUBLE: return { self.DAN_RANK_7_KYU: self.GAME_DP_DAN_RANK_7_KYU, self.DAN_RANK_6_KYU: self.GAME_DP_DAN_RANK_6_KYU, self.DAN_RANK_5_KYU: self.GAME_DP_DAN_RANK_5_KYU, self.DAN_RANK_4_KYU: self.GAME_DP_DAN_RANK_4_KYU, self.DAN_RANK_3_KYU: self.GAME_DP_DAN_RANK_3_KYU, self.DAN_RANK_2_KYU: self.GAME_DP_DAN_RANK_2_KYU, self.DAN_RANK_1_KYU: self.GAME_DP_DAN_RANK_1_KYU, self.DAN_RANK_1_DAN: self.GAME_DP_DAN_RANK_1_DAN, self.DAN_RANK_2_DAN: self.GAME_DP_DAN_RANK_2_DAN, self.DAN_RANK_3_DAN: self.GAME_DP_DAN_RANK_3_DAN, self.DAN_RANK_4_DAN: self.GAME_DP_DAN_RANK_4_DAN, self.DAN_RANK_5_DAN: self.GAME_DP_DAN_RANK_5_DAN, self.DAN_RANK_6_DAN: self.GAME_DP_DAN_RANK_6_DAN, self.DAN_RANK_7_DAN: self.GAME_DP_DAN_RANK_7_DAN, self.DAN_RANK_8_DAN: self.GAME_DP_DAN_RANK_8_DAN, self.DAN_RANK_9_DAN: self.GAME_DP_DAN_RANK_9_DAN, self.DAN_RANK_10_DAN: self.GAME_DP_DAN_RANK_10_DAN, self.DAN_RANK_CHUDEN: self.GAME_DP_DAN_RANK_CHUDEN, self.DAN_RANK_KAIDEN: self.GAME_DP_DAN_RANK_KAIDEN, }[db_dan] else: raise Exception("Invalid cltype!") def game_to_db_rank(self, game_dan: int, cltype: int) -> int: # Special case for no DAN rank if game_dan == -1: return -1 if cltype == self.GAME_CLTYPE_SINGLE: return { self.GAME_SP_DAN_RANK_7_KYU: self.DAN_RANK_7_KYU, self.GAME_SP_DAN_RANK_6_KYU: self.DAN_RANK_6_KYU, self.GAME_SP_DAN_RANK_5_KYU: self.DAN_RANK_5_KYU, self.GAME_SP_DAN_RANK_4_KYU: self.DAN_RANK_4_KYU, self.GAME_SP_DAN_RANK_3_KYU: self.DAN_RANK_3_KYU, self.GAME_SP_DAN_RANK_2_KYU: self.DAN_RANK_2_KYU, self.GAME_SP_DAN_RANK_1_KYU: self.DAN_RANK_1_KYU, self.GAME_SP_DAN_RANK_1_DAN: self.DAN_RANK_1_DAN, self.GAME_SP_DAN_RANK_2_DAN: self.DAN_RANK_2_DAN, self.GAME_SP_DAN_RANK_3_DAN: self.DAN_RANK_3_DAN, self.GAME_SP_DAN_RANK_4_DAN: self.DAN_RANK_4_DAN, self.GAME_SP_DAN_RANK_5_DAN: self.DAN_RANK_5_DAN, self.GAME_SP_DAN_RANK_6_DAN: self.DAN_RANK_6_DAN, self.GAME_SP_DAN_RANK_7_DAN: self.DAN_RANK_7_DAN, self.GAME_SP_DAN_RANK_8_DAN: self.DAN_RANK_8_DAN, self.GAME_SP_DAN_RANK_9_DAN: self.DAN_RANK_9_DAN, self.GAME_SP_DAN_RANK_10_DAN: self.DAN_RANK_10_DAN, self.GAME_SP_DAN_RANK_CHUDEN: self.DAN_RANK_CHUDEN, self.GAME_SP_DAN_RANK_KAIDEN: self.DAN_RANK_KAIDEN, }[game_dan] elif cltype == self.GAME_CLTYPE_DOUBLE: return { self.GAME_DP_DAN_RANK_7_KYU: self.DAN_RANK_7_KYU, self.GAME_DP_DAN_RANK_6_KYU: self.DAN_RANK_6_KYU, self.GAME_DP_DAN_RANK_5_KYU: self.DAN_RANK_5_KYU, self.GAME_DP_DAN_RANK_4_KYU: self.DAN_RANK_4_KYU, self.GAME_DP_DAN_RANK_3_KYU: self.DAN_RANK_3_KYU, self.GAME_DP_DAN_RANK_2_KYU: self.DAN_RANK_2_KYU, self.GAME_DP_DAN_RANK_1_KYU: self.DAN_RANK_1_KYU, self.GAME_DP_DAN_RANK_1_DAN: self.DAN_RANK_1_DAN, self.GAME_DP_DAN_RANK_2_DAN: self.DAN_RANK_2_DAN, self.GAME_DP_DAN_RANK_3_DAN: self.DAN_RANK_3_DAN, self.GAME_DP_DAN_RANK_4_DAN: self.DAN_RANK_4_DAN, self.GAME_DP_DAN_RANK_5_DAN: self.DAN_RANK_5_DAN, self.GAME_DP_DAN_RANK_6_DAN: self.DAN_RANK_6_DAN, self.GAME_DP_DAN_RANK_7_DAN: self.DAN_RANK_7_DAN, self.GAME_DP_DAN_RANK_8_DAN: self.DAN_RANK_8_DAN, self.GAME_DP_DAN_RANK_9_DAN: self.DAN_RANK_9_DAN, self.GAME_DP_DAN_RANK_10_DAN: self.DAN_RANK_10_DAN, self.GAME_DP_DAN_RANK_CHUDEN: self.DAN_RANK_CHUDEN, self.GAME_DP_DAN_RANK_KAIDEN: self.DAN_RANK_KAIDEN, }[game_dan] else: raise Exception("Invalid cltype!") def game_to_db_chart(self, db_chart: int) -> int: return { self.GAME_CHART_TYPE_B7: self.CHART_TYPE_B7, self.GAME_CHART_TYPE_N7: self.CHART_TYPE_N7, self.GAME_CHART_TYPE_H7: self.CHART_TYPE_H7, self.GAME_CHART_TYPE_A7: self.CHART_TYPE_A7, self.GAME_CHART_TYPE_N14: self.CHART_TYPE_N14, self.GAME_CHART_TYPE_H14: self.CHART_TYPE_H14, self.GAME_CHART_TYPE_A14: self.CHART_TYPE_A14, }[db_chart] def handle_IIDX23shop_getname_request(self, request: Node) -> Node: root = Node.void("IIDX23shop") root.set_attribute("cls_opt", "0") machine = self.data.local.machine.get_machine(self.config.machine.pcbid) root.set_attribute("opname", machine.name) root.set_attribute("pid", str(self.get_machine_region())) return root def handle_IIDX23shop_savename_request(self, request: Node) -> Node: self.update_machine_name(request.attribute("opname")) root = Node.void("IIDX23shop") return root def handle_IIDX23shop_sentinfo_request(self, request: Node) -> Node: root = Node.void("IIDX23shop") return root def handle_IIDX23shop_getconvention_request(self, request: Node) -> Node: root = Node.void("IIDX23shop") machine = self.data.local.machine.get_machine(self.config.machine.pcbid) if machine.arcade is not None: course = self.data.local.machine.get_settings( machine.arcade, self.game, self.music_version, "shop_course" ) else: course = None if course is None: course = ValidatedDict() root.set_attribute("music_0", str(course.get_int("music_0", 20032))) root.set_attribute("music_1", str(course.get_int("music_1", 20009))) root.set_attribute("music_2", str(course.get_int("music_2", 20015))) root.set_attribute("music_3", str(course.get_int("music_3", 20064))) root.add_child(Node.bool("valid", course.get_bool("valid"))) return root def handle_IIDX23shop_setconvention_request(self, request: Node) -> Node: root = Node.void("IIDX23shop") machine = self.data.local.machine.get_machine(self.config.machine.pcbid) if machine.arcade is not None: course = ValidatedDict() course.replace_int("music_0", request.child_value("music_0")) course.replace_int("music_1", request.child_value("music_1")) course.replace_int("music_2", request.child_value("music_2")) course.replace_int("music_3", request.child_value("music_3")) course.replace_bool("valid", request.child_value("valid")) self.data.local.machine.put_settings( machine.arcade, self.game, self.music_version, "shop_course", course ) return root def handle_IIDX23shop_sendescapepackageinfo_request(self, request: Node) -> Node: root = Node.void("IIDX23shop") root.set_attribute("expire", str((Time.now() + 86400 * 365) * 1000)) return root def handle_IIDX23ranking_getranker_request(self, request: Node) -> Node: root = Node.void("IIDX23ranking") chart = self.game_to_db_chart(int(request.attribute("clid"))) if chart not in [ self.CHART_TYPE_N7, self.CHART_TYPE_H7, self.CHART_TYPE_A7, self.CHART_TYPE_N14, self.CHART_TYPE_H14, self.CHART_TYPE_A14, ]: # Chart type 6 is presumably beginner mode, but it crashes the game return root machine = self.data.local.machine.get_machine(self.config.machine.pcbid) if machine.arcade is not None: course = self.data.local.machine.get_settings( machine.arcade, self.game, self.music_version, "shop_course" ) else: course = None if course is None: course = ValidatedDict() if not course.get_bool("valid"): # Shop course not enabled or not present return root convention = Node.void("convention") root.add_child(convention) convention.set_attribute("clid", str(chart)) convention.set_attribute("update_date", str(Time.now() * 1000)) # Grab all scores for each of the four songs, filter out people who haven't # set us as their arcade and then return the top 20 scores (adding all 4 songs). songids = [ course.get_int("music_0"), course.get_int("music_1"), course.get_int("music_2"), course.get_int("music_3"), ] totalscores: Dict[UserID, int] = {} profiles: Dict[UserID, Profile] = {} for songid in songids: scores = self.data.local.music.get_all_scores( self.game, self.music_version, songid=songid, songchart=chart, ) for score in scores: if score[0] not in totalscores: totalscores[score[0]] = 0 profile = self.get_any_profile(score[0]) if profile is None: profile = Profile(self.game, self.version, "", 0) profiles[score[0]] = profile totalscores[score[0]] += score[1].points topscores = sorted( [ (totalscores[userid], profiles[userid]) for userid in totalscores if self.user_joined_arcade(machine, profiles[userid]) ], key=lambda tup: tup[0], reverse=True, )[:20] rank = 0 for topscore in topscores: rank = rank + 1 detail = Node.void("detail") convention.add_child(detail) detail.set_attribute("name", topscore[1].get_str("name")) detail.set_attribute("rank", str(rank)) detail.set_attribute("score", str(topscore[0])) detail.set_attribute("pid", str(topscore[1].get_int("pid"))) qpro = topscore[1].get_dict("qpro") detail.set_attribute("head", str(qpro.get_int("head"))) detail.set_attribute("hair", str(qpro.get_int("hair"))) detail.set_attribute("face", str(qpro.get_int("face"))) detail.set_attribute("body", str(qpro.get_int("body"))) detail.set_attribute("hand", str(qpro.get_int("hand"))) return root def handle_IIDX23ranking_entry_request(self, request: Node) -> Node: extid = int(request.attribute("iidxid")) courseid = int(request.attribute("coid")) chart = self.game_to_db_chart(int(request.attribute("clid"))) course_type = int(request.attribute("regist_type")) clear_status = self.game_to_db_status(int(request.attribute("clr"))) pgreats = int(request.attribute("pgnum")) greats = int(request.attribute("gnum")) if course_type == 0: index = self.COURSE_TYPE_INTERNET_RANKING elif course_type == 1: index = self.COURSE_TYPE_SECRET else: raise Exception("Unknown registration type for course entry!") userid = self.data.remote.user.from_extid(self.game, self.version, extid) if userid is not None: # Update achievement to track course statistics self.update_course( userid, index, courseid, chart, clear_status, pgreats, greats, ) # We should return the user's position, but its not displayed anywhere # so fuck it. root = Node.void("IIDX23ranking") root.set_attribute("anum", "1") root.set_attribute("jun", "1") return root def handle_IIDX23music_crate_request(self, request: Node) -> Node: root = Node.void("IIDX23music") attempts = self.get_clear_rates() all_songs = list( set( [ song.id for song in self.data.local.music.get_all_songs( self.game, self.music_version ) ] ) ) for song in all_songs: clears = [] fcs = [] for chart in [0, 1, 2, 3, 4, 5]: placed = False if song in attempts and chart in attempts[song]: values = attempts[song][chart] if values["total"] > 0: clears.append(int((100 * values["clears"]) / values["total"])) fcs.append(int((100 * values["fcs"]) / values["total"])) placed = True if not placed: clears.append(101) fcs.append(101) clearnode = Node.u8_array("c", clears + fcs) clearnode.set_attribute("mid", str(song)) root.add_child(clearnode) return root def handle_IIDX23music_getrank_request(self, request: Node) -> Node: cltype = int(request.attribute("cltype")) root = Node.void("IIDX23music") style = Node.void("style") root.add_child(style) style.set_attribute("type", str(cltype)) for rivalid in [-1, 0, 1, 2, 3, 4]: if rivalid == -1: attr = "iidxid" else: attr = f"iidxid{rivalid}" try: extid = int(request.attribute(attr)) except Exception: # Invalid extid continue userid = self.data.remote.user.from_extid(self.game, self.version, extid) if userid is not None: scores = self.data.remote.music.get_scores( self.game, self.music_version, userid ) # Grab score data for user/rival scoredata = self.make_score_struct( scores, self.CLEAR_TYPE_SINGLE if cltype == self.GAME_CLTYPE_SINGLE else self.CLEAR_TYPE_DOUBLE, rivalid, ) for s in scoredata: root.add_child(Node.s16_array("m", s)) # Grab most played for user/rival most_played = [ play[0] for play in self.data.local.music.get_most_played( self.game, self.music_version, userid, 20 ) ] if len(most_played) < 20: most_played.extend([0] * (20 - len(most_played))) best = Node.u16_array("best", most_played) best.set_attribute("rno", str(rivalid)) root.add_child(best) if rivalid == -1: # Grab beginner statuses for user only beginnerdata = self.make_beginner_struct(scores) for b in beginnerdata: root.add_child(Node.u16_array("b", b)) return root def handle_IIDX23music_reg_request(self, request: Node) -> Node: extid = int(request.attribute("iidxid")) musicid = int(request.attribute("mid")) chart = self.game_to_db_chart(int(request.attribute("clid"))) userid = self.data.remote.user.from_extid(self.game, self.version, extid) # See if we need to report global or shop scores if self.machine_joined_arcade(): game_config = self.get_game_config() global_scores = game_config.get_bool("global_shop_ranking") machine = self.data.local.machine.get_machine(self.config.machine.pcbid) else: # If we aren't in an arcade, we can only show global scores global_scores = True machine = None # First, determine our current ranking before saving the new score all_scores = sorted( self.data.remote.music.get_all_scores( game=self.game, version=self.music_version, songid=musicid, songchart=chart, ), key=lambda s: (s[1].points, s[1].timestamp), reverse=True, ) all_players = { uid: prof for (uid, prof) in self.get_any_profiles([s[0] for s in all_scores]) } if not global_scores: all_scores = [ score for score in all_scores if ( score[0] == userid or self.user_joined_arcade(machine, all_players[score[0]]) ) ] # Find our actual index oldindex = None for i in range(len(all_scores)): if all_scores[i][0] == userid: oldindex = i break if userid is not None: clear_status = self.game_to_db_status(int(request.attribute("cflg"))) pgreats = int(request.attribute("pgnum")) greats = int(request.attribute("gnum")) miss_count = int(request.attribute("mnum")) ghost = request.child_value("ghost") shopid = ID.parse_machine_id(request.attribute("shopconvid")) self.update_score( userid, musicid, chart, clear_status, pgreats, greats, miss_count, ghost, shopid, ) # Calculate and return statistics about this song root = Node.void("IIDX23music") root.set_attribute("clid", request.attribute("clid")) root.set_attribute("mid", request.attribute("mid")) attempts = self.get_clear_rates(musicid, chart) count = attempts[musicid][chart]["total"] clear = attempts[musicid][chart]["clears"] full_combo = attempts[musicid][chart]["fcs"] if count > 0: root.set_attribute("crate", str(int((100 * clear) / count))) root.set_attribute("frate", str(int((100 * full_combo) / count))) else: root.set_attribute("crate", "0") root.set_attribute("frate", "0") root.set_attribute("rankside", "0") if userid is not None: # Shop ranking shopdata = Node.void("shopdata") root.add_child(shopdata) shopdata.set_attribute( "rank", "-1" if oldindex is None else str(oldindex + 1) ) # Grab the rank of some other players on this song ranklist = Node.void("ranklist") root.add_child(ranklist) all_scores = sorted( self.data.remote.music.get_all_scores( game=self.game, version=self.music_version, songid=musicid, songchart=chart, ), key=lambda s: (s[1].points, s[1].timestamp), reverse=True, ) missing_players = [uid for (uid, _) in all_scores if uid not in all_players] for uid, prof in self.get_any_profiles(missing_players): all_players[uid] = prof if not global_scores: all_scores = [ score for score in all_scores if ( score[0] == userid or self.user_joined_arcade(machine, all_players[score[0]]) ) ] # Find our actual index ourindex = None for i in range(len(all_scores)): if all_scores[i][0] == userid: ourindex = i break if ourindex is None: raise Exception("Cannot find our own score after saving to DB!") start = ourindex - 4 end = ourindex + 4 if start < 0: start = 0 if end >= len(all_scores): end = len(all_scores) - 1 relevant_scores = all_scores[start : (end + 1)] record_num = start + 1 for score in relevant_scores: profile = all_players[score[0]] data = Node.void("data") ranklist.add_child(data) data.set_attribute("iidx_id", str(profile.extid)) data.set_attribute("name", profile.get_str("name")) machine_name = "" if "shop_location" in profile: shop_id = profile.get_int("shop_location") machine = self.get_machine_by_id(shop_id) if machine is not None: machine_name = machine.name data.set_attribute("opname", machine_name) data.set_attribute("rnum", str(record_num)) data.set_attribute("score", str(score[1].points)) data.set_attribute( "clflg", str(self.db_to_game_status(score[1].data.get_int("clear_status"))), ) data.set_attribute("pid", str(profile.get_int("pid"))) data.set_attribute("myFlg", "1" if score[0] == userid else "0") data.set_attribute("update", "0") data.set_attribute( "sgrade", str( self.db_to_game_rank( profile.get_int(self.DAN_RANKING_SINGLE, -1), self.GAME_CLTYPE_SINGLE, ), ), ) data.set_attribute( "dgrade", str( self.db_to_game_rank( profile.get_int(self.DAN_RANKING_DOUBLE, -1), self.GAME_CLTYPE_DOUBLE, ), ), ) qpro = profile.get_dict("qpro") data.set_attribute("head", str(qpro.get_int("head"))) data.set_attribute("hair", str(qpro.get_int("hair"))) data.set_attribute("face", str(qpro.get_int("face"))) data.set_attribute("body", str(qpro.get_int("body"))) data.set_attribute("hand", str(qpro.get_int("hand"))) record_num = record_num + 1 return root def handle_IIDX23music_breg_request(self, request: Node) -> Node: extid = int(request.attribute("iidxid")) musicid = int(request.attribute("mid")) userid = self.data.remote.user.from_extid(self.game, self.version, extid) if userid is not None: clear_status = self.game_to_db_status(int(request.attribute("cflg"))) pgreats = int(request.attribute("pgnum")) greats = int(request.attribute("gnum")) self.update_score( userid, musicid, self.CHART_TYPE_B7, clear_status, pgreats, greats, -1, b"", None, ) # Return nothing. root = Node.void("IIDX23music") return root def handle_IIDX23music_play_request(self, request: Node) -> Node: musicid = int(request.attribute("mid")) chart = self.game_to_db_chart(int(request.attribute("clid"))) clear_status = self.game_to_db_status(int(request.attribute("cflg"))) self.update_score( None, # No userid since its anonymous musicid, chart, clear_status, 0, # No ex score 0, # No ex score 0, # No miss count None, # No ghost None, # No shop for this user ) # Calculate and return statistics about this song root = Node.void("IIDX23music") root.set_attribute("clid", request.attribute("clid")) root.set_attribute("mid", request.attribute("mid")) attempts = self.get_clear_rates(musicid, chart) count = attempts[musicid][chart]["total"] clear = attempts[musicid][chart]["clears"] full_combo = attempts[musicid][chart]["fcs"] if count > 0: root.set_attribute("crate", str(int((100 * clear) / count))) root.set_attribute("frate", str(int((100 * full_combo) / count))) else: root.set_attribute("crate", "0") root.set_attribute("frate", "0") return root def handle_IIDX23music_appoint_request(self, request: Node) -> Node: musicid = int(request.attribute("mid")) chart = self.game_to_db_chart(int(request.attribute("clid"))) ghost_type = int(request.attribute("ctype")) extid = int(request.attribute("iidxid")) userid = self.data.remote.user.from_extid(self.game, self.version, extid) root = Node.void("IIDX23music") if userid is not None: # Try to look up previous ghost for user my_score = self.data.remote.music.get_score( self.game, self.music_version, userid, musicid, chart ) if my_score is not None: mydata = Node.binary("mydata", my_score.data.get_bytes("ghost")) mydata.set_attribute("score", str(my_score.points)) root.add_child(mydata) ghost_score = self.get_ghost( { self.GAME_GHOST_TYPE_RIVAL: self.GHOST_TYPE_RIVAL, self.GAME_GHOST_TYPE_GLOBAL_TOP: self.GHOST_TYPE_GLOBAL_TOP, self.GAME_GHOST_TYPE_GLOBAL_AVERAGE: self.GHOST_TYPE_GLOBAL_AVERAGE, self.GAME_GHOST_TYPE_LOCAL_TOP: self.GHOST_TYPE_LOCAL_TOP, self.GAME_GHOST_TYPE_LOCAL_AVERAGE: self.GHOST_TYPE_LOCAL_AVERAGE, self.GAME_GHOST_TYPE_DAN_TOP: self.GHOST_TYPE_DAN_TOP, self.GAME_GHOST_TYPE_DAN_AVERAGE: self.GHOST_TYPE_DAN_AVERAGE, self.GAME_GHOST_TYPE_RIVAL_TOP: self.GHOST_TYPE_RIVAL_TOP, self.GAME_GHOST_TYPE_RIVAL_AVERAGE: self.GHOST_TYPE_RIVAL_AVERAGE, }.get(ghost_type, self.GHOST_TYPE_NONE), request.attribute("subtype"), self.GAME_GHOST_LENGTH, musicid, chart, userid, ) # Add ghost score if we support it if ghost_score is not None: sdata = Node.binary("sdata", ghost_score["ghost"]) sdata.set_attribute("score", str(ghost_score["score"])) if "name" in ghost_score: sdata.set_attribute("name", ghost_score["name"]) if "pid" in ghost_score: sdata.set_attribute("pid", str(ghost_score["pid"])) if "extid" in ghost_score: sdata.set_attribute("riidxid", str(ghost_score["extid"])) root.add_child(sdata) return root def handle_IIDX23grade_raised_request(self, request: Node) -> Node: extid = int(request.attribute("iidxid")) cltype = int(request.attribute("gtype")) rank = self.game_to_db_rank(int(request.attribute("gid")), cltype) userid = self.data.remote.user.from_extid(self.game, self.version, extid) if userid is not None: percent = int(request.attribute("achi")) stages_cleared = int(request.attribute("cstage")) cleared = stages_cleared == self.DAN_STAGES if cltype == self.GAME_CLTYPE_SINGLE: index = self.DAN_RANKING_SINGLE else: index = self.DAN_RANKING_DOUBLE self.update_rank( userid, index, rank, percent, cleared, stages_cleared, ) # Figure out number of players that played this ranking all_achievements = self.data.local.user.get_all_achievements( self.game, self.version, achievementid=rank, achievementtype=index ) root = Node.void("IIDX23grade") root.set_attribute("pnum", str(len(all_achievements))) return root def handle_IIDX23pc_common_request(self, request: Node) -> Node: root = Node.void("IIDX23pc") root.set_attribute("expire", "600") ir = Node.void("ir") root.add_child(ir) ir.set_attribute("beat", "2") # See if we configured event overrides if self.machine_joined_arcade(): game_config = self.get_game_config() event_phase = game_config.get_int("event_phase") omni_events = game_config.get_bool("omnimix_events_enabled") else: # If we aren't in an arcade, we turn off events event_phase = 0 omni_events = False if event_phase == 0 or (self.omnimix and (not omni_events)): boss_phase = 0 event1 = 0 event2 = 0 elif event_phase == 1: boss_phase = 1 event1 = 1 event2 = 0 elif event_phase in [2, 3, 4]: boss_phase = 2 event1 = 0 event2 = event_phase - 1 boss = Node.void("boss") root.add_child(boss) boss.set_attribute("phase", str(boss_phase)) event1_phase = Node.void("event1_phase") root.add_child(event1_phase) event1_phase.set_attribute("phase", str(event1)) event2_phase = Node.void("event2_phase") root.add_child(event2_phase) event2_phase.set_attribute("phase", str(event2)) extra_boss_event = Node.void("extra_boss_event") root.add_child(extra_boss_event) extra_boss_event.set_attribute("phase", "1") bemani_summer2016 = Node.void("bemani_summer2016") root.add_child(bemani_summer2016) bemani_summer2016.set_attribute("phase", "1") vip_black_pass = Node.void("vip_pass_black") root.add_child(vip_black_pass) event1_rainbow_ticket = Node.void("event1_rainbow_ticket") root.add_child(event1_rainbow_ticket) djlevel_result = Node.void("djlevel_result") root.add_child(djlevel_result) newsong_another = Node.void("newsong_another") root.add_child(newsong_another) newsong_another.set_attribute("open", "1") # Course definitions courses: List[Dict[str, Any]] = [ { "name": "POP", "id": 1, "songs": [ 23034, 23012, 23011, 23032, ], }, { "name": "TRANCE", "id": 2, "songs": [ 23014, 23033, 23038, 23013, ], }, { "name": "DJ", "id": 3, "songs": [ 23025, 23026, 23024, 23061, ], }, { "name": "HCN", "id": 4, "songs": [ 23016, 23010, 23057, 23000, ], }, { "name": "TRAIN", "id": 5, "songs": [ 23023, 6029, 23047, 7008, ], }, { "name": "USAO", "id": 6, "songs": [ 20064, 21015, 22044, 23027, ], }, { "name": "New Face", "id": 7, "songs": [ 23054, 23039, 23036, 23045, ], }, { "name": "CANDY", "id": 8, "songs": [ 16021, 18069, 19073, 23008, ], }, { "name": "ROCK", "id": 9, "songs": [ 8026, 23002, 22072, 23050, ], }, { "name": "JAZZ", "id": 10, "songs": [ 1017, 20088, 23035, 23030, ], }, { "name": "TAIYO", "id": 11, "songs": [ 11007, 23020, 13029, 23051, ], }, { "name": "RHYZE", "id": 12, "songs": [ 23029, 21070, 21071, 20063, ], }, { "name": "COLLABORATION", "id": 13, "songs": [ 20094, 20008, 20020, 21082, ], }, ] # Secret course definitions secret_courses: List[Dict[str, Any]] = [ { "name": "COLORS", "id": 1, "songs": [ 20038, 20012, 20007, 22012, ], }, { "name": "BROKEN", "id": 2, "songs": [ 4003, 18028, 18068, 22001, ], }, { "name": "PENDUAL", "id": 3, "songs": [ 22013, 22008, 22054, 22100, ], }, { "name": "SYMMETRY", "id": 4, "songs": [ 9052, 10024, 12054, 22017, ], }, { "name": "SEVEN", "id": 5, "songs": [ 13014, 11014, 17059, 22011, ], }, { "name": "RAVE", "id": 6, "songs": [ 21051, 22078, 21083, 23078, ], }, { "name": "P*Light", "id": 7, "songs": [ 22061, 23031, 22080, 23079, ], }, { "name": "GRAND FINAL", "id": 8, "songs": [ 14021, 21032, 21045, 23075, ], }, { "name": "SAY RYU", "id": 9, "songs": [ 13038, 15026, 21007, 23082, ], }, { "name": "SUMMER", "id": 10, "songs": [ 15048, 18005, 18021, 23091, ], }, { "name": "ART CORE", "id": 11, "songs": [ 10023, 22051, 20097, 23090, ], }, { "name": "HAPPY", "id": 12, "songs": [ 11036, 19070, 20040, 23093, ], }, { "name": "TAG", "id": 13, "songs": [ 21058, 20015, 18056, 23095, ], }, ] # For some reason, copula omnimix crashes on course mode, so don't enable it if not self.omnimix: internet_ranking = Node.void("internet_ranking") root.add_child(internet_ranking) used_ids: List[int] = [] for c in courses: if c["id"] in used_ids: raise Exception("Cannot have multiple courses with the same ID!") elif c["id"] < 0 or c["id"] >= 20: raise Exception("Course ID is out of bounds!") else: used_ids.append(c["id"]) course = Node.void("course") internet_ranking.add_child(course) course.set_attribute("opflg", "1") course.set_attribute("course_id", str(c["id"])) course.set_attribute("mid0", str(c["songs"][0])) course.set_attribute("mid1", str(c["songs"][1])) course.set_attribute("mid2", str(c["songs"][2])) course.set_attribute("mid3", str(c["songs"][3])) course.set_attribute("name", c["name"]) secret_ex_course = Node.void("secret_ex_course") root.add_child(secret_ex_course) used_secret_ids: List[int] = [] for c in secret_courses: if c["id"] in used_secret_ids: raise Exception( "Cannot have multiple secret courses with the same ID!" ) elif c["id"] < 0 or c["id"] >= 20: raise Exception("Secret course ID is out of bounds!") else: used_secret_ids.append(c["id"]) course = Node.void("course") secret_ex_course.add_child(course) course.set_attribute("course_id", str(c["id"])) course.set_attribute("mid0", str(c["songs"][0])) course.set_attribute("mid1", str(c["songs"][1])) course.set_attribute("mid2", str(c["songs"][2])) course.set_attribute("mid3", str(c["songs"][3])) course.set_attribute("name", c["name"]) expert = Node.void("expert") root.add_child(expert) expert.set_attribute("phase", "1") expert_random_select = Node.void("expert_random_select") root.add_child(expert_random_select) expert_random_select.set_attribute("phase", "1") expert_full = Node.void("expert_secret_full_open") root.add_child(expert_full) return root def handle_IIDX23pc_delete_request(self, request: Node) -> Node: return Node.void("IIDX23pc") def handle_IIDX23pc_playstart_request(self, request: Node) -> Node: return Node.void("IIDX23pc") def handle_IIDX23pc_playend_request(self, request: Node) -> Node: return Node.void("IIDX23pc") def handle_IIDX23pc_oldget_request(self, request: Node) -> Node: refid = request.attribute("rid") userid = self.data.remote.user.from_refid(self.game, self.version, refid) if userid is not None: oldversion = self.previous_version() profile = oldversion.get_profile(userid) else: profile = None root = Node.void("IIDX23pc") root.set_attribute("status", "1" if profile is None else "0") return root def handle_IIDX23pc_getname_request(self, request: Node) -> Node: refid = request.attribute("rid") userid = self.data.remote.user.from_refid(self.game, self.version, refid) if userid is not None: oldversion = self.previous_version() profile = oldversion.get_profile(userid) else: profile = None if profile is None: raise Exception( "Should not get here if we have no profile, we should " + "have returned '1' in the 'oldget' method above " + "which should tell the game not to present a migration." ) root = Node.void("IIDX23pc") root.set_attribute("name", profile.get_str("name")) root.set_attribute("idstr", ID.format_extid(profile.extid)) root.set_attribute("pid", str(profile.get_int("pid"))) return root def handle_IIDX23pc_takeover_request(self, request: Node) -> Node: refid = request.attribute("rid") name = request.attribute("name") pid = int(request.attribute("pid")) newprofile = self.new_profile_by_refid(refid, name, pid) root = Node.void("IIDX23pc") if newprofile is not None: root.set_attribute("id", str(newprofile.extid)) return root def handle_IIDX23pc_reg_request(self, request: Node) -> Node: refid = request.attribute("rid") name = request.attribute("name") pid = int(request.attribute("pid")) profile = self.new_profile_by_refid(refid, name, pid) root = Node.void("IIDX23pc") if profile is not None: root.set_attribute("id", str(profile.extid)) root.set_attribute("id_str", ID.format_extid(profile.extid)) return root def handle_IIDX23pc_get_request(self, request: Node) -> Node: refid = request.attribute("rid") root = self.get_profile_by_refid(refid) if root is None: root = Node.void("IIDX23pc") return root def handle_IIDX23pc_save_request(self, request: Node) -> Node: extid = int(request.attribute("iidxid")) self.put_profile_by_extid(extid, request) root = Node.void("IIDX23pc") return root def handle_IIDX23pc_visit_request(self, request: Node) -> Node: root = Node.void("IIDX23pc") root.set_attribute("anum", "0") root.set_attribute("pnum", "0") root.set_attribute("sflg", "0") root.set_attribute("pflg", "0") root.set_attribute("aflg", "0") root.set_attribute("snum", "0") return root def handle_IIDX23pc_shopregister_request(self, request: Node) -> Node: extid = int(request.child_value("iidx_id")) location = ID.parse_machine_id(request.child_value("location_id")) userid = self.data.remote.user.from_extid(self.game, self.version, extid) if userid is not None: profile = self.get_profile(userid) if profile is None: profile = Profile(self.game, self.version, "", extid) profile.replace_int("shop_location", location) self.put_profile(userid, profile) root = Node.void("IIDX23pc") return root def handle_IIDX23pc_eaappliresult_request(self, request: Node) -> Node: clear_map = { self.GAME_CLEAR_STATUS_NO_PLAY: "NO PLAY", self.GAME_CLEAR_STATUS_FAILED: "FAILED", self.GAME_CLEAR_STATUS_ASSIST_CLEAR: "ASSIST CLEAR", self.GAME_CLEAR_STATUS_EASY_CLEAR: "EASY CLEAR", self.GAME_CLEAR_STATUS_CLEAR: "CLEAR", self.GAME_CLEAR_STATUS_HARD_CLEAR: "HARD CLEAR", self.GAME_CLEAR_STATUS_EX_HARD_CLEAR: "EX HARD CLEAR", self.GAME_CLEAR_STATUS_FULL_COMBO: "FULL COMBO", } # first we'll grab the data from the packet # did = request.child_value('did') # rid = request.child_value('rid') name = request.child_value("name") # qpro_hair = request.child_value('qpro_hair') # qpro_head = request.child_value('qpro_head') # qpro_body = request.child_value('qpro_body') # qpro_hand = request.child_value('qpro_hand') music_id = request.child_value("music_id") class_id = request.child_value("class_id") # no_save = request.child_value('no_save') # is_couple = request.child_value('is_couple') # target_graph = request.child_value('target_graph') target_exscore = request.child_value("target_exscore") # pacemaker = request.child_value('pacemaker') best_clear = request.child_value("best_clear") # best_djlevel = request.child_value('best_djlevel') # best_exscore = request.child_value('best_exscore') # best_misscount = request.child_value('best_misscount') now_clear = request.child_value("now_clear") # now_djlevel = request.child_value('now_djlevel') now_exscore = request.child_value("now_exscore") # now_misscount = request.child_value('now_misscount') now_pgreat = request.child_value("now_pgreat") now_great = request.child_value("now_great") now_good = request.child_value("now_good") now_bad = request.child_value("now_bad") now_poor = request.child_value("now_poor") now_combo = request.child_value("now_combo") now_fast = request.child_value("now_fast") now_slow = request.child_value("now_slow") best_clear_string = clear_map.get(best_clear, "NO PLAY") now_clear_string = clear_map.get(now_clear, "NO PLAY") # let's get the song info first song = self.data.local.music.get_song( self.game, self.music_version, music_id, self.game_to_db_chart(class_id) ) notecount = song.data.get("notecount", 0) # Construct the dictionary for the broadcast card_data = { BroadcastConstants.DJ_NAME: name, BroadcastConstants.SONG_NAME: song.name, BroadcastConstants.ARTIST_NAME: song.artist, BroadcastConstants.DIFFICULTY: song.data.get("difficulty", 0), BroadcastConstants.TARGET_EXSCORE: target_exscore, BroadcastConstants.EXSCORE: now_exscore, BroadcastConstants.BEST_CLEAR_STATUS: best_clear_string, BroadcastConstants.CLEAR_STATUS: now_clear_string, BroadcastConstants.PLAY_STATS_HEADER: "How did you do?", BroadcastConstants.PERFECT_GREATS: now_pgreat, BroadcastConstants.GREATS: now_great, BroadcastConstants.GOODS: now_good, BroadcastConstants.BADS: now_bad, BroadcastConstants.POORS: now_poor, BroadcastConstants.COMBO_BREAKS: now_combo, BroadcastConstants.SLOWS: now_slow, BroadcastConstants.FASTS: now_fast, } if notecount != 0: max_score = notecount * 2 percent = now_exscore / max_score grade = int(9 * percent) grades = ["F", "F", "E", "D", "C", "B", "A", "AA", "AAA", "MAX"] card_data[BroadcastConstants.GRADE] = grades[grade] card_data[BroadcastConstants.RATE] = str(round(percent, 2)) # Try to broadcast out the score to our webhook(s) self.data.triggers.broadcast_score(card_data, self.game, song) return Node.void("IIDX23pc") def format_profile(self, userid: UserID, profile: Profile) -> Node: root = Node.void("IIDX23pc") # Look up play stats we bridge to every mix play_stats = self.get_play_statistics(userid) # Look up judge window adjustments judge_dict = profile.get_dict("machine_judge_adjust") machine_judge = judge_dict.get_dict(self.config.machine.pcbid) # Profile data pcdata = Node.void("pcdata") root.add_child(pcdata) pcdata.set_attribute("id", str(profile.extid)) pcdata.set_attribute("idstr", ID.format_extid(profile.extid)) pcdata.set_attribute("name", profile.get_str("name")) pcdata.set_attribute("pid", str(profile.get_int("pid"))) pcdata.set_attribute("spnum", str(play_stats.get_int("single_plays"))) pcdata.set_attribute("dpnum", str(play_stats.get_int("double_plays"))) pcdata.set_attribute("sach", str(play_stats.get_int("single_dj_points"))) pcdata.set_attribute("dach", str(play_stats.get_int("double_dj_points"))) pcdata.set_attribute("mode", str(profile.get_int("mode"))) pcdata.set_attribute("pmode", str(profile.get_int("pmode"))) pcdata.set_attribute("rtype", str(profile.get_int("rtype"))) pcdata.set_attribute("sp_opt", str(profile.get_int("sp_opt"))) pcdata.set_attribute("dp_opt", str(profile.get_int("dp_opt"))) pcdata.set_attribute("dp_opt2", str(profile.get_int("dp_opt2"))) pcdata.set_attribute("gpos", str(profile.get_int("gpos"))) pcdata.set_attribute("s_sorttype", str(profile.get_int("s_sorttype"))) pcdata.set_attribute("d_sorttype", str(profile.get_int("d_sorttype"))) pcdata.set_attribute("s_pace", str(profile.get_int("s_pace"))) pcdata.set_attribute("d_pace", str(profile.get_int("d_pace"))) pcdata.set_attribute("s_gno", str(profile.get_int("s_gno"))) pcdata.set_attribute("d_gno", str(profile.get_int("d_gno"))) pcdata.set_attribute("s_gtype", str(profile.get_int("s_gtype"))) pcdata.set_attribute("d_gtype", str(profile.get_int("d_gtype"))) pcdata.set_attribute("s_sdlen", str(profile.get_int("s_sdlen"))) pcdata.set_attribute("d_sdlen", str(profile.get_int("d_sdlen"))) pcdata.set_attribute("s_sdtype", str(profile.get_int("s_sdtype"))) pcdata.set_attribute("d_sdtype", str(profile.get_int("d_sdtype"))) pcdata.set_attribute("s_timing", str(profile.get_int("s_timing"))) pcdata.set_attribute("d_timing", str(profile.get_int("d_timing"))) pcdata.set_attribute("s_notes", str(profile.get_float("s_notes"))) pcdata.set_attribute("d_notes", str(profile.get_float("d_notes"))) pcdata.set_attribute("s_judge", str(profile.get_int("s_judge"))) pcdata.set_attribute("d_judge", str(profile.get_int("d_judge"))) pcdata.set_attribute("s_judgeAdj", str(machine_judge.get_int("single"))) pcdata.set_attribute("d_judgeAdj", str(machine_judge.get_int("double"))) pcdata.set_attribute("s_hispeed", str(profile.get_float("s_hispeed"))) pcdata.set_attribute("d_hispeed", str(profile.get_float("d_hispeed"))) pcdata.set_attribute("s_liflen", str(profile.get_int("s_lift"))) pcdata.set_attribute("d_liflen", str(profile.get_int("d_lift"))) pcdata.set_attribute("s_disp_judge", str(profile.get_int("s_disp_judge"))) pcdata.set_attribute("d_disp_judge", str(profile.get_int("d_disp_judge"))) pcdata.set_attribute("s_opstyle", str(profile.get_int("s_opstyle"))) pcdata.set_attribute("d_opstyle", str(profile.get_int("d_opstyle"))) pcdata.set_attribute("s_exscore", str(profile.get_int("s_exscore"))) pcdata.set_attribute("d_exscore", str(profile.get_int("d_exscore"))) pcdata.set_attribute("s_largejudge", str(profile.get_int("s_largejudge"))) pcdata.set_attribute("d_largejudge", str(profile.get_int("d_largejudge"))) premium_unlocks = Node.void("ea_premium_course") root.add_child(premium_unlocks) # Secret flags (shh!) secret_dict = profile.get_dict("secret") secret = Node.void("secret") root.add_child(secret) secret.add_child(Node.s64_array("flg1", secret_dict.get_int_array("flg1", 4))) secret.add_child(Node.s64_array("flg2", secret_dict.get_int_array("flg2", 4))) secret.add_child(Node.s64_array("flg3", secret_dict.get_int_array("flg3", 4))) # Favorites for folder in ["favorite1", "favorite2", "favorite3"]: favorite_dict = profile.get_dict(folder) sp_mlist = b"" sp_clist = b"" singles_list = favorite_dict["single"] if "single" in favorite_dict else [] for single in singles_list: sp_mlist = sp_mlist + struct.pack(" Profile: newprofile = oldprofile.clone() play_stats = self.get_play_statistics(userid) # Track play counts cltype = int(request.attribute("cltype")) if cltype == self.GAME_CLTYPE_SINGLE: play_stats.increment_int("single_plays") if cltype == self.GAME_CLTYPE_DOUBLE: play_stats.increment_int("double_plays") # Track DJ points play_stats.replace_int("single_dj_points", int(request.attribute("s_achi"))) play_stats.replace_int("double_dj_points", int(request.attribute("d_achi"))) # Profile settings newprofile.replace_int("mode", int(request.attribute("mode"))) newprofile.replace_int("pmode", int(request.attribute("pmode"))) newprofile.replace_int("rtype", int(request.attribute("rtype"))) newprofile.replace_int("s_lift", int(request.attribute("s_lift"))) newprofile.replace_int("d_lift", int(request.attribute("d_lift"))) newprofile.replace_int("sp_opt", int(request.attribute("sp_opt"))) newprofile.replace_int("dp_opt", int(request.attribute("dp_opt"))) newprofile.replace_int("dp_opt2", int(request.attribute("dp_opt2"))) newprofile.replace_int("gpos", int(request.attribute("gpos"))) newprofile.replace_int("s_sorttype", int(request.attribute("s_sorttype"))) newprofile.replace_int("d_sorttype", int(request.attribute("d_sorttype"))) newprofile.replace_int("s_pace", int(request.attribute("s_pace"))) newprofile.replace_int("d_pace", int(request.attribute("d_pace"))) newprofile.replace_int("s_gno", int(request.attribute("s_gno"))) newprofile.replace_int("d_gno", int(request.attribute("d_gno"))) newprofile.replace_int("s_gtype", int(request.attribute("s_gtype"))) newprofile.replace_int("d_gtype", int(request.attribute("d_gtype"))) newprofile.replace_int("s_sdlen", int(request.attribute("s_sdlen"))) newprofile.replace_int("d_sdlen", int(request.attribute("d_sdlen"))) newprofile.replace_int("s_sdtype", int(request.attribute("s_sdtype"))) newprofile.replace_int("d_sdtype", int(request.attribute("d_sdtype"))) newprofile.replace_int("s_timing", int(request.attribute("s_timing"))) newprofile.replace_int("d_timing", int(request.attribute("d_timing"))) newprofile.replace_float("s_notes", float(request.attribute("s_notes"))) newprofile.replace_float("d_notes", float(request.attribute("d_notes"))) newprofile.replace_int("s_judge", int(request.attribute("s_judge"))) newprofile.replace_int("d_judge", int(request.attribute("d_judge"))) newprofile.replace_float("s_hispeed", float(request.attribute("s_hispeed"))) newprofile.replace_float("d_hispeed", float(request.attribute("d_hispeed"))) newprofile.replace_int("s_disp_judge", int(request.attribute("s_disp_judge"))) newprofile.replace_int("d_disp_judge", int(request.attribute("d_disp_judge"))) newprofile.replace_int("s_opstyle", int(request.attribute("s_opstyle"))) newprofile.replace_int("d_opstyle", int(request.attribute("d_opstyle"))) newprofile.replace_int("s_exscore", int(request.attribute("s_exscore"))) newprofile.replace_int("d_exscore", int(request.attribute("d_exscore"))) newprofile.replace_int("s_largejudge", int(request.attribute("s_largejudge"))) newprofile.replace_int("d_largejudge", int(request.attribute("d_largejudge"))) # Update judge window adjustments per-machine judge_dict = newprofile.get_dict("machine_judge_adjust") machine_judge = judge_dict.get_dict(self.config.machine.pcbid) machine_judge.replace_int("single", int(request.attribute("s_judgeAdj"))) machine_judge.replace_int("double", int(request.attribute("d_judgeAdj"))) judge_dict.replace_dict(self.config.machine.pcbid, machine_judge) newprofile.replace_dict("machine_judge_adjust", judge_dict) # Secret flags saving secret = request.child("secret") if secret is not None: secret_dict = newprofile.get_dict("secret") secret_dict.replace_int_array("flg1", 4, secret.child_value("flg1")) secret_dict.replace_int_array("flg2", 4, secret.child_value("flg2")) secret_dict.replace_int_array("flg3", 4, secret.child_value("flg3")) newprofile.replace_dict("secret", secret_dict) # Basic achievements achievements = request.child("achievements") if achievements is not None: newprofile.replace_int( "visit_flg", int(achievements.attribute("visit_flg")) ) newprofile.replace_int( "last_weekly", int(achievements.attribute("last_weekly")) ) newprofile.replace_int( "weekly_num", int(achievements.attribute("weekly_num")) ) pack_id = int(achievements.attribute("pack_id")) if pack_id > 0: self.data.local.user.put_achievement( self.game, self.version, userid, pack_id, "daily", { "pack_flg": int(achievements.attribute("pack_flg")), "pack_comp": int(achievements.attribute("pack_comp")), }, ) trophies = achievements.child("trophy") if trophies is not None: # We only load the first 10 in profile load. newprofile.replace_int_array("trophy", 10, trophies.value[:10]) # Deller saving deller = request.child("deller") if deller is not None: newprofile.replace_int( "deller", newprofile.get_int("deller") + int(deller.attribute("deller")) ) # Secret course expert point saving expert_point = request.child("expert_point") if expert_point is not None: courseid = int(expert_point.attribute("course_id")) # Update achievement to track expert points expert_point_achievement = self.data.local.user.get_achievement( self.game, self.version, userid, courseid, "expert_point", ) if expert_point_achievement is None: expert_point_achievement = ValidatedDict() expert_point_achievement.replace_int( "normal_points", int(expert_point.attribute("n_point")), ) expert_point_achievement.replace_int( "hyper_points", int(expert_point.attribute("h_point")), ) expert_point_achievement.replace_int( "another_points", int(expert_point.attribute("a_point")), ) self.data.local.user.put_achievement( self.game, self.version, userid, courseid, "expert_point", expert_point_achievement, ) # Favorites saving for favorite in request.children: singles = [] doubles = [] name = None if favorite.name in ["favorite", "extra_favorite"]: if favorite.name == "favorite": name = "favorite1" elif favorite.name == "extra_favorite": folder = favorite.attribute("folder_id") if folder == "0": name = "favorite2" if folder == "1": name = "favorite3" if name is None: continue single_music_bin = favorite.child_value("sp_mlist") single_chart_bin = favorite.child_value("sp_clist") double_music_bin = favorite.child_value("dp_mlist") double_chart_bin = favorite.child_value("dp_clist") for i in range(self.FAVORITE_LIST_LENGTH): singles.append( { "id": struct.unpack( "