"""The pyLoad integration."""

from __future__ import annotations

from aiohttp import CookieJar
from pyloadapi.api import PyLoadAPI
from pyloadapi.exceptions import CannotConnect, InvalidAuth, ParserError

from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
    CONF_HOST,
    CONF_PASSWORD,
    CONF_PORT,
    CONF_SSL,
    CONF_USERNAME,
    CONF_VERIFY_SSL,
    Platform,
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryError, ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_create_clientsession

from .coordinator import PyLoadCoordinator

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

type PyLoadConfigEntry = ConfigEntry[PyLoadCoordinator]


async def async_setup_entry(hass: HomeAssistant, entry: PyLoadConfigEntry) -> bool:
    """Set up pyLoad from a config entry."""

    url = (
        f"{"https" if entry.data[CONF_SSL] else "http"}://"
        f"{entry.data[CONF_HOST]}:{entry.data[CONF_PORT]}/"
    )

    session = async_create_clientsession(
        hass,
        verify_ssl=entry.data[CONF_VERIFY_SSL],
        cookie_jar=CookieJar(unsafe=True),
    )
    pyloadapi = PyLoadAPI(
        session,
        api_url=url,
        username=entry.data[CONF_USERNAME],
        password=entry.data[CONF_PASSWORD],
    )

    try:
        await pyloadapi.login()
    except CannotConnect as e:
        raise ConfigEntryNotReady(
            "Unable to connect and retrieve data from pyLoad API"
        ) from e
    except ParserError as e:
        raise ConfigEntryNotReady("Unable to parse data from pyLoad API") from e
    except InvalidAuth as e:
        raise ConfigEntryError(
            f"Authentication failed for {entry.data[CONF_USERNAME]}, check your login credentials"
        ) from e
    coordinator = PyLoadCoordinator(hass, pyloadapi)

    await coordinator.async_config_entry_first_refresh()

    entry.runtime_data = coordinator
    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)

    return True


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