forked from home-assistant/core
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add System Bridge Media Source (home-assistant#72865)
- Loading branch information
Showing
4 changed files
with
245 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,209 @@ | ||
| """System Bridge Media Source Implementation.""" | ||
| from __future__ import annotations | ||
|
|
||
| from systembridgeconnector.models.media_directories import MediaDirectories | ||
| from systembridgeconnector.models.media_files import File as MediaFile, MediaFiles | ||
|
|
||
| from homeassistant.components.media_player.const import MEDIA_CLASS_DIRECTORY | ||
| from homeassistant.components.media_source.const import ( | ||
| MEDIA_CLASS_MAP, | ||
| MEDIA_MIME_TYPES, | ||
| ) | ||
| from homeassistant.components.media_source.models import ( | ||
| BrowseMediaSource, | ||
| MediaSource, | ||
| MediaSourceItem, | ||
| PlayMedia, | ||
| ) | ||
| from homeassistant.config_entries import ConfigEntry | ||
| from homeassistant.const import CONF_API_KEY, CONF_HOST, CONF_PORT | ||
| from homeassistant.core import HomeAssistant | ||
|
|
||
| from .const import DOMAIN | ||
| from .coordinator import SystemBridgeDataUpdateCoordinator | ||
|
|
||
|
|
||
| async def async_get_media_source(hass: HomeAssistant) -> MediaSource: | ||
| """Set up SystemBridge media source.""" | ||
| return SystemBridgeSource(hass) | ||
|
|
||
|
|
||
| class SystemBridgeSource(MediaSource): | ||
| """Provide System Bridge media files as a media source.""" | ||
|
|
||
| def __init__( | ||
| self, | ||
| hass: HomeAssistant, | ||
| ) -> None: | ||
| """Initialize source.""" | ||
| super().__init__(DOMAIN) | ||
| self.name = "System Bridge" | ||
| self.hass: HomeAssistant = hass | ||
|
|
||
| async def async_resolve_media( | ||
| self, | ||
| item: MediaSourceItem, | ||
| ) -> PlayMedia: | ||
| """Resolve media to a url.""" | ||
| entry_id, path, mime_type = item.identifier.split("~~", 2) | ||
| entry = self.hass.config_entries.async_get_entry(entry_id) | ||
| if entry is None: | ||
| raise ValueError("Invalid entry") | ||
| path_split = path.split("/", 1) | ||
| return PlayMedia( | ||
| f"{_build_base_url(entry)}&base={path_split[0]}&path={path_split[1]}", | ||
| mime_type, | ||
| ) | ||
|
|
||
| async def async_browse_media( | ||
| self, | ||
| item: MediaSourceItem, | ||
| ) -> BrowseMediaSource: | ||
| """Return media.""" | ||
| if not item.identifier: | ||
| return self._build_bridges() | ||
|
|
||
| if "~~" not in item.identifier: | ||
| entry = self.hass.config_entries.async_get_entry(item.identifier) | ||
| if entry is None: | ||
| raise ValueError("Invalid entry") | ||
| coordinator: SystemBridgeDataUpdateCoordinator = self.hass.data[DOMAIN].get( | ||
| entry.entry_id | ||
| ) | ||
| directories = await coordinator.async_get_media_directories() | ||
| return _build_root_paths(entry, directories) | ||
|
|
||
| entry_id, path = item.identifier.split("~~", 1) | ||
| entry = self.hass.config_entries.async_get_entry(entry_id) | ||
| if entry is None: | ||
| raise ValueError("Invalid entry") | ||
|
|
||
| coordinator = self.hass.data[DOMAIN].get(entry.entry_id) | ||
|
|
||
| path_split = path.split("/", 1) | ||
|
|
||
| files = await coordinator.async_get_media_files( | ||
| path_split[0], path_split[1] if len(path_split) > 1 else None | ||
| ) | ||
|
|
||
| return _build_media_items(entry, files, path, item.identifier) | ||
|
|
||
| def _build_bridges(self) -> BrowseMediaSource: | ||
| """Build bridges for System Bridge media.""" | ||
| children = [] | ||
| for entry in self.hass.config_entries.async_entries(DOMAIN): | ||
| if entry.entry_id is not None: | ||
| children.append( | ||
| BrowseMediaSource( | ||
| domain=DOMAIN, | ||
| identifier=entry.entry_id, | ||
| media_class=MEDIA_CLASS_DIRECTORY, | ||
| media_content_type="", | ||
| title=entry.title, | ||
| can_play=False, | ||
| can_expand=True, | ||
| children=[], | ||
| children_media_class=MEDIA_CLASS_DIRECTORY, | ||
| ) | ||
| ) | ||
|
|
||
| return BrowseMediaSource( | ||
| domain=DOMAIN, | ||
| identifier="", | ||
| media_class=MEDIA_CLASS_DIRECTORY, | ||
| media_content_type="", | ||
| title=self.name, | ||
| can_play=False, | ||
| can_expand=True, | ||
| children=children, | ||
| children_media_class=MEDIA_CLASS_DIRECTORY, | ||
| ) | ||
|
|
||
|
|
||
| def _build_base_url( | ||
| entry: ConfigEntry, | ||
| ) -> str: | ||
| """Build base url for System Bridge media.""" | ||
| return f"http://{entry.data[CONF_HOST]}:{entry.data[CONF_PORT]}/api/media/file/data?apiKey={entry.data[CONF_API_KEY]}" | ||
|
|
||
|
|
||
| def _build_root_paths( | ||
| entry: ConfigEntry, | ||
| media_directories: MediaDirectories, | ||
| ) -> BrowseMediaSource: | ||
| """Build base categories for System Bridge media.""" | ||
| return BrowseMediaSource( | ||
| domain=DOMAIN, | ||
| identifier="", | ||
| media_class=MEDIA_CLASS_DIRECTORY, | ||
| media_content_type="", | ||
| title=entry.title, | ||
| can_play=False, | ||
| can_expand=True, | ||
| children=[ | ||
| BrowseMediaSource( | ||
| domain=DOMAIN, | ||
| identifier=f"{entry.entry_id}~~{directory.key}", | ||
| media_class=MEDIA_CLASS_DIRECTORY, | ||
| media_content_type="", | ||
| title=f"{directory.key[:1].capitalize()}{directory.key[1:]}", | ||
| can_play=False, | ||
| can_expand=True, | ||
| children=[], | ||
| children_media_class=MEDIA_CLASS_DIRECTORY, | ||
| ) | ||
| for directory in media_directories.directories | ||
| ], | ||
| children_media_class=MEDIA_CLASS_DIRECTORY, | ||
| ) | ||
|
|
||
|
|
||
| def _build_media_items( | ||
| entry: ConfigEntry, | ||
| media_files: MediaFiles, | ||
| path: str, | ||
| identifier: str, | ||
| ) -> BrowseMediaSource: | ||
| """Fetch requested files.""" | ||
| return BrowseMediaSource( | ||
| domain=DOMAIN, | ||
| identifier=identifier, | ||
| media_class=MEDIA_CLASS_DIRECTORY, | ||
| media_content_type="", | ||
| title=f"{entry.title} - {path}", | ||
| can_play=False, | ||
| can_expand=True, | ||
| children=[ | ||
| _build_media_item(identifier, file) | ||
| for file in media_files.files | ||
| if file.is_directory | ||
| or ( | ||
| file.is_file | ||
| and file.mime_type is not None | ||
| and file.mime_type.startswith(MEDIA_MIME_TYPES) | ||
| ) | ||
| ], | ||
| ) | ||
|
|
||
|
|
||
| def _build_media_item( | ||
| path: str, | ||
| media_file: MediaFile, | ||
| ) -> BrowseMediaSource: | ||
| """Build individual media item.""" | ||
| ext = ( | ||
| f"~~{media_file.mime_type}" | ||
| if media_file.is_file and media_file.mime_type is not None | ||
| else "" | ||
| ) | ||
| return BrowseMediaSource( | ||
| domain=DOMAIN, | ||
| identifier=f"{path}/{media_file.name}{ext}", | ||
| media_class=MEDIA_CLASS_DIRECTORY | ||
| if media_file.is_directory or media_file.mime_type is None | ||
| else MEDIA_CLASS_MAP[media_file.mime_type.split("/", 1)[0]], | ||
| media_content_type=media_file.mime_type, | ||
| title=media_file.name, | ||
| can_play=media_file.is_file, | ||
| can_expand=media_file.is_directory, | ||
| ) |