class MrpPairingHandler(PairingHandler):
"""Base class for API used to pair with an Apple TV."""
def __init__(self, config, session_manager: ClientSessionManager, loop):
"""Initialize a new MrpPairingHandler."""
super().__init__(session_manager, config.get_service(Protocol.MRP))
self.connection = MrpConnection(config.address, self.service.port,
loop)
self.srp = SRPAuthHandler()
self.protocol = MrpProtocol(self.connection, self.srp, self.service)
self.pairing_procedure = MrpPairingProcedure(self.protocol, self.srp)
self.pin_code = None
self._has_paired = False
async def close(self):
"""Call to free allocated resources after pairing."""
self.connection.close()
await super().close()
@property
def has_paired(self):
"""If a successful pairing has been performed."""
return self._has_paired
async def begin(self):
"""Start pairing process."""
return await error_handler(self.pairing_procedure.start_pairing,
exceptions.PairingError)
async def finish(self):
"""Stop pairing process."""
if not self.pin_code:
raise exceptions.PairingError("no pin given")
credentials = str(await error_handler(
self.pairing_procedure.finish_pairing,
exceptions.PairingError,
self.pin_code,
))
_LOGGER.debug("Verifying credentials %s", credentials)
verifier = MrpPairingVerifier(self.protocol, self.srp,
Credentials.parse(credentials))
await error_handler(verifier.verify_credentials,
exceptions.PairingError)
self.service.credentials = credentials
self._has_paired = True
@property
def device_provides_pin(self):
"""Return True if remote device presents PIN code, else False."""
return True
def pin(self, pin):
"""Pin code used for pairing."""
self.pin_code = str(pin).zfill(4)
_LOGGER.debug("MRP PIN changed to %s", self.pin_code)
class MrpPairingHandler(PairingHandler):
"""Base class for API used to pair with an Apple TV."""
def __init__(self, config, session, loop):
"""Initialize a new MrpPairingHandler."""
super().__init__(session, config.get_service(Protocol.MRP))
self.connection = MrpConnection(config.address, self.service.port,
loop)
self.srp = SRPAuthHandler()
self.protocol = MrpProtocol(loop, self.connection, self.srp,
self.service)
self.pairing_procedure = MrpPairingProcedure(self.protocol, self.srp)
self.pin_code = None
async def close(self):
"""Call to free allocated resources after pairing."""
self.connection.close()
await super().close()
@property
def has_paired(self):
"""If a successful pairing has been performed."""
return self.service.credentials is not None
def begin(self):
"""Start pairing process."""
return error_handler(self.pairing_procedure.start_pairing,
exceptions.PairingError)
async def finish(self):
"""Stop pairing process."""
if not self.pin_code:
raise exceptions.PairingError("no pin given")
self.service.credentials = str(await error_handler(
self.pairing_procedure.finish_pairing,
exceptions.PairingError,
self.pin_code,
))
@property
def device_provides_pin(self):
"""Return True if remote device presents PIN code, else False."""
return True
def pin(self, pin):
"""Pin code used for pairing."""
self.pin_code = str(pin).zfill(4)