mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-06-23 10:26:30 +00:00
bf75a8a439
* Cleanup discovery data * Update API.md * Update validate.py * Update discovery.py * Update const.py
92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
"""Init file for Hass.io network RESTful API."""
|
|
import voluptuous as vol
|
|
|
|
from .utils import api_process, api_validate
|
|
from ..const import (
|
|
ATTR_ADDON, ATTR_UUID, ATTR_CONFIG, ATTR_DISCOVERY, ATTR_SERVICE,
|
|
REQUEST_FROM)
|
|
from ..coresys import CoreSysAttributes
|
|
from ..exceptions import APIError, APIForbidden
|
|
from ..validate import SERVICE_ALL
|
|
|
|
|
|
SCHEMA_DISCOVERY = vol.Schema({
|
|
vol.Required(ATTR_SERVICE): SERVICE_ALL,
|
|
vol.Optional(ATTR_CONFIG): vol.Maybe(dict),
|
|
})
|
|
|
|
|
|
class APIDiscovery(CoreSysAttributes):
|
|
"""Handle RESTful API for discovery functions."""
|
|
|
|
def _extract_message(self, request):
|
|
"""Extract discovery message from URL."""
|
|
message = self.sys_discovery.get(request.match_info.get('uuid'))
|
|
if not message:
|
|
raise APIError("Discovery message not found")
|
|
return message
|
|
|
|
def _check_permission_ha(self, request):
|
|
"""Check permission for API call / Home Assistant."""
|
|
if request[REQUEST_FROM] != self.sys_homeassistant:
|
|
raise APIForbidden("Only HomeAssistant can use this API!")
|
|
|
|
@api_process
|
|
async def list(self, request):
|
|
"""Show register services."""
|
|
self._check_permission_ha(request)
|
|
|
|
discovery = []
|
|
for message in self.sys_discovery.list_messages:
|
|
discovery.append({
|
|
ATTR_ADDON: message.addon,
|
|
ATTR_SERVICE: message.service,
|
|
ATTR_UUID: message.uuid,
|
|
ATTR_CONFIG: message.config,
|
|
})
|
|
|
|
return {ATTR_DISCOVERY: discovery}
|
|
|
|
@api_process
|
|
async def set_discovery(self, request):
|
|
"""Write data into a discovery pipeline."""
|
|
body = await api_validate(SCHEMA_DISCOVERY, request)
|
|
addon = request[REQUEST_FROM]
|
|
|
|
# Access?
|
|
if body[ATTR_SERVICE] not in addon.discovery:
|
|
raise APIForbidden(f"Can't use discovery!")
|
|
|
|
# Process discovery message
|
|
message = self.sys_discovery.send(addon, **body)
|
|
|
|
return {ATTR_UUID: message.uuid}
|
|
|
|
@api_process
|
|
async def get_discovery(self, request):
|
|
"""Read data into a discovery message."""
|
|
message = self._extract_message(request)
|
|
|
|
# HomeAssistant?
|
|
self._check_permission_ha(request)
|
|
|
|
return {
|
|
ATTR_ADDON: message.addon,
|
|
ATTR_SERVICE: message.service,
|
|
ATTR_UUID: message.uuid,
|
|
ATTR_CONFIG: message.config,
|
|
}
|
|
|
|
@api_process
|
|
async def del_discovery(self, request):
|
|
"""Delete data into a discovery message."""
|
|
message = self._extract_message(request)
|
|
addon = request[REQUEST_FROM]
|
|
|
|
# Permission
|
|
if message.addon != addon.slug:
|
|
raise APIForbidden(f"Can't remove discovery message")
|
|
|
|
self.sys_discovery.remove(message)
|
|
return True
|