"""Spook - Your homie."""

from __future__ import annotations

from typing import TYPE_CHECKING

import voluptuous as vol

from homeassistant.components.zone import (
    DOMAIN,
    ENTITY_ID_HOME,
    UPDATE_FIELDS,
    Zone,
    ZoneStorageCollection,
)
from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE, CONF_RADIUS
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_component import DATA_INSTANCES, EntityComponent

from ....services import AbstractSpookAdminService

if TYPE_CHECKING:
    from homeassistant.core import ServiceCall


def _is_the_house(entity: Zone) -> bool:
    """Tell whether a zone is the home zone drawn from the house's location.

    Home Assistant only draws it when no zone has taken `zone.home` already:
    zones from YAML and storage are set up first, and one named Home gets that
    entity ID. Those are ordinary zones, and moving the house would not move
    them. The drawn one is the only editable zone without a stored ID.
    """
    return (
        entity.entity_id == ENTITY_ID_HOME
        and entity.editable
        # pylint: disable-next=protected-access
        and "id" not in entity._config  # noqa: SLF001
    )


class SpookService(AbstractSpookAdminService):
    """Zone service to update a zone on the fly, the home zone included."""

    domain = DOMAIN
    service = "update"
    schema = {
        vol.Required("entity_id"): cv.entity_domain(DOMAIN),
    } | UPDATE_FIELDS

    async def async_handle_service(self, call: ServiceCall) -> None:
        """Handle the service call."""
        entity_component: EntityComponent[Zone] = self.hass.data[DATA_INSTANCES][DOMAIN]

        collection: ZoneStorageCollection
        if DOMAIN in self.hass.data:
            collection = self.hass.data[DOMAIN]
        else:
            # Home zone is set in YAML, as a result Home Assistant doesn't
            # set the storage collection into hass data.
            # Major hack to get around this. 👻
            collection = self.hass.data["websocket_api"]["zone/list"][
                0
            ].__self__.storage_collection

        if not (entity := entity_component.get_entity(call.data["entity_id"])):
            message = f"Could not find entity_id: {call.data['entity_id']}"
            raise HomeAssistantError(message)

        if _is_the_house(entity):
            await self._async_update_home(call)
            return

        # pylint: disable-next=protected-access
        if not entity.editable or "id" not in entity._config:  # noqa: SLF001
            message = f"This zone is not editable: {call.data['entity_id']}"
            raise HomeAssistantError(message)

        data = call.data.copy()
        data.pop("entity_id")

        # pylint: disable-next=protected-access
        await collection.async_update_item(entity._config["id"], data)  # noqa: SLF001

    async def _async_update_home(self, call: ServiceCall) -> None:
        """Move or resize the home zone.

        It is not a stored zone: Home Assistant draws it from the location
        set for the whole house, and redraws it when that changes. So this
        changes the location, the same way Home Assistant's own
        `homeassistant.set_location` does, and the radius along with it.

        Only those. Its name is the name of the whole house, and a home zone
        is never passive.
        """
        data = dict(call.data)
        data.pop("entity_id")

        if unsupported := set(data) - {CONF_LATITUDE, CONF_LONGITUDE, CONF_RADIUS}:
            message = (
                "The home zone only takes a latitude, longitude and radius, "
                f"not: {', '.join(sorted(unsupported))}"
            )
            raise HomeAssistantError(message)

        # Home Assistant keeps the radius of the house in whole meters, and
        # never below zero. Cutting a fraction off quietly would be a
        # different zone than was asked for, and the location is stored
        # without going through the checks its configuration normally does.
        if CONF_RADIUS in data:
            radius = data[CONF_RADIUS]
            if radius != int(radius) or radius < 0:
                message = "The home zone takes its radius in whole meters, zero or more"
                raise HomeAssistantError(message)
            data[CONF_RADIUS] = int(radius)

        await self.hass.config.async_update(**data)