bemaniutils/bemani/backend/jubeat/common.py

import time

from bemani.backend.jubeat.base import JubeatBase
from bemani.protocol import Node


class JubeatLoggerReportHandler(JubeatBase):
    def handle_logger_report_request(self, request: Node) -> Node:
        # Handle this by returning nothing, game doesn't care
        root = Node.void("logger")
        return root


class JubeatDemodataGetNewsHandler(JubeatBase):
    def handle_demodata_get_news_request(self, request: Node) -> Node:
        demodata = Node.void("demodata")
        data = Node.void("data")
        demodata.add_child(data)

        officialnews = Node.void("officialnews")
        data.add_child(officialnews)
        officialnews.set_attribute("count", "0")

        return demodata


class JubeatDemodataGetHitchartHandler(JubeatBase):
    def handle_demodata_get_hitchart_request(self, request: Node) -> Node:
        demodata = Node.void("demodata")
        data = Node.void("data")
        demodata.add_child(data)

        # Not sure what this is, maybe date?
        data.add_child(Node.string("update", time.strftime("%d/%m/%Y")))

        # No idea which songs are licensed or regular, so only return hit chart
        # for all songs on regular mode.
        hitchart_lic = Node.void("hitchart_lic")
        data.add_child(hitchart_lic)
        hitchart_lic.set_attribute("count", "0")

        songs = self.data.local.music.get_hit_chart(self.game, self.music_version, 10)
        hitchart_org = Node.void("hitchart_org")
        data.add_child(hitchart_org)
        hitchart_org.set_attribute("count", str(len(songs)))
        rank = 1
        for song in songs:
            rankdata = Node.void("rankdata")
            hitchart_org.add_child(rankdata)
            rankdata.add_child(Node.s32("music_id", song[0]))
            rankdata.add_child(Node.s16("rank", rank))
            rankdata.add_child(Node.s16("prev", rank))
            rank = rank + 1

        return demodata


class JubeatLobbyCheckHandler(JubeatBase):
    def handle_lobby_check_request(self, request: Node) -> Node:
        root = Node.void("lobby")
        data = Node.void("data")
        root.add_child(data)

        data.add_child(Node.s16("interval", 0))
        data.add_child(Node.s16("entry_timeout", 0))
        entrant_nr = Node.u32("entrant_nr", 0)
        entrant_nr.set_attribute("time", "0")
        data.add_child(entrant_nr)

        return root


class JubeatGamendRegisterHandler(JubeatBase):
    def handle_gameend_regist_request(self, request: Node) -> Node:
        data = request.child("data")
        player = data.child("player")

        if player is not None:
            refid = player.child_value("refid")
        else:
            refid = None

        if refid is not None:
            userid = self.data.remote.user.from_refid(self.game, self.version, refid)
        else:
            userid = None

        if userid is not None:
            oldprofile = self.get_profile(userid)
            newprofile = self.unformat_profile(userid, request, oldprofile)
        else:
            newprofile = None

        if userid is not None and newprofile is not None:
            self.put_profile(userid, newprofile)

        gameend = Node.void("gameend")
        data = Node.void("data")
        gameend.add_child(data)
        player = Node.void("player")
        data.add_child(player)
        player.add_child(Node.s32("session_id", 1))
        player.add_child(Node.s32("end_final_session_id", 1))
        return gameend


class JubeatGametopGetMeetingHandler(JubeatBase):
    def handle_gametop_get_meeting_request(self, request: Node) -> Node:
        gametop = Node.void("gametop")
        data = Node.void("data")
        gametop.add_child(data)
        meeting = Node.void("meeting")
        data.add_child(meeting)
        single = Node.void("single")
        meeting.add_child(single)
        single.set_attribute("count", "0")
        tag = Node.void("tag")
        meeting.add_child(tag)
        tag.set_attribute("count", "0")
        reward = Node.void("reward")
        data.add_child(reward)
        reward.add_child(Node.s32("total", -1))
        reward.add_child(Node.s32("point", -1))
        return gametop