117 lines
5.1 KiB
Python
117 lines
5.1 KiB
Python
|
from typing import Optional
|
||
|
|
||
|
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.
|
||
|
"""
|
||
|
|
||
|
def handle_cardmng_request(self, request: Node) -> Optional[Node]:
|
||
|
"""
|
||
|
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.
|
||
|
"""
|
||
|
method = request.attribute('method')
|
||
|
|
||
|
if method == 'inquire':
|
||
|
# 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
|
||
|
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 None:
|
||
|
bound = False
|
||
|
else:
|
||
|
bound = oldgame.has_profile(userid)
|
||
|
expired = True
|
||
|
|
||
|
refid = self.data.local.user.get_refid(self.game, self.version, userid)
|
||
|
paseli_enabled = self.supports_paseli() and self.config['paseli']['enabled']
|
||
|
|
||
|
root = Node.void('cardmng')
|
||
|
root.set_attribute('refid', refid)
|
||
|
root.set_attribute('dataid', refid)
|
||
|
root.set_attribute('newflag', '1') # Always seems to be set to 1
|
||
|
root.set_attribute('binded', '1' if bound else '0') # Whether we've bound to this version of the game or not
|
||
|
root.set_attribute('expired', '1' if expired else '0') # Whether we're expired
|
||
|
root.set_attribute('ecflag', '1' if paseli_enabled else '0') # Whether to allow paseli
|
||
|
root.set_attribute('useridflag', '1')
|
||
|
root.set_attribute('extidflag', '1')
|
||
|
return root
|
||
|
|
||
|
elif method == 'authpass':
|
||
|
# 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
|
||
|
|
||
|
elif method == 'getrefid':
|
||
|
# 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
|
||
|
|
||
|
elif method == 'bindmodel':
|
||
|
# 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
|
||
|
|
||
|
elif method == 'getkeepspan':
|
||
|
# Unclear what this method does, return an arbitrary span
|
||
|
root = Node.void('cardmng')
|
||
|
root.set_attribute('keepspan', '30')
|
||
|
return root
|
||
|
|
||
|
elif method == 'getdatalist':
|
||
|
# Unclear what this method does, return a dummy response
|
||
|
root = Node.void('cardmng')
|
||
|
return root
|
||
|
|
||
|
# Invalid method
|
||
|
return None
|