961 lines
41 KiB
Python
961 lines
41 KiB
Python
import copy
|
|
from typing import Optional, Dict, Any, Tuple
|
|
|
|
from bemani.backend.reflec.base import ReflecBeatBase
|
|
from bemani.backend.reflec.reflecbeat import ReflecBeat
|
|
|
|
from bemani.common import ValidatedDict, VersionConstants, ID, Time
|
|
from bemani.data import UserID
|
|
from bemani.protocol import Node
|
|
|
|
|
|
class ReflecBeatLimelight(ReflecBeatBase):
|
|
|
|
name = "REFLEC BEAT limelight"
|
|
version = VersionConstants.REFLEC_BEAT_LIMELIGHT
|
|
|
|
# Clear types according to the game
|
|
GAME_CLEAR_TYPE_NO_PLAY = 0
|
|
GAME_CLEAR_TYPE_FAILED = 2
|
|
GAME_CLEAR_TYPE_CLEARED = 3
|
|
GAME_CLEAR_TYPE_FULL_COMBO = 4
|
|
|
|
def previous_version(self) -> Optional[ReflecBeatBase]:
|
|
return ReflecBeat(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 in [
|
|
self.COMBO_TYPE_NONE,
|
|
self.COMBO_TYPE_ALMOST_COMBO,
|
|
]:
|
|
return self.GAME_CLEAR_TYPE_CLEARED
|
|
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_FULL_COMBO:
|
|
return (self.CLEAR_TYPE_CLEARED, self.COMBO_TYPE_FULL_COMBO)
|
|
|
|
raise Exception(f'Invalid game_clear_type {game_clear_type}')
|
|
|
|
def handle_log_exception_request(self, request: Node) -> Node:
|
|
return Node.void('log')
|
|
|
|
def handle_log_pcb_status_request(self, request: Node) -> Node:
|
|
return Node.void('log')
|
|
|
|
def handle_log_opsetting_request(self, request: Node) -> Node:
|
|
return Node.void('log')
|
|
|
|
def handle_log_play_request(self, request: Node) -> Node:
|
|
return Node.void('log')
|
|
|
|
def handle_pcbinfo_get_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')
|
|
pref = machine.data.get_int('pref', 51)
|
|
else:
|
|
machine_name = ''
|
|
close = False
|
|
hour = 0
|
|
minute = 0
|
|
pref = 51
|
|
|
|
root = Node.void('pcbinfo')
|
|
info = Node.void('info')
|
|
root.add_child(info)
|
|
|
|
info.add_child(Node.string('name', machine_name))
|
|
info.add_child(Node.s16('pref', pref))
|
|
info.add_child(Node.bool('close', close))
|
|
info.add_child(Node.u8('hour', hour))
|
|
info.add_child(Node.u8('min', minute))
|
|
|
|
return root
|
|
|
|
def handle_pcbinfo_set_request(self, request: Node) -> Node:
|
|
self.update_machine_name(request.child_value('info/name'))
|
|
self.update_machine_data({
|
|
'close': request.child_value('info/close'),
|
|
'hour': request.child_value('info/hour'),
|
|
'minute': request.child_value('info/min'),
|
|
'pref': request.child_value('info/pref'),
|
|
})
|
|
return Node.void('pcbinfo')
|
|
|
|
def __add_event_info(self, request: Node) -> None:
|
|
events: Dict[int, int] = {}
|
|
|
|
for (_eventid, _phase) in events.items():
|
|
data = Node.void('data')
|
|
request.add_child(data)
|
|
data.add_child(Node.s32('type', -1))
|
|
data.add_child(Node.s32('value', -1))
|
|
|
|
def handle_sysinfo_get_request(self, request: Node) -> Node:
|
|
root = Node.void('sysinfo')
|
|
trd = Node.void('trd')
|
|
root.add_child(trd)
|
|
|
|
# Add event info
|
|
self.__add_event_info(trd)
|
|
|
|
return root
|
|
|
|
def handle_ranking_read_request(self, request: Node) -> Node:
|
|
root = Node.void('ranking')
|
|
|
|
licenses = Node.void('lic_10')
|
|
root.add_child(licenses)
|
|
originals = Node.void('org_10')
|
|
root.add_child(originals)
|
|
|
|
licenses.add_child(Node.time('time', Time.now()))
|
|
originals.add_child(Node.time('time', Time.now()))
|
|
|
|
hitchart = self.data.local.music.get_hit_chart(self.game, self.version, 10)
|
|
rank = 1
|
|
for (mid, _plays) in hitchart:
|
|
record = Node.void('record')
|
|
originals.add_child(record)
|
|
record.add_child(Node.s16('id', mid))
|
|
record.add_child(Node.s16('rank', rank))
|
|
rank = rank + 1
|
|
|
|
return root
|
|
|
|
def handle_event_r_get_all_request(self, request: Node) -> Node:
|
|
limit = request.child_value('limit')
|
|
|
|
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)
|
|
statuses = self.data.local.lobby.get_all_play_session_infos(self.game, self.version)
|
|
statuses.sort(key=lambda x: x[1]['time'], reverse=True)
|
|
|
|
# Cap all comment blocks to the limit
|
|
if limit >= 0:
|
|
comments = comments[:limit]
|
|
statuses = statuses[:limit]
|
|
|
|
# Mapping of profiles to userIDs
|
|
uid_mapping = {
|
|
uid: prof for (uid, prof) in self.get_any_profiles(
|
|
[c[0] for c in comments] +
|
|
[s[0] for s in statuses]
|
|
)
|
|
}
|
|
|
|
# Mapping of location ID to machine name
|
|
lid_mapping: Dict[int, str] = {}
|
|
|
|
root = Node.void('event_r')
|
|
root.add_child(Node.s32('time', Time.now()))
|
|
statusnode = Node.void('status')
|
|
root.add_child(statusnode)
|
|
commentnode = Node.void('comment')
|
|
root.add_child(commentnode)
|
|
|
|
for (uid, comment) in comments:
|
|
lid = ID.parse_machine_id(comment.data.get_str('lid'))
|
|
|
|
# Look up external data for the request
|
|
if lid not in lid_mapping:
|
|
machine = self.get_machine_by_id(lid)
|
|
if machine is not None:
|
|
lid_mapping[lid] = machine.name
|
|
else:
|
|
lid_mapping[lid] = ''
|
|
|
|
c = Node.void('c')
|
|
commentnode.add_child(c)
|
|
c.add_child(Node.s32('uid', uid_mapping[uid].get_int('extid')))
|
|
c.add_child(Node.string('p_name', uid_mapping[uid].get_str('name')))
|
|
c.add_child(Node.s32('exp', uid_mapping[uid].get_int('exp')))
|
|
c.add_child(Node.s32('customize', comment.data.get_int('customize')))
|
|
c.add_child(Node.s32('tid', comment.data.get_int('teamid')))
|
|
c.add_child(Node.string('t_name', comment.data.get_str('teamname')))
|
|
c.add_child(Node.string('lid', comment.data.get_str('lid')))
|
|
c.add_child(Node.string('s_name', lid_mapping[lid]))
|
|
c.add_child(Node.s8('pref', comment.data.get_int('prefecture')))
|
|
c.add_child(Node.s32('time', comment.timestamp))
|
|
c.add_child(Node.string('comment', comment.data.get_str('comment')))
|
|
c.add_child(Node.bool('is_tweet', comment.data.get_bool('tweet')))
|
|
|
|
for (uid, status) in statuses:
|
|
lid = ID.parse_machine_id(status.get_str('lid'))
|
|
|
|
# Look up external data for the request
|
|
if lid not in lid_mapping:
|
|
machine = self.get_machine_by_id(lid)
|
|
if machine is not None:
|
|
lid_mapping[lid] = machine.name
|
|
else:
|
|
lid_mapping[lid] = ''
|
|
|
|
s = Node.void('s')
|
|
statusnode.add_child(s)
|
|
s.add_child(Node.s32('uid', uid_mapping[uid].get_int('extid')))
|
|
s.add_child(Node.string('p_name', uid_mapping[uid].get_str('name')))
|
|
s.add_child(Node.s32('exp', uid_mapping[uid].get_int('exp')))
|
|
s.add_child(Node.s32('customize', status.get_int('customize')))
|
|
s.add_child(Node.s32('tid', uid_mapping[uid].get_int('team_id', -1)))
|
|
s.add_child(Node.string('t_name', uid_mapping[uid].get_str('team_name', '')))
|
|
s.add_child(Node.string('lid', status.get_str('lid')))
|
|
s.add_child(Node.string('s_name', lid_mapping[lid]))
|
|
s.add_child(Node.s8('pref', status.get_int('prefecture')))
|
|
s.add_child(Node.s32('time', status.get_int('time')))
|
|
s.add_child(Node.s8('status', status.get_int('status')))
|
|
s.add_child(Node.s8('stage', status.get_int('stage')))
|
|
s.add_child(Node.s32('mid', status.get_int('mid')))
|
|
s.add_child(Node.s8('ng', status.get_int('ng')))
|
|
|
|
return root
|
|
|
|
def handle_event_w_add_comment_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)
|
|
|
|
customize = request.child_value('customize')
|
|
lid = request.child_value('lid')
|
|
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',
|
|
{
|
|
'customize': customize,
|
|
'lid': lid,
|
|
'teamid': teamid,
|
|
'teamname': teamname,
|
|
'prefecture': prefecture,
|
|
'comment': comment,
|
|
'tweet': is_tweet,
|
|
},
|
|
)
|
|
|
|
return Node.void('event_w')
|
|
|
|
def handle_event_w_update_status_request(self, request: Node) -> Node:
|
|
# Update user status so puzzle comments can show it
|
|
extid = request.child_value('uid')
|
|
userid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
if userid is not None:
|
|
customize = request.child_value('customize')
|
|
status = request.child_value('status')
|
|
stage = request.child_value('stage')
|
|
mid = request.child_value('mid')
|
|
ng = request.child_value('ng')
|
|
lid = request.child_value('lid')
|
|
prefecture = request.child_value('pref')
|
|
|
|
self.data.local.lobby.put_play_session_info(
|
|
self.game,
|
|
self.version,
|
|
userid,
|
|
{
|
|
'customize': customize,
|
|
'status': status,
|
|
'stage': stage,
|
|
'mid': mid,
|
|
'ng': ng,
|
|
'lid': lid,
|
|
'prefecture': prefecture,
|
|
},
|
|
)
|
|
return Node.void('event_w')
|
|
|
|
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'),
|
|
}
|
|
)
|
|
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.get_int('extid')))
|
|
e.add_child(Node.string('pn', profile.get_str('name')))
|
|
e.add_child(Node.s32('uattr', profile.get_int('uattr')))
|
|
e.add_child(Node.s32('mopt', lobby.get_int('mopt')))
|
|
e.add_child(Node.s16('mg', profile.get_int('mg')))
|
|
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)))
|
|
|
|
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
|
|
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
|
|
|
|
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.get_int('extid')))
|
|
e.add_child(Node.string('pn', profile.get_str('name')))
|
|
e.add_child(Node.s32('uattr', profile.get_int('uattr')))
|
|
e.add_child(Node.s32('mopt', lobby.get_int('mopt')))
|
|
e.add_child(Node.s16('mg', profile.get_int('mg')))
|
|
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)))
|
|
|
|
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:
|
|
# Add a dummy entry into the lobby setup so we can clean up on end play
|
|
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,
|
|
{}
|
|
)
|
|
|
|
root = Node.void('player')
|
|
root.add_child(Node.bool('is_suc', True))
|
|
|
|
unlock_music = Node.void('unlock_music')
|
|
root.add_child(unlock_music)
|
|
unlock_item = Node.void('unlock_item')
|
|
root.add_child(unlock_item)
|
|
item_lock_ctrl = Node.void('item_lock_ctrl')
|
|
root.add_child(item_lock_ctrl)
|
|
|
|
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))
|
|
|
|
# Add event info
|
|
self.__add_event_info(root)
|
|
|
|
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_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('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.get_int('extid')))
|
|
root.add_child(Node.s32('time', Time.now()))
|
|
return root
|
|
|
|
def format_profile(self, userid: UserID, profile: ValidatedDict) -> 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)
|
|
|
|
base = Node.void('base')
|
|
pdata.add_child(base)
|
|
base.add_child(Node.s32('uid', profile.get_int('extid')))
|
|
base.add_child(Node.string('name', profile.get_str('name')))
|
|
base.add_child(Node.s16('icon_id', profile.get_int('icon')))
|
|
base.add_child(Node.s16('lv', profile.get_int('lvl')))
|
|
base.add_child(Node.s32('exp', profile.get_int('exp')))
|
|
base.add_child(Node.s16('mg', profile.get_int('mg')))
|
|
base.add_child(Node.s16('ap', profile.get_int('ap')))
|
|
base.add_child(Node.s32('pc', profile.get_int('pc')))
|
|
base.add_child(Node.s32('uattr', profile.get_int('uattr')))
|
|
|
|
last_play_date = statistics.get_int_array('last_play_date', 3)
|
|
today_play_date = Time.todays_date()
|
|
if (
|
|
last_play_date[0] == today_play_date[0] and
|
|
last_play_date[1] == today_play_date[1] and
|
|
last_play_date[2] == today_play_date[2]
|
|
):
|
|
today_count = statistics.get_int('today_plays', 0)
|
|
else:
|
|
today_count = 0
|
|
|
|
con = Node.void('con')
|
|
pdata.add_child(con)
|
|
con.add_child(Node.s32('day', today_count))
|
|
con.add_child(Node.s32('cnt', statistics.get_int('total_plays')))
|
|
con.add_child(Node.s32('total_cnt', statistics.get_int('total_plays')))
|
|
con.add_child(Node.s32('last', statistics.get_int('last_play_timestamp')))
|
|
con.add_child(Node.s32('now', Time.now()))
|
|
|
|
team = Node.void('team')
|
|
pdata.add_child(team)
|
|
team.add_child(Node.s32('id', profile.get_int('team_id', -1)))
|
|
team.add_child(Node.string('name', profile.get_str('team_name', '')))
|
|
|
|
custom = Node.void('custom')
|
|
customdict = profile.get_dict('custom')
|
|
pdata.add_child(custom)
|
|
custom.add_child(Node.u8('s_gls', customdict.get_int('s_gls')))
|
|
custom.add_child(Node.u8('bgm_m', customdict.get_int('bgm_m')))
|
|
custom.add_child(Node.u8('st_f', customdict.get_int('st_f')))
|
|
custom.add_child(Node.u8('st_bg', customdict.get_int('st_bg')))
|
|
custom.add_child(Node.u8('st_bg_b', customdict.get_int('st_bg_b')))
|
|
custom.add_child(Node.u8('eff_e', customdict.get_int('eff_e')))
|
|
custom.add_child(Node.u8('se_s', customdict.get_int('se_s')))
|
|
custom.add_child(Node.u8('se_s_v', customdict.get_int('se_s_v')))
|
|
custom.add_child(Node.s16('last_music_id', customdict.get_int('last_music_id')))
|
|
custom.add_child(Node.u8('last_note_grade', customdict.get_int('last_note_grade')))
|
|
custom.add_child(Node.u8('sort_type', customdict.get_int('sort_type')))
|
|
custom.add_child(Node.u8('narrowdown_type', customdict.get_int('narrowdown_type')))
|
|
custom.add_child(Node.bool('is_begginer', customdict.get_bool('is_begginer'))) # Yes, this is spelled right
|
|
custom.add_child(Node.bool('is_tut', customdict.get_bool('is_tut')))
|
|
custom.add_child(Node.s16_array('symbol_chat_0', customdict.get_int_array('symbol_chat_0', 6)))
|
|
custom.add_child(Node.s16_array('symbol_chat_1', customdict.get_int_array('symbol_chat_1', 6)))
|
|
custom.add_child(Node.u8('gauge_style', customdict.get_int('gauge_style')))
|
|
custom.add_child(Node.u8('obj_shade', customdict.get_int('obj_shade')))
|
|
custom.add_child(Node.u8('obj_size', customdict.get_int('obj_size')))
|
|
custom.add_child(Node.s16_array('byword', customdict.get_int_array('byword', 2)))
|
|
custom.add_child(Node.bool_array('is_auto_byword', customdict.get_bool_array('is_auto_byword', 2)))
|
|
custom.add_child(Node.bool('is_tweet', customdict.get_bool('is_tweet')))
|
|
custom.add_child(Node.bool('is_link_twitter', customdict.get_bool('is_link_twitter')))
|
|
custom.add_child(Node.s16('mrec_type', customdict.get_int('mrec_type')))
|
|
custom.add_child(Node.s16('card_disp_type', customdict.get_int('card_disp_type')))
|
|
custom.add_child(Node.s16('tab_sel', customdict.get_int('tab_sel')))
|
|
custom.add_child(Node.s32_array('hidden_param', customdict.get_int_array('hidden_param', 20)))
|
|
|
|
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]))
|
|
|
|
# Scores
|
|
record = Node.void('record')
|
|
pdata.add_child(record)
|
|
|
|
for score in scores:
|
|
rec = Node.void('rec')
|
|
record.add_child(rec)
|
|
rec.add_child(Node.u16('mid', score.id))
|
|
rec.add_child(Node.u8('ng', score.chart))
|
|
rec.add_child(Node.s32('point', score.data.get_dict('stats').get_int('earned_points')))
|
|
rec.add_child(Node.s32('played_time', score.timestamp))
|
|
|
|
mrec_0 = Node.void('mrec_0')
|
|
rec.add_child(mrec_0)
|
|
mrec_0.add_child(Node.s32('win', score.data.get_dict('stats').get_int('win')))
|
|
mrec_0.add_child(Node.s32('lose', score.data.get_dict('stats').get_int('lose')))
|
|
mrec_0.add_child(Node.s32('draw', score.data.get_dict('stats').get_int('draw')))
|
|
mrec_0.add_child(Node.u8('ct', self.__db_to_game_clear_type(score.data.get_int('clear_type'), score.data.get_int('combo_type'))))
|
|
mrec_0.add_child(Node.s16('ar', int(score.data.get_int('achievement_rate') / 10)))
|
|
mrec_0.add_child(Node.s32('bs', score.points))
|
|
mrec_0.add_child(Node.s16('mc', score.data.get_int('combo')))
|
|
mrec_0.add_child(Node.s16('bmc', score.data.get_int('miss_count')))
|
|
|
|
mrec_1 = Node.void('mrec_1')
|
|
rec.add_child(mrec_1)
|
|
mrec_1.add_child(Node.s32('win', 0))
|
|
mrec_1.add_child(Node.s32('lose', 0))
|
|
mrec_1.add_child(Node.s32('draw', 0))
|
|
mrec_1.add_child(Node.u8('ct', 0))
|
|
mrec_1.add_child(Node.s16('ar', 0))
|
|
mrec_1.add_child(Node.s32('bs', 0))
|
|
mrec_1.add_child(Node.s16('mc', 0))
|
|
mrec_1.add_child(Node.s16('bmc', -1))
|
|
|
|
# Comment (seems unused?)
|
|
pdata.add_child(Node.string('cmnt', ''))
|
|
|
|
# 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.u8('slot_id', slotid))
|
|
r.add_child(Node.string('name', rprofile.get_str('name')))
|
|
r.add_child(Node.s32('id', rprofile.get_int('extid')))
|
|
r.add_child(Node.bool('friend', True))
|
|
r.add_child(Node.bool('locked', False))
|
|
r.add_child(Node.s32('rc', 0))
|
|
slotid = slotid + 1
|
|
|
|
# Glass points
|
|
glass = Node.void('glass')
|
|
pdata.add_child(glass)
|
|
|
|
for item in achievements:
|
|
if item.type != 'glass':
|
|
continue
|
|
|
|
g = Node.void('g')
|
|
glass.add_child(g)
|
|
g.add_child(Node.s32('id', item.id))
|
|
g.add_child(Node.s32('exp', item.data.get_int('exp')))
|
|
|
|
# Favorite music
|
|
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')))
|
|
|
|
narrow_down = Node.void('narrow_down')
|
|
pdata.add_child(narrow_down)
|
|
narrow_down.add_child(Node.s32_array('adv_param', [-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1]))
|
|
|
|
return root
|
|
|
|
def unformat_profile(self, userid: UserID, request: Node, oldprofile: ValidatedDict) -> ValidatedDict:
|
|
game_config = self.get_game_config()
|
|
newprofile = copy.deepcopy(oldprofile)
|
|
|
|
newprofile.replace_int('lid', ID.parse_machine_id(request.child_value('lid')))
|
|
newprofile.replace_str('name', request.child_value('pdata/base/name'))
|
|
newprofile.replace_int('icon', request.child_value('pdata/base/icon_id'))
|
|
newprofile.replace_int('lvl', request.child_value('pdata/base/lv'))
|
|
newprofile.replace_int('exp', request.child_value('pdata/base/exp'))
|
|
newprofile.replace_int('mg', request.child_value('pdata/base/mg'))
|
|
newprofile.replace_int('ap', request.child_value('pdata/base/ap'))
|
|
newprofile.replace_int('pc', request.child_value('pdata/base/pc'))
|
|
newprofile.replace_int('uattr', request.child_value('pdata/base/uattr'))
|
|
|
|
customdict = newprofile.get_dict('custom')
|
|
custom = request.child('pdata/custom')
|
|
if custom:
|
|
customdict.replace_int('s_gls', custom.child_value('s_gls'))
|
|
customdict.replace_int('bgm_m', custom.child_value('bgm_m'))
|
|
customdict.replace_int('st_f', custom.child_value('st_f'))
|
|
customdict.replace_int('st_bg', custom.child_value('st_bg'))
|
|
customdict.replace_int('st_bg_b', custom.child_value('st_bg_b'))
|
|
customdict.replace_int('eff_e', custom.child_value('eff_e'))
|
|
customdict.replace_int('se_s', custom.child_value('se_s'))
|
|
customdict.replace_int('se_s_v', custom.child_value('se_s_v'))
|
|
customdict.replace_int('last_music_id', custom.child_value('last_music_id'))
|
|
customdict.replace_int('last_note_grade', custom.child_value('last_note_grade'))
|
|
customdict.replace_int('sort_type', custom.child_value('sort_type'))
|
|
customdict.replace_int('narrowdown_type', custom.child_value('narrowdown_type'))
|
|
customdict.replace_bool('is_begginer', custom.child_value('is_begginer')) # Yes, this is spelled right
|
|
customdict.replace_bool('is_tut', custom.child_value('is_tut'))
|
|
customdict.replace_int_array('symbol_chat_0', 6, custom.child_value('symbol_chat_0'))
|
|
customdict.replace_int_array('symbol_chat_1', 6, custom.child_value('symbol_chat_1'))
|
|
customdict.replace_int('gauge_style', custom.child_value('gauge_style'))
|
|
customdict.replace_int('obj_shade', custom.child_value('obj_shade'))
|
|
customdict.replace_int('obj_size', custom.child_value('obj_size'))
|
|
customdict.replace_int_array('byword', 2, custom.child_value('byword'))
|
|
customdict.replace_bool_array('is_auto_byword', 2, custom.child_value('is_auto_byword'))
|
|
customdict.replace_bool('is_tweet', custom.child_value('is_tweet'))
|
|
customdict.replace_bool('is_link_twitter', custom.child_value('is_link_twitter'))
|
|
customdict.replace_int('mrec_type', custom.child_value('mrec_type'))
|
|
customdict.replace_int('card_disp_type', custom.child_value('card_disp_type'))
|
|
customdict.replace_int('tab_sel', custom.child_value('tab_sel'))
|
|
customdict.replace_int_array('hidden_param', 20, custom.child_value('hidden_param'))
|
|
newprofile.replace_dict('custom', customdict)
|
|
|
|
# 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,
|
|
},
|
|
)
|
|
|
|
# Grab any new records set during this play session. Reflec Beat Limelight only sends
|
|
# the top record back for songs that were played at least once during the session.
|
|
# Note that it sends the top record, so if you play the song twice, it will return
|
|
# only one record. Also, if you get a lower score than a previous try, it will return
|
|
# the previous try. So, we must also look at the battle log for the actual play scores,
|
|
# and combine the data if we can.
|
|
savedrecords: Dict[int, Dict[int, Dict[str, int]]] = {}
|
|
songplays = request.child('pdata/record')
|
|
if songplays:
|
|
for child in songplays.children:
|
|
if child.name != 'rec':
|
|
continue
|
|
|
|
songid = child.child_value('mid')
|
|
chart = child.child_value('ng')
|
|
|
|
# These don't get sent with the battle logs, so we try to construct
|
|
# the values here.
|
|
if songid not in savedrecords:
|
|
savedrecords[songid] = {}
|
|
savedrecords[songid][chart] = {
|
|
'achievement_rate': child.child_value('mrec_0/ar') * 10,
|
|
'points': child.child_value('mrec_0/bs'),
|
|
'combo': child.child_value('mrec_0/mc'),
|
|
'miss_count': child.child_value('mrec_0/bmc'),
|
|
'win': child.child_value('mrec_0/win'),
|
|
'lose': child.child_value('mrec_0/lose'),
|
|
'draw': child.child_value('mrec_0/draw'),
|
|
'earned_points': child.child_value('point'),
|
|
}
|
|
|
|
# Now, see the actual battles that were played. If we can, unify the data with a record.
|
|
# We only do that when the record achievement rate and score matches the battle achievement
|
|
# rate and score, so we know for a fact that that record was generated by this battle.
|
|
battlelogs = request.child('pdata/blog')
|
|
if battlelogs:
|
|
for child in battlelogs.children:
|
|
if child.name != 'log':
|
|
continue
|
|
|
|
songid = child.child_value('mid')
|
|
chart = child.child_value('ng')
|
|
|
|
clear_type = child.child_value('myself/ct')
|
|
achievement_rate = child.child_value('myself/ar') * 10
|
|
points = child.child_value('myself/s')
|
|
|
|
clear_type, combo_type = self.__game_to_db_clear_type(clear_type)
|
|
|
|
combo = None
|
|
miss_count = -1
|
|
stats = None
|
|
|
|
if songid in savedrecords:
|
|
if chart in savedrecords[songid]:
|
|
data = savedrecords[songid][chart]
|
|
|
|
if (
|
|
data['achievement_rate'] == achievement_rate and
|
|
data['points'] == points
|
|
):
|
|
# This is the same record! Use the stats from it to update our
|
|
# internal representation.
|
|
combo = data['combo']
|
|
miss_count = data['miss_count']
|
|
stats = {
|
|
'win': data['win'],
|
|
'lose': data['lose'],
|
|
'draw': data['draw'],
|
|
'earned_points': data['earned_points'],
|
|
}
|
|
|
|
self.update_score(
|
|
userid,
|
|
songid,
|
|
chart,
|
|
points,
|
|
achievement_rate,
|
|
clear_type,
|
|
combo_type,
|
|
miss_count,
|
|
combo=combo,
|
|
stats=stats,
|
|
)
|
|
|
|
# Keep track of glass points so unlocks work
|
|
glass = request.child('pdata/glass')
|
|
if glass:
|
|
for child in glass.children:
|
|
if child.name != 'g':
|
|
continue
|
|
|
|
gid = child.child_value('id')
|
|
exp = child.child_value('exp')
|
|
self.data.local.user.put_achievement(
|
|
self.game,
|
|
self.version,
|
|
userid,
|
|
gid,
|
|
'glass',
|
|
{
|
|
'exp': exp,
|
|
},
|
|
)
|
|
|
|
# Keep track of favorite music selections
|
|
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,
|
|
},
|
|
)
|
|
|
|
# Keep track of play statistics
|
|
self.update_play_statistics(userid)
|
|
|
|
return newprofile
|