#!/usr/bin/env python3
"""Controls the LEDs on the ReSpeaker Mic Array v2.0 (USB) ."""
import argparse
import asyncio
import logging
import time
from functools import partial

from pixel_ring import pixel_ring
from wyoming.event import Event
from wyoming.satellite import (
    SatelliteConnected,
    SatelliteDisconnected,
    StreamingStarted,
    StreamingStopped,
)
from wyoming.server import AsyncEventHandler, AsyncServer
from wyoming.snd import Played
from wyoming.vad import VoiceStarted, VoiceStopped
from wyoming.wake import Detection

_LOGGER = logging.getLogger()


async def main() -> None:
    """Main entry point."""
    parser = argparse.ArgumentParser()
    parser.add_argument("--uri", required=True, help="unix:// or tcp://")
    parser.add_argument("--debug", action="store_true", help="Log DEBUG messages")
    args = parser.parse_args()

    logging.basicConfig(level=logging.DEBUG if args.debug else logging.INFO)
    _LOGGER.debug(args)

    _LOGGER.info("Ready")

    # Turn on power to LEDs
    pixel_ring.set_color_palette(0x0080FF, 0x007A37)
    pixel_ring.think()
    await asyncio.sleep(3)
    pixel_ring.off()

    # Start server
    server = AsyncServer.from_uri(args.uri)

    try:
        await server.run(partial(LEDsEventHandler, args))
    except KeyboardInterrupt:
        pass
    finally:
        pixel_ring.off()


class LEDsEventHandler(AsyncEventHandler):
    """Event handler for clients."""

    def __init__(
        self,
        cli_args: argparse.Namespace,
        *args,
        **kwargs,
    ) -> None:
        super().__init__(*args, **kwargs)

        self.cli_args = cli_args
        self.client_id = str(time.monotonic_ns())

        _LOGGER.debug("Client connected: %s", self.client_id)

    async def handle_event(self, event: Event) -> bool:
        _LOGGER.debug(event)

        if Detection.is_type(event.type):
            _LOGGER.debug("Detection")
            pixel_ring.wakeup()
        elif VoiceStarted.is_type(event.type):
            _LOGGER.debug("VoiceStarted")
            pixel_ring.speak()
        elif VoiceStopped.is_type(event.type):
            _LOGGER.debug("VoiceStopped")
            pixel_ring.spin()
        elif StreamingStopped.is_type(event.type):
            _LOGGER.debug("StreamingStopped")
            pixel_ring.off()
        elif SatelliteConnected.is_type(event.type):
            _LOGGER.debug("SatelliteConnected")
            pixel_ring.think()
            await asyncio.sleep(2)
            pixel_ring.off()
        elif Played.is_type(event.type):
            _LOGGER.debug("Played")
            pixel_ring.off()
        elif SatelliteDisconnected.is_type(event.type):
            _LOGGER.debug("SatelliteDisconnected")
            pixel_ring.mono(0xFF0000)

        return True


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        pass