312 lines
9.9 KiB
Python
312 lines
9.9 KiB
Python
"""Support for Almond."""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from datetime import timedelta
|
|
import logging
|
|
import time
|
|
|
|
from aiohttp import ClientError, ClientSession
|
|
import async_timeout
|
|
from pyalmond import AbstractAlmondWebAuth, AlmondLocalAuth, WebAlmondAPI
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.auth.const import GROUP_ID_ADMIN
|
|
from homeassistant.components import conversation
|
|
from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
|
|
from homeassistant.const import (
|
|
CONF_CLIENT_ID,
|
|
CONF_CLIENT_SECRET,
|
|
CONF_HOST,
|
|
CONF_TYPE,
|
|
EVENT_HOMEASSISTANT_START,
|
|
)
|
|
from homeassistant.core import Context, CoreState, HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
from homeassistant.helpers import (
|
|
aiohttp_client,
|
|
config_entry_oauth2_flow,
|
|
config_validation as cv,
|
|
event,
|
|
intent,
|
|
network,
|
|
storage,
|
|
)
|
|
|
|
from . import config_flow
|
|
from .const import DOMAIN, TYPE_LOCAL, TYPE_OAUTH2
|
|
|
|
STORAGE_VERSION = 1
|
|
STORAGE_KEY = DOMAIN
|
|
|
|
ALMOND_SETUP_DELAY = 30
|
|
|
|
DEFAULT_OAUTH2_HOST = "https://almond.stanford.edu"
|
|
DEFAULT_LOCAL_HOST = "http://localhost:3000"
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: vol.Any(
|
|
vol.Schema(
|
|
{
|
|
vol.Required(CONF_TYPE): TYPE_OAUTH2,
|
|
vol.Required(CONF_CLIENT_ID): cv.string,
|
|
vol.Required(CONF_CLIENT_SECRET): cv.string,
|
|
vol.Optional(CONF_HOST, default=DEFAULT_OAUTH2_HOST): cv.url,
|
|
}
|
|
),
|
|
vol.Schema(
|
|
{vol.Required(CONF_TYPE): TYPE_LOCAL, vol.Required(CONF_HOST): cv.url}
|
|
),
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup(hass, config):
|
|
"""Set up the Almond component."""
|
|
hass.data[DOMAIN] = {}
|
|
|
|
if DOMAIN not in config:
|
|
return True
|
|
|
|
conf = config[DOMAIN]
|
|
|
|
host = conf[CONF_HOST]
|
|
|
|
if conf[CONF_TYPE] == TYPE_OAUTH2:
|
|
config_flow.AlmondFlowHandler.async_register_implementation(
|
|
hass,
|
|
config_entry_oauth2_flow.LocalOAuth2Implementation(
|
|
hass,
|
|
DOMAIN,
|
|
conf[CONF_CLIENT_ID],
|
|
conf[CONF_CLIENT_SECRET],
|
|
f"{host}/me/api/oauth2/authorize",
|
|
f"{host}/me/api/oauth2/token",
|
|
),
|
|
)
|
|
return True
|
|
|
|
if not hass.config_entries.async_entries(DOMAIN):
|
|
hass.async_create_task(
|
|
hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_IMPORT},
|
|
data={"type": TYPE_LOCAL, "host": conf[CONF_HOST]},
|
|
)
|
|
)
|
|
return True
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up Almond config entry."""
|
|
websession = aiohttp_client.async_get_clientsession(hass)
|
|
|
|
if entry.data["type"] == TYPE_LOCAL:
|
|
auth = AlmondLocalAuth(entry.data["host"], websession)
|
|
else:
|
|
# OAuth2
|
|
implementation = (
|
|
await config_entry_oauth2_flow.async_get_config_entry_implementation(
|
|
hass, entry
|
|
)
|
|
)
|
|
oauth_session = config_entry_oauth2_flow.OAuth2Session(
|
|
hass, entry, implementation
|
|
)
|
|
auth = AlmondOAuth(entry.data["host"], websession, oauth_session)
|
|
|
|
api = WebAlmondAPI(auth)
|
|
agent = AlmondAgent(hass, api, entry)
|
|
|
|
# Hass.io does its own configuration.
|
|
if not entry.data.get("is_hassio"):
|
|
# If we're not starting or local, set up Almond right away
|
|
if hass.state != CoreState.not_running or entry.data["type"] == TYPE_LOCAL:
|
|
await _configure_almond_for_ha(hass, entry, api)
|
|
|
|
else:
|
|
# OAuth2 implementations can potentially rely on the HA Cloud url.
|
|
# This url is not be available until 30 seconds after boot.
|
|
|
|
async def configure_almond(_now):
|
|
try:
|
|
await _configure_almond_for_ha(hass, entry, api)
|
|
except ConfigEntryNotReady:
|
|
_LOGGER.warning(
|
|
"Unable to configure Almond to connect to Home Assistant"
|
|
)
|
|
|
|
async def almond_hass_start(_event):
|
|
event.async_call_later(hass, ALMOND_SETUP_DELAY, configure_almond)
|
|
|
|
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_START, almond_hass_start)
|
|
|
|
conversation.async_set_agent(hass, agent)
|
|
return True
|
|
|
|
|
|
async def _configure_almond_for_ha(
|
|
hass: HomeAssistant, entry: ConfigEntry, api: WebAlmondAPI
|
|
):
|
|
"""Configure Almond to connect to HA."""
|
|
try:
|
|
if entry.data["type"] == TYPE_OAUTH2:
|
|
# If we're connecting over OAuth2, we will only set up connection
|
|
# with Home Assistant if we're remotely accessible.
|
|
hass_url = network.get_url(hass, allow_internal=False, prefer_cloud=True)
|
|
else:
|
|
hass_url = network.get_url(hass)
|
|
except network.NoURLAvailableError:
|
|
# If no URL is available, we're not going to configure Almond to connect to HA.
|
|
return
|
|
|
|
_LOGGER.debug("Configuring Almond to connect to Home Assistant at %s", hass_url)
|
|
store = storage.Store(hass, STORAGE_VERSION, STORAGE_KEY)
|
|
data = await store.async_load()
|
|
|
|
if data is None:
|
|
data = {}
|
|
|
|
user = None
|
|
if "almond_user" in data:
|
|
user = await hass.auth.async_get_user(data["almond_user"])
|
|
|
|
if user is None:
|
|
user = await hass.auth.async_create_system_user("Almond", [GROUP_ID_ADMIN])
|
|
data["almond_user"] = user.id
|
|
await store.async_save(data)
|
|
|
|
refresh_token = await hass.auth.async_create_refresh_token(
|
|
user,
|
|
# Almond will be fine as long as we restart once every 5 years
|
|
access_token_expiration=timedelta(days=365 * 5),
|
|
)
|
|
|
|
# Create long lived access token
|
|
access_token = hass.auth.async_create_access_token(refresh_token)
|
|
|
|
# Store token in Almond
|
|
try:
|
|
with async_timeout.timeout(30):
|
|
await api.async_create_device(
|
|
{
|
|
"kind": "io.home-assistant",
|
|
"hassUrl": hass_url,
|
|
"accessToken": access_token,
|
|
"refreshToken": "",
|
|
# 5 years from now in ms.
|
|
"accessTokenExpires": (time.time() + 60 * 60 * 24 * 365 * 5) * 1000,
|
|
}
|
|
)
|
|
except (asyncio.TimeoutError, ClientError) as err:
|
|
if isinstance(err, asyncio.TimeoutError):
|
|
msg = "Request timeout"
|
|
else:
|
|
msg = err
|
|
_LOGGER.warning("Unable to configure Almond: %s", msg)
|
|
await hass.auth.async_remove_refresh_token(refresh_token)
|
|
raise ConfigEntryNotReady from err
|
|
|
|
# Clear all other refresh tokens
|
|
for token in list(user.refresh_tokens.values()):
|
|
if token.id != refresh_token.id:
|
|
await hass.auth.async_remove_refresh_token(token)
|
|
|
|
|
|
async def async_unload_entry(hass, entry):
|
|
"""Unload Almond."""
|
|
conversation.async_set_agent(hass, None)
|
|
return True
|
|
|
|
|
|
class AlmondOAuth(AbstractAlmondWebAuth):
|
|
"""Almond Authentication using OAuth2."""
|
|
|
|
def __init__(
|
|
self,
|
|
host: str,
|
|
websession: ClientSession,
|
|
oauth_session: config_entry_oauth2_flow.OAuth2Session,
|
|
) -> None:
|
|
"""Initialize Almond auth."""
|
|
super().__init__(host, websession)
|
|
self._oauth_session = oauth_session
|
|
|
|
async def async_get_access_token(self):
|
|
"""Return a valid access token."""
|
|
if not self._oauth_session.valid_token:
|
|
await self._oauth_session.async_ensure_token_valid()
|
|
|
|
return self._oauth_session.token["access_token"]
|
|
|
|
|
|
class AlmondAgent(conversation.AbstractConversationAgent):
|
|
"""Almond conversation agent."""
|
|
|
|
def __init__(
|
|
self, hass: HomeAssistant, api: WebAlmondAPI, entry: ConfigEntry
|
|
) -> None:
|
|
"""Initialize the agent."""
|
|
self.hass = hass
|
|
self.api = api
|
|
self.entry = entry
|
|
|
|
@property
|
|
def attribution(self):
|
|
"""Return the attribution."""
|
|
return {"name": "Powered by Almond", "url": "https://almond.stanford.edu/"}
|
|
|
|
async def async_get_onboarding(self):
|
|
"""Get onboard url if not onboarded."""
|
|
if self.entry.data.get("onboarded"):
|
|
return None
|
|
|
|
host = self.entry.data["host"]
|
|
if self.entry.data.get("is_hassio"):
|
|
host = "/core_almond"
|
|
return {
|
|
"text": "Would you like to opt-in to share your anonymized commands with Stanford to improve Almond's responses?",
|
|
"url": f"{host}/conversation",
|
|
}
|
|
|
|
async def async_set_onboarding(self, shown):
|
|
"""Set onboarding status."""
|
|
self.hass.config_entries.async_update_entry(
|
|
self.entry, data={**self.entry.data, "onboarded": shown}
|
|
)
|
|
|
|
return True
|
|
|
|
async def async_process(
|
|
self, text: str, context: Context, conversation_id: str | None = None
|
|
) -> intent.IntentResponse:
|
|
"""Process a sentence."""
|
|
response = await self.api.async_converse_text(text, conversation_id)
|
|
|
|
first_choice = True
|
|
buffer = ""
|
|
for message in response["messages"]:
|
|
if message["type"] == "text":
|
|
buffer += f"\n{message['text']}"
|
|
elif message["type"] == "picture":
|
|
buffer += f"\n Picture: {message['url']}"
|
|
elif message["type"] == "rdl":
|
|
buffer += (
|
|
f"\n Link: {message['rdl']['displayTitle']} "
|
|
f"{message['rdl']['webCallback']}"
|
|
)
|
|
elif message["type"] == "choice":
|
|
if first_choice:
|
|
first_choice = False
|
|
else:
|
|
buffer += ","
|
|
buffer += f" {message['title']}"
|
|
|
|
intent_result = intent.IntentResponse()
|
|
intent_result.async_set_speech(buffer.strip())
|
|
return intent_result
|