"""Support for scenes provided by WMS WebControl pro."""

from __future__ import annotations

from typing import Any

from wmspro.scene import Scene as WMS_Scene

from homeassistant.components.scene import Scene
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback

from . import WebControlProConfigEntry
from .const import ATTRIBUTION, DOMAIN, MANUFACTURER


async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: WebControlProConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up the WMS based scenes from a config entry."""
    hub = config_entry.runtime_data

    async_add_entities(
        WebControlProScene(config_entry.entry_id, scene)
        for scene in hub.scenes.values()
    )


class WebControlProScene(Scene):
    """Representation of a WMS based scene."""

    _attr_attribution = ATTRIBUTION
    _attr_has_entity_name = True

    def __init__(self, config_entry_id: str, scene: WMS_Scene) -> None:
        """Initialize the entity with the configured scene."""
        super().__init__()

        # Scene information
        self._scene = scene
        self._attr_name = scene.name
        self._attr_unique_id = str(scene.id)

        # Room information
        room = scene.room
        room_name = room.name
        room_id_str = str(room.id)
        self._attr_device_info = DeviceInfo(
            identifiers={(DOMAIN, room_id_str)},
            manufacturer=MANUFACTURER,
            model="Room",
            name=room_name,
            serial_number=room_id_str,
            suggested_area=room_name,
            via_device=(DOMAIN, config_entry_id),
            configuration_url=f"http://{scene.host}/control",
        )

    async def async_activate(self, **kwargs: Any) -> None:
        """Activate scene. Try to get entities into requested state."""
        await self._scene()