"""Balboa entities."""
from __future__ import annotations

from pybalboa import EVENT_UPDATE, SpaClient

from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
from homeassistant.helpers.entity import DeviceClassName, DeviceInfo, Entity

from .const import DOMAIN


class BalboaBaseEntity(Entity):
    """Balboa base entity."""

    def __init__(
        self, client: SpaClient, name: str | DeviceClassName | None = None
    ) -> None:
        """Initialize the control."""
        mac = client.mac_address
        model = client.model

        self._attr_should_poll = False
        self._attr_unique_id = f'{model}-{name}-{mac.replace(":","")[-6:]}'
        self._attr_name = name
        self._attr_has_entity_name = True
        self._attr_device_info = DeviceInfo(
            identifiers={(DOMAIN, mac)},
            name=model,
            manufacturer="Balboa Water Group",
            model=model,
            sw_version=client.software_version,
            connections={(CONNECTION_NETWORK_MAC, mac)},
        )
        self._client = client

    @property
    def assumed_state(self) -> bool:
        """Return whether the state is based on actual reading from device."""
        return not self._client.available


class BalboaEntity(BalboaBaseEntity):
    """Balboa entity."""

    async def async_added_to_hass(self) -> None:
        """Run when entity about to be added to hass."""
        self.async_on_remove(self._client.on(EVENT_UPDATE, self.async_write_ha_state))