reorganization - created package

2020-02-01 21:13:06 +09:00
parent 6c94f31dec
commit 88d94b78a0
16 changed files with 100 additions and 211 deletions
@@ -0,0 +1,122 @@
+import asyncio
+import logging
+from asyncio import BaseTransport, BaseProtocol
+from typing import Optional, Union, Tuple, Text
+
+from joycontrol.controller import Controller
+from joycontrol.report import OutputReport, SubCommand, InputReport
+
+logger = logging.getLogger(__name__)
+
+
+def controller_protocol_factory(controller: Controller):
+    def create_controller_protocol():
+        return ControllerProtocol(controller)
+    return create_controller_protocol
+
+
+class ControllerProtocol(BaseProtocol):
+    def __init__(self, controller: Controller):
+        self.controller = controller
+
+        self.transport = None
+
+        # This must always be an 0x21 input report to be compatible with button events
+        self.button_input_report = InputReport()
+        self.button_input_report.set_input_report_id(0x21)
+        self.button_input_report.set_misc()
+
+        self._data_received = asyncio.Event()
+
+    async def wait_for_output_report(self):
+        self._data_received.clear()
+        await self._data_received.wait()
+
+    def connection_made(self, transport: BaseTransport) -> None:
+        logger.debug('Connection established.')
+        self.transport = transport
+
+    def connection_lost(self, exc: Optional[Exception]) -> None:
+        raise NotImplementedError()
+
+    def error_received(self, exc: Exception) -> None:
+        raise NotImplementedError()
+
+    async def report_received(self, data: Union[bytes, Text], addr: Tuple[str, int]) -> None:
+        self._data_received.set()
+
+        try:
+            report = OutputReport(list(data))
+        except ValueError as v_err:
+            logger.warning(f'Report parsing error "{v_err}" - IGNORE')
+            return
+
+        # classify sub command
+        sc_byte, sub_command = report.get_sub_command()
+        logging.info(f'received output report - {sub_command}')
+        if sub_command is None:
+            logger.warning(f'Received output report does not contain a sub command')
+        elif sub_command == SubCommand.REQUEST_DEVICE_INFO:
+            await self._command_request_device_info(report)
+
+        elif sub_command == SubCommand.SET_SHIPMENT_STATE:
+            await self._command_set_shipment_state(report)
+
+        elif sub_command == SubCommand.SPI_FLASH_READ:
+            await self._command_spi_flash_read(report)
+
+        elif sub_command == SubCommand.SET_INPUT_REPORT_MODE:
+            await self._command_set_input_report_mode(report)
+
+        elif sub_command == SubCommand.TRIGGER_BUTTONS_ELAPSED_TIME:
+            await self._command_trigger_buttons_elapsed_time(report)
+
+        elif sub_command == SubCommand.NOT_IMPLEMENTED:
+            logger.warning(f'Sub command 0x{sc_byte:02x} not implemented - ignoring')
+
+    async def _command_request_device_info(self, output_report):
+        address = self.transport.get_extra_info('sockname')
+        assert address is not None
+        bd_address = list(map(lambda x: int(x, 16), address[0].split(':')))
+
+        self.button_input_report.set_misc()
+        self.button_input_report.set_ack(0x82)
+        self.button_input_report.sub_0x02_device_info(bd_address, controller=self.controller)
+
+        await self.button_input_report.write(self.transport)
+
+    async def _command_set_shipment_state(self, output_report):
+        self.button_input_report.set_misc()
+        self.button_input_report.set_ack(0x80)
+        self.button_input_report.sub_0x08_shipment()
+
+        await self.button_input_report.write(self.transport)
+
+    async def _command_spi_flash_read(self, output_report):
+        self.button_input_report.set_misc()
+        self.button_input_report.set_ack(0x90)
+        self.button_input_report.sub_0x10_spi_flash_read(output_report)
+
+        await self.button_input_report.write(self.transport)
+
+    async def _command_set_input_report_mode(self, output_report):
+        self.button_input_report.set_misc()
+        self.button_input_report.set_ack(0x80)
+        self.button_input_report.sub_0x03_set_input_report_mode()
+
+        await self.button_input_report.write(self.transport)
+
+    async def _command_trigger_buttons_elapsed_time(self, output_report):
+        self.button_input_report.set_misc()
+        self.button_input_report.set_ack(0x83)
+        self.button_input_report.sub_0x04_trigger_buttons_elapsed_time()
+
+        await self.button_input_report.write(self.transport)
+
+    async def _enable_6axis_sensor(self, output_report):
+        self.button_input_report.set_misc()
+        self.button_input_report.set_ack(0x80)
+
+        self.button_input_report.reply_to_subcommand_id(0x40)
+
+        await self.button_input_report.write(self.transport)