from __future__ import annotations

import logging

import homeassistant.core as ha_core
from homeassistant.components import spotify as ha_spotify
from homeassistant.components.media_player import BrowseMedia
from pychromecast import Chromecast

_LOGGER = logging.getLogger(__name__)


async def async_get_media_browser_root_object(
    hass: ha_core.HomeAssistant, cast_type: str
) -> list[BrowseMedia]:
    """Create a root object for media browsing."""
    try:
        result = await ha_spotify.async_browse_media(hass, None, None)
    except KeyError:
        _LOGGER.debug(
            "failed to call spotify.async_browse_media, the Home Assistant spotify "
            "integration may not be setup"
        )
        return []
    _LOGGER.debug("async_get_media_browser_root_object return %s", result.children)
    return result.children


async def async_browse_media(
    hass: ha_core.HomeAssistant,
    media_content_type: str,
    media_content_id: str,
    cast_type: str,
) -> BrowseMedia | None:
    """Browse media."""
    _LOGGER.debug("async_browse_media %s, %s", media_content_type, media_content_id)
    result = None
    # Check if this media is handled by Spotify, if it isn't just return None.
    if ha_spotify.is_spotify_media_type(media_content_type):
        # Browse deeper in the tree
        result = await ha_spotify.async_browse_media(
            hass, media_content_type, media_content_id, can_play_artist=False
        )
    _LOGGER.debug("async_browse_media return: %s", result)
    return result


async def async_play_media(
    hass: ha_core.HomeAssistant,
    cast_entity_id,
    chromecast: Chromecast,
    media_type: str,
    media_id: str,
) -> bool:
    """Play media."""
    _LOGGER.debug("async_browse_media %s, %s", media_type, media_id)
    # If this is a spotify URI, forward to the the spotcast.start service, if not return
    # False
    if media_id and media_id.startswith("spotify:"):
        # Get the spotify URI
        spotify_uri = ha_spotify.spotify_uri_from_media_browser_url(media_id)
        data = {"entity_id": cast_entity_id, "uri": spotify_uri}
        await hass.services.async_call("spotcast", "start", data, blocking=False)
        return True
    return False