1309 lines
54 KiB
Python
1309 lines
54 KiB
Python
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:
|
||
# <item>
|
||
# <type __type="u8">any</type>
|
||
# <id __type="u16">any</id>
|
||
# <param __type="u16">0-3</param>
|
||
# </item>
|
||
|
||
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
|