"""The Powerfox integration."""

import asyncio

from powerfox import (
    DeviceType,
    Powerfox,
    PowerfoxAuthenticationError,
    PowerfoxConnectionError,
)

from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, Platform
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession

from .const import DOMAIN
from .coordinator import (
    PowerfoxConfigEntry,
    PowerfoxDataUpdateCoordinator,
    PowerfoxReportDataUpdateCoordinator,
)

PLATFORMS: list[Platform] = [Platform.SENSOR]


async def async_setup_entry(hass: HomeAssistant, entry: PowerfoxConfigEntry) -> bool:
    """Set up Powerfox from a config entry."""
    client = Powerfox(
        username=entry.data[CONF_EMAIL],
        password=entry.data[CONF_PASSWORD],
        session=async_get_clientsession(hass),
    )

    try:
        devices = await client.all_devices()
    except PowerfoxAuthenticationError as err:
        await client.close()
        raise ConfigEntryAuthFailed(
            translation_domain=DOMAIN,
            translation_key="auth_failed",
        ) from err
    except PowerfoxConnectionError as err:
        await client.close()
        raise ConfigEntryNotReady(
            translation_domain=DOMAIN,
            translation_key="connection_error",
        ) from err

    coordinators: list[
        PowerfoxDataUpdateCoordinator | PowerfoxReportDataUpdateCoordinator
    ] = []
    for device in devices:
        if device.type == DeviceType.GAS_METER:
            coordinators.append(
                PowerfoxReportDataUpdateCoordinator(hass, entry, client, device)
            )
            continue
        coordinators.append(PowerfoxDataUpdateCoordinator(hass, entry, client, device))

    await asyncio.gather(
        *[
            coordinator.async_config_entry_first_refresh()
            for coordinator in coordinators
        ]
    )

    entry.runtime_data = coordinators

    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
    return True


async def async_unload_entry(hass: HomeAssistant, entry: PowerfoxConfigEntry) -> bool:
    """Unload a config entry."""
    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)