mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-06-26 11:56:30 +00:00
d7df423deb
* Allow event stream over api proxy * fix lint * fix lint * cleanup code * fix bug * fix prepare * Fix stream bug * fix api request
180 lines
7.5 KiB
Python
180 lines
7.5 KiB
Python
"""Init file for HassIO rest api."""
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
from aiohttp import web
|
|
|
|
from .addons import APIAddons
|
|
from .homeassistant import APIHomeAssistant
|
|
from .host import APIHost
|
|
from .network import APINetwork
|
|
from .supervisor import APISupervisor
|
|
from .security import APISecurity
|
|
from .snapshots import APISnapshots
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class RestAPI(object):
|
|
"""Handle rest api for hassio."""
|
|
|
|
def __init__(self, config, loop):
|
|
"""Initialize docker base wrapper."""
|
|
self.config = config
|
|
self.loop = loop
|
|
self.webapp = web.Application(loop=self.loop)
|
|
|
|
# service stuff
|
|
self._handler = None
|
|
self.server = None
|
|
|
|
def register_host(self, host_control, hardware):
|
|
"""Register hostcontrol function."""
|
|
api_host = APIHost(self.config, self.loop, host_control, hardware)
|
|
|
|
self.webapp.router.add_get('/host/info', api_host.info)
|
|
self.webapp.router.add_get('/host/hardware', api_host.hardware)
|
|
self.webapp.router.add_post('/host/reboot', api_host.reboot)
|
|
self.webapp.router.add_post('/host/shutdown', api_host.shutdown)
|
|
self.webapp.router.add_post('/host/update', api_host.update)
|
|
self.webapp.router.add_post('/host/options', api_host.options)
|
|
|
|
def register_network(self, host_control):
|
|
"""Register network function."""
|
|
api_net = APINetwork(self.config, self.loop, host_control)
|
|
|
|
self.webapp.router.add_get('/network/info', api_net.info)
|
|
self.webapp.router.add_post('/network/options', api_net.options)
|
|
|
|
def register_supervisor(self, supervisor, snapshots, addons, host_control,
|
|
updater):
|
|
"""Register supervisor function."""
|
|
api_supervisor = APISupervisor(
|
|
self.config, self.loop, supervisor, snapshots, addons,
|
|
host_control, updater)
|
|
|
|
self.webapp.router.add_get('/supervisor/ping', api_supervisor.ping)
|
|
self.webapp.router.add_get('/supervisor/info', api_supervisor.info)
|
|
self.webapp.router.add_post(
|
|
'/supervisor/update', api_supervisor.update)
|
|
self.webapp.router.add_post(
|
|
'/supervisor/reload', api_supervisor.reload)
|
|
self.webapp.router.add_post(
|
|
'/supervisor/options', api_supervisor.options)
|
|
self.webapp.router.add_get('/supervisor/logs', api_supervisor.logs)
|
|
|
|
def register_homeassistant(self, dock_homeassistant):
|
|
"""Register homeassistant function."""
|
|
api_hass = APIHomeAssistant(self.config, self.loop, dock_homeassistant)
|
|
|
|
self.webapp.router.add_get('/homeassistant/info', api_hass.info)
|
|
self.webapp.router.add_get('/homeassistant/logs', api_hass.logs)
|
|
self.webapp.router.add_post('/homeassistant/options', api_hass.options)
|
|
self.webapp.router.add_post('/homeassistant/update', api_hass.update)
|
|
self.webapp.router.add_post('/homeassistant/restart', api_hass.restart)
|
|
self.webapp.router.add_post('/homeassistant/stop', api_hass.stop)
|
|
self.webapp.router.add_post('/homeassistant/start', api_hass.start)
|
|
self.webapp.router.add_post('/homeassistant/check', api_hass.check)
|
|
self.webapp.router.add_post(
|
|
'/homeassistant/api/{path:.+}', api_hass.api)
|
|
self.webapp.router.add_get(
|
|
'/homeassistant/api/{path:.+}', api_hass.api)
|
|
self.webapp.router.add_get(
|
|
'/homeassistant/api', api_hass.api)
|
|
|
|
def register_addons(self, addons):
|
|
"""Register homeassistant function."""
|
|
api_addons = APIAddons(self.config, self.loop, addons)
|
|
|
|
self.webapp.router.add_get('/addons', api_addons.list)
|
|
self.webapp.router.add_post('/addons/reload', api_addons.reload)
|
|
|
|
self.webapp.router.add_get('/addons/{addon}/info', api_addons.info)
|
|
self.webapp.router.add_post(
|
|
'/addons/{addon}/install', api_addons.install)
|
|
self.webapp.router.add_post(
|
|
'/addons/{addon}/uninstall', api_addons.uninstall)
|
|
self.webapp.router.add_post('/addons/{addon}/start', api_addons.start)
|
|
self.webapp.router.add_post('/addons/{addon}/stop', api_addons.stop)
|
|
self.webapp.router.add_post(
|
|
'/addons/{addon}/restart', api_addons.restart)
|
|
self.webapp.router.add_post(
|
|
'/addons/{addon}/update', api_addons.update)
|
|
self.webapp.router.add_post(
|
|
'/addons/{addon}/options', api_addons.options)
|
|
self.webapp.router.add_post(
|
|
'/addons/{addon}/rebuild', api_addons.rebuild)
|
|
self.webapp.router.add_get('/addons/{addon}/logs', api_addons.logs)
|
|
self.webapp.router.add_get('/addons/{addon}/logo', api_addons.logo)
|
|
self.webapp.router.add_get(
|
|
'/addons/{addon}/changelog', api_addons.changelog)
|
|
self.webapp.router.add_post('/addons/{addon}/stdin', api_addons.stdin)
|
|
|
|
def register_security(self):
|
|
"""Register security function."""
|
|
api_security = APISecurity(self.config, self.loop)
|
|
|
|
self.webapp.router.add_get('/security/info', api_security.info)
|
|
self.webapp.router.add_post('/security/options', api_security.options)
|
|
self.webapp.router.add_post('/security/totp', api_security.totp)
|
|
self.webapp.router.add_post('/security/session', api_security.session)
|
|
|
|
def register_snapshots(self, snapshots):
|
|
"""Register snapshots function."""
|
|
api_snapshots = APISnapshots(self.config, self.loop, snapshots)
|
|
|
|
self.webapp.router.add_get('/snapshots', api_snapshots.list)
|
|
self.webapp.router.add_post('/snapshots/reload', api_snapshots.reload)
|
|
|
|
self.webapp.router.add_post(
|
|
'/snapshots/new/full', api_snapshots.snapshot_full)
|
|
self.webapp.router.add_post(
|
|
'/snapshots/new/partial', api_snapshots.snapshot_partial)
|
|
|
|
self.webapp.router.add_get(
|
|
'/snapshots/{snapshot}/info', api_snapshots.info)
|
|
self.webapp.router.add_post(
|
|
'/snapshots/{snapshot}/remove', api_snapshots.remove)
|
|
self.webapp.router.add_post(
|
|
'/snapshots/{snapshot}/restore/full', api_snapshots.restore_full)
|
|
self.webapp.router.add_post(
|
|
'/snapshots/{snapshot}/restore/partial',
|
|
api_snapshots.restore_partial)
|
|
|
|
def register_panel(self):
|
|
"""Register panel for homeassistant."""
|
|
def create_panel_response(build_type):
|
|
"""Create a function to generate a response."""
|
|
path = Path(__file__).parents[1].joinpath(
|
|
'panel/hassio-main-{}.html'.format(build_type))
|
|
|
|
return lambda request: web.FileResponse(path)
|
|
|
|
# This route is for backwards compatibility with HA < 0.58
|
|
self.webapp.router.add_get('/panel', create_panel_response('es5'))
|
|
self.webapp.router.add_get('/panel_es5', create_panel_response('es5'))
|
|
self.webapp.router.add_get(
|
|
'/panel_latest', create_panel_response('latest'))
|
|
|
|
async def start(self):
|
|
"""Run rest api webserver."""
|
|
self._handler = self.webapp.make_handler(loop=self.loop)
|
|
|
|
try:
|
|
self.server = await self.loop.create_server(
|
|
self._handler, "0.0.0.0", "80")
|
|
except OSError as err:
|
|
_LOGGER.fatal(
|
|
"Failed to create HTTP server at 0.0.0.0:80 -> %s", err)
|
|
|
|
async def stop(self):
|
|
"""Stop rest api webserver."""
|
|
if self.server:
|
|
self.server.close()
|
|
await self.server.wait_closed()
|
|
await self.webapp.shutdown()
|
|
|
|
if self._handler:
|
|
await self._handler.shutdown(60)
|
|
await self.webapp.cleanup()
|