"""Support for Bluesound devices."""

from asyncio import Task
from datetime import datetime, timedelta
import logging
from typing import TYPE_CHECKING, Any, override

from pyblu import Input, Player, Preset, Status, SyncStatus

from homeassistant.components import media_source
from homeassistant.components.media_player import (
    BrowseMedia,
    MediaPlayerEntity,
    MediaPlayerEntityFeature,
    MediaPlayerState,
    MediaType,
    async_process_play_media_url,
)
from homeassistant.const import CONF_HOST, CONF_PORT
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers import (
    device_registry as dr,
    entity_registry as er,
    issue_registry as ir,
)
from homeassistant.helpers.device_registry import (
    CONNECTION_NETWORK_MAC,
    DeviceInfo,
    format_mac,
)
from homeassistant.helpers.dispatcher import (
    async_dispatcher_connect,
    async_dispatcher_send,
)
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util import dt as dt_util, slugify

from .const import (
    ATTR_BLUESOUND_GROUP,
    ATTR_MASTER,
    DOMAIN,
    SERVICE_JOIN,
    SERVICE_UNJOIN,
)
from .coordinator import BluesoundCoordinator
from .utils import (
    dispatcher_join_signal,
    dispatcher_unjoin_signal,
    format_unique_id,
    id_to_paired_player,
)

if TYPE_CHECKING:
    from . import BluesoundConfigEntry

_LOGGER = logging.getLogger(__name__)

SCAN_INTERVAL = timedelta(minutes=15)

DATA_BLUESOUND = DOMAIN
DEFAULT_PORT = 11000

POLL_TIMEOUT = 120


async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: BluesoundConfigEntry,
    async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
    """Set up the Bluesound entry."""
    bluesound_player = BluesoundPlayer(
        config_entry.runtime_data.coordinator,
        config_entry.data[CONF_HOST],
        config_entry.data[CONF_PORT],
        config_entry.runtime_data.player,
    )

    async_add_entities([bluesound_player], update_before_add=True)


class BluesoundPlayer(CoordinatorEntity[BluesoundCoordinator], MediaPlayerEntity):
    """Representation of a Bluesound Player."""

    _attr_media_content_type = MediaType.MUSIC
    _attr_has_entity_name = True
    _attr_name = None
    _attr_volume_step = 0.01

    def __init__(
        self,
        coordinator: BluesoundCoordinator,
        host: str,
        port: int,
        player: Player,
    ) -> None:
        """Initialize the media player."""
        super().__init__(coordinator)
        sync_status = coordinator.data.sync_status

        self.host = host
        self.port = port
        self._poll_status_loop_task: Task[None] | None = None
        self._poll_sync_status_loop_task: Task[None] | None = None
        self._id = sync_status.id
        self._last_status_update: datetime | None = None
        self._sync_status = sync_status
        self._status: Status = coordinator.data.status
        self._inputs: list[Input] = coordinator.data.inputs
        self._presets: list[Preset] = coordinator.data.presets
        self._group_name: str | None = None
        self._group_list: list[str] = []
        self._group_members: list[str] | None = None
        self._bluesound_device_name = sync_status.name
        self._player = player
        self._last_status_update = dt_util.utcnow()

        self._attr_unique_id = format_unique_id(sync_status.mac, port)
        # there should always be one player with the default port per mac
        if port == DEFAULT_PORT:
            self._attr_device_info = DeviceInfo(
                identifiers={(DOMAIN, format_mac(sync_status.mac))},
                connections={(CONNECTION_NETWORK_MAC, sync_status.mac)},
                name=sync_status.name,
                manufacturer=sync_status.brand,
                model=sync_status.model_name,
                model_id=sync_status.model,
            )
        else:
            self._attr_device_info = DeviceInfo(
                identifiers={(DOMAIN, format_unique_id(sync_status.mac, port))},
                name=sync_status.name,
                manufacturer=sync_status.brand,
                model=sync_status.model_name,
                model_id=sync_status.model,
            )
            # The leader (default-port) device is a separate config entry that may
            # not be loaded yet; link to it only if it already exists.
            if leader_devices := dr.async_get(coordinator.hass).async_get_devices(
                identifiers={(DOMAIN, format_mac(sync_status.mac))}
            ):
                self._attr_device_info["via_device_id"] = leader_devices[0].id

    @override
    async def async_added_to_hass(self) -> None:
        """Start the polling task."""
        await super().async_added_to_hass()

        assert self._sync_status.id is not None
        self.async_on_remove(
            async_dispatcher_connect(
                self.hass,
                dispatcher_join_signal(self.entity_id),
                self.async_add_follower,
            )
        )
        self.async_on_remove(
            async_dispatcher_connect(
                self.hass,
                dispatcher_unjoin_signal(self._sync_status.id),
                self.async_remove_follower,
            )
        )

    @override
    async def async_will_remove_from_hass(self) -> None:
        """Stop the polling task."""
        await super().async_will_remove_from_hass()

    @callback
    @override
    def _handle_coordinator_update(self) -> None:
        """Handle updated data from the coordinator."""
        self._sync_status = self.coordinator.data.sync_status
        self._status = self.coordinator.data.status
        self._inputs = self.coordinator.data.inputs
        self._presets = self.coordinator.data.presets

        self._last_status_update = dt_util.utcnow()

        self._group_list = self.rebuild_bluesound_group()
        self._group_members = self.rebuild_group_members()

        self.async_write_ha_state()

    @property
    @override
    def state(self) -> MediaPlayerState:
        """Return the state of the device."""
        if self.available is False:
            return MediaPlayerState.OFF

        if self.is_grouped and not self.is_leader:
            return MediaPlayerState.IDLE

        match self._status.state:
            case "pause":
                return MediaPlayerState.PAUSED
            case "stream" | "play":
                return MediaPlayerState.PLAYING
            case _:
                return MediaPlayerState.IDLE

    @property
    @override
    def media_title(self) -> str | None:
        """Title of current playing media."""
        if self.available is False or (self.is_grouped and not self.is_leader):
            return None

        return self._status.name

    @property
    @override
    def media_artist(self) -> str | None:
        """Artist of current playing media (Music track only)."""
        if self.available is False:
            return None

        if self.is_grouped and not self.is_leader:
            return self._group_name

        return self._status.artist

    @property
    @override
    def media_album_name(self) -> str | None:
        """Artist of current playing media (Music track only)."""
        if self.available is False or (self.is_grouped and not self.is_leader):
            return None

        return self._status.album

    @property
    @override
    def media_image_url(self) -> str | None:
        """Image url of current playing media."""
        if self.available is False or (self.is_grouped and not self.is_leader):
            return None

        url = self._status.image
        if url is None:
            return None

        if url[0] == "/":
            url = f"http://{self.host}:{self.port}{url}"

        return url

    @property
    @override
    def media_position(self) -> int | None:
        """Position of current playing media in seconds."""
        if self.available is False or (self.is_grouped and not self.is_leader):
            return None

        mediastate = self.state
        if self._last_status_update is None or mediastate == MediaPlayerState.IDLE:
            return None

        position = self._status.seconds
        if position is None:
            return None

        if mediastate == MediaPlayerState.PLAYING:
            position += (dt_util.utcnow() - self._last_status_update).total_seconds()

        return int(position)

    @property
    @override
    def media_duration(self) -> int | None:
        """Duration of current playing media in seconds."""
        if self.available is False or (self.is_grouped and not self.is_leader):
            return None

        duration = self._status.total_seconds
        if duration is None:
            return None

        return int(duration)

    @property
    @override
    def media_position_updated_at(self) -> datetime | None:
        """Last time status was updated."""
        return self._last_status_update

    @property
    @override
    def volume_level(self) -> float | None:
        """Volume level of the media player (0..1)."""
        volume = self._status.volume

        if self.is_grouped:
            volume = self._sync_status.volume

        return volume / 100

    @property
    @override
    def is_volume_muted(self) -> bool:
        """Boolean if volume is currently muted."""
        mute = False

        if self._status is not None:
            mute = self._status.mute
        if self.is_grouped:
            mute = self._sync_status.mute_volume is not None

        return mute

    @property
    def id(self) -> str | None:
        """Get id of device."""
        return self._id

    @property
    def bluesound_device_name(self) -> str | None:
        """Return the device name as returned by the device."""
        return self._bluesound_device_name

    @property
    def sync_status(self) -> SyncStatus:
        """Return the sync status."""
        return self._sync_status

    @property
    @override
    def source_list(self) -> list[str] | None:
        """List of available input sources."""
        if self.available is False or (self.is_grouped and not self.is_leader):
            return None

        sources = [x.name for x in self._presets]

        # ignore if both id and text are None
        for input_ in self._inputs:
            if input_.text is not None:
                sources.append(input_.text)
            elif input_.id is not None:
                sources.append(input_.id)

        return sources

    @property
    @override
    def source(self) -> str | None:
        """Name of the current input source."""
        if self.available is False or (self.is_grouped and not self.is_leader):
            return None

        if self._status.input_id is not None:
            for input_ in self._inputs:
                # the input might not have an id
                # => also try to match on the stream_url/url
                # we have to use both because neither matches all the time
                if (
                    input_.id == self._status.input_id
                    or input_.url == self._status.stream_url
                ):
                    return input_.text if input_.text is not None else input_.id

        for preset in self._presets:
            if preset.url == self._status.stream_url:
                return preset.name

        return self._status.service

    @property
    @override
    def supported_features(self) -> MediaPlayerEntityFeature:
        """Flag of media commands that are supported."""
        if self.available is False:
            return MediaPlayerEntityFeature(0)

        if self.is_grouped and not self.is_leader:
            return (
                MediaPlayerEntityFeature.VOLUME_STEP
                | MediaPlayerEntityFeature.VOLUME_SET
                | MediaPlayerEntityFeature.VOLUME_MUTE
                | MediaPlayerEntityFeature.GROUPING
            )

        supported = (
            MediaPlayerEntityFeature.CLEAR_PLAYLIST
            | MediaPlayerEntityFeature.BROWSE_MEDIA
            | MediaPlayerEntityFeature.GROUPING
        )

        if not self._status.indexing:
            supported = (
                supported
                | MediaPlayerEntityFeature.PAUSE
                | MediaPlayerEntityFeature.PREVIOUS_TRACK
                | MediaPlayerEntityFeature.NEXT_TRACK
                | MediaPlayerEntityFeature.PLAY_MEDIA
                | MediaPlayerEntityFeature.STOP
                | MediaPlayerEntityFeature.PLAY
                | MediaPlayerEntityFeature.SELECT_SOURCE
                | MediaPlayerEntityFeature.SHUFFLE_SET
            )

        current_vol = self.volume_level
        if current_vol is not None and current_vol >= 0:
            supported = (
                supported
                | MediaPlayerEntityFeature.VOLUME_STEP
                | MediaPlayerEntityFeature.VOLUME_SET
                | MediaPlayerEntityFeature.VOLUME_MUTE
            )

        if self._status.can_seek:
            supported = supported | MediaPlayerEntityFeature.SEEK

        return supported

    @property
    def is_leader(self) -> bool:
        """Return true if player is leader of a group."""
        return self._sync_status.followers is not None

    @property
    def is_grouped(self) -> bool:
        """Return true if player is member or leader of a group."""
        return (
            self._sync_status.followers is not None
            or self._sync_status.leader is not None
        )

    @property
    @override
    def shuffle(self) -> bool:
        """Return true if shuffle is active."""
        shuffle = False
        if self._status is not None:
            shuffle = self._status.shuffle

        return shuffle

    @property
    @override
    def group_members(self) -> list[str] | None:
        """Get list of group members. Leader is always first."""
        return self._group_members

    @override
    async def async_join_players(self, group_members: list[str]) -> None:
        """Join `group_members` as a player group with the current player."""
        if self.entity_id in group_members:
            raise ServiceValidationError("Cannot join player to itself")

        entity_ids_with_sync_status = self._entity_ids_with_sync_status()

        paired_players = []
        for group_member in group_members:
            sync_status = entity_ids_with_sync_status.get(group_member)
            if sync_status is None:
                continue
            paired_player = id_to_paired_player(sync_status.id)
            if paired_player:
                paired_players.append(paired_player)

        if paired_players:
            await self._player.add_followers(paired_players)

    @override
    async def async_unjoin_player(self) -> None:
        """Remove this player from any group."""
        if self._sync_status.leader is not None:
            leader_id = f"{self._sync_status.leader.ip}:{self._sync_status.leader.port}"
            async_dispatcher_send(
                self.hass, dispatcher_unjoin_signal(leader_id), self.host, self.port
            )

        if self._sync_status.followers is not None:
            await self._player.remove_follower(self.host, self.port)

    async def async_bluesound_join(self, master: str) -> None:
        """Join the player to a group."""
        ir.async_create_issue(
            self.hass,
            DOMAIN,
            f"deprecated_service_{SERVICE_JOIN}",
            is_fixable=False,
            breaks_in_ha_version="2026.7.0",
            issue_domain=DOMAIN,
            severity=ir.IssueSeverity.WARNING,
            translation_key="deprecated_service_join",
            translation_placeholders={
                "name": slugify(self.sync_status.name),
            },
        )

        if master == self.entity_id:
            raise ServiceValidationError("Cannot join player to itself")

        _LOGGER.debug("Trying to join player: %s", self.id)
        async_dispatcher_send(
            self.hass, dispatcher_join_signal(master), self.host, self.port
        )

    async def async_bluesound_unjoin(self) -> None:
        """Unjoin the player from a group."""
        ir.async_create_issue(
            self.hass,
            DOMAIN,
            f"deprecated_service_{SERVICE_UNJOIN}",
            is_fixable=False,
            breaks_in_ha_version="2026.7.0",
            issue_domain=DOMAIN,
            severity=ir.IssueSeverity.WARNING,
            translation_key="deprecated_service_unjoin",
            translation_placeholders={
                "name": slugify(self.sync_status.name),
            },
        )

        await self.async_unjoin_player()

    @property
    @override
    def extra_state_attributes(self) -> dict[str, Any] | None:
        """List members in group."""
        attributes: dict[str, Any] = {}
        if self._group_list:
            attributes = {ATTR_BLUESOUND_GROUP: self._group_list}

        attributes[ATTR_MASTER] = self.is_leader

        return attributes

    def rebuild_bluesound_group(self) -> list[str]:
        """Rebuild the list of entities in speaker group."""
        if self.sync_status.leader is None and self.sync_status.followers is None:
            return []

        config_entries: list[BluesoundConfigEntry] = (
            self.hass.config_entries.async_entries(DOMAIN)
        )
        sync_status_list = [
            x.runtime_data.coordinator.data.sync_status for x in config_entries
        ]

        leader_sync_status: SyncStatus | None = None
        if self.sync_status.leader is None:
            leader_sync_status = self.sync_status
        else:
            required_id = f"{self.sync_status.leader.ip}:{self.sync_status.leader.port}"
            for sync_status in sync_status_list:
                if sync_status.id == required_id:
                    leader_sync_status = sync_status
                    break

        if leader_sync_status is None or leader_sync_status.followers is None:
            return []

        follower_ids = [f"{x.ip}:{x.port}" for x in leader_sync_status.followers]
        follower_names = [
            sync_status.name
            for sync_status in sync_status_list
            if sync_status.id in follower_ids
        ]
        follower_names.insert(0, leader_sync_status.name)
        return follower_names

    def rebuild_group_members(self) -> list[str] | None:
        """Get list of group members. Leader is always first."""
        if self.sync_status.leader is None and self.sync_status.followers is None:
            return None

        entity_ids_with_sync_status = self._entity_ids_with_sync_status()

        leader_entity_id = None
        followers = None
        if self.sync_status.followers is not None:
            leader_entity_id = self.entity_id
            followers = self.sync_status.followers
        elif self.sync_status.leader is not None:
            leader_id = f"{self.sync_status.leader.ip}:{self.sync_status.leader.port}"
            for entity_id, sync_status in entity_ids_with_sync_status.items():
                if sync_status.id == leader_id:
                    leader_entity_id = entity_id
                    followers = sync_status.followers
                    break

        if leader_entity_id is None or followers is None:
            return None

        grouped_entity_ids = [leader_entity_id]
        for follower in followers:
            follower_id = f"{follower.ip}:{follower.port}"
            entity_ids = [
                entity_id
                for entity_id, sync_status in entity_ids_with_sync_status.items()
                if sync_status.id == follower_id
            ]
            match entity_ids:
                case [entity_id]:
                    grouped_entity_ids.append(entity_id)

        return grouped_entity_ids

    def _entity_ids_with_sync_status(self) -> dict[str, SyncStatus]:
        result = {}

        entity_registry = er.async_get(self.hass)

        config_entries: list[BluesoundConfigEntry] = (
            self.hass.config_entries.async_entries(DOMAIN)
        )
        for config_entry in config_entries:
            entity_entries = er.async_entries_for_config_entry(
                entity_registry, config_entry.entry_id
            )
            for entity_entry in entity_entries:
                if entity_entry.domain == "media_player":
                    result[entity_entry.entity_id] = (
                        config_entry.runtime_data.coordinator.data.sync_status
                    )

        return result

    async def async_add_follower(self, host: str, port: int) -> None:
        """Add follower to leader."""
        await self._player.add_follower(host, port)

    async def async_remove_follower(self, host: str, port: int) -> None:
        """Remove follower to leader."""
        await self._player.remove_follower(host, port)

    @override
    async def async_set_shuffle(self, shuffle: bool) -> None:
        """Enable or disable shuffle mode."""
        await self._player.shuffle(shuffle)

    @override
    async def async_select_source(self, source: str) -> None:
        """Select input source."""
        if self.is_grouped and not self.is_leader:
            return

        # presets and inputs might have the same name; presets have priority
        for input_ in self._inputs:
            if source in (input_.text, input_.id):
                await self._player.play_url(input_.url)
                return
        for preset in self._presets:
            if preset.name == source:
                await self._player.load_preset(preset.id)
                return

        raise ServiceValidationError(f"Source {source} not found")

    @override
    async def async_clear_playlist(self) -> None:
        """Clear players playlist."""
        if self.is_grouped and not self.is_leader:
            return

        await self._player.clear()

    @override
    async def async_media_next_track(self) -> None:
        """Send media_next command to media player."""
        if self.is_grouped and not self.is_leader:
            return

        await self._player.skip()

    @override
    async def async_media_previous_track(self) -> None:
        """Send media_previous command to media player."""
        if self.is_grouped and not self.is_leader:
            return

        await self._player.back()

    @override
    async def async_media_play(self) -> None:
        """Send media_play command to media player."""
        if self.is_grouped and not self.is_leader:
            return

        await self._player.play()

    @override
    async def async_media_pause(self) -> None:
        """Send media_pause command to media player."""
        if self.is_grouped and not self.is_leader:
            return

        await self._player.pause()

    @override
    async def async_media_stop(self) -> None:
        """Send stop command."""
        if self.is_grouped and not self.is_leader:
            return

        await self._player.stop()

    @override
    async def async_media_seek(self, position: float) -> None:
        """Send media_seek command to media player."""
        if self.is_grouped and not self.is_leader:
            return

        await self._player.play(seek=int(position))

    @override
    async def async_play_media(
        self, media_type: MediaType | str, media_id: str, **kwargs: Any
    ) -> None:
        """Send the play_media command to the media player."""
        if self.is_grouped and not self.is_leader:
            return

        if media_source.is_media_source_id(media_id):
            play_item = await media_source.async_resolve_media(
                self.hass, media_id, self.entity_id
            )
            media_id = play_item.url

        url = async_process_play_media_url(self.hass, media_id)

        await self._player.play_url(url)

    @override
    async def async_set_volume_level(self, volume: float) -> None:
        """Send volume_up command to media player."""
        volume = round(volume * 100)
        volume = min(100, volume)
        volume = max(0, volume)

        await self._player.volume(level=volume)

    @override
    async def async_mute_volume(self, mute: bool) -> None:
        """Send mute command to media player."""
        await self._player.volume(mute=mute)

    @override
    async def async_browse_media(
        self,
        media_content_type: MediaType | str | None = None,
        media_content_id: str | None = None,
    ) -> BrowseMedia:
        """Implement the websocket media browsing helper."""
        return await media_source.async_browse_media(
            self.hass,
            media_content_id,
            content_filter=lambda item: item.media_content_type.startswith("audio/"),
        )
