import asyncio

import json

import logging

import time

from typing import Any, Awaitable, Dict, List, Protocol, Tuple

from light9.collector.output import Output as OutputInstance

from light9.newtypes import (DeviceUri, DmxIndex, DmxMessageIndex, OutputAttr, OutputUri, OutputValue)

log = logging.getLogger('weblisteners')

class UiListener(Protocol):

    async def sendMessage(self, msg):

        ...

class WebListeners(object):

    def __init__(self) -> None:

        self.clients: List[Tuple[UiListener, Dict[DeviceUri, Dict[OutputAttr, OutputValue]]]] = []

        self.pendingMessageForDev: Dict[DeviceUri, Tuple[Dict[OutputAttr, OutputValue], Dict[Tuple[DeviceUri, OutputAttr], Tuple[OutputUri,

                                                                                                                                 DmxMessageIndex]]]] = {}

                if msg is None:

                    msg = self.makeMsg(dev, attrs, outputMap)

                seen[dev] = attrs

                sendAwaits.append(client.sendMessage(msg))

            await asyncio.gather(*sendAwaits)

    def makeMsg(self, dev: DeviceUri, attrs: Dict[OutputAttr, Any], outputMap: Dict[Tuple[DeviceUri, OutputAttr], Tuple[OutputUri, DmxMessageIndex]]):

        attrRows = []

        for attr, val in attrs.items():

            outputUri, bufIndex = outputMap[(dev, attr)]

            dmxIndex = DmxIndex(bufIndex + 1)

        attrRows.sort(key=lambda r: r['chan'])

        for row in attrRows:

            row['chan'] = '%s %s' % (row['chan'][0], row['chan'][1])

        msg = json.dumps({'outputAttrsSet': {'dev': dev, 'attrs': attrRows}}, sort_keys=True)

        return msg