"""Oasis Mini coordinator."""

from __future__ import annotations

from datetime import datetime, timedelta
import logging

import async_timeout

from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed

from .const import DOMAIN
from .pyoasismini import OasisMini

_LOGGER = logging.getLogger(__name__)


class OasisMiniCoordinator(DataUpdateCoordinator[str]):
    """Oasis Mini data update coordinator."""

    attempt: int = 0
    last_updated: datetime | None = None

    def __init__(self, hass: HomeAssistant, device: OasisMini) -> None:
        """Initialize."""
        super().__init__(
            hass,
            _LOGGER,
            name=DOMAIN,
            update_interval=timedelta(seconds=10),
            always_update=False,
        )
        self.device = device

    async def _async_update_data(self):
        """Update the data."""
        data: str | None = None
        self.attempt += 1

        try:
            async with async_timeout.timeout(10):
                if not self.device.mac_address:
                    await self.device.async_get_mac_address()
                if not self.device.serial_number:
                    await self.device.async_get_serial_number()
                if not self.device.software_version:
                    await self.device.async_get_software_version()
                data = await self.device.async_get_status()
                self.attempt = 0
                await self.device.async_get_current_track_details()
                await self.device.async_get_playlist_details()
                await self.device.async_cloud_get_playlists()
        except Exception as ex:  # pylint:disable=broad-except
            if self.attempt > 2 or not (data or self.data):
                raise UpdateFailed(
                    f"Couldn't read from the Oasis Mini after {self.attempt} attempts"
                ) from ex

        if data != self.data:
            self.last_updated = datetime.now()
        return data