mirror of
https://github.com/home-assistant/core
synced 2024-08-02 23:40:32 +02:00
127 lines
3.9 KiB
Python
127 lines
3.9 KiB
Python
"""Implement the services discovery feature from Hass.io for Add-ons."""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from dataclasses import dataclass
|
|
import logging
|
|
from typing import Any
|
|
|
|
from aiohttp import web
|
|
from aiohttp.web_exceptions import HTTPServiceUnavailable
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.http import HomeAssistantView
|
|
from homeassistant.const import ATTR_NAME, ATTR_SERVICE, EVENT_HOMEASSISTANT_START
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.data_entry_flow import BaseServiceInfo
|
|
|
|
from .const import ATTR_ADDON, ATTR_CONFIG, ATTR_DISCOVERY, ATTR_UUID
|
|
from .handler import HassioAPIError
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass
|
|
class HassioServiceInfo(BaseServiceInfo):
|
|
"""Prepared info from hassio entries."""
|
|
|
|
config: dict[str, Any]
|
|
|
|
|
|
@callback
|
|
def async_setup_discovery_view(hass: HomeAssistant, hassio):
|
|
"""Discovery setup."""
|
|
hassio_discovery = HassIODiscovery(hass, hassio)
|
|
hass.http.register_view(hassio_discovery)
|
|
|
|
# Handle exists discovery messages
|
|
async def _async_discovery_start_handler(event):
|
|
"""Process all exists discovery on startup."""
|
|
try:
|
|
data = await hassio.retrieve_discovery_messages()
|
|
except HassioAPIError as err:
|
|
_LOGGER.error("Can't read discover info: %s", err)
|
|
return
|
|
|
|
jobs = [
|
|
hassio_discovery.async_process_new(discovery)
|
|
for discovery in data[ATTR_DISCOVERY]
|
|
]
|
|
if jobs:
|
|
await asyncio.wait(jobs)
|
|
|
|
hass.bus.async_listen_once(
|
|
EVENT_HOMEASSISTANT_START, _async_discovery_start_handler
|
|
)
|
|
|
|
|
|
class HassIODiscovery(HomeAssistantView):
|
|
"""Hass.io view to handle base part."""
|
|
|
|
name = "api:hassio_push:discovery"
|
|
url = "/api/hassio_push/discovery/{uuid}"
|
|
|
|
def __init__(self, hass: HomeAssistant, hassio):
|
|
"""Initialize WebView."""
|
|
self.hass = hass
|
|
self.hassio = hassio
|
|
|
|
async def post(self, request, uuid):
|
|
"""Handle new discovery requests."""
|
|
# Fetch discovery data and prevent injections
|
|
try:
|
|
data = await self.hassio.get_discovery_message(uuid)
|
|
except HassioAPIError as err:
|
|
_LOGGER.error("Can't read discovery data: %s", err)
|
|
raise HTTPServiceUnavailable() from None
|
|
|
|
await self.async_process_new(data)
|
|
return web.Response()
|
|
|
|
async def delete(self, request, uuid):
|
|
"""Handle remove discovery requests."""
|
|
data = await request.json()
|
|
|
|
await self.async_process_del(data)
|
|
return web.Response()
|
|
|
|
async def async_process_new(self, data):
|
|
"""Process add discovery entry."""
|
|
service = data[ATTR_SERVICE]
|
|
config_data = data[ATTR_CONFIG]
|
|
|
|
# Read additional Add-on info
|
|
try:
|
|
addon_info = await self.hassio.get_addon_info(data[ATTR_ADDON])
|
|
except HassioAPIError as err:
|
|
_LOGGER.error("Can't read add-on info: %s", err)
|
|
return
|
|
config_data[ATTR_ADDON] = addon_info[ATTR_NAME]
|
|
|
|
# Use config flow
|
|
await self.hass.config_entries.flow.async_init(
|
|
service,
|
|
context={"source": config_entries.SOURCE_HASSIO},
|
|
data=HassioServiceInfo(config=config_data),
|
|
)
|
|
|
|
async def async_process_del(self, data):
|
|
"""Process remove discovery entry."""
|
|
service = data[ATTR_SERVICE]
|
|
uuid = data[ATTR_UUID]
|
|
|
|
# Check if really deletet / prevent injections
|
|
try:
|
|
data = await self.hassio.get_discovery_message(uuid)
|
|
except HassioAPIError:
|
|
pass
|
|
else:
|
|
_LOGGER.warning("Retrieve wrong unload for %s", service)
|
|
return
|
|
|
|
# Use config flow
|
|
for entry in self.hass.config_entries.async_entries(service):
|
|
if entry.source != config_entries.SOURCE_HASSIO:
|
|
continue
|
|
await self.hass.config_entries.async_remove(entry)
|