mirror of
https://github.com/drmext/MonkeyBusiness.git
synced 2024-11-27 17:00:59 +01:00
483 lines
17 KiB
Python
483 lines
17 KiB
Python
import xml.etree.ElementTree as ET
|
|
from os import path
|
|
|
|
from tinydb import Query, where
|
|
|
|
import config
|
|
import random
|
|
import time
|
|
|
|
from fastapi import APIRouter, Request, Response
|
|
|
|
from core_common import core_process_request, core_prepare_response, E
|
|
from core_database import get_db
|
|
|
|
router = APIRouter(prefix="/local", tags=["local"])
|
|
router.model_whitelist = ["REC"]
|
|
|
|
|
|
def get_profile(cid):
|
|
return get_db().table("dancerush_profile").get(where("card") == cid)
|
|
|
|
|
|
def get_game_profile(cid, game_version):
|
|
profile = get_profile(cid)
|
|
|
|
return profile["version"].get(str(game_version), None)
|
|
|
|
|
|
def get_id_from_profile(cid):
|
|
profile = get_db().table("dancerush_profile").get(where("card") == cid)
|
|
|
|
djid = "%08d" % profile["drs_id"]
|
|
djid_split = "-".join([djid[:4], djid[4:]])
|
|
|
|
return profile["drs_id"], djid_split
|
|
|
|
|
|
@router.post("/{gameinfo}/game/get_common")
|
|
async def drs_game_get_common(request: Request):
|
|
request_info = await core_process_request(request)
|
|
|
|
songs = {}
|
|
|
|
# TODO: server side song unlock is incomplete, use hex edits for now
|
|
for f in (
|
|
path.join("modules", "drs", "music-info-base.xml"),
|
|
path.join("music-info-base.xml"),
|
|
):
|
|
if path.exists(f):
|
|
with open(f, "r", encoding="utf-8") as fp:
|
|
tree = ET.parse(fp, ET.XMLParser())
|
|
root = tree.getroot()
|
|
|
|
for entry in root:
|
|
mid = entry.get("id")
|
|
songs[mid] = {}
|
|
for atr in (
|
|
"title_name",
|
|
"title_yomigana",
|
|
"artist_name",
|
|
"artist_yomigana",
|
|
"bpm_max",
|
|
"bpm_min",
|
|
# "distribution_date",
|
|
"volume",
|
|
"bg_no",
|
|
"region",
|
|
# "limitation_type",
|
|
# "price",
|
|
"genre",
|
|
"play_video_flags",
|
|
"is_fixed",
|
|
"version",
|
|
"demo_pri",
|
|
"license",
|
|
"color1",
|
|
"color2",
|
|
"color3",
|
|
):
|
|
songs[mid][atr] = entry.find(f"info/{atr}").text
|
|
if songs[mid][atr] == None:
|
|
songs[mid][atr] = ""
|
|
for atr in (
|
|
"1b",
|
|
"1a",
|
|
"2b",
|
|
"2a",
|
|
):
|
|
songs[mid][f"{atr}_difnum"] = entry.find(
|
|
f"difficulty/fumen_{atr}/difnum"
|
|
).text
|
|
# songs[mid][f"{atr}_playable"] = entry.find(f"difficulty/fumen_{atr}/playable").text
|
|
break
|
|
|
|
response = E.response(
|
|
E.game(
|
|
E.mdb(
|
|
*[
|
|
E.music(
|
|
E.info(
|
|
E.title_name(songs[s]["title_name"], __type="str"),
|
|
E.title_yomigana(songs[s]["title_yomigana"], __type="str"),
|
|
E.artist_name(songs[s]["artist_name"], __type="str"),
|
|
E.artist_yomigana(
|
|
songs[s]["artist_yomigana"], __type="str"
|
|
),
|
|
E.bpm_max(songs[s]["bpm_max"], __type="u32"),
|
|
E.bpm_min(songs[s]["bpm_min"], __type="u32"),
|
|
E.distribution_date(20180427, __type="u32"),
|
|
E.volume(songs[s]["volume"], __type="u16"),
|
|
E.bg_no(songs[s]["bg_no"], __type="u16"),
|
|
E.region("JUAKYC", __type="str"),
|
|
E.limitation_type(3, __type="u8"),
|
|
E.price(0, __type="s32"),
|
|
E.genre(songs[s]["genre"], __type="u32"),
|
|
E.play_video_flags(
|
|
songs[s]["play_video_flags"], __type="u32"
|
|
),
|
|
E.is_fixed(songs[s]["is_fixed"], __type="u8"),
|
|
E.version(songs[s]["version"], __type="u8"),
|
|
E.demo_pri(songs[s]["demo_pri"], __type="u8"),
|
|
E.license(songs[s]["license"], __type="str"),
|
|
E.color1(int(songs[s]["color1"], 16), __type="u32"),
|
|
E.color2(int(songs[s]["color2"], 16), __type="u32"),
|
|
E.color3(int(songs[s]["color3"], 16), __type="u32"),
|
|
),
|
|
E.difficulty(
|
|
E.fumen_1b(
|
|
E.difnum(songs[s]["1b_difnum"], __type="u8"),
|
|
E.playable(1, __type="u8"),
|
|
),
|
|
E.fumen_1a(
|
|
E.difnum(songs[s]["1a_difnum"], __type="u8"),
|
|
E.playable(1, __type="u8"),
|
|
),
|
|
E.fumen_2b(
|
|
E.difnum(songs[s]["2b_difnum"], __type="u8"),
|
|
E.playable(1, __type="u8"),
|
|
),
|
|
E.fumen_2a(
|
|
E.difnum(songs[s]["2a_difnum"], __type="u8"),
|
|
E.playable(1, __type="u8"),
|
|
),
|
|
),
|
|
id=s,
|
|
)
|
|
for s in songs
|
|
],
|
|
),
|
|
E.extra(*[E.info(E.music_id(i, __type="s32")) for i in songs]),
|
|
E.contest(
|
|
*[
|
|
E.info(
|
|
E.contest_id(i, __type="s32"),
|
|
E.start_date(1683422123358, __type="u64"),
|
|
E.end_date(1693422123358, __type="u64"),
|
|
E.title("", __type="str"),
|
|
E.regulation(i, __type="s32"),
|
|
E.target_music(
|
|
E.music(
|
|
E.music_id(1, __type="s32"),
|
|
E.music_type("1b", __type="str"),
|
|
)
|
|
),
|
|
)
|
|
for i in range(1, 3)
|
|
]
|
|
),
|
|
E.event(
|
|
*[
|
|
E.info(
|
|
E.event_id(e, __type="s32"),
|
|
E.start_date(1683422123358, __type="u64"),
|
|
E.end_date(1693422123358, __type="u64"),
|
|
E.param("", __type="str"),
|
|
)
|
|
for e in range(1, 14)
|
|
]
|
|
),
|
|
# E.kac2020(
|
|
# E.reward(
|
|
# E.data(
|
|
# E.music_id(1, __type="s32"),
|
|
# E.is_available(1, __type="bool"),
|
|
# )
|
|
# )
|
|
# ),
|
|
# E.silhouette(E.info(E.silhouette_id(i, __type="s32"))),
|
|
# E.music_condition(*[E.music(E.conditions(), id=s) for s in songs]),
|
|
)
|
|
)
|
|
|
|
response_body, response_headers = await core_prepare_response(request, response)
|
|
return Response(content=response_body, headers=response_headers)
|
|
|
|
|
|
@router.post("/{gameinfo}/game/get_playdata_{player}")
|
|
async def drs_game_get_playdata(player: str, request: Request):
|
|
request_info = await core_process_request(request)
|
|
game_version = request_info["game_version"]
|
|
|
|
dataid = request_info["root"][0].find("userid/refid").text
|
|
profile = get_game_profile(dataid, game_version)
|
|
|
|
if profile:
|
|
djid, djid_split = get_id_from_profile(dataid)
|
|
|
|
response = E.response(
|
|
E.game(
|
|
E.result(0, __type="s32"),
|
|
E.userid(E.code(djid, __type="s32")),
|
|
E.profile(E.name(profile["name"], __type="str")),
|
|
E.playinfo(
|
|
E.softcode("", __type="str"),
|
|
E.start_date(1683422123358, __type="u64"),
|
|
E.end_date(1683422123358, __type="u64"),
|
|
E.mode_id(profile["mode_id"], __type="s32"),
|
|
E.music_id(profile["music_id"], __type="s32"),
|
|
E.music_type(profile["music_type"], __type="str"),
|
|
E.pcbid("0", __type="str"),
|
|
E.locid("EA000001", __type="str"),
|
|
),
|
|
E.paramdata(
|
|
*[
|
|
E.data(
|
|
E.data_type(p[0], __type="s32"),
|
|
E.data_id(p[1], __type="s32"),
|
|
E.param_list(p[2], __type="s32"),
|
|
)
|
|
for p in profile["params"]
|
|
]
|
|
),
|
|
E.dance_dance_rush(E.data()),
|
|
E.summer_dance_damp(E.data()),
|
|
E.kac2020(),
|
|
E.hidden_param(0, __type="s32"),
|
|
E.play_count(1001, __type="u32"),
|
|
E.daily_count(301, __type="u32"),
|
|
E.play_chain(31, __type="u32"),
|
|
)
|
|
)
|
|
|
|
else:
|
|
response = E.response(
|
|
E.game(
|
|
E.result(1, __type="s32"),
|
|
)
|
|
)
|
|
|
|
response_body, response_headers = await core_prepare_response(request, response)
|
|
return Response(content=response_body, headers=response_headers)
|
|
|
|
|
|
@router.post("/{gameinfo}/game/lock_multi_login_{player}")
|
|
async def drs_game_lock_multi_login(player: str, request: Request):
|
|
request_info = await core_process_request(request)
|
|
|
|
response = E.response(E.game())
|
|
|
|
response_body, response_headers = await core_prepare_response(request, response)
|
|
return Response(content=response_body, headers=response_headers)
|
|
|
|
|
|
@router.post("/{gameinfo}/game/sign_up_{player}")
|
|
async def drs_game_sign_up(player: str, request: Request):
|
|
request_info = await core_process_request(request)
|
|
game_version = request_info["game_version"]
|
|
|
|
root = request_info["root"][0]
|
|
|
|
dataid = root.find("userid/dataid").text
|
|
cardno = root.find("userid/cardno").text
|
|
name = root.find("profile/name").text
|
|
|
|
db = get_db().table("dancerush_profile")
|
|
all_profiles_for_card = db.get(Query().card == dataid)
|
|
|
|
if all_profiles_for_card is None:
|
|
all_profiles_for_card = {"card": dataid, "version": {}}
|
|
|
|
if "drs_id" not in all_profiles_for_card:
|
|
drs_id = random.randint(10000000, 99999999)
|
|
all_profiles_for_card["drs_id"] = drs_id
|
|
|
|
all_profiles_for_card["version"][str(game_version)] = {
|
|
"game_version": game_version,
|
|
"name": name,
|
|
"mode_id": 0,
|
|
"music_id": 1,
|
|
"music_type": "1a",
|
|
"params": [],
|
|
}
|
|
|
|
db.upsert(all_profiles_for_card, where("card") == dataid)
|
|
|
|
response = E.response(E.game())
|
|
|
|
response_body, response_headers = await core_prepare_response(request, response)
|
|
return Response(content=response_body, headers=response_headers)
|
|
|
|
|
|
@router.post("/{gameinfo}/game/get_musicscore_{player}")
|
|
async def drs_get_musicscore(player: str, request: Request):
|
|
request_info = await core_process_request(request)
|
|
game_version = request_info["game_version"]
|
|
|
|
scores = []
|
|
db = get_db()
|
|
for record in db.table("drs_scores_best").search(
|
|
(where("game_version") == game_version)
|
|
):
|
|
scores.append(
|
|
[
|
|
record["music_id"],
|
|
record["music_type"],
|
|
record["score"],
|
|
record["rank"],
|
|
record["combo"],
|
|
record["param"],
|
|
]
|
|
)
|
|
|
|
response = E.response(
|
|
E.game(
|
|
E.scoredata(
|
|
*[
|
|
E.music(
|
|
E.music_id(s[0], __type="s32"),
|
|
E.music_type(s[1], __type="str"),
|
|
E.play_cnt(1, __type="s32"),
|
|
E.score(s[2], __type="s32"),
|
|
E.rank(s[3], __type="s32"),
|
|
E.combo(s[4], __type="s32"),
|
|
E.param(s[5], __type="s32"),
|
|
E.bestscore_date(1683422123358, __type="u64"),
|
|
E.lastplay_date(1683422123358, __type="u64"),
|
|
)
|
|
for s in scores
|
|
],
|
|
),
|
|
),
|
|
)
|
|
|
|
response_body, response_headers = await core_prepare_response(request, response)
|
|
return Response(content=response_body, headers=response_headers)
|
|
|
|
|
|
@router.post("/{gameinfo}/game/save_musicscore")
|
|
async def drs_save_musicscore(request: Request):
|
|
request_info = await core_process_request(request)
|
|
game_version = request_info["game_version"]
|
|
|
|
timestamp = time.time()
|
|
|
|
root = request_info["root"][0][0]
|
|
|
|
dataid = root.find("userid/refid").text
|
|
profile = get_game_profile(dataid, game_version)
|
|
djid, djid_split = get_id_from_profile(dataid)
|
|
|
|
music_id = int(root.find("music_id").text)
|
|
music_type = root.find("music_type").text
|
|
mode = int(root.find("mode").text)
|
|
score = int(root.find("score").text)
|
|
rank = int(root.find("rank").text)
|
|
combo = int(root.find("combo").text)
|
|
param = int(root.find("param").text)
|
|
perfect = int(root.find("member/perfect").text)
|
|
great = int(root.find("member/great").text)
|
|
good = int(root.find("member/good").text)
|
|
bad = int(root.find("member/bad").text)
|
|
|
|
db = get_db()
|
|
db.table("drs_scores").insert(
|
|
{
|
|
"timestamp": timestamp,
|
|
"game_version": game_version,
|
|
"drs_id": djid,
|
|
"music_id": music_id,
|
|
"music_type": music_type,
|
|
"mode": mode,
|
|
"score": score,
|
|
"rank": rank,
|
|
"combo": combo,
|
|
"param": param,
|
|
"perfect": perfect,
|
|
"great": great,
|
|
"good": good,
|
|
"bad": bad,
|
|
},
|
|
)
|
|
|
|
best = db.table("drs_scores_best").get(
|
|
(where("drs_id") == djid)
|
|
& (where("game_version") == game_version)
|
|
& (where("music_id") == music_id)
|
|
& (where("music_type") == music_type)
|
|
)
|
|
best = {} if best is None else best
|
|
|
|
best_score_data = {
|
|
"game_version": game_version,
|
|
"drs_id": djid,
|
|
"name": profile["name"],
|
|
"music_id": music_id,
|
|
"music_type": music_type,
|
|
"score": max(score, best.get("score", score)),
|
|
"rank": max(rank, best.get("rank", rank)),
|
|
"combo": max(combo, best.get("combo", combo)),
|
|
"param": param,
|
|
}
|
|
|
|
db.table("drs_scores_best").upsert(
|
|
best_score_data,
|
|
(where("drs_id") == djid)
|
|
& (where("game_version") == game_version)
|
|
& (where("music_id") == music_id)
|
|
& (where("music_type") == music_type),
|
|
)
|
|
|
|
response = E.response(E.game())
|
|
|
|
response_body, response_headers = await core_prepare_response(request, response)
|
|
return Response(content=response_body, headers=response_headers)
|
|
|
|
|
|
@router.post("/{gameinfo}/game/save_playdata")
|
|
async def drs_save_musicscore(request: Request):
|
|
request_info = await core_process_request(request)
|
|
game_version = request_info["game_version"]
|
|
|
|
root = request_info["root"][0][0]
|
|
|
|
dataid = root.find("userid/refid").text
|
|
|
|
profile = get_profile(dataid)
|
|
game_profile = profile["version"].get(str(game_version), {})
|
|
|
|
game_profile["mode_id"] = int(root.find("playinfo/mode_id").text)
|
|
game_profile["music_id"] = int(root.find("playinfo/music_id").text)
|
|
game_profile["music_type"] = root.find("playinfo/music_type").text
|
|
|
|
old_params = game_profile["params"]
|
|
params = {}
|
|
|
|
for old in old_params:
|
|
t = str(old[0])
|
|
i = str(old[1])
|
|
p = old[2]
|
|
if t not in params:
|
|
params[t] = {}
|
|
if i not in params[t]:
|
|
params[t][i] = {}
|
|
params[t][i] = p
|
|
|
|
for info in root.find("paramdata"):
|
|
t = info.find("data_type").text
|
|
i = info.find("data_id").text
|
|
p = info.find("param_list")
|
|
|
|
if t not in params:
|
|
params[t] = {}
|
|
if i not in params[t]:
|
|
params[t][i] = {}
|
|
params[t][i] = [int(x) for x in p.text.split(" ")]
|
|
|
|
params_list = []
|
|
|
|
for t in params:
|
|
for i in params[t]:
|
|
params_list.append([int(t), int(i), params[t][i]])
|
|
|
|
game_profile["params"] = params_list
|
|
|
|
profile["version"][str(game_version)] = game_profile
|
|
|
|
get_db().table("dancerush_profile").upsert(profile, where("card") == dataid)
|
|
|
|
response = E.response(E.game())
|
|
|
|
response_body, response_headers = await core_prepare_response(request, response)
|
|
return Response(content=response_body, headers=response_headers)
|