899 lines
32 KiB
Python
899 lines
32 KiB
Python
"""Provide functionality to interact with Cast devices on the network."""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from contextlib import suppress
|
|
from datetime import datetime, timedelta
|
|
import functools as ft
|
|
import json
|
|
import logging
|
|
from urllib.parse import quote
|
|
|
|
import pychromecast
|
|
from pychromecast.controllers.homeassistant import HomeAssistantController
|
|
from pychromecast.controllers.multizone import MultizoneManager
|
|
from pychromecast.controllers.plex import PlexController
|
|
from pychromecast.controllers.receiver import VOLUME_CONTROL_TYPE_FIXED
|
|
from pychromecast.quick_play import quick_play
|
|
from pychromecast.socket_client import (
|
|
CONNECTION_STATUS_CONNECTED,
|
|
CONNECTION_STATUS_DISCONNECTED,
|
|
)
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.auth.models import RefreshToken
|
|
from homeassistant.components import media_source, zeroconf
|
|
from homeassistant.components.http.auth import async_sign_path
|
|
from homeassistant.components.media_player import MediaPlayerEntity
|
|
from homeassistant.components.media_player.const import (
|
|
ATTR_MEDIA_EXTRA,
|
|
MEDIA_TYPE_MOVIE,
|
|
MEDIA_TYPE_MUSIC,
|
|
MEDIA_TYPE_TVSHOW,
|
|
SUPPORT_BROWSE_MEDIA,
|
|
SUPPORT_NEXT_TRACK,
|
|
SUPPORT_PAUSE,
|
|
SUPPORT_PLAY,
|
|
SUPPORT_PLAY_MEDIA,
|
|
SUPPORT_PREVIOUS_TRACK,
|
|
SUPPORT_SEEK,
|
|
SUPPORT_STOP,
|
|
SUPPORT_TURN_OFF,
|
|
SUPPORT_TURN_ON,
|
|
SUPPORT_VOLUME_MUTE,
|
|
SUPPORT_VOLUME_SET,
|
|
)
|
|
from homeassistant.components.plex.const import PLEX_URI_SCHEME
|
|
from homeassistant.components.plex.services import lookup_plex_media
|
|
from homeassistant.const import (
|
|
CAST_APP_ID_HOMEASSISTANT_LOVELACE,
|
|
CAST_APP_ID_HOMEASSISTANT_MEDIA,
|
|
EVENT_HOMEASSISTANT_STOP,
|
|
STATE_IDLE,
|
|
STATE_OFF,
|
|
STATE_PAUSED,
|
|
STATE_PLAYING,
|
|
)
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers import config_validation as cv
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
from homeassistant.helpers.entity import DeviceInfo
|
|
from homeassistant.helpers.network import NoURLAvailableError, get_url
|
|
import homeassistant.util.dt as dt_util
|
|
from homeassistant.util.logging import async_create_catching_coro
|
|
|
|
from .const import (
|
|
ADDED_CAST_DEVICES_KEY,
|
|
CAST_MULTIZONE_MANAGER_KEY,
|
|
CONF_IGNORE_CEC,
|
|
CONF_UUID,
|
|
DOMAIN as CAST_DOMAIN,
|
|
SIGNAL_CAST_DISCOVERED,
|
|
SIGNAL_CAST_REMOVED,
|
|
SIGNAL_HASS_CAST_SHOW_VIEW,
|
|
)
|
|
from .discovery import setup_internal_discovery
|
|
from .helpers import CastStatusListener, ChromecastInfo, ChromeCastZeroconf
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CAST_SPLASH = "https://www.home-assistant.io/images/cast/splash.png"
|
|
|
|
SUPPORT_CAST = (
|
|
SUPPORT_PAUSE | SUPPORT_PLAY | SUPPORT_PLAY_MEDIA | SUPPORT_STOP | SUPPORT_TURN_OFF
|
|
)
|
|
|
|
STATE_CASTING = "casting"
|
|
|
|
ENTITY_SCHEMA = vol.All(
|
|
vol.Schema(
|
|
{
|
|
vol.Optional(CONF_UUID): cv.string,
|
|
vol.Optional(CONF_IGNORE_CEC): vol.All(cv.ensure_list, [cv.string]),
|
|
}
|
|
),
|
|
)
|
|
|
|
|
|
@callback
|
|
def _async_create_cast_device(hass: HomeAssistant, info: ChromecastInfo):
|
|
"""Create a CastDevice Entity from the chromecast object.
|
|
|
|
Returns None if the cast device has already been added.
|
|
"""
|
|
_LOGGER.debug("_async_create_cast_device: %s", info)
|
|
if info.uuid is None:
|
|
_LOGGER.error("_async_create_cast_device uuid none: %s", info)
|
|
return None
|
|
|
|
# Found a cast with UUID
|
|
added_casts = hass.data[ADDED_CAST_DEVICES_KEY]
|
|
if info.uuid in added_casts:
|
|
# Already added this one, the entity will take care of moved hosts
|
|
# itself
|
|
return None
|
|
# -> New cast device
|
|
added_casts.add(info.uuid)
|
|
|
|
if info.is_dynamic_group:
|
|
# This is a dynamic group, do not add it but connect to the service.
|
|
group = DynamicCastGroup(hass, info)
|
|
group.async_setup()
|
|
return None
|
|
|
|
return CastDevice(info)
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
"""Set up Cast from a config entry."""
|
|
hass.data.setdefault(ADDED_CAST_DEVICES_KEY, set())
|
|
|
|
# Import CEC IGNORE attributes
|
|
pychromecast.IGNORE_CEC += config_entry.data.get(CONF_IGNORE_CEC) or []
|
|
|
|
wanted_uuids = config_entry.data.get(CONF_UUID) or None
|
|
|
|
@callback
|
|
def async_cast_discovered(discover: ChromecastInfo) -> None:
|
|
"""Handle discovery of a new chromecast."""
|
|
# If wanted_uuids is set, we're only accepting specific cast devices identified
|
|
# by UUID
|
|
if wanted_uuids is not None and discover.uuid not in wanted_uuids:
|
|
# UUID not matching, ignore.
|
|
return
|
|
|
|
cast_device = _async_create_cast_device(hass, discover)
|
|
if cast_device is not None:
|
|
async_add_entities([cast_device])
|
|
|
|
async_dispatcher_connect(hass, SIGNAL_CAST_DISCOVERED, async_cast_discovered)
|
|
ChromeCastZeroconf.set_zeroconf(await zeroconf.async_get_instance(hass))
|
|
hass.async_add_executor_job(setup_internal_discovery, hass, config_entry)
|
|
|
|
|
|
class CastDevice(MediaPlayerEntity):
|
|
"""Representation of a Cast device on the network.
|
|
|
|
This class is the holder of the pychromecast.Chromecast object and its
|
|
socket client. It therefore handles all reconnects and audio group changing
|
|
"elected leader" itself.
|
|
"""
|
|
|
|
_attr_should_poll = False
|
|
_attr_media_image_remotely_accessible = True
|
|
|
|
def __init__(self, cast_info: ChromecastInfo) -> None:
|
|
"""Initialize the cast device."""
|
|
|
|
self._cast_info = cast_info
|
|
self.services = cast_info.services
|
|
self._chromecast: pychromecast.Chromecast | None = None
|
|
self.cast_status = None
|
|
self.media_status = None
|
|
self.media_status_received = None
|
|
self.mz_media_status: dict[str, pychromecast.controllers.media.MediaStatus] = {}
|
|
self.mz_media_status_received: dict[str, datetime] = {}
|
|
self.mz_mgr = None
|
|
self._attr_available = False
|
|
self._status_listener: CastStatusListener | None = None
|
|
self._hass_cast_controller: HomeAssistantController | None = None
|
|
|
|
self._add_remove_handler = None
|
|
self._cast_view_remove_handler = None
|
|
self._attr_unique_id = cast_info.uuid
|
|
self._attr_name = cast_info.friendly_name
|
|
if cast_info.model_name != "Google Cast Group":
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(CAST_DOMAIN, str(cast_info.uuid).replace("-", ""))},
|
|
manufacturer=str(cast_info.manufacturer),
|
|
model=cast_info.model_name,
|
|
name=str(cast_info.friendly_name),
|
|
)
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Create chromecast object when added to hass."""
|
|
self._add_remove_handler = async_dispatcher_connect(
|
|
self.hass, SIGNAL_CAST_DISCOVERED, self._async_cast_discovered
|
|
)
|
|
self.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, self._async_stop)
|
|
self.async_set_cast_info(self._cast_info)
|
|
# asyncio.create_task is used to avoid delaying startup wrapup if the device
|
|
# is discovered already during startup but then fails to respond
|
|
asyncio.create_task(
|
|
async_create_catching_coro(self.async_connect_to_chromecast())
|
|
)
|
|
|
|
self._cast_view_remove_handler = async_dispatcher_connect(
|
|
self.hass, SIGNAL_HASS_CAST_SHOW_VIEW, self._handle_signal_show_view
|
|
)
|
|
|
|
async def async_will_remove_from_hass(self) -> None:
|
|
"""Disconnect Chromecast object when removed."""
|
|
await self._async_disconnect()
|
|
if self._add_remove_handler:
|
|
self._add_remove_handler()
|
|
self._add_remove_handler = None
|
|
if self._cast_view_remove_handler:
|
|
self._cast_view_remove_handler()
|
|
self._cast_view_remove_handler = None
|
|
|
|
def async_set_cast_info(self, cast_info):
|
|
"""Set the cast information."""
|
|
self._cast_info = cast_info
|
|
|
|
async def async_connect_to_chromecast(self):
|
|
"""Set up the chromecast object."""
|
|
|
|
_LOGGER.debug(
|
|
"[%s %s] Connecting to cast device by service %s",
|
|
self.entity_id,
|
|
self._cast_info.friendly_name,
|
|
self.services,
|
|
)
|
|
chromecast = await self.hass.async_add_executor_job(
|
|
pychromecast.get_chromecast_from_cast_info,
|
|
pychromecast.discovery.CastInfo(
|
|
self.services,
|
|
self._cast_info.uuid,
|
|
self._cast_info.model_name,
|
|
self._cast_info.friendly_name,
|
|
None,
|
|
None,
|
|
),
|
|
ChromeCastZeroconf.get_zeroconf(),
|
|
)
|
|
chromecast.media_controller.app_id = CAST_APP_ID_HOMEASSISTANT_MEDIA
|
|
self._chromecast = chromecast
|
|
|
|
if CAST_MULTIZONE_MANAGER_KEY not in self.hass.data:
|
|
self.hass.data[CAST_MULTIZONE_MANAGER_KEY] = MultizoneManager()
|
|
|
|
self.mz_mgr = self.hass.data[CAST_MULTIZONE_MANAGER_KEY]
|
|
|
|
self._status_listener = CastStatusListener(self, chromecast, self.mz_mgr)
|
|
self._attr_available = False
|
|
self.cast_status = chromecast.status
|
|
self.media_status = chromecast.media_controller.status
|
|
self._chromecast.start()
|
|
self.async_write_ha_state()
|
|
|
|
async def _async_disconnect(self):
|
|
"""Disconnect Chromecast object if it is set."""
|
|
if self._chromecast is None:
|
|
# Can't disconnect if not connected.
|
|
return
|
|
_LOGGER.debug(
|
|
"[%s %s] Disconnecting from chromecast socket",
|
|
self.entity_id,
|
|
self._cast_info.friendly_name,
|
|
)
|
|
self._attr_available = False
|
|
self.async_write_ha_state()
|
|
|
|
await self.hass.async_add_executor_job(self._chromecast.disconnect)
|
|
|
|
self._invalidate()
|
|
|
|
self.async_write_ha_state()
|
|
|
|
def _invalidate(self):
|
|
"""Invalidate some attributes."""
|
|
self._chromecast = None
|
|
self.cast_status = None
|
|
self.media_status = None
|
|
self.media_status_received = None
|
|
self.mz_media_status = {}
|
|
self.mz_media_status_received = {}
|
|
self.mz_mgr = None
|
|
self._hass_cast_controller = None
|
|
if self._status_listener is not None:
|
|
self._status_listener.invalidate()
|
|
self._status_listener = None
|
|
|
|
# ========== Callbacks ==========
|
|
def new_cast_status(self, cast_status):
|
|
"""Handle updates of the cast status."""
|
|
self.cast_status = cast_status
|
|
self._attr_volume_level = cast_status.volume_level if cast_status else None
|
|
self._attr_is_volume_muted = (
|
|
cast_status.volume_muted if self.cast_status else None
|
|
)
|
|
self.schedule_update_ha_state()
|
|
|
|
def new_media_status(self, media_status):
|
|
"""Handle updates of the media status."""
|
|
if (
|
|
media_status
|
|
and media_status.player_is_idle
|
|
and media_status.idle_reason == "ERROR"
|
|
):
|
|
external_url = None
|
|
internal_url = None
|
|
tts_base_url = None
|
|
url_description = ""
|
|
if "tts" in self.hass.config.components:
|
|
with suppress(KeyError): # base_url not configured
|
|
tts_base_url = self.hass.components.tts.get_base_url(self.hass)
|
|
|
|
with suppress(NoURLAvailableError): # external_url not configured
|
|
external_url = get_url(self.hass, allow_internal=False)
|
|
|
|
with suppress(NoURLAvailableError): # internal_url not configured
|
|
internal_url = get_url(self.hass, allow_external=False)
|
|
|
|
if media_status.content_id:
|
|
if tts_base_url and media_status.content_id.startswith(tts_base_url):
|
|
url_description = f" from tts.base_url ({tts_base_url})"
|
|
if external_url and media_status.content_id.startswith(external_url):
|
|
url_description = f" from external_url ({external_url})"
|
|
if internal_url and media_status.content_id.startswith(internal_url):
|
|
url_description = f" from internal_url ({internal_url})"
|
|
|
|
_LOGGER.error(
|
|
"Failed to cast media %s%s. Please make sure the URL is: "
|
|
"Reachable from the cast device and either a publicly resolvable "
|
|
"hostname or an IP address",
|
|
media_status.content_id,
|
|
url_description,
|
|
)
|
|
|
|
self.media_status = media_status
|
|
self.media_status_received = dt_util.utcnow()
|
|
self.schedule_update_ha_state()
|
|
|
|
def new_connection_status(self, connection_status):
|
|
"""Handle updates of connection status."""
|
|
_LOGGER.debug(
|
|
"[%s %s] Received cast device connection status: %s",
|
|
self.entity_id,
|
|
self._cast_info.friendly_name,
|
|
connection_status.status,
|
|
)
|
|
if connection_status.status == CONNECTION_STATUS_DISCONNECTED:
|
|
self._attr_available = False
|
|
self._invalidate()
|
|
self.schedule_update_ha_state()
|
|
return
|
|
|
|
new_available = connection_status.status == CONNECTION_STATUS_CONNECTED
|
|
if new_available != self.available:
|
|
# Connection status callbacks happen often when disconnected.
|
|
# Only update state when availability changed to put less pressure
|
|
# on state machine.
|
|
_LOGGER.debug(
|
|
"[%s %s] Cast device availability changed: %s",
|
|
self.entity_id,
|
|
self._cast_info.friendly_name,
|
|
connection_status.status,
|
|
)
|
|
self._attr_available = new_available
|
|
self.schedule_update_ha_state()
|
|
|
|
def multizone_new_media_status(self, group_uuid, media_status):
|
|
"""Handle updates of audio group media status."""
|
|
_LOGGER.debug(
|
|
"[%s %s] Multizone %s media status: %s",
|
|
self.entity_id,
|
|
self._cast_info.friendly_name,
|
|
group_uuid,
|
|
media_status,
|
|
)
|
|
self.mz_media_status[group_uuid] = media_status
|
|
self.mz_media_status_received[group_uuid] = dt_util.utcnow()
|
|
self.schedule_update_ha_state()
|
|
|
|
# ========== Service Calls ==========
|
|
def _media_controller(self):
|
|
"""
|
|
Return media controller.
|
|
|
|
First try from our own cast, then groups which our cast is a member in.
|
|
"""
|
|
media_status = self.media_status
|
|
media_controller = self._chromecast.media_controller
|
|
|
|
if media_status is None or media_status.player_state == "UNKNOWN":
|
|
groups = self.mz_media_status
|
|
for k, val in groups.items():
|
|
if val and val.player_state != "UNKNOWN":
|
|
media_controller = self.mz_mgr.get_multizone_mediacontroller(k)
|
|
break
|
|
|
|
return media_controller
|
|
|
|
def turn_on(self):
|
|
"""Turn on the cast device."""
|
|
|
|
if not self._chromecast.is_idle:
|
|
# Already turned on
|
|
return
|
|
|
|
if self._chromecast.app_id is not None:
|
|
# Quit the previous app before starting splash screen
|
|
self._chromecast.quit_app()
|
|
|
|
# The only way we can turn the Chromecast is on is by launching an app
|
|
self._chromecast.play_media(CAST_SPLASH, pychromecast.STREAM_TYPE_BUFFERED)
|
|
|
|
def turn_off(self):
|
|
"""Turn off the cast device."""
|
|
self._chromecast.quit_app()
|
|
|
|
def mute_volume(self, mute):
|
|
"""Mute the volume."""
|
|
self._chromecast.set_volume_muted(mute)
|
|
|
|
def set_volume_level(self, volume):
|
|
"""Set volume level, range 0..1."""
|
|
self._chromecast.set_volume(volume)
|
|
|
|
def media_play(self):
|
|
"""Send play command."""
|
|
media_controller = self._media_controller()
|
|
media_controller.play()
|
|
|
|
def media_pause(self):
|
|
"""Send pause command."""
|
|
media_controller = self._media_controller()
|
|
media_controller.pause()
|
|
|
|
def media_stop(self):
|
|
"""Send stop command."""
|
|
media_controller = self._media_controller()
|
|
media_controller.stop()
|
|
|
|
def media_previous_track(self):
|
|
"""Send previous track command."""
|
|
media_controller = self._media_controller()
|
|
media_controller.queue_prev()
|
|
|
|
def media_next_track(self):
|
|
"""Send next track command."""
|
|
media_controller = self._media_controller()
|
|
media_controller.queue_next()
|
|
|
|
def media_seek(self, position):
|
|
"""Seek the media to a specific location."""
|
|
media_controller = self._media_controller()
|
|
media_controller.seek(position)
|
|
|
|
async def async_browse_media(self, media_content_type=None, media_content_id=None):
|
|
"""Implement the websocket media browsing helper."""
|
|
result = await media_source.async_browse_media(self.hass, media_content_id)
|
|
return result
|
|
|
|
async def async_play_media(self, media_type, media_id, **kwargs):
|
|
"""Play a piece of media."""
|
|
# Handle media_source
|
|
if media_source.is_media_source_id(media_id):
|
|
sourced_media = await media_source.async_resolve_media(self.hass, media_id)
|
|
media_type = sourced_media.mime_type
|
|
media_id = sourced_media.url
|
|
|
|
# If media ID is a relative URL, we serve it from HA.
|
|
# Create a signed path.
|
|
if media_id[0] == "/":
|
|
# Sign URL with Home Assistant Cast User
|
|
config_entry_id = self.registry_entry.config_entry_id
|
|
config_entry = self.hass.config_entries.async_get_entry(config_entry_id)
|
|
user_id = config_entry.data["user_id"]
|
|
user = await self.hass.auth.async_get_user(user_id)
|
|
if user.refresh_tokens:
|
|
refresh_token: RefreshToken = list(user.refresh_tokens.values())[0]
|
|
|
|
media_id = async_sign_path(
|
|
self.hass,
|
|
refresh_token.id,
|
|
quote(media_id),
|
|
timedelta(seconds=media_source.DEFAULT_EXPIRY_TIME),
|
|
)
|
|
|
|
# prepend external URL
|
|
hass_url = get_url(self.hass, prefer_external=True)
|
|
media_id = f"{hass_url}{media_id}"
|
|
|
|
await self.hass.async_add_executor_job(
|
|
ft.partial(self.play_media, media_type, media_id, **kwargs)
|
|
)
|
|
|
|
def play_media(self, media_type, media_id, **kwargs):
|
|
"""Play media from a URL."""
|
|
extra = kwargs.get(ATTR_MEDIA_EXTRA, {})
|
|
metadata = extra.get("metadata")
|
|
|
|
# We do not want this to be forwarded to a group
|
|
if media_type == CAST_DOMAIN:
|
|
try:
|
|
app_data = json.loads(media_id)
|
|
if metadata is not None:
|
|
app_data["metadata"] = extra.get("metadata")
|
|
except json.JSONDecodeError:
|
|
_LOGGER.error("Invalid JSON in media_content_id")
|
|
raise
|
|
|
|
# Special handling for passed `app_id` parameter. This will only launch
|
|
# an arbitrary cast app, generally for UX.
|
|
if "app_id" in app_data:
|
|
app_id = app_data.pop("app_id")
|
|
_LOGGER.info("Starting Cast app by ID %s", app_id)
|
|
self._chromecast.start_app(app_id)
|
|
if app_data:
|
|
_LOGGER.warning(
|
|
"Extra keys %s were ignored. Please use app_name to cast media",
|
|
app_data.keys(),
|
|
)
|
|
return
|
|
|
|
app_name = app_data.pop("app_name")
|
|
try:
|
|
quick_play(self._chromecast, app_name, app_data)
|
|
except NotImplementedError:
|
|
_LOGGER.error("App %s not supported", app_name)
|
|
# Handle plex
|
|
elif media_id and media_id.startswith(PLEX_URI_SCHEME):
|
|
media_id = media_id[len(PLEX_URI_SCHEME) :]
|
|
media, _ = lookup_plex_media(self.hass, media_type, media_id)
|
|
if media is None:
|
|
return
|
|
controller = PlexController()
|
|
self._chromecast.register_handler(controller)
|
|
controller.play_media(media)
|
|
else:
|
|
self._chromecast.media_controller.play_media(
|
|
media_id, media_type, **kwargs.get(ATTR_MEDIA_EXTRA, {})
|
|
)
|
|
|
|
def _media_status(self):
|
|
"""
|
|
Return media status.
|
|
|
|
First try from our own cast, then groups which our cast is a member in.
|
|
"""
|
|
media_status = self.media_status
|
|
media_status_received = self.media_status_received
|
|
|
|
if media_status is None or media_status.player_state == "UNKNOWN":
|
|
groups = self.mz_media_status
|
|
for k, val in groups.items():
|
|
if val and val.player_state != "UNKNOWN":
|
|
media_status = val
|
|
media_status_received = self.mz_media_status_received[k]
|
|
break
|
|
|
|
return (media_status, media_status_received)
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the player."""
|
|
# The lovelace app loops media to prevent timing out, don't show that
|
|
if self.app_id == CAST_APP_ID_HOMEASSISTANT_LOVELACE:
|
|
return STATE_CASTING
|
|
if (media_status := self._media_status()[0]) is not None:
|
|
if media_status.player_is_playing:
|
|
return STATE_PLAYING
|
|
if media_status.player_is_paused:
|
|
return STATE_PAUSED
|
|
if media_status.player_is_idle:
|
|
return STATE_IDLE
|
|
if self.app_id is not None and self.app_id != pychromecast.IDLE_APP_ID:
|
|
return STATE_CASTING
|
|
if self._chromecast is not None and self._chromecast.is_idle:
|
|
return STATE_OFF
|
|
return None
|
|
|
|
@property
|
|
def media_content_id(self):
|
|
"""Content ID of current playing media."""
|
|
# The lovelace app loops media to prevent timing out, don't show that
|
|
if self.app_id == CAST_APP_ID_HOMEASSISTANT_LOVELACE:
|
|
return None
|
|
media_status = self._media_status()[0]
|
|
return media_status.content_id if media_status else None
|
|
|
|
@property
|
|
def media_content_type(self):
|
|
"""Content type of current playing media."""
|
|
# The lovelace app loops media to prevent timing out, don't show that
|
|
if self.app_id == CAST_APP_ID_HOMEASSISTANT_LOVELACE:
|
|
return None
|
|
if (media_status := self._media_status()[0]) is None:
|
|
return None
|
|
if media_status.media_is_tvshow:
|
|
return MEDIA_TYPE_TVSHOW
|
|
if media_status.media_is_movie:
|
|
return MEDIA_TYPE_MOVIE
|
|
if media_status.media_is_musictrack:
|
|
return MEDIA_TYPE_MUSIC
|
|
return None
|
|
|
|
@property
|
|
def media_duration(self):
|
|
"""Duration of current playing media in seconds."""
|
|
# The lovelace app loops media to prevent timing out, don't show that
|
|
if self.app_id == CAST_APP_ID_HOMEASSISTANT_LOVELACE:
|
|
return None
|
|
media_status = self._media_status()[0]
|
|
return media_status.duration if media_status else None
|
|
|
|
@property
|
|
def media_image_url(self):
|
|
"""Image url of current playing media."""
|
|
if (media_status := self._media_status()[0]) is None:
|
|
return None
|
|
|
|
images = media_status.images
|
|
|
|
return images[0].url if images and images[0].url else None
|
|
|
|
@property
|
|
def media_title(self):
|
|
"""Title of current playing media."""
|
|
media_status = self._media_status()[0]
|
|
return media_status.title if media_status else None
|
|
|
|
@property
|
|
def media_artist(self):
|
|
"""Artist of current playing media (Music track only)."""
|
|
media_status = self._media_status()[0]
|
|
return media_status.artist if media_status else None
|
|
|
|
@property
|
|
def media_album_name(self):
|
|
"""Album of current playing media (Music track only)."""
|
|
media_status = self._media_status()[0]
|
|
return media_status.album_name if media_status else None
|
|
|
|
@property
|
|
def media_album_artist(self):
|
|
"""Album artist of current playing media (Music track only)."""
|
|
media_status = self._media_status()[0]
|
|
return media_status.album_artist if media_status else None
|
|
|
|
@property
|
|
def media_track(self):
|
|
"""Track number of current playing media (Music track only)."""
|
|
media_status = self._media_status()[0]
|
|
return media_status.track if media_status else None
|
|
|
|
@property
|
|
def media_series_title(self):
|
|
"""Return the title of the series of current playing media."""
|
|
media_status = self._media_status()[0]
|
|
return media_status.series_title if media_status else None
|
|
|
|
@property
|
|
def media_season(self):
|
|
"""Season of current playing media (TV Show only)."""
|
|
media_status = self._media_status()[0]
|
|
return media_status.season if media_status else None
|
|
|
|
@property
|
|
def media_episode(self):
|
|
"""Episode of current playing media (TV Show only)."""
|
|
media_status = self._media_status()[0]
|
|
return media_status.episode if media_status else None
|
|
|
|
@property
|
|
def app_id(self):
|
|
"""Return the ID of the current running app."""
|
|
return self._chromecast.app_id if self._chromecast else None
|
|
|
|
@property
|
|
def app_name(self):
|
|
"""Name of the current running app."""
|
|
return self._chromecast.app_display_name if self._chromecast else None
|
|
|
|
@property
|
|
def supported_features(self):
|
|
"""Flag media player features that are supported."""
|
|
support = SUPPORT_CAST
|
|
media_status = self._media_status()[0]
|
|
|
|
if (
|
|
self._chromecast
|
|
and self._chromecast.cast_type == pychromecast.const.CAST_TYPE_CHROMECAST
|
|
):
|
|
support |= SUPPORT_TURN_ON
|
|
|
|
if (
|
|
self.cast_status
|
|
and self.cast_status.volume_control_type != VOLUME_CONTROL_TYPE_FIXED
|
|
):
|
|
support |= SUPPORT_VOLUME_MUTE | SUPPORT_VOLUME_SET
|
|
|
|
if media_status:
|
|
if media_status.supports_queue_next:
|
|
support |= SUPPORT_PREVIOUS_TRACK | SUPPORT_NEXT_TRACK
|
|
if media_status.supports_seek:
|
|
support |= SUPPORT_SEEK
|
|
|
|
if "media_source" in self.hass.config.components:
|
|
support |= SUPPORT_BROWSE_MEDIA
|
|
|
|
return support
|
|
|
|
@property
|
|
def media_position(self):
|
|
"""Position of current playing media in seconds."""
|
|
# The lovelace app loops media to prevent timing out, don't show that
|
|
if self.app_id == CAST_APP_ID_HOMEASSISTANT_LOVELACE:
|
|
return None
|
|
media_status = self._media_status()[0]
|
|
if media_status is None or not (
|
|
media_status.player_is_playing
|
|
or media_status.player_is_paused
|
|
or media_status.player_is_idle
|
|
):
|
|
return None
|
|
return media_status.current_time
|
|
|
|
@property
|
|
def media_position_updated_at(self):
|
|
"""When was the position of the current playing media valid.
|
|
|
|
Returns value from homeassistant.util.dt.utcnow().
|
|
"""
|
|
if self.app_id == CAST_APP_ID_HOMEASSISTANT_LOVELACE:
|
|
return None
|
|
media_status_recevied = self._media_status()[1]
|
|
return media_status_recevied
|
|
|
|
async def _async_cast_discovered(self, discover: ChromecastInfo):
|
|
"""Handle discovery of new Chromecast."""
|
|
if self._cast_info.uuid != discover.uuid:
|
|
# Discovered is not our device.
|
|
return
|
|
|
|
_LOGGER.debug("Discovered chromecast with same UUID: %s", discover)
|
|
self.async_set_cast_info(discover)
|
|
|
|
async def _async_stop(self, event):
|
|
"""Disconnect socket on Home Assistant stop."""
|
|
await self._async_disconnect()
|
|
|
|
def _handle_signal_show_view(
|
|
self,
|
|
controller: HomeAssistantController,
|
|
entity_id: str,
|
|
view_path: str,
|
|
url_path: str | None,
|
|
):
|
|
"""Handle a show view signal."""
|
|
if entity_id != self.entity_id or self._chromecast is None:
|
|
return
|
|
|
|
if self._hass_cast_controller is None:
|
|
self._hass_cast_controller = controller
|
|
self._chromecast.register_handler(controller)
|
|
|
|
self._hass_cast_controller.show_lovelace_view(view_path, url_path)
|
|
|
|
|
|
class DynamicCastGroup:
|
|
"""Representation of a Cast device on the network - for dynamic cast groups."""
|
|
|
|
def __init__(self, hass, cast_info: ChromecastInfo):
|
|
"""Initialize the cast device."""
|
|
|
|
self.hass = hass
|
|
self._cast_info = cast_info
|
|
self.services = cast_info.services
|
|
self._chromecast: pychromecast.Chromecast | None = None
|
|
self.mz_mgr = None
|
|
self._status_listener: CastStatusListener | None = None
|
|
|
|
self._add_remove_handler = None
|
|
self._del_remove_handler = None
|
|
|
|
def async_setup(self):
|
|
"""Create chromecast object."""
|
|
self._add_remove_handler = async_dispatcher_connect(
|
|
self.hass, SIGNAL_CAST_DISCOVERED, self._async_cast_discovered
|
|
)
|
|
self._del_remove_handler = async_dispatcher_connect(
|
|
self.hass, SIGNAL_CAST_REMOVED, self._async_cast_removed
|
|
)
|
|
self.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, self._async_stop)
|
|
self.async_set_cast_info(self._cast_info)
|
|
self.hass.async_create_task(
|
|
async_create_catching_coro(self.async_connect_to_chromecast())
|
|
)
|
|
|
|
async def async_tear_down(self) -> None:
|
|
"""Disconnect Chromecast object."""
|
|
await self._async_disconnect()
|
|
if self._cast_info.uuid is not None:
|
|
# Remove the entity from the added casts so that it can dynamically
|
|
# be re-added again.
|
|
self.hass.data[ADDED_CAST_DEVICES_KEY].remove(self._cast_info.uuid)
|
|
if self._add_remove_handler:
|
|
self._add_remove_handler()
|
|
self._add_remove_handler = None
|
|
if self._del_remove_handler:
|
|
self._del_remove_handler()
|
|
self._del_remove_handler = None
|
|
|
|
def async_set_cast_info(self, cast_info):
|
|
"""Set the cast information and set up the chromecast object."""
|
|
|
|
self._cast_info = cast_info
|
|
|
|
async def async_connect_to_chromecast(self):
|
|
"""Set the cast information and set up the chromecast object."""
|
|
|
|
_LOGGER.debug(
|
|
"[%s %s] Connecting to cast device by service %s",
|
|
"Dynamic group",
|
|
self._cast_info.friendly_name,
|
|
self.services,
|
|
)
|
|
chromecast = await self.hass.async_add_executor_job(
|
|
pychromecast.get_chromecast_from_cast_info,
|
|
pychromecast.discovery.CastInfo(
|
|
self.services,
|
|
self._cast_info.uuid,
|
|
self._cast_info.model_name,
|
|
self._cast_info.friendly_name,
|
|
None,
|
|
None,
|
|
),
|
|
ChromeCastZeroconf.get_zeroconf(),
|
|
)
|
|
chromecast.media_controller.app_id = CAST_APP_ID_HOMEASSISTANT_MEDIA
|
|
self._chromecast = chromecast
|
|
|
|
if CAST_MULTIZONE_MANAGER_KEY not in self.hass.data:
|
|
self.hass.data[CAST_MULTIZONE_MANAGER_KEY] = MultizoneManager()
|
|
|
|
self.mz_mgr = self.hass.data[CAST_MULTIZONE_MANAGER_KEY]
|
|
|
|
self._status_listener = CastStatusListener(self, chromecast, self.mz_mgr, True)
|
|
self._chromecast.start()
|
|
|
|
async def _async_disconnect(self):
|
|
"""Disconnect Chromecast object if it is set."""
|
|
if self._chromecast is None:
|
|
# Can't disconnect if not connected.
|
|
return
|
|
_LOGGER.debug(
|
|
"[%s %s] Disconnecting from chromecast socket",
|
|
"Dynamic group",
|
|
self._cast_info.friendly_name,
|
|
)
|
|
|
|
await self.hass.async_add_executor_job(self._chromecast.disconnect)
|
|
|
|
self._invalidate()
|
|
|
|
def _invalidate(self):
|
|
"""Invalidate some attributes."""
|
|
self._chromecast = None
|
|
self.mz_mgr = None
|
|
if self._status_listener is not None:
|
|
self._status_listener.invalidate()
|
|
self._status_listener = None
|
|
|
|
async def _async_cast_discovered(self, discover: ChromecastInfo):
|
|
"""Handle discovery of new Chromecast."""
|
|
if self._cast_info.uuid != discover.uuid:
|
|
# Discovered is not our device.
|
|
return
|
|
|
|
_LOGGER.debug("Discovered dynamic group with same UUID: %s", discover)
|
|
self.async_set_cast_info(discover)
|
|
|
|
async def _async_cast_removed(self, discover: ChromecastInfo):
|
|
"""Handle removal of Chromecast."""
|
|
if self._cast_info.uuid != discover.uuid:
|
|
# Removed is not our device.
|
|
return
|
|
|
|
if not discover.services:
|
|
# Clean up the dynamic group
|
|
_LOGGER.debug("Clean up dynamic group: %s", discover)
|
|
await self.async_tear_down()
|
|
|
|
async def _async_stop(self, event):
|
|
"""Disconnect socket on Home Assistant stop."""
|
|
await self._async_disconnect()
|