RelayTV/RelayTV-HA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Home Assitant Integration

Browse Source
This commit is contained in:
mark
2026-02-23 15:23:00 -06:00
parent 3204f07947
commit 87de29797e
7 changed files with 606 additions and 176 deletions
Download Patch File Download Diff File Expand all files Collapse all files
custom_components/relaytv/__init__.py
+191 -102
View File
@@ -1,43 +1,36 @@
"""RelayTV integration.
This integration:
1) Registers a Home Assistant sidebar iframe panel that embeds the RelayTV web UI.
2) Exposes a RelayTV media_player entity backed by RelayTV's local HTTP API.
3) Provides services (e.g., relaytv.smart_url) for automations and mobile share flows.
The API layer targets RelayTV's canonical endpoints (see RelayTV server docs/API.md).
"""
"""RelayTV integration."""
from __future__ import annotations
import logging
from urllib.parse import urlparse
from homeassistant.core import HomeAssistant
from homeassistant.config_entries import ConfigEntry
from homeassistant.components import frontend
from homeassistant.helpers import aiohttp_client
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ENTITY_ID
from homeassistant.core import HomeAssistant, ServiceCall
from homeassistant.helpers import aiohttp_client, entity_registry as er
from homeassistant.helpers.storage import Store
from .const import (
CONF_BASE_URL,
CONF_PANEL_ICON,
CONF_PANEL_PATH,
CONF_PANEL_TITLE,
CONF_PANEL_ENABLED,
CONF_PANEL_TARGET_ENTRY_ID,
DATA_API,
DATA_COORDINATOR,
DATA_PANEL_SETTINGS,
DATA_STORE,
DEFAULT_PANEL_ICON,
DEFAULT_PANEL_PATH,
DEFAULT_PANEL_TITLE,
DATA_API,
DATA_COORDINATOR,
DOMAIN,
PLATFORMS,
SERVICE_SMART_URL,
SERVICE_PLAY_NOW,
SERVICE_ANNOUNCE,
SERVICE_PLAY_NOW,
SERVICE_SMART_URL,
)
from .relaytv_api import RelayTVApi
from .coordinator import RelayTVCoordinator
from .relaytv_api import RelayTVApi
_LOGGER = logging.getLogger(__name__)
@@ -47,72 +40,188 @@ def _normalize_base_url(raw: str) -> str:
raw = (raw or "").strip()
if not raw:
return ""
# Allow users to paste host:port, add scheme.
if "://" not in raw:
raw = f"http://{raw}"
# Basic parse/normalize; keep path if user provided one.
p = urlparse(raw)
if not p.netloc:
parsed = urlparse(raw)
if not parsed.netloc:
return raw
# Remove trailing slash to avoid double slashes when HA appends.
normalized = f"{p.scheme}://{p.netloc}{p.path}".rstrip("/")
if p.query:
normalized += f"?{p.query}"
normalized = f"{parsed.scheme}://{parsed.netloc}{parsed.path}".rstrip("/")
if parsed.query:
normalized += f"?{parsed.query}"
return normalized
async def async_setup(hass: HomeAssistant, config: dict) -> bool:
"""Set up the RelayTV domain."""
hass.data.setdefault(DOMAIN, {})
return True
async def _async_ensure_settings(hass: HomeAssistant) -> dict:
domain_data = hass.data.setdefault(DOMAIN, {})
if DATA_STORE not in domain_data:
domain_data[DATA_STORE] = Store(hass, 1, f"{DOMAIN}_panel_settings")
if DATA_PANEL_SETTINGS not in domain_data:
domain_data[DATA_PANEL_SETTINGS] = await domain_data[DATA_STORE].async_load() or {
CONF_PANEL_ENABLED: True,
CONF_PANEL_TARGET_ENTRY_ID: None,
}
return domain_data[DATA_PANEL_SETTINGS]
async def _async_save_settings(hass: HomeAssistant) -> None:
data = hass.data.setdefault(DOMAIN, {})
await data[DATA_STORE].async_save(data[DATA_PANEL_SETTINGS])
def _get_entry_data(hass: HomeAssistant, entry_id: str) -> dict | None:
return hass.data.get(DOMAIN, {}).get(entry_id)
def _fallback_entry_id(hass: HomeAssistant) -> str | None:
for entry_id, value in hass.data.get(DOMAIN, {}).items():
if isinstance(value, dict) and DATA_API in value:
return entry_id
return None
def _resolve_entry_id_for_call(hass: HomeAssistant, call: ServiceCall) -> str | None:
entity_ids: list[str] = []
raw_entity = call.data.get(CONF_ENTITY_ID)
if isinstance(raw_entity, str):
entity_ids = [raw_entity]
elif isinstance(raw_entity, list):
entity_ids = [item for item in raw_entity if isinstance(item, str)]
registry = er.async_get(hass)
if not entity_ids:
device_id = call.data.get("device_id")
device_ids = [device_id] if isinstance(device_id, str) else device_id
if isinstance(device_ids, list):
for item in device_ids:
if not isinstance(item, str):
continue
for reg_entry in er.async_entries_for_device(registry, item):
if reg_entry.entity_id.startswith("media_player."):
entity_ids.append(reg_entry.entity_id)
for entity_id in entity_ids:
reg_entry = registry.async_get(entity_id)
if reg_entry and _get_entry_data(hass, reg_entry.config_entry_id):
return reg_entry.config_entry_id
panel_target = hass.data.get(DOMAIN, {}).get(DATA_PANEL_SETTINGS, {}).get(CONF_PANEL_TARGET_ENTRY_ID)
if panel_target and _get_entry_data(hass, panel_target):
return panel_target
return _fallback_entry_id(hass)
def _register_panel(hass: HomeAssistant, *, path: str, title: str, icon: str, url: str) -> None:
frontend.async_register_built_in_panel(
hass,
component_name="iframe",
sidebar_title=title,
sidebar_icon=icon,
frontend_url_path=path,
config={"url": url},
require_admin=False,
)
def _async_unregister_panel(hass: HomeAssistant) -> None:
try:
frontend.async_remove_panel(hass, DEFAULT_PANEL_PATH)
except Exception:
_LOGGER.debug("Panel removal failed (it may not exist)", exc_info=True)
async def _async_update_panel(hass: HomeAssistant) -> None:
settings = await _async_ensure_settings(hass)
_async_unregister_panel(hass)
if not settings.get(CONF_PANEL_ENABLED, True):
return
target_entry_id = settings.get(CONF_PANEL_TARGET_ENTRY_ID)
target = _get_entry_data(hass, target_entry_id) if target_entry_id else None
if target is None:
fallback_id = _fallback_entry_id(hass)
if not fallback_id:
return
settings[CONF_PANEL_TARGET_ENTRY_ID] = fallback_id
await _async_save_settings(hass)
target = _get_entry_data(hass, fallback_id)
target_entry_id = fallback_id
if not target:
return
url = target[DATA_API].base_url
_register_panel(
hass,
path=DEFAULT_PANEL_PATH,
title=DEFAULT_PANEL_TITLE,
icon=DEFAULT_PANEL_ICON,
url=url,
)
_LOGGER.info("Registered RelayTV panel to entry %s (%s)", target_entry_id, url)
async def _async_set_default_sidebar_target(hass: HomeAssistant, entry: ConfigEntry) -> None:
settings = await _async_ensure_settings(hass)
settings[CONF_PANEL_TARGET_ENTRY_ID] = entry.entry_id
await _async_save_settings(hass)
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up RelayTV from a config entry."""
base_url = _normalize_base_url(entry.data.get(CONF_BASE_URL, ""))
if not base_url:
_LOGGER.error("RelayTV base URL is empty; panel will not be registered")
_LOGGER.error("RelayTV base URL is empty")
return False
# Store runtime objects
hass.data.setdefault(DOMAIN, {})
await _async_ensure_settings(hass)
session = aiohttp_client.async_get_clientsession(hass)
api = RelayTVApi(session=session, base_url=base_url)
coordinator = RelayTVCoordinator(hass=hass, api=api)
hass.data[DOMAIN][entry.entry_id] = {DATA_API: api, DATA_COORDINATOR: coordinator}
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = {DATA_API: api, DATA_COORDINATOR: coordinator}
# Prime coordinator (non-fatal if it fails; entity will show unavailable).
await coordinator.async_config_entry_first_refresh()
title = entry.options.get(CONF_PANEL_TITLE, DEFAULT_PANEL_TITLE)
icon = entry.options.get(CONF_PANEL_ICON, DEFAULT_PANEL_ICON)
path = entry.options.get(CONF_PANEL_PATH, DEFAULT_PANEL_PATH)
_register_panel(hass, path=path, title=title, icon=icon, url=base_url)
# Register platforms/entities
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
# Register services
async def _handle_smart_url(call):
await _async_set_default_sidebar_target(hass, entry)
await _async_update_panel(hass)
async def _handle_smart_url(call: ServiceCall):
url = (call.data.get("url") or "").strip()
if not url:
return
await api.smart_url(url)
await coordinator.async_request_refresh()
entry_id = _resolve_entry_id_for_call(hass, call)
store = _get_entry_data(hass, entry_id) if entry_id else None
if not store:
return
await store[DATA_API].smart_url(url)
await store[DATA_COORDINATOR].async_request_refresh()
async def _handle_play_now(call):
"""Play a URL immediately (maps to RelayTV POST /play).
Note: RelayTV /play clears the queue.
"""
async def _handle_play_now(call: ServiceCall):
url = (call.data.get("url") or "").strip()
if not url:
return
use_ytdlp = call.data.get("use_ytdlp")
cec = call.data.get("cec")
await api.play(url=url, use_ytdlp=use_ytdlp, cec=cec)
await coordinator.async_request_refresh()
entry_id = _resolve_entry_id_for_call(hass, call)
store = _get_entry_data(hass, entry_id) if entry_id else None
if not store:
return
await store[DATA_API].play(
url=url,
use_ytdlp=call.data.get("use_ytdlp"),
cec=call.data.get("cec"),
)
await store[DATA_COORDINATOR].async_request_refresh()
async def _handle_announce(call):
# RelayTV doesn't have a distinct "announce" mode; alias to play_now.
async def _handle_announce(call: ServiceCall):
await _handle_play_now(call)
# Register services once (first config entry wins).
if not hass.services.has_service(DOMAIN, SERVICE_SMART_URL):
hass.services.async_register(DOMAIN, SERVICE_SMART_URL, _handle_smart_url)
if not hass.services.has_service(DOMAIN, SERVICE_PLAY_NOW):
@@ -125,56 +234,36 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload RelayTV Web UI config entry."""
"""Unload RelayTV entry."""
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
path = entry.options.get(CONF_PANEL_PATH, DEFAULT_PANEL_PATH)
try:
frontend.async_remove_panel(hass, path)
except Exception: # pragma: no cover
_LOGGER.debug("Panel removal failed (it may not exist)", exc_info=True)
# Remove services only if this is the last entry.
hass.data.get(DOMAIN, {}).pop(entry.entry_id, None)
if not hass.data.get(DOMAIN):
try:
hass.services.async_remove(DOMAIN, SERVICE_SMART_URL)
except Exception:
pass
settings = await _async_ensure_settings(hass)
if settings.get(CONF_PANEL_TARGET_ENTRY_ID) == entry.entry_id:
replacement_id = _fallback_entry_id(hass)
settings[CONF_PANEL_TARGET_ENTRY_ID] = replacement_id
await _async_save_settings(hass)
_LOGGER.info(
"RelayTV sidebar target removed (%s); switched to %s",
entry.entry_id,
replacement_id,
)
if not any(cfg_entry.entry_id in hass.data.get(DOMAIN, {}) for cfg_entry in hass.config_entries.async_entries(DOMAIN)):
_async_unregister_panel(hass)
for service_name in (SERVICE_SMART_URL, SERVICE_PLAY_NOW, SERVICE_ANNOUNCE):
if hass.services.has_service(DOMAIN, service_name):
hass.services.async_remove(DOMAIN, service_name)
return unload_ok
await _async_update_panel(hass)
return unload_ok
async def _async_update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Handle options updates by re-registering the panel."""
"""Handle entry updates."""
base_url = _normalize_base_url(entry.data.get(CONF_BASE_URL, ""))
title = entry.options.get(CONF_PANEL_TITLE, DEFAULT_PANEL_TITLE)
icon = entry.options.get(CONF_PANEL_ICON, DEFAULT_PANEL_ICON)
path = entry.options.get(CONF_PANEL_PATH, DEFAULT_PANEL_PATH)
try:
frontend.async_remove_panel(hass, path)
except Exception:
pass
_register_panel(hass, path=path, title=title, icon=icon, url=base_url)
# Update API base_url if needed
store = hass.data.get(DOMAIN, {}).get(entry.entry_id)
store = _get_entry_data(hass, entry.entry_id)
if store and base_url:
store[DATA_API].base_url = base_url
def _register_panel(hass: HomeAssistant, *, path: str, title: str, icon: str, url: str) -> None:
"""Register the sidebar iframe panel."""
# We use the built-in iframe panel. Keyword args protect against HA signature drift.
frontend.async_register_built_in_panel(
hass,
component_name="iframe",
sidebar_title=title,
sidebar_icon=icon,
frontend_url_path=path,
config={"url": url},
require_admin=False,
)
_LOGGER.info("Registered RelayTV Web UI panel at /%s → %s", path, url)
await _async_update_panel(hass)