125 lines
5.5 KiB
Python
125 lines
5.5 KiB
Python
from bemani.backend.base import Base, Status
|
|
from bemani.protocol import Node
|
|
from bemani.common import Model
|
|
|
|
|
|
class CardManagerHandler(Base):
|
|
"""
|
|
The class that handles card management. This assumes it is attached as a mixin to a game
|
|
class so that it can understand if there's a profile for a game or not.
|
|
|
|
Handle a request for card management. This is independent of a game's profile handling,
|
|
but still gives the game information as to whether or not a profile exists for a game.
|
|
These methods handle looking up a card, handling binding a profile to a game version,
|
|
returning whether a game profile exists or should be migrated, and creating a new account
|
|
when no account is associated with a card.
|
|
"""
|
|
|
|
def handle_cardmng_inquire_request(self, request: Node) -> Node:
|
|
# Given a cardid, look up the dataid/refid (same thing in this system).
|
|
# If the card doesn't exist or isn't allowed, return a status specifying this
|
|
# instead of the results of the dataid/refid lookup.
|
|
cardid = request.attribute("cardid")
|
|
modelstring = request.attribute("model")
|
|
userid = self.data.local.user.from_cardid(cardid)
|
|
|
|
if userid is None:
|
|
# This user doesn't exist, force system to create new account
|
|
root = Node.void("cardmng")
|
|
root.set_attribute("status", str(Status.NOT_REGISTERED))
|
|
return root
|
|
|
|
# Special handling for looking up whether the previous game's profile existed. If we
|
|
# don't do this then some games won't present the user with a migration.
|
|
bound = self.has_profile(userid)
|
|
expired = False
|
|
if bound is False:
|
|
if modelstring is not None:
|
|
model = Model.from_modelstring(modelstring)
|
|
oldgame = Base.create(self.data, self.config, model, self.model)
|
|
if oldgame is not None:
|
|
bound = oldgame.has_profile(userid)
|
|
expired = self.supports_expired_profiles
|
|
|
|
refid = self.data.local.user.get_refid(self.game, self.version, userid)
|
|
paseli_enabled = self.supports_paseli and self.config.paseli.enabled
|
|
newflag = (
|
|
self.data.remote.user.get_any_profile(self.game, self.version, userid)
|
|
is None
|
|
)
|
|
|
|
root = Node.void("cardmng")
|
|
root.set_attribute("refid", refid)
|
|
root.set_attribute("dataid", refid)
|
|
|
|
# Unsure what this does, but it appears not to matter so we set it to my best guess.
|
|
root.set_attribute("newflag", "1" if newflag else "0")
|
|
|
|
# Whether we've bound a profile to this refid/dataid or not. This includes current profiles and any
|
|
# older game profiles that might exist that we should do a conversion from.
|
|
root.set_attribute("binded", "1" if bound else "0")
|
|
|
|
# Whether this version of the profile is expired (was converted to newer version). We support forwards
|
|
# and backwards compatibility so some games will always set this to 0.
|
|
root.set_attribute("expired", "1" if expired else "0")
|
|
|
|
# Whether to allow paseli, as enabled by the operator and arcade owner.
|
|
root.set_attribute("ecflag", "1" if paseli_enabled else "0")
|
|
|
|
# I have absolutely no idea what these do.
|
|
root.set_attribute("useridflag", "1")
|
|
root.set_attribute("extidflag", "1")
|
|
return root
|
|
|
|
def handle_cardmng_authpass_request(self, request: Node) -> Node:
|
|
# Given a dataid/refid previously found via inquire, verify the pin
|
|
refid = request.attribute("refid")
|
|
pin = request.attribute("pass")
|
|
userid = self.data.local.user.from_refid(self.game, self.version, refid)
|
|
if userid is not None:
|
|
valid = self.data.local.user.validate_pin(userid, pin)
|
|
else:
|
|
valid = False
|
|
root = Node.void("cardmng")
|
|
root.set_attribute(
|
|
"status", str(Status.SUCCESS if valid else Status.INVALID_PIN)
|
|
)
|
|
return root
|
|
|
|
def handle_cardmng_getrefid_request(self, request: Node) -> Node:
|
|
# Given a cardid and a pin, register the card with the system and generate a new dataid/refid + extid
|
|
cardid = request.attribute("cardid")
|
|
pin = request.attribute("passwd")
|
|
userid = self.data.local.user.create_account(cardid, pin)
|
|
if userid is None:
|
|
# This user can't be created
|
|
root = Node.void("cardmng")
|
|
root.set_attribute("status", str(Status.NOT_ALLOWED))
|
|
return root
|
|
|
|
refid = self.data.local.user.create_refid(self.game, self.version, userid)
|
|
root = Node.void("cardmng")
|
|
root.set_attribute("dataid", refid)
|
|
root.set_attribute("refid", refid)
|
|
return root
|
|
|
|
def handle_cardmng_bindmodel_request(self, request: Node) -> Node:
|
|
# Given a refid, bind the user's card to the current version of the game
|
|
refid = request.attribute("refid")
|
|
userid = self.data.local.user.from_refid(self.game, self.version, refid)
|
|
self.bind_profile(userid)
|
|
root = Node.void("cardmng")
|
|
root.set_attribute("dataid", refid)
|
|
return root
|
|
|
|
def handle_cardmng_getkeepspan_request(self, request: Node) -> Node:
|
|
# Unclear what this method does, return an arbitrary span
|
|
root = Node.void("cardmng")
|
|
root.set_attribute("keepspan", "30")
|
|
return root
|
|
|
|
def handle_cardmng_getdatalist_request(self, request: Node) -> Node:
|
|
# Unclear what this method does, return a dummy response
|
|
root = Node.void("cardmng")
|
|
return root
|