mirror of https://github.com/home-assistant/core
146 lines
4.5 KiB
Python
146 lines
4.5 KiB
Python
"""Account linking via the cloud."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
import logging
|
|
from typing import Any
|
|
|
|
import aiohttp
|
|
from awesomeversion import AwesomeVersion
|
|
from hass_nabucasa import account_link
|
|
|
|
from homeassistant.const import __version__ as HA_VERSION
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers import config_entry_oauth2_flow, event
|
|
|
|
from .const import DATA_CLOUD, DOMAIN
|
|
|
|
DATA_SERVICES = "cloud_account_link_services"
|
|
CACHE_TIMEOUT = 3600
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CURRENT_VERSION = AwesomeVersion(HA_VERSION)
|
|
CURRENT_PLAIN_VERSION = AwesomeVersion(
|
|
CURRENT_VERSION.string.removesuffix(f"{CURRENT_VERSION.modifier}")
|
|
)
|
|
|
|
|
|
@callback
|
|
def async_setup(hass: HomeAssistant) -> None:
|
|
"""Set up cloud account link."""
|
|
config_entry_oauth2_flow.async_add_implementation_provider(
|
|
hass, DOMAIN, async_provide_implementation
|
|
)
|
|
|
|
|
|
async def async_provide_implementation(
|
|
hass: HomeAssistant, domain: str
|
|
) -> list[config_entry_oauth2_flow.AbstractOAuth2Implementation]:
|
|
"""Provide an implementation for a domain."""
|
|
services = await _get_services(hass)
|
|
|
|
for service in services:
|
|
if (
|
|
service["service"] == domain
|
|
and service["min_version"] <= CURRENT_PLAIN_VERSION
|
|
and (
|
|
service.get("accepts_new_authorizations", True)
|
|
or (
|
|
(entries := hass.config_entries.async_entries(domain))
|
|
and any(
|
|
entry.data.get("auth_implementation") == DOMAIN
|
|
for entry in entries
|
|
)
|
|
)
|
|
)
|
|
):
|
|
return [CloudOAuth2Implementation(hass, domain)]
|
|
|
|
return []
|
|
|
|
|
|
async def _get_services(hass: HomeAssistant) -> list[dict[str, Any]]:
|
|
"""Get the available services."""
|
|
services: list[dict[str, Any]]
|
|
if DATA_SERVICES in hass.data:
|
|
services = hass.data[DATA_SERVICES]
|
|
return services # noqa: RET504
|
|
|
|
try:
|
|
services = await account_link.async_fetch_available_services(
|
|
hass.data[DATA_CLOUD]
|
|
)
|
|
except (aiohttp.ClientError, TimeoutError):
|
|
return []
|
|
|
|
hass.data[DATA_SERVICES] = services
|
|
|
|
@callback
|
|
def clear_services(_now: datetime) -> None:
|
|
"""Clear services cache."""
|
|
hass.data.pop(DATA_SERVICES, None)
|
|
|
|
event.async_call_later(hass, CACHE_TIMEOUT, clear_services)
|
|
|
|
return services
|
|
|
|
|
|
class CloudOAuth2Implementation(config_entry_oauth2_flow.AbstractOAuth2Implementation):
|
|
"""Cloud implementation of the OAuth2 flow."""
|
|
|
|
def __init__(self, hass: HomeAssistant, service: str) -> None:
|
|
"""Initialize cloud OAuth2 implementation."""
|
|
self.hass = hass
|
|
self.service = service
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
"""Name of the implementation."""
|
|
return "Home Assistant Cloud"
|
|
|
|
@property
|
|
def domain(self) -> str:
|
|
"""Domain that is providing the implementation."""
|
|
return DOMAIN
|
|
|
|
async def async_generate_authorize_url(self, flow_id: str) -> str:
|
|
"""Generate a url for the user to authorize."""
|
|
helper = account_link.AuthorizeAccountHelper(
|
|
self.hass.data[DATA_CLOUD], self.service
|
|
)
|
|
authorize_url = await helper.async_get_authorize_url()
|
|
|
|
async def await_tokens() -> None:
|
|
"""Wait for tokens and pass them on when received."""
|
|
try:
|
|
tokens = await helper.async_get_tokens()
|
|
|
|
except TimeoutError:
|
|
_LOGGER.info("Timeout fetching tokens for flow %s", flow_id)
|
|
except account_link.AccountLinkException as err:
|
|
_LOGGER.info(
|
|
"Failed to fetch tokens for flow %s: %s", flow_id, err.code
|
|
)
|
|
else:
|
|
await self.hass.config_entries.flow.async_configure(
|
|
flow_id=flow_id, user_input=tokens
|
|
)
|
|
|
|
self.hass.async_create_task(await_tokens())
|
|
|
|
return authorize_url
|
|
|
|
async def async_resolve_external_data(self, external_data: Any) -> dict:
|
|
"""Resolve external data to tokens."""
|
|
# We already passed in tokens
|
|
dict_data: dict = external_data
|
|
return dict_data
|
|
|
|
async def _async_refresh_token(self, token: dict) -> dict:
|
|
"""Refresh a token."""
|
|
new_token = await account_link.async_fetch_access_token(
|
|
self.hass.data[DATA_CLOUD], self.service, token["refresh_token"]
|
|
)
|
|
return {**token, **new_token}
|