mirror of https://github.com/home-assistant/core
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
"""The Jellyfin integration."""
|
|
|
|
from typing import Any
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
|
|
from homeassistant.helpers import device_registry as dr
|
|
|
|
from .client_wrapper import CannotConnect, InvalidAuth, create_client, validate_input
|
|
from .const import CONF_CLIENT_DEVICE_ID, DOMAIN, PLATFORMS
|
|
from .coordinator import JellyfinDataUpdateCoordinator
|
|
|
|
type JellyfinConfigEntry = ConfigEntry[JellyfinDataUpdateCoordinator]
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: JellyfinConfigEntry) -> bool:
|
|
"""Set up Jellyfin from a config entry."""
|
|
if CONF_CLIENT_DEVICE_ID not in entry.data:
|
|
entry_data = entry.data.copy()
|
|
entry_data[CONF_CLIENT_DEVICE_ID] = entry.entry_id
|
|
hass.config_entries.async_update_entry(entry, data=entry_data)
|
|
|
|
device_id = entry.data[CONF_CLIENT_DEVICE_ID]
|
|
device_name = ascii(hass.config.location_name)
|
|
|
|
client = create_client(device_id=device_id, device_name=device_name)
|
|
|
|
try:
|
|
user_id, connect_result = await validate_input(hass, dict(entry.data), client)
|
|
except CannotConnect as ex:
|
|
raise ConfigEntryNotReady("Cannot connect to Jellyfin server") from ex
|
|
except InvalidAuth as ex:
|
|
raise ConfigEntryAuthFailed(ex) from ex
|
|
|
|
server_info: dict[str, Any] = connect_result["Servers"][0]
|
|
|
|
coordinator = JellyfinDataUpdateCoordinator(hass, client, server_info, user_id)
|
|
|
|
await coordinator.async_config_entry_first_refresh()
|
|
|
|
entry.runtime_data = coordinator
|
|
entry.async_on_unload(client.stop)
|
|
|
|
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: JellyfinConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
|
|
|
|
|
|
async def async_remove_config_entry_device(
|
|
hass: HomeAssistant, config_entry: JellyfinConfigEntry, device_entry: dr.DeviceEntry
|
|
) -> bool:
|
|
"""Remove device from a config entry."""
|
|
coordinator = config_entry.runtime_data
|
|
|
|
return not device_entry.identifiers.intersection(
|
|
(
|
|
(DOMAIN, coordinator.server_id),
|
|
*((DOMAIN, device_id) for device_id in coordinator.device_ids),
|
|
)
|
|
)
|