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
        ))