1044 lines
42 KiB
Python
1044 lines
42 KiB
Python
# vim: set fileencoding=utf-8
|
|
import base64
|
|
from typing import Any, Dict, List, Optional, Tuple
|
|
from typing_extensions import Final
|
|
|
|
from bemani.backend.ess import EventLogHandler
|
|
from bemani.backend.ddr.base import DDRBase
|
|
from bemani.backend.ddr.ddr2014 import DDR2014
|
|
from bemani.common import (
|
|
Profile,
|
|
ValidatedDict,
|
|
VersionConstants,
|
|
CardCipher,
|
|
Time,
|
|
ID,
|
|
intish,
|
|
)
|
|
from bemani.data import Data, Achievement, Machine, Score, UserID
|
|
from bemani.protocol import Node
|
|
|
|
|
|
class DDRAce(
|
|
DDRBase,
|
|
EventLogHandler,
|
|
):
|
|
name: str = "DanceDanceRevolution A"
|
|
version: int = VersionConstants.DDR_ACE
|
|
|
|
GAME_STYLE_SINGLE: Final[int] = 0
|
|
GAME_STYLE_DOUBLE: Final[int] = 1
|
|
GAME_STYLE_VERSUS: Final[int] = 2
|
|
|
|
GAME_RIVAL_TYPE_RIVAL3: Final[int] = 32
|
|
GAME_RIVAL_TYPE_RIVAL2: Final[int] = 16
|
|
GAME_RIVAL_TYPE_RIVAL1: Final[int] = 8
|
|
GAME_RIVAL_TYPE_WORLD: Final[int] = 4
|
|
GAME_RIVAL_TYPE_AREA: Final[int] = 2
|
|
GAME_RIVAL_TYPE_MACHINE: Final[int] = 1
|
|
|
|
GAME_CHART_SINGLE_BEGINNER: Final[int] = 0
|
|
GAME_CHART_SINGLE_BASIC: Final[int] = 1
|
|
GAME_CHART_SINGLE_DIFFICULT: Final[int] = 2
|
|
GAME_CHART_SINGLE_EXPERT: Final[int] = 3
|
|
GAME_CHART_SINGLE_CHALLENGE: Final[int] = 4
|
|
GAME_CHART_DOUBLE_BASIC: Final[int] = 5
|
|
GAME_CHART_DOUBLE_DIFFICULT: Final[int] = 6
|
|
GAME_CHART_DOUBLE_EXPERT: Final[int] = 7
|
|
GAME_CHART_DOUBLE_CHALLENGE: Final[int] = 8
|
|
|
|
GAME_HALO_NONE: Final[int] = 6
|
|
GAME_HALO_GOOD_COMBO: Final[int] = 7
|
|
GAME_HALO_GREAT_COMBO: Final[int] = 8
|
|
GAME_HALO_PERFECT_COMBO: Final[int] = 9
|
|
GAME_HALO_MARVELOUS_COMBO: Final[int] = 10
|
|
|
|
GAME_RANK_E: Final[int] = 15
|
|
GAME_RANK_D: Final[int] = 14
|
|
GAME_RANK_D_PLUS: Final[int] = 13
|
|
GAME_RANK_C_MINUS: Final[int] = 12
|
|
GAME_RANK_C: Final[int] = 11
|
|
GAME_RANK_C_PLUS: Final[int] = 10
|
|
GAME_RANK_B_MINUS: Final[int] = 9
|
|
GAME_RANK_B: Final[int] = 8
|
|
GAME_RANK_B_PLUS: Final[int] = 7
|
|
GAME_RANK_A_MINUS: Final[int] = 6
|
|
GAME_RANK_A: Final[int] = 5
|
|
GAME_RANK_A_PLUS: Final[int] = 4
|
|
GAME_RANK_AA_MINUS: Final[int] = 3
|
|
GAME_RANK_AA: Final[int] = 2
|
|
GAME_RANK_AA_PLUS: Final[int] = 1
|
|
GAME_RANK_AAA: Final[int] = 0
|
|
|
|
GAME_MAX_SONGS: Final[int] = 1024
|
|
|
|
GAME_COMMON_AREA_OFFSET: Final[int] = 1
|
|
GAME_COMMON_WEIGHT_DISPLAY_OFFSET: Final[int] = 3
|
|
GAME_COMMON_CHARACTER_OFFSET: Final[int] = 4
|
|
GAME_COMMON_EXTRA_CHARGE_OFFSET: Final[int] = 5
|
|
GAME_COMMON_TOTAL_PLAYS_OFFSET: Final[int] = 9
|
|
GAME_COMMON_SINGLE_PLAYS_OFFSET: Final[int] = 11
|
|
GAME_COMMON_DOUBLE_PLAYS_OFFSET: Final[int] = 12
|
|
GAME_COMMON_WEIGHT_OFFSET: Final[int] = 17
|
|
GAME_COMMON_NAME_OFFSET: Final[int] = 25
|
|
GAME_COMMON_SEQ_OFFSET: Final[int] = 26
|
|
|
|
GAME_OPTION_SPEED_OFFSET: Final[int] = 1
|
|
GAME_OPTION_BOOST_OFFSET: Final[int] = 2
|
|
GAME_OPTION_APPEARANCE_OFFSET: Final[int] = 3
|
|
GAME_OPTION_TURN_OFFSET: Final[int] = 4
|
|
GAME_OPTION_STEP_ZONE_OFFSET: Final[int] = 5
|
|
GAME_OPTION_SCROLL_OFFSET: Final[int] = 6
|
|
GAME_OPTION_ARROW_COLOR_OFFSET: Final[int] = 7
|
|
GAME_OPTION_CUT_OFFSET: Final[int] = 8
|
|
GAME_OPTION_FREEZE_OFFSET: Final[int] = 9
|
|
GAME_OPTION_JUMPS_OFFSET: Final[int] = 10
|
|
GAME_OPTION_ARROW_SKIN_OFFSET: Final[int] = 11
|
|
GAME_OPTION_FILTER_OFFSET: Final[int] = 12
|
|
GAME_OPTION_GUIDELINE_OFFSET: Final[int] = 13
|
|
GAME_OPTION_GAUGE_OFFSET: Final[int] = 14
|
|
GAME_OPTION_COMBO_POSITION_OFFSET: Final[int] = 15
|
|
GAME_OPTION_FAST_SLOW_OFFSET: Final[int] = 16
|
|
|
|
GAME_LAST_CALORIES_OFFSET: Final[int] = 10
|
|
|
|
GAME_RIVAL_SLOT_1_ACTIVE_OFFSET: Final[int] = 1
|
|
GAME_RIVAL_SLOT_2_ACTIVE_OFFSET: Final[int] = 2
|
|
GAME_RIVAL_SLOT_3_ACTIVE_OFFSET: Final[int] = 3
|
|
GAME_RIVAL_SLOT_1_DDRCODE_OFFSET: Final[int] = 9
|
|
GAME_RIVAL_SLOT_2_DDRCODE_OFFSET: Final[int] = 10
|
|
GAME_RIVAL_SLOT_3_DDRCODE_OFFSET: Final[int] = 11
|
|
|
|
def previous_version(self) -> Optional[DDRBase]:
|
|
return DDR2014(self.data, self.config, self.model)
|
|
|
|
@classmethod
|
|
def run_scheduled_work(
|
|
cls, data: Data, config: Dict[str, Any]
|
|
) -> List[Tuple[str, Dict[str, Any]]]:
|
|
# DDR Ace has a weird bug where it sends a profile save for a blank
|
|
# profile before reading it back when creating a new profile. If there
|
|
# is no profile on read-back, it errors out, and it also uses the name
|
|
# and area ID as the takeover/succession data if the user had previous
|
|
# data on an old game. However, if for some reason the user cancels out
|
|
# of the name entry, loses power or disconnects from the network at the
|
|
# right time, then the profile exists in a broken state forever until they
|
|
# edit it on the front-end. As a work-around to this, we remember the last
|
|
# time each profile was written to, and we look up profiles that are older
|
|
# than a few minutes (the maximum possible time for DDR Ace to write back
|
|
# a new profile after creating a blank one) and have blank names and delete
|
|
# them in order to keep the profiles on the network in sane order. This
|
|
# should normally never delete any profiles.
|
|
profiles = data.local.user.get_all_profiles(cls.game, cls.version)
|
|
several_minutes_ago = Time.now() - (Time.SECONDS_IN_MINUTE * 5)
|
|
events = []
|
|
|
|
for userid, profile in profiles:
|
|
if (
|
|
profile.get_str("name") == ""
|
|
and profile.get_int("write_time") < several_minutes_ago
|
|
):
|
|
data.local.user.delete_profile(cls.game, cls.version, userid)
|
|
events.append(
|
|
(
|
|
"ddr_profile_purge",
|
|
{
|
|
"userid": userid,
|
|
},
|
|
)
|
|
)
|
|
|
|
return events
|
|
|
|
@property
|
|
def supports_paseli(self) -> bool:
|
|
if self.model.dest != "J":
|
|
# DDR Ace in USA mode doesn't support PASELI properly.
|
|
# When in Asia mode it shows PASELI but won't let you select it.
|
|
return False
|
|
else:
|
|
# All other modes should work with PASELI.
|
|
return True
|
|
|
|
def game_to_db_rank(self, game_rank: int) -> int:
|
|
return {
|
|
self.GAME_RANK_AAA: self.RANK_AAA,
|
|
self.GAME_RANK_AA_PLUS: self.RANK_AA_PLUS,
|
|
self.GAME_RANK_AA: self.RANK_AA,
|
|
self.GAME_RANK_AA_MINUS: self.RANK_AA_MINUS,
|
|
self.GAME_RANK_A_PLUS: self.RANK_A_PLUS,
|
|
self.GAME_RANK_A: self.RANK_A,
|
|
self.GAME_RANK_A_MINUS: self.RANK_A_MINUS,
|
|
self.GAME_RANK_B_PLUS: self.RANK_B_PLUS,
|
|
self.GAME_RANK_B: self.RANK_B,
|
|
self.GAME_RANK_B_MINUS: self.RANK_B_MINUS,
|
|
self.GAME_RANK_C_PLUS: self.RANK_C_PLUS,
|
|
self.GAME_RANK_C: self.RANK_C,
|
|
self.GAME_RANK_C_MINUS: self.RANK_C_MINUS,
|
|
self.GAME_RANK_D_PLUS: self.RANK_D_PLUS,
|
|
self.GAME_RANK_D: self.RANK_D,
|
|
self.GAME_RANK_E: self.RANK_E,
|
|
}[game_rank]
|
|
|
|
def db_to_game_rank(self, db_rank: int) -> int:
|
|
return {
|
|
self.RANK_AAA: self.GAME_RANK_AAA,
|
|
self.RANK_AA_PLUS: self.GAME_RANK_AA_PLUS,
|
|
self.RANK_AA: self.GAME_RANK_AA,
|
|
self.RANK_AA_MINUS: self.GAME_RANK_AA_MINUS,
|
|
self.RANK_A_PLUS: self.GAME_RANK_A_PLUS,
|
|
self.RANK_A: self.GAME_RANK_A,
|
|
self.RANK_A_MINUS: self.GAME_RANK_A_MINUS,
|
|
self.RANK_B_PLUS: self.GAME_RANK_B_PLUS,
|
|
self.RANK_B: self.GAME_RANK_B,
|
|
self.RANK_B_MINUS: self.GAME_RANK_B_MINUS,
|
|
self.RANK_C_PLUS: self.GAME_RANK_C_PLUS,
|
|
self.RANK_C: self.GAME_RANK_C,
|
|
self.RANK_C_MINUS: self.GAME_RANK_C_MINUS,
|
|
self.RANK_D_PLUS: self.GAME_RANK_D_PLUS,
|
|
self.RANK_D: self.GAME_RANK_D,
|
|
self.RANK_E: self.GAME_RANK_E,
|
|
}[db_rank]
|
|
|
|
def game_to_db_chart(self, game_chart: int) -> int:
|
|
return {
|
|
self.GAME_CHART_SINGLE_BEGINNER: self.CHART_SINGLE_BEGINNER,
|
|
self.GAME_CHART_SINGLE_BASIC: self.CHART_SINGLE_BASIC,
|
|
self.GAME_CHART_SINGLE_DIFFICULT: self.CHART_SINGLE_DIFFICULT,
|
|
self.GAME_CHART_SINGLE_EXPERT: self.CHART_SINGLE_EXPERT,
|
|
self.GAME_CHART_SINGLE_CHALLENGE: self.CHART_SINGLE_CHALLENGE,
|
|
self.GAME_CHART_DOUBLE_BASIC: self.CHART_DOUBLE_BASIC,
|
|
self.GAME_CHART_DOUBLE_DIFFICULT: self.CHART_DOUBLE_DIFFICULT,
|
|
self.GAME_CHART_DOUBLE_EXPERT: self.CHART_DOUBLE_EXPERT,
|
|
self.GAME_CHART_DOUBLE_CHALLENGE: self.CHART_DOUBLE_CHALLENGE,
|
|
}[game_chart]
|
|
|
|
def db_to_game_chart(self, db_chart: int) -> int:
|
|
return {
|
|
self.CHART_SINGLE_BEGINNER: self.GAME_CHART_SINGLE_BEGINNER,
|
|
self.CHART_SINGLE_BASIC: self.GAME_CHART_SINGLE_BASIC,
|
|
self.CHART_SINGLE_DIFFICULT: self.GAME_CHART_SINGLE_DIFFICULT,
|
|
self.CHART_SINGLE_EXPERT: self.GAME_CHART_SINGLE_EXPERT,
|
|
self.CHART_SINGLE_CHALLENGE: self.GAME_CHART_SINGLE_CHALLENGE,
|
|
self.CHART_DOUBLE_BASIC: self.GAME_CHART_DOUBLE_BASIC,
|
|
self.CHART_DOUBLE_DIFFICULT: self.GAME_CHART_DOUBLE_DIFFICULT,
|
|
self.CHART_DOUBLE_EXPERT: self.GAME_CHART_DOUBLE_EXPERT,
|
|
self.CHART_DOUBLE_CHALLENGE: self.GAME_CHART_DOUBLE_CHALLENGE,
|
|
}[db_chart]
|
|
|
|
def game_to_db_halo(self, game_halo: int) -> int:
|
|
if game_halo == self.GAME_HALO_MARVELOUS_COMBO:
|
|
return self.HALO_MARVELOUS_FULL_COMBO
|
|
elif game_halo == self.GAME_HALO_PERFECT_COMBO:
|
|
return self.HALO_PERFECT_FULL_COMBO
|
|
elif game_halo == self.GAME_HALO_GREAT_COMBO:
|
|
return self.HALO_GREAT_FULL_COMBO
|
|
elif game_halo == self.GAME_HALO_GOOD_COMBO:
|
|
return self.HALO_GOOD_FULL_COMBO
|
|
else:
|
|
return self.HALO_NONE
|
|
|
|
def db_to_game_halo(self, db_halo: int) -> int:
|
|
if db_halo == self.HALO_MARVELOUS_FULL_COMBO:
|
|
return self.GAME_HALO_MARVELOUS_COMBO
|
|
elif db_halo == self.HALO_PERFECT_FULL_COMBO:
|
|
return self.GAME_HALO_PERFECT_COMBO
|
|
elif db_halo == self.HALO_GREAT_FULL_COMBO:
|
|
return self.GAME_HALO_GREAT_COMBO
|
|
elif db_halo == self.HALO_GOOD_FULL_COMBO:
|
|
return self.GAME_HALO_GOOD_COMBO
|
|
else:
|
|
return self.GAME_HALO_NONE
|
|
|
|
def handle_tax_get_phase_request(self, request: Node) -> Node:
|
|
tax = Node.void("tax")
|
|
tax.add_child(Node.s32("phase", 0))
|
|
return tax
|
|
|
|
def __handle_userload(
|
|
self, userid: Optional[UserID], requestdata: Node, response: Node
|
|
) -> None:
|
|
has_profile: bool = False
|
|
achievements: List[Achievement] = []
|
|
scores: List[Score] = []
|
|
|
|
if userid is not None:
|
|
has_profile = self.has_profile(userid)
|
|
achievements = self.data.local.user.get_achievements(
|
|
self.game, self.version, userid
|
|
)
|
|
scores = self.data.remote.music.get_scores(
|
|
self.game, self.music_version, userid
|
|
)
|
|
|
|
# Place scores into an arrangement for easier distribution to Ace.
|
|
scores_by_mcode: Dict[int, List[Optional[Score]]] = {}
|
|
for score in scores:
|
|
if score.id not in scores_by_mcode:
|
|
scores_by_mcode[score.id] = [None] * 9
|
|
|
|
scores_by_mcode[score.id][self.db_to_game_chart(score.chart)] = score
|
|
|
|
# First, set new flag
|
|
response.add_child(Node.bool("is_new", not has_profile))
|
|
|
|
# Now, return the scores to Ace
|
|
for mcode in scores_by_mcode:
|
|
music = Node.void("music")
|
|
response.add_child(music)
|
|
music.add_child(Node.u32("mcode", mcode))
|
|
|
|
scores_that_matter = scores_by_mcode[mcode]
|
|
while scores_that_matter[-1] is None:
|
|
scores_that_matter = scores_that_matter[:-1]
|
|
|
|
for score in scores_that_matter:
|
|
note = Node.void("note")
|
|
music.add_child(note)
|
|
|
|
if score is None:
|
|
note.add_child(Node.u16("count", 0))
|
|
note.add_child(Node.u8("rank", 0))
|
|
note.add_child(Node.u8("clearkind", 0))
|
|
note.add_child(Node.s32("score", 0))
|
|
note.add_child(Node.s32("ghostid", 0))
|
|
else:
|
|
note.add_child(Node.u16("count", score.plays))
|
|
note.add_child(
|
|
Node.u8(
|
|
"rank", self.db_to_game_rank(score.data.get_int("rank"))
|
|
)
|
|
)
|
|
note.add_child(
|
|
Node.u8(
|
|
"clearkind",
|
|
self.db_to_game_halo(score.data.get_int("halo")),
|
|
)
|
|
)
|
|
note.add_child(Node.s32("score", score.points))
|
|
note.add_child(Node.s32("ghostid", score.key))
|
|
|
|
# Active event settings
|
|
activeevents = [
|
|
1,
|
|
3,
|
|
5,
|
|
9,
|
|
10,
|
|
11,
|
|
12,
|
|
13,
|
|
15,
|
|
16,
|
|
17,
|
|
18,
|
|
19,
|
|
20,
|
|
21,
|
|
22,
|
|
23,
|
|
24,
|
|
25,
|
|
26,
|
|
27,
|
|
28,
|
|
29,
|
|
30,
|
|
31,
|
|
32,
|
|
33,
|
|
34,
|
|
35,
|
|
36,
|
|
37,
|
|
38,
|
|
39,
|
|
40,
|
|
41,
|
|
42,
|
|
]
|
|
|
|
# Event reward settings
|
|
rewards = {
|
|
"30": {
|
|
999: 5,
|
|
}
|
|
}
|
|
|
|
# Now handle event progress and activation.
|
|
events = {ach.id: ach.data for ach in achievements if ach.type == "9999"}
|
|
progress = [ach for ach in achievements if ach.type != "9999"]
|
|
|
|
# Make sure we always send a babylon's adventure save event or the game won't send progress
|
|
babylon_included = False
|
|
for evtprogress in progress:
|
|
if evtprogress.id == 999 and evtprogress.type == "30":
|
|
babylon_included = True
|
|
break
|
|
|
|
if not babylon_included:
|
|
progress.append(
|
|
Achievement(
|
|
999,
|
|
"30",
|
|
None,
|
|
{
|
|
"completed": False,
|
|
"progress": 0,
|
|
},
|
|
)
|
|
)
|
|
|
|
for event in activeevents:
|
|
# Get completion data
|
|
playerstats = events.get(event, ValidatedDict({"completed": False}))
|
|
|
|
# Return the data
|
|
eventdata = Node.void("eventdata")
|
|
response.add_child(eventdata)
|
|
eventdata.add_child(Node.u32("eventid", event))
|
|
eventdata.add_child(Node.s32("eventtype", 9999))
|
|
eventdata.add_child(Node.u32("eventno", 0))
|
|
eventdata.add_child(Node.s64("condition", 0))
|
|
eventdata.add_child(Node.u32("reward", 0))
|
|
eventdata.add_child(
|
|
Node.s32("comptime", 1 if playerstats.get_bool("completed") else 0)
|
|
)
|
|
eventdata.add_child(Node.s64("savedata", 0))
|
|
|
|
for evtprogress in progress:
|
|
# Babylon's adventure progres and anything else the game sends
|
|
eventdata = Node.void("eventdata")
|
|
response.add_child(eventdata)
|
|
eventdata.add_child(Node.u32("eventid", evtprogress.id))
|
|
eventdata.add_child(Node.s32("eventtype", int(evtprogress.type)))
|
|
eventdata.add_child(Node.u32("eventno", 0))
|
|
eventdata.add_child(Node.s64("condition", 0))
|
|
eventdata.add_child(
|
|
Node.u32(
|
|
"reward", rewards.get(evtprogress.type, {}).get(evtprogress.id)
|
|
)
|
|
)
|
|
eventdata.add_child(
|
|
Node.s32("comptime", 1 if evtprogress.data.get_bool("completed") else 0)
|
|
)
|
|
eventdata.add_child(
|
|
Node.s64("savedata", evtprogress.data.get_int("progress"))
|
|
)
|
|
|
|
def __handle_usersave(
|
|
self, userid: Optional[UserID], requestdata: Node, response: Node
|
|
) -> None:
|
|
if userid is None:
|
|
# the game sends us empty user ID strings when a guest is playing.
|
|
# Return early so it doesn't wait a minute and a half to show the
|
|
# results screen.
|
|
return
|
|
|
|
if requestdata.child_value("isgameover"):
|
|
style = int(requestdata.child_value("playstyle"))
|
|
is_dp = style == self.GAME_STYLE_DOUBLE
|
|
|
|
# We don't save anything for gameover requests, since we
|
|
# already saved scores on individual ones. So, just use this
|
|
# as a spot to bump play counts and such
|
|
play_stats = self.get_play_statistics(userid)
|
|
if is_dp:
|
|
play_stats.increment_int("double_plays")
|
|
else:
|
|
play_stats.increment_int("single_plays")
|
|
self.update_play_statistics(userid, play_stats)
|
|
|
|
# Now is a good time to check if we have workout mode enabled,
|
|
# and if so, store the calories earned for this set.
|
|
profile = self.get_profile(userid)
|
|
enabled = profile.get_bool("workout_mode")
|
|
weight = profile.get_int("weight")
|
|
|
|
if enabled and weight > 0:
|
|
# We enabled weight display, find the calories and save them
|
|
total = 0
|
|
for child in requestdata.children:
|
|
if child.name != "note":
|
|
continue
|
|
|
|
total = total + (child.child_value("calorie") or 0)
|
|
|
|
self.data.local.user.put_time_based_achievement(
|
|
self.game,
|
|
self.version,
|
|
userid,
|
|
0,
|
|
"workout",
|
|
{
|
|
"calories": total,
|
|
"weight": weight,
|
|
},
|
|
)
|
|
|
|
# Find any event updates
|
|
for child in requestdata.children:
|
|
if child.name != "event":
|
|
continue
|
|
|
|
# Skip empty events or events we don't support
|
|
eventid = child.child_value("eventid")
|
|
eventtype = child.child_value("eventtype")
|
|
if eventid == 0 or eventtype == 0:
|
|
continue
|
|
|
|
# Save data to replay to the client later
|
|
completed = child.child_value("comptime") != 0
|
|
progress = child.child_value("savedata")
|
|
|
|
self.data.local.user.put_achievement(
|
|
self.game,
|
|
self.version,
|
|
userid,
|
|
eventid,
|
|
str(eventtype),
|
|
{
|
|
"completed": completed,
|
|
"progress": progress,
|
|
},
|
|
)
|
|
|
|
return
|
|
|
|
# Find the highest stagenum played
|
|
score = None
|
|
stagenum = 0
|
|
for child in requestdata.children:
|
|
if child.name != "note":
|
|
continue
|
|
|
|
if child.child_value("stagenum") > stagenum:
|
|
score = child
|
|
stagenum = child.child_value("stagenum")
|
|
|
|
if score is None:
|
|
raise Exception("Couldn't find newest score to save!")
|
|
|
|
songid = score.child_value("mcode")
|
|
chart = self.game_to_db_chart(score.child_value("notetype"))
|
|
rank = self.game_to_db_rank(score.child_value("rank"))
|
|
halo = self.game_to_db_halo(score.child_value("clearkind"))
|
|
points = score.child_value("score")
|
|
combo = score.child_value("maxcombo")
|
|
ghost = score.child_value("ghost")
|
|
self.update_score(
|
|
userid,
|
|
songid,
|
|
chart,
|
|
points,
|
|
rank,
|
|
halo,
|
|
combo,
|
|
ghost=ghost,
|
|
)
|
|
|
|
def __handle_rivalload(
|
|
self, userid: Optional[UserID], requestdata: Node, response: Node
|
|
) -> None:
|
|
data = Node.void("data")
|
|
response.add_child(data)
|
|
data.add_child(Node.s32("recordtype", requestdata.child_value("loadflag")))
|
|
|
|
thismachine = self.data.local.machine.get_machine(self.config.machine.pcbid)
|
|
machines_by_id: Dict[int, Optional[Machine]] = {thismachine.id: thismachine}
|
|
|
|
loadkind = requestdata.child_value("loadflag")
|
|
profiles_by_userid: Dict[UserID, Profile] = {}
|
|
|
|
def get_machine(lid: int) -> Optional[Machine]:
|
|
if lid not in machines_by_id:
|
|
pcbid = self.data.local.machine.from_machine_id(lid)
|
|
if pcbid is None:
|
|
machines_by_id[lid] = None
|
|
return None
|
|
|
|
machine = self.data.local.machine.get_machine(pcbid)
|
|
if machine is None:
|
|
machines_by_id[lid] = None
|
|
return None
|
|
|
|
machines_by_id[lid] = machine
|
|
return machines_by_id[lid]
|
|
|
|
if loadkind == self.GAME_RIVAL_TYPE_WORLD:
|
|
# Just load all scores for this network
|
|
scores = self.data.remote.music.get_all_records(
|
|
self.game, self.music_version
|
|
)
|
|
elif loadkind == self.GAME_RIVAL_TYPE_AREA:
|
|
if thismachine.arcade is not None:
|
|
match_arcade = thismachine.arcade
|
|
match_machine = None
|
|
else:
|
|
match_arcade = None
|
|
match_machine = thismachine.id
|
|
|
|
# Load up all scores by any user registered on a machine in the same arcade
|
|
profiles = self.data.local.user.get_all_profiles(self.game, self.version)
|
|
userids: List[UserID] = []
|
|
for userid, profiledata in profiles:
|
|
profiles_by_userid[userid] = profiledata
|
|
|
|
# If we have an arcade to match, see if this user's location matches the arcade.
|
|
# If we don't, just match lid directly
|
|
if match_arcade is not None:
|
|
theirmachine = get_machine(profiledata.get_int("lid"))
|
|
if theirmachine is not None and theirmachine.arcade == match_arcade:
|
|
userids.append(userid)
|
|
elif match_machine is not None:
|
|
if profiledata.get_int("lid") == match_machine:
|
|
userids.append(userid)
|
|
|
|
# Load all scores for users in the area
|
|
scores = self.data.local.music.get_all_records(
|
|
self.game, self.music_version, userlist=userids
|
|
)
|
|
elif loadkind == self.GAME_RIVAL_TYPE_MACHINE:
|
|
# Load up all scores and filter them by those earned at this location
|
|
scores = self.data.local.music.get_all_records(
|
|
self.game, self.music_version, locationlist=[thismachine.id]
|
|
)
|
|
elif loadkind in [
|
|
self.GAME_RIVAL_TYPE_RIVAL1,
|
|
self.GAME_RIVAL_TYPE_RIVAL2,
|
|
self.GAME_RIVAL_TYPE_RIVAL3,
|
|
]:
|
|
# Load up this user's highscores, format the way the below code expects it
|
|
extid = requestdata.child_value("ddrcode")
|
|
otherid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
userscores = self.data.remote.music.get_scores(
|
|
self.game, self.music_version, otherid
|
|
)
|
|
scores = [(otherid, score) for score in userscores]
|
|
else:
|
|
# Nothing here
|
|
scores = []
|
|
|
|
missing_users = [
|
|
userid for (userid, _) in scores if userid not in profiles_by_userid
|
|
]
|
|
for userid, profile in self.get_any_profiles(missing_users):
|
|
profiles_by_userid[userid] = profile
|
|
|
|
for userid, score in scores:
|
|
if profiles_by_userid.get(userid) is None:
|
|
raise Exception(f"Logic error, couldn't find any profile for {userid}")
|
|
profiledata = profiles_by_userid[userid]
|
|
|
|
record = Node.void("record")
|
|
data.add_child(record)
|
|
record.add_child(Node.u32("mcode", score.id))
|
|
record.add_child(Node.u8("notetype", self.db_to_game_chart(score.chart)))
|
|
record.add_child(
|
|
Node.u8("rank", self.db_to_game_rank(score.data.get_int("rank")))
|
|
)
|
|
record.add_child(
|
|
Node.u8("clearkind", self.db_to_game_halo(score.data.get_int("halo")))
|
|
)
|
|
record.add_child(Node.u8("flagdata", 0))
|
|
record.add_child(Node.string("name", profiledata.get_str("name")))
|
|
record.add_child(Node.s32("area", profiledata.get_int("area", 58)))
|
|
record.add_child(Node.s32("code", profiledata.extid))
|
|
record.add_child(Node.s32("score", score.points))
|
|
record.add_child(Node.s32("ghostid", score.key))
|
|
|
|
def __handle_usernew(
|
|
self, userid: Optional[UserID], requestdata: Node, response: Node
|
|
) -> None:
|
|
if userid is None:
|
|
raise Exception("Expecting valid UserID to create new profile!")
|
|
|
|
machine = self.data.local.machine.get_machine(self.config.machine.pcbid)
|
|
profile = Profile(
|
|
self.game,
|
|
self.version,
|
|
"",
|
|
0,
|
|
{
|
|
"lid": machine.id,
|
|
},
|
|
)
|
|
self.put_profile(userid, profile)
|
|
|
|
response.add_child(Node.string("seq", ID.format_extid(profile.extid)))
|
|
response.add_child(Node.s32("code", profile.extid))
|
|
response.add_child(Node.string("shoparea", ""))
|
|
|
|
def __handle_inheritance(
|
|
self, userid: Optional[UserID], requestdata: Node, response: Node
|
|
) -> None:
|
|
if userid is not None:
|
|
previous_version = self.previous_version()
|
|
profile = previous_version.get_profile(userid)
|
|
else:
|
|
profile = None
|
|
|
|
response.add_child(
|
|
Node.s32("InheritanceStatus", 1 if profile is not None else 0)
|
|
)
|
|
|
|
def __handle_ghostload(
|
|
self, userid: Optional[UserID], requestdata: Node, response: Node
|
|
) -> None:
|
|
ghostid = requestdata.child_value("ghostid")
|
|
ghost = self.data.local.music.get_score_by_key(
|
|
self.game, self.music_version, ghostid
|
|
)
|
|
if ghost is None:
|
|
return
|
|
|
|
userid, score = ghost
|
|
profile = self.get_profile(userid)
|
|
if profile is None:
|
|
return
|
|
|
|
if "ghost" not in score.data:
|
|
return
|
|
|
|
ghostdata = Node.void("ghostdata")
|
|
response.add_child(ghostdata)
|
|
ghostdata.add_child(Node.s32("code", profile.extid))
|
|
ghostdata.add_child(Node.u32("mcode", score.id))
|
|
ghostdata.add_child(Node.u8("notetype", self.db_to_game_chart(score.chart)))
|
|
ghostdata.add_child(Node.s32("ghostsize", len(score.data["ghost"])))
|
|
ghostdata.add_child(Node.string("ghost", score.data["ghost"]))
|
|
|
|
def handle_playerdata_usergamedata_advanced_request(
|
|
self, request: Node
|
|
) -> Optional[Node]:
|
|
playerdata = Node.void("playerdata")
|
|
|
|
# DDR Ace decides to be difficult and have a third level of packet switching
|
|
mode = request.child_value("data/mode")
|
|
refid = request.child_value("data/refid")
|
|
extid = request.child_value("data/ddrcode")
|
|
|
|
userid = self.data.remote.user.from_refid(self.game, self.version, refid)
|
|
if userid is None:
|
|
# Possibly look up by extid instead
|
|
userid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
|
|
if mode == "userload":
|
|
self.__handle_userload(userid, request.child("data"), playerdata)
|
|
elif mode == "usersave":
|
|
self.__handle_usersave(userid, request.child("data"), playerdata)
|
|
elif mode == "rivalload":
|
|
self.__handle_rivalload(userid, request.child("data"), playerdata)
|
|
elif mode == "usernew":
|
|
self.__handle_usernew(userid, request.child("data"), playerdata)
|
|
elif mode == "inheritance":
|
|
self.__handle_inheritance(userid, request.child("data"), playerdata)
|
|
elif mode == "ghostload":
|
|
self.__handle_ghostload(userid, request.child("data"), playerdata)
|
|
else:
|
|
# We don't support this
|
|
return None
|
|
|
|
playerdata.add_child(Node.s32("result", 0))
|
|
return playerdata
|
|
|
|
def handle_playerdata_usergamedata_send_request(self, request: Node) -> Node:
|
|
playerdata = Node.void("playerdata")
|
|
refid = request.child_value("data/refid")
|
|
|
|
userid = self.data.remote.user.from_refid(self.game, self.version, refid)
|
|
if userid is not None:
|
|
profile = self.get_profile(userid) or Profile(
|
|
self.game, self.version, refid, 0
|
|
)
|
|
usergamedata = profile.get_dict("usergamedata")
|
|
|
|
for record in request.child("data/record").children:
|
|
if record.name != "d":
|
|
continue
|
|
|
|
strdata = base64.b64decode(record.value)
|
|
bindata = base64.b64decode(record.child_value("bin1"))
|
|
|
|
# Grab and format the profile objects
|
|
strdatalist = strdata.split(b",")
|
|
profiletype = strdatalist[1].decode("utf-8")
|
|
strdatalist = strdatalist[2:]
|
|
|
|
# Extract relevant bits for frontend/API
|
|
if profiletype == "COMMON":
|
|
profile.replace_str(
|
|
"name",
|
|
strdatalist[self.GAME_COMMON_NAME_OFFSET].decode("ascii"),
|
|
)
|
|
profile.replace_int(
|
|
"area",
|
|
intish(
|
|
strdatalist[self.GAME_COMMON_AREA_OFFSET].decode("ascii"),
|
|
16,
|
|
),
|
|
)
|
|
profile.replace_bool(
|
|
"workout_mode",
|
|
int(
|
|
strdatalist[self.GAME_COMMON_WEIGHT_DISPLAY_OFFSET].decode(
|
|
"ascii"
|
|
),
|
|
16,
|
|
)
|
|
!= 0,
|
|
)
|
|
profile.replace_int(
|
|
"weight",
|
|
int(
|
|
float(
|
|
strdatalist[self.GAME_COMMON_WEIGHT_OFFSET].decode(
|
|
"ascii"
|
|
)
|
|
)
|
|
* 10
|
|
),
|
|
)
|
|
profile.replace_int(
|
|
"character",
|
|
int(
|
|
strdatalist[self.GAME_COMMON_CHARACTER_OFFSET].decode(
|
|
"ascii"
|
|
),
|
|
16,
|
|
),
|
|
)
|
|
if profiletype == "OPTION":
|
|
profile.replace_int(
|
|
"combo",
|
|
int(
|
|
strdatalist[self.GAME_OPTION_COMBO_POSITION_OFFSET].decode(
|
|
"ascii"
|
|
),
|
|
16,
|
|
),
|
|
)
|
|
profile.replace_int(
|
|
"early_late",
|
|
int(
|
|
strdatalist[self.GAME_OPTION_FAST_SLOW_OFFSET].decode(
|
|
"ascii"
|
|
),
|
|
16,
|
|
),
|
|
)
|
|
profile.replace_int(
|
|
"arrowskin",
|
|
int(
|
|
strdatalist[self.GAME_OPTION_ARROW_SKIN_OFFSET].decode(
|
|
"ascii"
|
|
),
|
|
16,
|
|
),
|
|
)
|
|
profile.replace_int(
|
|
"guidelines",
|
|
int(
|
|
strdatalist[self.GAME_OPTION_GUIDELINE_OFFSET].decode(
|
|
"ascii"
|
|
),
|
|
16,
|
|
),
|
|
)
|
|
profile.replace_int(
|
|
"filter",
|
|
int(
|
|
strdatalist[self.GAME_OPTION_FILTER_OFFSET].decode("ascii"),
|
|
16,
|
|
),
|
|
)
|
|
|
|
usergamedata[profiletype] = {
|
|
"strdata": b",".join(strdatalist),
|
|
"bindata": bindata,
|
|
}
|
|
|
|
profile.replace_dict("usergamedata", usergamedata)
|
|
profile.replace_int("write_time", Time.now())
|
|
self.put_profile(userid, profile)
|
|
|
|
playerdata.add_child(Node.s32("result", 0))
|
|
return playerdata
|
|
|
|
def handle_playerdata_usergamedata_recv_request(self, request: Node) -> Node:
|
|
playerdata = Node.void("playerdata")
|
|
|
|
player = Node.void("player")
|
|
playerdata.add_child(player)
|
|
|
|
refid = request.child_value("data/refid")
|
|
userid = self.data.remote.user.from_refid(self.game, self.version, refid)
|
|
if userid is not None:
|
|
profile = self.get_profile(userid)
|
|
links = self.data.local.user.get_links(self.game, self.version, userid)
|
|
records = 0
|
|
|
|
record = Node.void("record")
|
|
player.add_child(record)
|
|
|
|
def acehex(val: int) -> str:
|
|
return hex(val)[2:]
|
|
|
|
if profile is None:
|
|
# Just return a default empty node
|
|
record.add_child(Node.string("d", "<NODATA>"))
|
|
records = 1
|
|
else:
|
|
# Figure out what profiles are being requested
|
|
profiletypes = request.child_value("data/recv_csv").split(",")[::2]
|
|
usergamedata = profile.get_dict("usergamedata")
|
|
for ptype in profiletypes:
|
|
if ptype in usergamedata:
|
|
records = records + 1
|
|
|
|
if ptype == "COMMON":
|
|
# Return basic profile options
|
|
name = profile.get_str("name")
|
|
area = profile.get_int("area", self.get_machine_region())
|
|
if name == "":
|
|
# This is a bogus profile created by the first login, substitute the
|
|
# previous version values so that profile succession works.
|
|
previous_version = self.previous_version()
|
|
old_profile = previous_version.get_profile(userid)
|
|
if old_profile is not None:
|
|
name = old_profile.get_str("name")
|
|
area = old_profile.get_int(
|
|
"area", self.get_machine_region()
|
|
)
|
|
else:
|
|
area = self.get_machine_region()
|
|
|
|
common = usergamedata[ptype]["strdata"].split(b",")
|
|
common[self.GAME_COMMON_NAME_OFFSET] = name.encode("ascii")
|
|
common[self.GAME_COMMON_AREA_OFFSET] = acehex(area).encode(
|
|
"ascii"
|
|
)
|
|
common[self.GAME_COMMON_WEIGHT_DISPLAY_OFFSET] = (
|
|
b"1" if profile.get_bool("workout_mode") else b"0"
|
|
)
|
|
common[self.GAME_COMMON_WEIGHT_OFFSET] = str(
|
|
float(profile.get_int("weight")) / 10.0
|
|
).encode("ascii")
|
|
common[self.GAME_COMMON_CHARACTER_OFFSET] = acehex(
|
|
profile.get_int("character")
|
|
).encode("ascii")
|
|
usergamedata[ptype]["strdata"] = b",".join(common)
|
|
if ptype == "OPTION":
|
|
# Return user settings for frontend
|
|
option = usergamedata[ptype]["strdata"].split(b",")
|
|
option[self.GAME_OPTION_FAST_SLOW_OFFSET] = acehex(
|
|
profile.get_int("early_late")
|
|
).encode("ascii")
|
|
option[self.GAME_OPTION_COMBO_POSITION_OFFSET] = acehex(
|
|
profile.get_int("combo")
|
|
).encode("ascii")
|
|
option[self.GAME_OPTION_ARROW_SKIN_OFFSET] = acehex(
|
|
profile.get_int("arrowskin")
|
|
).encode("ascii")
|
|
option[self.GAME_OPTION_GUIDELINE_OFFSET] = acehex(
|
|
profile.get_int("guidelines")
|
|
).encode("ascii")
|
|
option[self.GAME_OPTION_FILTER_OFFSET] = acehex(
|
|
profile.get_int("filter")
|
|
).encode("ascii")
|
|
usergamedata[ptype]["strdata"] = b",".join(option)
|
|
if ptype == "LAST":
|
|
# Return the number of calories expended in the last day
|
|
workouts = self.data.local.user.get_time_based_achievements(
|
|
self.game,
|
|
self.version,
|
|
userid,
|
|
achievementtype="workout",
|
|
since=Time.now() - Time.SECONDS_IN_DAY,
|
|
)
|
|
total = sum([w.data.get_int("calories") for w in workouts])
|
|
|
|
last = usergamedata[ptype]["strdata"].split(b",")
|
|
last[self.GAME_LAST_CALORIES_OFFSET] = acehex(total).encode(
|
|
"ascii"
|
|
)
|
|
usergamedata[ptype]["strdata"] = b",".join(last)
|
|
if ptype == "RIVAL":
|
|
# Fill in the DDR code and active status of the three active
|
|
# rivals.
|
|
rival = usergamedata[ptype]["strdata"].split(b",")
|
|
lastdict = profile.get_dict("last")
|
|
|
|
friends: Dict[int, Optional[Profile]] = {}
|
|
for link in links:
|
|
if link.type[:7] != "friend_":
|
|
continue
|
|
|
|
pos = int(link.type[7:])
|
|
friends[pos] = self.get_profile(link.other_userid)
|
|
|
|
for rivalno in [1, 2, 3]:
|
|
activeslot = {
|
|
1: self.GAME_RIVAL_SLOT_1_ACTIVE_OFFSET,
|
|
2: self.GAME_RIVAL_SLOT_2_ACTIVE_OFFSET,
|
|
3: self.GAME_RIVAL_SLOT_3_ACTIVE_OFFSET,
|
|
}[rivalno]
|
|
|
|
whichfriend = lastdict.get_int(f"rival{rivalno}") - 1
|
|
if whichfriend < 0:
|
|
# This rival isn't active
|
|
rival[activeslot] = b"0"
|
|
continue
|
|
|
|
friendprofile = friends.get(whichfriend)
|
|
if friendprofile is None:
|
|
# This rival doesn't exist
|
|
rival[activeslot] = b"0"
|
|
continue
|
|
|
|
ddrcodeslot = {
|
|
1: self.GAME_RIVAL_SLOT_1_DDRCODE_OFFSET,
|
|
2: self.GAME_RIVAL_SLOT_2_DDRCODE_OFFSET,
|
|
3: self.GAME_RIVAL_SLOT_3_DDRCODE_OFFSET,
|
|
}[rivalno]
|
|
|
|
rival[activeslot] = acehex(rivalno).encode("ascii")
|
|
rival[ddrcodeslot] = acehex(friendprofile.extid).encode(
|
|
"ascii"
|
|
)
|
|
|
|
usergamedata[ptype]["strdata"] = b",".join(rival)
|
|
|
|
dnode = Node.string(
|
|
"d",
|
|
base64.b64encode(usergamedata[ptype]["strdata"]).decode(
|
|
"ascii"
|
|
),
|
|
)
|
|
dnode.add_child(
|
|
Node.string(
|
|
"bin1",
|
|
base64.b64encode(usergamedata[ptype]["bindata"]).decode(
|
|
"ascii"
|
|
),
|
|
)
|
|
)
|
|
record.add_child(dnode)
|
|
|
|
player.add_child(Node.u32("record_num", records))
|
|
|
|
playerdata.add_child(Node.s32("result", 0))
|
|
return playerdata
|
|
|
|
def handle_system_convcardnumber_request(self, request: Node) -> Node:
|
|
cardid = request.child_value("data/card_id")
|
|
cardnumber = CardCipher.encode(cardid)
|
|
|
|
system = Node.void("system")
|
|
data = Node.void("data")
|
|
system.add_child(data)
|
|
|
|
system.add_child(Node.s32("result", 0))
|
|
data.add_child(Node.string("card_number", cardnumber))
|
|
return system
|