forked from mirror/joycontrol
Robert Martin
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
import asyncio
|
|
|
|
from button_state import ButtonState
|
|
from protocol import ControllerProtocol
|
|
|
|
|
|
class ControllerState:
|
|
def __init__(self, transport: asyncio.Transport, protocol: ControllerProtocol):
|
|
super().__init__()
|
|
self.transport = transport
|
|
|
|
self.protocol = protocol
|
|
|
|
async def send(self):
|
|
await self.protocol.button_input_report.write(self.transport)
|
|
|
|
async def connect(self):
|
|
"""
|
|
Waits until the switch is paired with the controller and accepts button commands
|
|
"""
|
|
# TODO HACK: Hard to say for now.
|
|
await self.protocol.wait_for_output_report()
|
|
# The switch sends data to our device, it shouldn't take long until the connection is fully established.
|
|
await asyncio.sleep(5)
|
|
|
|
def set_button_state(self, button_state: ButtonState):
|
|
"""
|
|
Sets the button status bytes in the input report
|
|
"""
|
|
self.protocol.button_input_report.set_button_status(button_state)
|
|
|
|
def set_stick_state(self):
|
|
"""
|
|
TODO
|
|
"""
|
|
raise NotImplementedError()
|
|
|
|
|