feat: send amiibo report

2020-04-23 23:07:21 +08:00
parent e90499393b
commit be8dce71a0
6 changed files with 387 additions and 37 deletions
@@ -0,0 +1,104 @@
+import argparse
+import asyncio
+import logging
+import os
+from contextlib import contextmanager
+
+from joycontrol import logging_default as log
+from joycontrol.command_line_interface import ControllerCLI
+from joycontrol.controller_state import ControllerState, button_push, set_nfc
+from joycontrol.protocol import controller_protocol_factory, Controller
+from joycontrol.server import create_hid_server
+
+logger = logging.getLogger(__name__)
+
+
+async def _main(controller, reconnect_bt_addr=None, capture_file=None, spi_flash=None, device_id=None, amiibo=None):
+    factory = controller_protocol_factory(controller, spi_flash=spi_flash)
+    ctl_psm, itr_psm = 17, 19    
+    transport, protocol = await create_hid_server(factory, 
+        reconnect_bt_addr=reconnect_bt_addr,
+        ctl_psm=ctl_psm, itr_psm=itr_psm, capture_file=capture_file, device_id=device_id)
+
+    controller_state = protocol.get_controller_state()
+    if amiibo:
+        await set_nfc(controller_state, amiibo.read())
+
+    await controller_state.connect()
+
+    async def amiibo(filename):
+        with open(filename, "rb") as amiibo_file:
+            content = amiibo_file.read()
+            await set_nfc(controller_state, content)
+
+    async def remove_amiibo():
+        await controller_state.set_nfc(None)
+
+    cli = ControllerCLI(controller_state)
+    cli.add_command('amiibo', amiibo)
+    cli.add_command('remove_amiibo', remove_amiibo)
+    await cli.run()
+
+    logger.info('Stopping communication...')
+    await transport.close()
+
+
+if __name__ == '__main__':
+    # check if root
+    if not os.geteuid() == 0:
+        raise PermissionError('Script must be run as root!')
+
+    # setup logging
+    log.configure()
+
+    parser = argparse.ArgumentParser()
+    #parser.add_argument('controller', help='JOYCON_R, JOYCON_L or PRO_CONTROLLER')
+    parser.add_argument('-l', '--log')
+    parser.add_argument('-d', '--device_id')
+    parser.add_argument('--spi_flash')
+    parser.add_argument('-r', '--reconnect_bt_addr', type=str, default=None, 
+        help='The Switch console bluetooth address, for reconnecting as an already paired controller')
+    parser.add_argument('-a', '--amiibo', type=argparse.FileType('rb'), default=None,
+        help='The amiibo dump file')
+    args = parser.parse_args()
+
+    """
+    if args.controller == 'JOYCON_R':
+        controller = Controller.JOYCON_R
+    elif args.controller == 'JOYCON_L':
+        controller = Controller.JOYCON_L
+    elif args.controller == 'PRO_CONTROLLER':
+        controller = Controller.PRO_CONTROLLER
+    else:
+        raise ValueError(f'Unknown controller "{args.controller}".')
+    """
+    controller = Controller.PRO_CONTROLLER
+
+    spi_flash = None
+    if args.spi_flash:
+        with open(args.spi_flash, 'rb') as spi_flash_file:
+            spi_flash = spi_flash_file.read()
+
+    # creates file if arg is given
+    @contextmanager
+    def get_output(path=None):
+        """
+        Opens file if path is given
+        """
+        if path is not None:
+            file = open(path, 'wb')
+            yield file
+            file.close()
+        else:
+            yield None
+
+    with get_output(args.log) as capture_file:
+        loop = asyncio.get_event_loop()
+        loop.run_until_complete(_main(
+            controller,
+            reconnect_bt_addr=args.reconnect_bt_addr,
+            capture_file=capture_file,
+            spi_flash=spi_flash,
+            device_id=args.device_id,
+            amiibo=args.amiibo
+        ))