first commit

hass/settings/custom_components/yandex_station/intent.py

@@ -0,0 +1,67 @@
+import asyncio
+import logging
+from typing import Optional
+
+from homeassistant.core import HomeAssistant, Event
+from homeassistant.helpers.intent import IntentHandler, Intent, IntentResponse
+
+from . import DOMAIN, DATA_SPEAKERS
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_intents(hass: HomeAssistant) -> None:
+    if DOMAIN not in hass.data:
+        return
+
+    handlers = []
+
+    for device in hass.data[DOMAIN][DATA_SPEAKERS].values():
+        platform = device['quasar_info']['platform']
+        if 'host' in device or platform.startswith('yandex'):
+            handler = YandexIntentHandler(device['entity'].entity_id)
+            hass.helpers.intent.async_register(handler)
+            handlers.append(handler)
+
+    if not handlers:
+        return
+
+    async def listener(event: Event):
+        request_id = event.data['request_id']
+        for handler in handlers:
+            if handler.request_id == request_id:
+                handler.response_text = event.data['text']
+                handler.response_waiter.set()
+                return
+
+    hass.bus.async_listen('yandex_station_response', listener)
+
+
+class YandexIntentHandler(IntentHandler):
+    request_id: str = None
+    response_text: Optional[str] = None
+    response_waiter = None
+
+    def __init__(self, entity_id: str):
+        self.intent_type = entity_id
+        self.response_waiter = asyncio.Event()
+
+        _LOGGER.debug(f"Init intent: {self.intent_type}")
+
+    async def async_handle(self, intent: Intent) -> IntentResponse:
+        self.request_id = intent.context.id
+        self.response_text = None
+        self.response_waiter.clear()
+
+        await intent.hass.services.async_call('media_player', 'play_media', {
+            'entity_id': self.intent_type,
+            'media_content_id': intent.text_input,
+            'media_content_type': 'question:' + self.request_id
+        })
+
+        await asyncio.wait_for(self.response_waiter.wait(), 2.0)
+
+        response = intent.create_response()
+        if self.response_text:
+            response.async_set_speech(self.response_text)
+        return response