core/homeassistant/components/opengarage/cover.py

"""Platform for the opengarage.io cover component."""

import logging
from typing import Any, cast, override

from homeassistant.components.cover import (
    CoverDeviceClass,
    CoverEntity,
    CoverEntityFeature,
    CoverState,
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback

from .coordinator import OpenGarageConfigEntry, OpenGarageDataUpdateCoordinator
from .entity import OpenGarageEntity

_LOGGER = logging.getLogger(__name__)

STATES_MAP = {0: CoverState.CLOSED, 1: CoverState.OPEN}


async def async_setup_entry(
    hass: HomeAssistant,
    entry: OpenGarageConfigEntry,
    async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
    """Set up the OpenGarage covers."""
    async_add_entities(
        [OpenGarageCover(entry.runtime_data, cast(str, entry.unique_id))]
    )


class OpenGarageCover(OpenGarageEntity, CoverEntity):
    """Representation of a OpenGarage cover."""

    _attr_device_class = CoverDeviceClass.GARAGE
    _attr_supported_features = CoverEntityFeature.OPEN | CoverEntityFeature.CLOSE
    _attr_name = None

    def __init__(
        self, coordinator: OpenGarageDataUpdateCoordinator, device_id: str
    ) -> None:
        """Initialize the cover."""
        self._state: str | None = None
        self._state_before_move: str | None = None

        super().__init__(coordinator, device_id)

    @property
    @override
    def is_closed(self) -> bool | None:
        """Return if the cover is closed."""
        if self._state is None:
            return None
        return self._state == CoverState.CLOSED

    @property
    @override
    def is_closing(self) -> bool | None:
        """Return if the cover is closing."""
        if self._state is None:
            return None
        return self._state == CoverState.CLOSING

    @property
    @override
    def is_opening(self) -> bool | None:
        """Return if the cover is opening."""
        if self._state is None:
            return None
        return self._state == CoverState.OPENING

    @override
    async def async_close_cover(self, **kwargs: Any) -> None:
        """Close the cover."""
        if self._state in [CoverState.CLOSED, CoverState.CLOSING]:
            return
        self._state_before_move = self._state
        self._state = CoverState.CLOSING
        self.async_write_ha_state()
        await self._push_button()

    @override
    async def async_open_cover(self, **kwargs: Any) -> None:
        """Open the cover."""
        if self._state in [CoverState.OPEN, CoverState.OPENING]:
            return
        self._state_before_move = self._state
        self._state = CoverState.OPENING
        self.async_write_ha_state()
        await self._push_button()

    @callback
    @override
    def _update_attr(self) -> None:
        """Update the state and attributes."""
        status = self.coordinator.data

        state = STATES_MAP.get(status.get("door"))  # type: ignore[arg-type]
        if self._state_before_move is not None:
            if self._state_before_move != state:
                self._state = state
                self._state_before_move = None
        else:
            self._state = state

    @property
    @override
    def current_cover_position(self) -> int | None:
        """Return current position of cover (0=closed, 100=open)."""
        if self._state == CoverState.CLOSED:
            return 0
        if self._state == CoverState.OPEN:
            return 100
        return None

    async def _push_button(self):
        """Send commands to API."""
        result = await self.coordinator.open_garage_connection.push_button()
        if result is None:
            _LOGGER.error("Unable to connect to OpenGarage device")
        if result == 1:
            return

        if result == 2:
            _LOGGER.error("Unable to control %s: Device key is incorrect", self.name)
        elif result > 2:
            _LOGGER.error("Unable to control %s: Error code %s", self.name, result)

        self._state = self._state_before_move
        self._state_before_move = None
        self.async_write_ha_state()