forked from mirror/joycontrol
spacemeowx2
105 lines
3.4 KiB
Python
105 lines
3.4 KiB
Python
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
|
|
))
|