mirror of
https://github.com/home-assistant/core
synced 2024-08-31 05:57:13 +02:00
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
"""Home Assistant Cast integration for Cast."""
|
|
from __future__ import annotations
|
|
|
|
from pychromecast.controllers.homeassistant import HomeAssistantController
|
|
import voluptuous as vol
|
|
|
|
from homeassistant import auth, config_entries, core
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.helpers import config_validation as cv, dispatcher
|
|
from homeassistant.helpers.network import get_url
|
|
|
|
from .const import DOMAIN, SIGNAL_HASS_CAST_SHOW_VIEW
|
|
|
|
SERVICE_SHOW_VIEW = "show_lovelace_view"
|
|
ATTR_VIEW_PATH = "view_path"
|
|
ATTR_URL_PATH = "dashboard_path"
|
|
|
|
|
|
async def async_setup_ha_cast(
|
|
hass: core.HomeAssistant, entry: config_entries.ConfigEntry
|
|
):
|
|
"""Set up Home Assistant Cast."""
|
|
user_id: str | None = entry.data.get("user_id")
|
|
user: auth.models.User | None = None
|
|
|
|
if user_id is not None:
|
|
user = await hass.auth.async_get_user(user_id)
|
|
|
|
if user is None:
|
|
user = await hass.auth.async_create_system_user(
|
|
"Home Assistant Cast", [auth.GROUP_ID_ADMIN]
|
|
)
|
|
hass.config_entries.async_update_entry(
|
|
entry, data={**entry.data, "user_id": user.id}
|
|
)
|
|
|
|
if user.refresh_tokens:
|
|
refresh_token: auth.models.RefreshToken = list(user.refresh_tokens.values())[0]
|
|
else:
|
|
refresh_token = await hass.auth.async_create_refresh_token(user)
|
|
|
|
async def handle_show_view(call: core.ServiceCall):
|
|
"""Handle a Show View service call."""
|
|
hass_url = get_url(hass, require_ssl=True, prefer_external=True)
|
|
|
|
controller = HomeAssistantController(
|
|
# If you are developing Home Assistant Cast, uncomment and set to your dev app id.
|
|
# app_id="5FE44367",
|
|
hass_url=hass_url,
|
|
client_id=None,
|
|
refresh_token=refresh_token.token,
|
|
)
|
|
|
|
dispatcher.async_dispatcher_send(
|
|
hass,
|
|
SIGNAL_HASS_CAST_SHOW_VIEW,
|
|
controller,
|
|
call.data[ATTR_ENTITY_ID],
|
|
call.data[ATTR_VIEW_PATH],
|
|
call.data.get(ATTR_URL_PATH),
|
|
)
|
|
|
|
hass.helpers.service.async_register_admin_service(
|
|
DOMAIN,
|
|
SERVICE_SHOW_VIEW,
|
|
handle_show_view,
|
|
vol.Schema(
|
|
{
|
|
ATTR_ENTITY_ID: cv.entity_id,
|
|
ATTR_VIEW_PATH: str,
|
|
vol.Optional(ATTR_URL_PATH): str,
|
|
}
|
|
),
|
|
)
|
|
|
|
|
|
async def async_remove_user(
|
|
hass: core.HomeAssistant, entry: config_entries.ConfigEntry
|
|
):
|
|
"""Remove Home Assistant Cast user."""
|
|
user_id: str | None = entry.data.get("user_id")
|
|
|
|
if user_id is not None:
|
|
user = await hass.auth.async_get_user(user_id)
|
|
await hass.auth.async_remove_user(user)
|