Added joycontrol-pluginloader

Author: Almtr

JoycontrolPlugin/__init__.py

@@ -0,0 +1,3 @@
+from JoycontrolPlugin.commands import JoycontrolCommands
+from JoycontrolPlugin.plugin import JoycontrolPlugin
+from JoycontrolPlugin.loader import load_plugin

JoycontrolPlugin/commands.py

@@ -0,0 +1,90 @@
+import math
+import asyncio
+import time
+import logging
+
+from joycontrol.controller_state import ControllerState
+from joycontrol.command_line_interface import ControllerCLI
+
+logger = logging.getLogger(__name__)
+
+MAX_STICK_POWER = 0xFFF/2
+
+class JoycontrolCommands:
+    def __init__(self, controller_state):
+        self.cli = ControllerCLI(controller_state)
+        self.controller_state = controller_state
+        self.max_stick_power = MAX_STICK_POWER
+
+    def __calc_stick_position(self, angle, power):
+        angle = (angle + 180) %  360 * -1
+
+        rad = math.radians(angle)
+        x = power * math.cos(rad)
+        y = power * math.sin(rad)
+
+        # Adjust the position of the circle.
+        stick_radius = MAX_STICK_POWER
+        h_val = x + stick_radius
+        v_val = y + stick_radius
+        return int(h_val), int(v_val)
+
+    async def stick(self, stick, direction=None, angle=None, power=MAX_STICK_POWER):
+        if direction == 'center':
+            angle = 0
+            power = 0
+        elif direction == 'left':
+            angle = 0
+        elif direction == 'up':
+            angle = 90
+        elif direction == 'right':
+            angle = 180
+        elif direction == 'down':
+            angle = 270
+        
+        if angle is None:
+            raise ValueError('Missing angle')
+
+        logger.debug(f'Stick: {stick}, {angle}, {power}')
+        h_val, v_val = self.__calc_stick_position(angle, power)
+        await self.cli.cmd_stick(stick, 'horizontal', h_val)
+        await self.cli.cmd_stick(stick, 'vertical', v_val)
+
+    async def left_stick(self, direction=None, angle=None, power=MAX_STICK_POWER):
+        await self.stick('left', direction, angle, power)
+
+    async def right_stick(self, direction=None, angle=None, power=MAX_STICK_POWER):
+        await self.stick('right', direction, angle, power)
+
+    async def button_press(self, *buttons):
+        if not buttons:
+            raise ValueError('No Buttons were given.')
+        
+        logger.debug('Press {}'.format(', '.join(buttons)))
+        button_state = self.controller_state.button_state
+        for button in buttons:
+            button_state.set_button(button)
+
+        # send report
+        await self.controller_state.send()
+    
+    async def button_release(self, *buttons):
+        if not buttons:
+            raise ValueError('No Buttons were given.')
+        
+        logger.debug('Release {}'.format(', '.join(buttons)))
+        button_state = self.controller_state.button_state
+        for button in buttons:
+            button_state.set_button(button, pushed=False)
+
+        # send report
+        await self.controller_state.send()
+
+    async def button_push(self, *buttons, press_time_sec=0.1):
+        await self.button_press(*buttons)
+        await self.wait(press_time_sec)
+        await self.button_release(*buttons)
+
+    async def wait(self, sec):
+        logger.debug(f'Wait: {sec}')
+        await asyncio.sleep(sec)

JoycontrolPlugin/loader.py

@@ -0,0 +1,13 @@
+import os
+import logging
+
+from importlib.machinery import SourceFileLoader
+
+logger = logging.getLogger(__name__)
+
+def load_plugin(plugin, controller_state, *plugin_options):
+    logger.info(f'Loading: {plugin}')
+    module = SourceFileLoader(plugin, plugin).load_module()
+    plugin_name = os.path.splitext(os.path.basename(plugin))[0]
+    joycontrol_plugin = module.__dict__[plugin_name](controller_state, *plugin_options)
+    return joycontrol_plugin

JoycontrolPlugin/plugin.py

@@ -0,0 +1,11 @@
+from JoycontrolPlugin.commands import JoycontrolCommands
+from abc import abstractmethod
+
+class JoycontrolPlugin(JoycontrolCommands):
+    def __init__(self, controller_state, options):
+        super().__init__(controller_state)
+        self.options = options
+
+    @abstractmethod
+    async def run(self):
+        pass

joycontrol-pluginloader.py

@@ -0,0 +1,70 @@
+#!/usr/bin/env python3
+
+import argparse
+import asyncio
+import logging
+import os
+
+from aioconsole import ainput
+from joycontrol import logging_default as log, utils
+from joycontrol.controller import Controller
+from joycontrol.controller_state import ControllerState, button_push
+from joycontrol.memory import FlashMemory
+from joycontrol.protocol import controller_protocol_factory
+from joycontrol.server import create_hid_server
+from importlib.machinery import SourceFileLoader
+from JoycontrolPlugin.loader import load_plugin
+
+logger = logging.getLogger(__name__)
+
+async def _main(args):
+    # Create memory containing default controller stick calibration
+    spi_flash = FlashMemory()
+
+    # Get controller name to emulate from arguments
+    controller = Controller.from_arg('PRO_CONTROLLER')
+
+    with utils.get_output(path=None, default=None) as capture_file:
+        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=args.reconnect_bt_addr,
+                                                      ctl_psm=ctl_psm,
+                                                      itr_psm=itr_psm, capture_file=capture_file,
+                                                      device_id=args.device_id)
+
+        controller_state = protocol.get_controller_state()
+
+        try:
+            # waits until controller is fully connected
+            await controller_state.connect()
+            joycontrol_plugin = load_plugin(args.plugin, controller_state, args.plugin_options)
+            await joycontrol_plugin.run()
+        except Exception as e:
+            logger.error(e)
+        finally:
+            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!')
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument('plugin', type=str, help='joycontrol plugin path')
+    parser.add_argument('plugin_options', nargs='*', help='joycontrol plugin options')
+    parser.add_argument('-d', '--device_id')
+    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('-v', '--verbose', action='store_true')
+    args = parser.parse_args()
+
+    if args.verbose:
+        log.configure()
+    else:
+        log.configure(console_level=logging.INFO)
+
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(
+        _main(args)
+    )

setup.py

@@ -0,0 +1,18 @@
+
+from setuptools import setup, find_packages
+
+README = open('README.md', 'r').read()
+
+setup(
+    name='joycontrol-pluginloader',
+    version='0.1',
+    long_description=README,
+    long_description_content_type='text/markdown',
+    url='https://github.com/almtr/joycontrol-pluginloader',
+    author='almtr',
+    description='PluginLoader for mart1nro/joycontrol',
+    packages=find_packages(),
+    install_requires=[
+        'hid', 'aioconsole', 'dbus-python', 'crc8'
+    ],
+)