ledgrab-haos-integration/custom_components/ledgrab/switch.py

"""Switch platform for LED Screen Controller."""
from __future__ import annotations

import logging
from typing import Any

from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity

from .const import DOMAIN, DATA_COORDINATOR
from .coordinator import LedGrabCoordinator

_LOGGER = logging.getLogger(__name__)


async def async_setup_entry(
    hass: HomeAssistant,
    entry: ConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up LED Screen Controller switches."""
    data = hass.data[DOMAIN][entry.entry_id]
    coordinator: LedGrabCoordinator = data[DATA_COORDINATOR]

    entities: list[SwitchEntity] = []
    if coordinator.data and "targets" in coordinator.data:
        for target_id, target_data in coordinator.data["targets"].items():
            entities.append(
                LedGrabSwitch(coordinator, target_id, entry.entry_id)
            )

    if coordinator.data:
        for clock in coordinator.data.get("sync_clocks", []):
            entities.append(
                LedGrabSyncClockSwitch(coordinator, clock["id"], entry.entry_id)
            )

    async_add_entities(entities)


class LedGrabSwitch(CoordinatorEntity, SwitchEntity):
    """Representation of a LED Screen Controller target processing switch."""

    _attr_has_entity_name = True

    def __init__(
        self,
        coordinator: LedGrabCoordinator,
        target_id: str,
        entry_id: str,
    ) -> None:
        """Initialize the switch."""
        super().__init__(coordinator)
        self._target_id = target_id
        self._entry_id = entry_id
        self._attr_unique_id = f"{target_id}_processing"
        self._attr_translation_key = "processing"
        self._attr_icon = "mdi:television-ambient-light"

    @property
    def device_info(self) -> dict[str, Any]:
        """Return device information."""
        return {"identifiers": {(DOMAIN, self._target_id)}}

    @property
    def is_on(self) -> bool:
        """Return true if processing is active."""
        target_data = self._get_target_data()
        if not target_data or not target_data.get("state"):
            return False
        return target_data["state"].get("processing", False)

    @property
    def available(self) -> bool:
        """Return if entity is available."""
        return self._get_target_data() is not None

    @property
    def extra_state_attributes(self) -> dict[str, Any]:
        """Return additional state attributes."""
        target_data = self._get_target_data()
        if not target_data:
            return {}

        attrs: dict[str, Any] = {"target_id": self._target_id}
        state = target_data.get("state") or {}
        metrics = target_data.get("metrics") or {}

        if state:
            attrs["fps_target"] = state.get("fps_target")
            attrs["fps_actual"] = state.get("fps_actual")

        if metrics:
            attrs["frames_processed"] = metrics.get("frames_processed")
            attrs["errors_count"] = metrics.get("errors_count")
            attrs["uptime_seconds"] = metrics.get("uptime_seconds")

        return attrs

    async def async_turn_on(self, **kwargs: Any) -> None:
        """Start processing."""
        await self.coordinator.start_processing(self._target_id)

    async def async_turn_off(self, **kwargs: Any) -> None:
        """Stop processing."""
        await self.coordinator.stop_processing(self._target_id)

    def _get_target_data(self) -> dict[str, Any] | None:
        """Get target data from coordinator."""
        if not self.coordinator.data:
            return None
        return self.coordinator.data.get("targets", {}).get(self._target_id)


class LedGrabSyncClockSwitch(CoordinatorEntity, SwitchEntity):
    """Running/paused control for a sync clock.

    On = clock running (linked animations advance), off = paused (frozen).
    """

    _attr_has_entity_name = True
    _attr_icon = "mdi:clock-outline"

    def __init__(
        self,
        coordinator: LedGrabCoordinator,
        clock_id: str,
        entry_id: str,
    ) -> None:
        super().__init__(coordinator)
        self._clock_id = clock_id
        self._entry_id = entry_id
        self._attr_unique_id = f"{clock_id}_running"
        self._attr_translation_key = "sync_clock_running"

    @property
    def device_info(self) -> dict[str, Any]:
        return {"identifiers": {(DOMAIN, self._clock_id)}}

    @property
    def is_on(self) -> bool:
        clock = self._get_clock()
        if not clock:
            return False
        return bool(clock.get("is_running", False))

    @property
    def available(self) -> bool:
        return self._get_clock() is not None

    async def async_turn_on(self, **kwargs: Any) -> None:
        await self.coordinator.resume_sync_clock(self._clock_id)

    async def async_turn_off(self, **kwargs: Any) -> None:
        await self.coordinator.pause_sync_clock(self._clock_id)

    def _get_clock(self) -> dict[str, Any] | None:
        if not self.coordinator.data:
            return None
        for clock in self.coordinator.data.get("sync_clocks", []):
            if clock.get("id") == self._clock_id:
                return clock
        return None