import time
import unittest

from pylgbst.movehub import MoveHub, MOVE_HUB_HARDWARE_HANDLE, PORT_LED, COLOR_RED
from pylgbst.peripherals import LED, TiltSensor, COLORS
from tests import log, HubMock, ConnectionMock, Thread

HANDLE = MOVE_HUB_HARDWARE_HANDLE


class GeneralTest(unittest.TestCase):
    def _wait_notifications_handled(self, hub):
        hub.connection.running = False
        for _ in range(1, 180):
            time.sleep(1)
            log.debug("Waiting for notifications to process...")
            if hub.connection.finished:
                log.debug("Done waiting for notifications to process")
                break

    def test_led(self):
        hub = HubMock()
        led = LED(hub, PORT_LED)
        led.set_color(COLOR_RED)
        self.assertEqual("0801813201510009", hub.writes[0][1])

    def test_tilt_sensor(self):
        hub = HubMock()
        hub.notify_mock.append((HANDLE, '0f00 04 3a 0128000000000100000001'))
        time.sleep(1)

        def callback(param1, param2=None, param3=None):
            if param2 is None:
                log.debug("Tilt: %s", TiltSensor.DUO_STATES[param1])
            else:
                log.debug("Tilt: %s %s %s", param1, param2, param3)

        self._inject_notification(hub, '0a00 47 3a 090100000001', 1)
        hub.tilt_sensor.subscribe(callback)
        hub.notify_mock.append((HANDLE, "0500453a05"))
        hub.notify_mock.append((HANDLE, "0a00473a010100000001"))
        time.sleep(1)
        self._inject_notification(hub, '0a00 47 3a 090100000001', 1)
        hub.tilt_sensor.subscribe(callback, TiltSensor.MODE_2AXIS_SIMPLE)

        hub.notify_mock.append((HANDLE, "0500453a09"))
        time.sleep(1)

        self._inject_notification(hub, '0a00 47 3a 090100000001', 1)
        hub.tilt_sensor.subscribe(callback, TiltSensor.MODE_2AXIS_FULL)
        hub.notify_mock.append((HANDLE, "0600453a04fe"))
        time.sleep(1)

        self._inject_notification(hub, '0a00 47 3a 090100000001', 1)
        hub.tilt_sensor.unsubscribe(callback)
        self._wait_notifications_handled(hub)
        # TODO: assert

    def test_motor(self):
        conn = ConnectionMock()
        conn.notifications.append((14, '0900 04 39 0227003738'))
        hub = HubMock(conn)
        time.sleep(0.1)

        conn.notifications.append((14, '050082390a'))
        hub.motor_AB.timed(1.5)
        self.assertEqual("0d018139110adc056464647f03", conn.writes[0][1])

        conn.notifications.append((14, '050082390a'))
        hub.motor_AB.angled(90)
        self.assertEqual("0f018139110c5a0000006464647f03", conn.writes[1][1])

    def test_capabilities(self):
        conn = ConnectionMock()
        conn.notifications.append((14, '0f00 04 01 0125000000001000000010'))
        conn.notifications.append((14, '0f00 04 02 0126000000001000000010'))
        conn.notifications.append((14, '0f00 04 37 0127000100000001000000'))
        conn.notifications.append((14, '0f00 04 38 0127000100000001000000'))
        conn.notifications.append((14, '0900 04 39 0227003738'))
        conn.notifications.append((14, '0f00 04 32 0117000100000001000000'))
        conn.notifications.append((14, '0f00 04 3a 0128000000000100000001'))
        conn.notifications.append((14, '0f00 04 3b 0115000200000002000000'))
        conn.notifications.append((14, '0f00 04 3c 0114000200000002000000'))
        conn.notifications.append((14, '0f00 8202 01'))
        conn.notifications.append((14, '0f00 8202 0a'))

        self._inject_notification(conn, '1200 0101 06 4c45474f204d6f766520487562', 1)
        self._inject_notification(conn, '1200 0108 06 4c45474f204d6f766520487562', 2)
        self._inject_notification(conn, '0900 47 3c 0227003738', 3)
        self._inject_notification(conn, '0600 45 3c 020d', 4)
        self._inject_notification(conn, '0900 47 3c 0227003738', 5)
        hub = MoveHub(conn)
        # demo_all(hub)
        self._wait_notifications_handled(hub)

    def test_color_sensor(self):
        #
        hub = HubMock()
        hub.notify_mock.append((HANDLE, '0f00 04010125000000001000000010'))
        time.sleep(1)

        def callback(color, unk1, unk2=None):
            name = COLORS[color] if color is not None else 'NONE'
            log.info("Color: %s %s %s", name, unk1, unk2)

        self._inject_notification(hub, '0a00 4701090100000001', 1)
        hub.color_distance_sensor.subscribe(callback)

        hub.notify_mock.append((HANDLE, "08004501ff0aff00"))
        time.sleep(1)
        # TODO: assert
        self._inject_notification(hub, '0a00 4701090100000001', 1)
        hub.color_distance_sensor.unsubscribe(callback)
        self._wait_notifications_handled(hub)

    def test_button(self):
        hub = HubMock()
        time.sleep(1)

        def callback(pressed):
            log.info("Pressed: %s", pressed)

        hub.notify_mock.append((HANDLE, "060001020600"))
        hub.button.subscribe(callback)

        hub.notify_mock.append((HANDLE, "060001020601"))
        hub.notify_mock.append((HANDLE, "060001020600"))
        time.sleep(1)
        # TODO: assert
        hub.button.unsubscribe(callback)
        self._wait_notifications_handled(hub)

    def _inject_notification(self, hub, notify, pause):
        def inject():
            time.sleep(pause)
            if isinstance(hub, ConnectionMock):
                hub.notifications.append((HANDLE, notify))
            else:
                hub.notify_mock.append((HANDLE, notify))

        Thread(target=inject).start()