-
-
Notifications
You must be signed in to change notification settings - Fork 52
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
af1b06c
commit ba0a34d
Showing
9 changed files
with
308 additions
and
10 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,205 @@ | ||
| """Intents for the client integration.""" | ||
| from __future__ import annotations | ||
|
|
||
| import json | ||
| from typing import TYPE_CHECKING, Any | ||
|
|
||
| import voluptuous as vol | ||
| from homeassistant.components.conversation import ATTR_AGENT_ID, ATTR_TEXT | ||
| from homeassistant.components.conversation import SERVICE_PROCESS as CONVERSATION_SERVICE | ||
| from homeassistant.components.conversation.const import DOMAIN as CONVERSATION_DOMAIN | ||
| from homeassistant.components.media_player.const import DOMAIN as MEDIA_PLAYER_DOMAIN | ||
| from homeassistant.config_entries import ConfigEntry, ConfigEntryState | ||
| from homeassistant.core import HomeAssistant | ||
| from homeassistant.helpers import area_registry as ar | ||
| from homeassistant.helpers import config_validation as cv | ||
| from homeassistant.helpers import device_registry as dr | ||
| from homeassistant.helpers import entity_registry as er | ||
| from homeassistant.helpers import intent | ||
|
|
||
| from . import DOMAIN | ||
| from .const import CONF_OPENAI_AGENT_ID | ||
| from .media_player import ATTR_MEDIA_ID, ATTR_MEDIA_TYPE, ATTR_RADIO_MODE, MassPlayer | ||
|
|
||
| if TYPE_CHECKING: | ||
| pass | ||
| if TYPE_CHECKING: | ||
| from music_assistant.client import MusicAssistantClient | ||
|
|
||
|
|
||
| INTENT_PLAY_MEDIA_ON_MEDIA_PLAYER = "MassPlayMediaOnMediaPlayer" | ||
| NAME_SLOT = "name" | ||
| AREA_SLOT = "area" | ||
| QUERY_SLOT = "query" | ||
| SLOT_VALUE = "value" | ||
|
|
||
|
|
||
| async def async_setup_intents(hass: HomeAssistant) -> None: | ||
| """Set up the climate intents.""" | ||
| intent.async_register(hass, MassPlayMediaOnMediaPlayerHandler()) | ||
|
|
||
|
|
||
| class MassPlayMediaOnMediaPlayerHandler(intent.IntentHandler): | ||
| """Handle PlayMediaOnMediaPlayer intents.""" | ||
|
|
||
| intent_type = INTENT_PLAY_MEDIA_ON_MEDIA_PLAYER | ||
| slot_schema = {vol.Any(NAME_SLOT, AREA_SLOT): cv.string, vol.Optional(QUERY_SLOT): cv.string} | ||
|
|
||
| async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse: | ||
| """Handle the intent.""" | ||
| slots = self.async_validate_slots(intent_obj.slots) | ||
| config_entry = await self._get_loaded_config_entry(intent_obj.hass) | ||
|
|
||
| query: str = slots.get(QUERY_SLOT, {}).get(SLOT_VALUE) | ||
| if query is not None: | ||
| service_data: dict[str, Any] = {} | ||
| service_data[ATTR_AGENT_ID] = config_entry.data.get(CONF_OPENAI_AGENT_ID) | ||
| service_data[ATTR_TEXT] = query | ||
|
|
||
| # Look up area to fail early | ||
| area_name = slots.get(AREA_SLOT, {}).get(SLOT_VALUE) | ||
| if area_name is not None: | ||
| areas = ar.async_get(intent_obj.hass) | ||
| area_name = area_name.casefold() | ||
| area = await self._find_area(area_name, areas) | ||
| if area is None: | ||
| raise intent.IntentHandleError(f"No area named {area_name}") | ||
| media_player_entity = await self._get_entity_by_area( | ||
| area, intent_obj.hass, config_entry | ||
| ) | ||
| if media_player_entity is None: | ||
| raise intent.IntentHandleError(f"No media player found matching area: {area_name}") | ||
|
|
||
| # Look up name to fail early | ||
| name: str = slots.get(NAME_SLOT, {}).get(SLOT_VALUE) | ||
| if name is not None: | ||
| name = name.casefold() | ||
| media_player_entity = await self._get_entity_from_registry( | ||
| name, intent_obj.hass, config_entry | ||
| ) | ||
| if media_player_entity is None: | ||
| raise intent.IntentHandleError(f"No media player found matching name: {name}") | ||
|
|
||
| actual_player = await self._get_mass_player_from_registry_entry( | ||
| intent_obj.hass, config_entry, media_player_entity | ||
| ) | ||
| if actual_player is None: | ||
| raise intent.IntentHandleError(f"No Mass media player found for name {name}") | ||
|
|
||
| media_id, media_type = await self._get_media_id_and_media_type_from_query_result( | ||
| intent_obj.hass, service_data, intent_obj | ||
| ) | ||
| await actual_player.async_play_media( | ||
| media_id=media_id, media_type=media_type, extra={ATTR_RADIO_MODE: False} | ||
| ) | ||
| response = intent_obj.create_response() | ||
| response.response_type = intent.IntentResponseType.ACTION_DONE | ||
| if area_name is not None: | ||
| response.async_set_speech(f"Playing selection in {area_name}") | ||
| if name is not None: | ||
| response.async_set_speech(f"Playing selection on {name}") | ||
| return response | ||
|
|
||
| async def _get_media_id_and_media_type_from_query_result( | ||
| self, hass: HomeAssistant, service_data: dict[str, Any], intent_obj: intent.Intent | ||
| ) -> str: | ||
| """Get from the query.""" | ||
| ai_response = await hass.services.async_call( | ||
| CONVERSATION_DOMAIN, | ||
| CONVERSATION_SERVICE, | ||
| {**service_data}, | ||
| blocking=True, | ||
| context=intent_obj.context, | ||
| return_response=True, | ||
| ) | ||
| json_payload = json.loads(ai_response["response"]["speech"]["plain"]["speech"]) | ||
| media_id = json_payload.get(ATTR_MEDIA_ID) | ||
| media_type = json_payload.get(ATTR_MEDIA_TYPE) | ||
| return media_id, media_type | ||
|
|
||
| async def _get_loaded_config_entry(self, hass: HomeAssistant) -> str: | ||
| """Get the correct config entry.""" | ||
| config_entries = hass.config_entries.async_entries(DOMAIN) | ||
| for config_entry in config_entries: | ||
| if config_entry.state == ConfigEntryState.LOADED: | ||
| return config_entry | ||
| return None | ||
|
|
||
| async def _get_entity_from_registry( | ||
| self, name: str, hass: HomeAssistant, config_entry: ConfigEntry | ||
| ) -> er.RegistryEntry: | ||
| """Get the entity from the registry.""" | ||
| entity_registry = er.async_get(hass) | ||
| entity_registry_entries = er.async_entries_for_config_entry( | ||
| entity_registry, config_entry.entry_id | ||
| ) | ||
| for entity_registry_entry in entity_registry_entries: | ||
| if await self._has_name(entity_registry_entry, name): | ||
| return entity_registry_entry | ||
| return None | ||
|
|
||
| async def _has_name(self, entity: er.RegistryEntry | None, name: str) -> bool: | ||
| """Return true if entity name or alias matches.""" | ||
| if entity is not None: | ||
| normalised_entity_id = ( | ||
| entity.entity_id.replace("_", " ").strip("media player.").casefold() | ||
| ) | ||
|
|
||
| if name in normalised_entity_id: | ||
| return True | ||
|
|
||
| # Check name/aliases | ||
| if (entity is None) or (not entity.aliases): | ||
| return False | ||
|
|
||
| return any(name == alias.casefold() for alias in entity.aliases) | ||
|
|
||
| async def _get_mass_player_from_registry_entry( | ||
| self, hass: HomeAssistant, config_entry: ConfigEntry, media_player_entity: er.RegistryEntry | ||
| ) -> MassPlayer: | ||
| """Return the mass player.""" | ||
| mass: MusicAssistantClient = hass.data[DOMAIN][config_entry.entry_id].mass | ||
| player_entity = hass.data[MEDIA_PLAYER_DOMAIN].get_entity(media_player_entity.entity_id) | ||
| mass_player = mass.players.get_player( | ||
| player_entity.extra_state_attributes.get("mass_player_id") | ||
| ) | ||
| actual_player = MassPlayer(mass, mass_player.player_id) | ||
| return actual_player | ||
|
|
||
| async def _find_area(self, area_name: str, areas: ar.AreaRegistry) -> ar.AreaEntry | None: | ||
| """Find an area by id or name, checking aliases too.""" | ||
| area = areas.async_get_area(area_name) or areas.async_get_area_by_name(area_name) | ||
| if area is not None: | ||
| return area | ||
|
|
||
| # Check area aliases | ||
| for maybe_area in areas.areas.values(): | ||
| if not maybe_area.aliases: | ||
| continue | ||
|
|
||
| for area_alias in maybe_area.aliases: | ||
| if area_name == area_alias.casefold(): | ||
| return maybe_area | ||
|
|
||
| return None | ||
|
|
||
| async def _get_entity_by_area( | ||
| self, area: ar.AreaEntry, hass: HomeAssistant, config_entry: ConfigEntry | ||
| ) -> er.RegistryEntry: | ||
| """Filter state/entity pairs by an area.""" | ||
| entity_registry = er.async_get(hass) | ||
| device_registry = dr.async_get(hass) | ||
| entity_registry_entries = er.async_entries_for_config_entry( | ||
| entity_registry, config_entry.entry_id | ||
| ) | ||
| for entity_registry_entry in entity_registry_entries: | ||
| if entity_registry_entry.area_id == area.id: | ||
| # Use entity's area id first | ||
| return entity_registry_entry | ||
| if entity_registry_entry.device_id is not None: | ||
| # Fall back to device area if not set on entity | ||
| device = device_registry.async_get(entity_registry_entry.device_id) | ||
| if device is not None and device.area_id == area.id: | ||
| return entity_registry_entry | ||
|
|
||
| return None |
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
Oops, something went wrong.