"""Base class for Automate Roller Blinds."""
import logging

import aiopulse2

from homeassistant.core import callback
from homeassistant.helpers import entity
from homeassistant.helpers.device_registry import async_get_registry as get_dev_reg
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity_registry import async_get_registry as get_ent_reg

from .const import AUTOMATE_ENTITY_REMOVE, DOMAIN

_LOGGER = logging.getLogger(__name__)


class AutomateBase(entity.Entity):
    """Base representation of an Automate roller."""

    def __init__(self, roller: aiopulse2.Roller) -> None:
        """Initialize the roller."""
        self.roller = roller

    @property
    def available(self) -> bool:
        """Return True if roller and hub is available."""
        return self.roller.online and self.roller.hub.connected

    async def async_remove_and_unregister(self):
        """Unregister from entity and device registry and call entity remove function."""
        _LOGGER.info("Removing %s %s", self.__class__.__name__, self.unique_id)

        ent_registry = await get_ent_reg(self.hass)
        if self.entity_id in ent_registry.entities:
            ent_registry.async_remove(self.entity_id)

        dev_registry = await get_dev_reg(self.hass)
        device = dev_registry.async_get_device(
            identifiers={(DOMAIN, self.unique_id)}, connections=set()
        )
        if device is not None:
            dev_registry.async_update_device(
                device.id, remove_config_entry_id=self.registry_entry.config_entry_id
            )

        await self.async_remove()

    async def async_added_to_hass(self):
        """Entity has been added to hass."""
        self.roller.callback_subscribe(self.notify_update)

        self.async_on_remove(
            async_dispatcher_connect(
                self.hass,
                AUTOMATE_ENTITY_REMOVE.format(self.roller.id),
                self.async_remove_and_unregister,
            )
        )

    async def async_will_remove_from_hass(self):
        """Entity being removed from hass."""
        self.roller.callback_unsubscribe(self.notify_update)

    @callback
    def notify_update(self, roller: aiopulse2.Roller):
        """Write updated device state information."""
        _LOGGER.debug(
            "Device update notification received: %s (%r)", roller.id, roller.name
        )
        self.async_write_ha_state()

    @property
    def should_poll(self):
        """Report that Automate entities do not need polling."""
        return False

    @property
    def unique_id(self):
        """Return the unique ID of this roller."""
        return self.roller.id

    @property
    def name(self):
        """Return the name of roller."""
        return self.roller.name

    @property
    def device_info(self):
        """Return the device info."""
        attrs = {
            "identifiers": {(DOMAIN, self.roller.id)},
        }
        return attrs