from __future__ import annotations
from typing import Callable, Awaitable
import asyncio
import logging

from duotecno.protocol import NodeType, EV_NODEDATABASEINFO_2, BaseMessage
from duotecno.unit import (
    BaseUnit,
    SwitchUnit,
    SensUnit,
    DimUnit,
    DuoswitchUnit,
    VirtualUnit,
    ControlUnit,
)


class Node:
    name: str
    index: int
    nodeType: NodeType
    address: int
    numUnits: int
    units: dict[int, BaseUnit]
    isloaded: asyncio.Event

    def __init__(
        self,
        name: str,
        address: int,
        index: int,
        nodeType: NodeType,
        numUnits: int,
        writer: Callable[[str], Awaitable[None]],
        pwaiter: Callable[[str], Awaitable[None]],
    ) -> None:
        self._log = logging.getLogger("pyduotecno-node")
        self.name = name
        self.address = address
        self.index = index
        self.numUnits = numUnits
        self.nodeType = nodeType
        self.writer = writer
        self.pwaiter = pwaiter
        self.isLoaded = asyncio.Event()
        self.isLoaded.clear()
        self.units = {}
        self._log.info(f"New node found: {self.name}")

    async def enable(self) -> None:
        for unit in self.units.values():
            await unit.enable()

    async def disable(self) -> None:
        for unit in self.units.values():
            await unit.disable()

    def get_name(self) -> str:
        return self.name

    def get_address(self) -> int:
        return self.address

    def __repr__(self) -> str:
        items = []
        for k, v in self.__dict__.items():
            if k not in ["_log", "writer"]:
                items.append(f"{k} = {v!r}")
        return "{}[{}]".format(type(self), ", ".join(items))

    def get_units(self) -> list[BaseUnit]:
        res = []
        for unit in self.units.values():
            res.append(unit)
        return res

    def get_unit_by_type(self, unit_type: list[str] | str) -> list[BaseUnit]:
        if isinstance(unit_type, str):
            unit_type = [unit_type]
        res = []
        for unit in self.units.values():
            for unitT in unit_type:
                if str(type(unit)) == f"<class 'duotecno.unit.{unitT}'>":
                    res.append(unit)
        return res

    async def load(self) -> None:
        self._log.debug(f"Node {self.name}: Requesting units")
        for i in range(self.numUnits):
            await self.writer(f"[209,2,{self.address},{i}]")
            await self.pwaiter(f"64,2,{self.address},{i}")

    async def handlePacket(self, packet: BaseMessage) -> None:
        if isinstance(packet, EV_NODEDATABASEINFO_2):
            if packet.unit not in self.units:
                u = BaseUnit
                if packet.unitTypeName == "SWITCH":
                    u = SwitchUnit
                elif packet.unitTypeName == "SENS":
                    u = SensUnit
                elif packet.unitTypeName == "DIM":
                    u = DimUnit
                elif packet.unitTypeName == "DUOSWITCH":
                    u = DuoswitchUnit
                elif packet.unitTypeName == "VIRTUAL":
                    u = VirtualUnit
                elif packet.unitTypeName == "CONTROL":
                    u = ControlUnit
                else:
                    self._log.warning(f"Unhandled unitType: {packet.unitTypeName}")
                self.units[packet.unit] = u(
                    self, name=packet.unitName, unit=packet.unit, writer=self.writer
                )
            if len(self.units) == self.numUnits:
                self.isLoaded.set()
            return
        if hasattr(packet, "unit") and packet.unit in self.units:
            await self.units[packet.unit].handlePacket(packet)
            return