"""The zcc integration."""

from __future__ import annotations

import logging

from zcc import ControlPoint, ControlPointError

from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC

from .const import DOMAIN
from .helpers import async_connect_to_controller

PLATFORMS = [
    Platform.COVER,
    Platform.FAN,
    Platform.LIGHT,
    Platform.SENSOR,
    Platform.SWITCH,
]

_LOGGER = logging.getLogger(__name__)


type ZimiConfigEntry = ConfigEntry[ControlPoint]


async def async_setup_entry(hass: HomeAssistant, entry: ZimiConfigEntry) -> bool:
    """Connect to Zimi Controller and register device."""

    try:
        api = await async_connect_to_controller(
            host=entry.data[CONF_HOST],
            port=entry.data[CONF_PORT],
        )

    except ControlPointError as error:
        raise ConfigEntryNotReady(f"Zimi setup failed: {error}") from error

    _LOGGER.debug("\n%s", api.describe())

    entry.runtime_data = api

    device_registry = dr.async_get(hass)
    device_registry.async_get_or_create(
        config_entry_id=entry.entry_id,
        identifiers={(DOMAIN, api.mac)},
        manufacturer=api.brand,
        name=api.network_name,
        model="Zimi Cloud Connect",
        sw_version=api.firmware_version,
        connections={(CONNECTION_NETWORK_MAC, api.mac)},
    )

    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)

    _LOGGER.debug("Zimi setup complete")

    return True


async def async_unload_entry(hass: HomeAssistant, entry: ZimiConfigEntry) -> bool:
    """Unload a config entry."""

    api = entry.runtime_data
    api.disconnect()

    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)