core/homeassistant/components/volumio/media_player.py

260 lines
8.4 KiB
Python

"""Volumio Platform.
Volumio rest API: https://volumio.github.io/docs/API/REST_API.html
"""
from __future__ import annotations
from datetime import timedelta
import json
from typing import Any
from homeassistant.components.media_player import (
BrowseMedia,
MediaPlayerEntity,
MediaPlayerEntityFeature,
MediaPlayerState,
MediaType,
RepeatMode,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID, CONF_NAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.util import Throttle
from .browse_media import browse_node, browse_top_level
from .const import DATA_INFO, DATA_VOLUMIO, DOMAIN
PLAYLIST_UPDATE_INTERVAL = timedelta(seconds=15)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up the Volumio media player platform."""
data = hass.data[DOMAIN][config_entry.entry_id]
volumio = data[DATA_VOLUMIO]
info = data[DATA_INFO]
uid = config_entry.data[CONF_ID]
name = config_entry.data[CONF_NAME]
entity = Volumio(volumio, uid, name, info)
async_add_entities([entity])
class Volumio(MediaPlayerEntity):
"""Volumio Player Object."""
_attr_has_entity_name = True
_attr_name = None
_attr_media_content_type = MediaType.MUSIC
_attr_supported_features = (
MediaPlayerEntityFeature.PAUSE
| MediaPlayerEntityFeature.VOLUME_SET
| MediaPlayerEntityFeature.VOLUME_MUTE
| MediaPlayerEntityFeature.PREVIOUS_TRACK
| MediaPlayerEntityFeature.NEXT_TRACK
| MediaPlayerEntityFeature.SEEK
| MediaPlayerEntityFeature.STOP
| MediaPlayerEntityFeature.PLAY
| MediaPlayerEntityFeature.PLAY_MEDIA
| MediaPlayerEntityFeature.VOLUME_STEP
| MediaPlayerEntityFeature.SELECT_SOURCE
| MediaPlayerEntityFeature.REPEAT_SET
| MediaPlayerEntityFeature.SHUFFLE_SET
| MediaPlayerEntityFeature.CLEAR_PLAYLIST
| MediaPlayerEntityFeature.BROWSE_MEDIA
)
_attr_source_list = []
def __init__(self, volumio, uid, name, info):
"""Initialize the media player."""
self._volumio = volumio
unique_id = uid
self._state = {}
self.thumbnail_cache = {}
self._attr_unique_id = unique_id
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, unique_id)},
manufacturer="Volumio",
model=info["hardware"],
name=name,
sw_version=info["systemversion"],
)
async def async_update(self) -> None:
"""Update state."""
self._state = await self._volumio.get_state()
await self._async_update_playlists()
@property
def state(self) -> MediaPlayerState:
"""Return the state of the device."""
status = self._state.get("status", None)
if status == "pause":
return MediaPlayerState.PAUSED
if status == "play":
return MediaPlayerState.PLAYING
return MediaPlayerState.IDLE
@property
def media_title(self):
"""Title of current playing media."""
return self._state.get("title", None)
@property
def media_artist(self):
"""Artist of current playing media (Music track only)."""
return self._state.get("artist", None)
@property
def media_album_name(self):
"""Artist of current playing media (Music track only)."""
return self._state.get("album", None)
@property
def media_image_url(self):
"""Image url of current playing media."""
url = self._state.get("albumart", None)
return self._volumio.canonic_url(url)
@property
def media_seek_position(self):
"""Time in seconds of current seek position."""
return self._state.get("seek", None)
@property
def media_duration(self):
"""Time in seconds of current song duration."""
return self._state.get("duration", None)
@property
def volume_level(self):
"""Volume level of the media player (0..1)."""
volume = self._state.get("volume", None)
if volume is not None and volume != "":
volume = int(volume) / 100
return volume
@property
def is_volume_muted(self):
"""Boolean if volume is currently muted."""
return self._state.get("mute", None)
@property
def shuffle(self):
"""Boolean if shuffle is enabled."""
return self._state.get("random", False)
@property
def repeat(self) -> RepeatMode:
"""Return current repeat mode."""
if self._state.get("repeat", None):
return RepeatMode.ALL
return RepeatMode.OFF
async def async_media_next_track(self) -> None:
"""Send media_next command to media player."""
await self._volumio.next()
async def async_media_previous_track(self) -> None:
"""Send media_previous command to media player."""
await self._volumio.previous()
async def async_media_play(self) -> None:
"""Send media_play command to media player."""
await self._volumio.play()
async def async_media_pause(self) -> None:
"""Send media_pause command to media player."""
if self._state.get("trackType") == "webradio":
await self._volumio.stop()
else:
await self._volumio.pause()
async def async_media_stop(self) -> None:
"""Send media_stop command to media player."""
await self._volumio.stop()
async def async_set_volume_level(self, volume: float) -> None:
"""Send volume_up command to media player."""
await self._volumio.set_volume_level(int(volume * 100))
async def async_volume_up(self) -> None:
"""Service to send the Volumio the command for volume up."""
await self._volumio.volume_up()
async def async_volume_down(self) -> None:
"""Service to send the Volumio the command for volume down."""
await self._volumio.volume_down()
async def async_mute_volume(self, mute: bool) -> None:
"""Send mute command to media player."""
if mute:
await self._volumio.mute()
else:
await self._volumio.unmute()
async def async_set_shuffle(self, shuffle: bool) -> None:
"""Enable/disable shuffle mode."""
await self._volumio.set_shuffle(shuffle)
async def async_set_repeat(self, repeat: RepeatMode) -> None:
"""Set repeat mode."""
if repeat == RepeatMode.OFF:
await self._volumio.repeatAll("false")
else:
await self._volumio.repeatAll("true")
async def async_select_source(self, source: str) -> None:
"""Choose an available playlist and play it."""
await self._volumio.play_playlist(source)
self._attr_source = source
async def async_clear_playlist(self) -> None:
"""Clear players playlist."""
await self._volumio.clear_playlist()
self._attr_source = None
@Throttle(PLAYLIST_UPDATE_INTERVAL)
async def _async_update_playlists(self, **kwargs):
"""Update available Volumio playlists."""
self._attr_source_list = await self._volumio.get_playlists()
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."""
await self._volumio.replace_and_play(json.loads(media_id))
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."""
self.thumbnail_cache = {}
if media_content_type in (None, "library"):
return await browse_top_level(self._volumio)
return await browse_node(
self, self._volumio, media_content_type, media_content_id
)
async def async_get_browse_image(
self,
media_content_type: MediaType | str,
media_content_id: str,
media_image_id: str | None = None,
) -> tuple[bytes | None, str | None]:
"""Get album art from Volumio."""
cached_url = self.thumbnail_cache.get(media_content_id)
image_url = self._volumio.canonic_url(cached_url)
return await self._async_fetch_image(image_url)