mirror of
https://github.com/home-assistant/supervisor.git
synced 2026-10-06 02:14:00 +01:00
Mike Degatano
d19166bb86
* Docker events based watchdog * Separate monitor from DockerAPI since it needs coresys * Move monitor into dockerAPI * Fix properties on coresys * Add watchdog tests * Added tests * pylint issue * Current state failures test * Thread-safe event processing * Use labels property
139 lines
4.3 KiB
Python
139 lines
4.3 KiB
Python
"""Home Assistant cli plugin.
|
|
|
|
Code: https://github.com/home-assistant/plugin-cli
|
|
"""
|
|
import asyncio
|
|
from contextlib import suppress
|
|
import logging
|
|
import secrets
|
|
from typing import Awaitable, Optional
|
|
|
|
from awesomeversion import AwesomeVersion
|
|
|
|
from ..const import ATTR_ACCESS_TOKEN
|
|
from ..coresys import CoreSys
|
|
from ..docker.cli import DockerCli
|
|
from ..docker.stats import DockerStats
|
|
from ..exceptions import CliError, CliUpdateError, DockerError
|
|
from .base import PluginBase
|
|
from .const import FILE_HASSIO_CLI
|
|
from .validate import SCHEMA_CLI_CONFIG
|
|
|
|
_LOGGER: logging.Logger = logging.getLogger(__name__)
|
|
|
|
|
|
class PluginCli(PluginBase):
|
|
"""HA cli interface inside supervisor."""
|
|
|
|
def __init__(self, coresys: CoreSys):
|
|
"""Initialize cli handler."""
|
|
super().__init__(FILE_HASSIO_CLI, SCHEMA_CLI_CONFIG)
|
|
self.slug = "cli"
|
|
self.coresys: CoreSys = coresys
|
|
self.instance: DockerCli = DockerCli(coresys)
|
|
|
|
@property
|
|
def latest_version(self) -> Optional[AwesomeVersion]:
|
|
"""Return version of latest cli."""
|
|
return self.sys_updater.version_cli
|
|
|
|
@property
|
|
def supervisor_token(self) -> str:
|
|
"""Return an access token for the Supervisor API."""
|
|
return self._data.get(ATTR_ACCESS_TOKEN)
|
|
|
|
async def install(self) -> None:
|
|
"""Install cli."""
|
|
_LOGGER.info("Running setup for CLI plugin")
|
|
while True:
|
|
# read cli tag and install it
|
|
if not self.latest_version:
|
|
await self.sys_updater.reload()
|
|
|
|
if self.latest_version:
|
|
with suppress(DockerError):
|
|
await self.instance.install(
|
|
self.latest_version,
|
|
image=self.sys_updater.image_cli,
|
|
)
|
|
break
|
|
_LOGGER.warning("Error on install cli plugin. Retry in 30sec")
|
|
await asyncio.sleep(30)
|
|
|
|
_LOGGER.info("CLI plugin is now installed")
|
|
self.version = self.instance.version
|
|
self.image = self.sys_updater.image_cli
|
|
self.save_data()
|
|
|
|
async def update(self, version: Optional[AwesomeVersion] = None) -> None:
|
|
"""Update local HA cli."""
|
|
version = version or self.latest_version
|
|
old_image = self.image
|
|
|
|
if version == self.version:
|
|
_LOGGER.warning("Version %s is already installed for CLI", version)
|
|
return
|
|
|
|
try:
|
|
await self.instance.update(version, image=self.sys_updater.image_cli)
|
|
except DockerError as err:
|
|
raise CliUpdateError("CLI update failed", _LOGGER.error) from err
|
|
else:
|
|
self.version = version
|
|
self.image = self.sys_updater.image_cli
|
|
self.save_data()
|
|
|
|
# Cleanup
|
|
with suppress(DockerError):
|
|
await self.instance.cleanup(old_image=old_image)
|
|
|
|
# Start cli
|
|
await self.start()
|
|
|
|
async def start(self) -> None:
|
|
"""Run cli."""
|
|
# Create new API token
|
|
self._data[ATTR_ACCESS_TOKEN] = secrets.token_hex(56)
|
|
self.save_data()
|
|
|
|
# Start Instance
|
|
_LOGGER.info("Starting CLI plugin")
|
|
try:
|
|
await self.instance.run()
|
|
except DockerError as err:
|
|
raise CliError("Can't start cli plugin", _LOGGER.error) from err
|
|
|
|
async def stop(self) -> None:
|
|
"""Stop cli."""
|
|
_LOGGER.info("Stopping cli plugin")
|
|
try:
|
|
await self.instance.stop()
|
|
except DockerError as err:
|
|
raise CliError("Can't stop cli plugin", _LOGGER.error) from err
|
|
|
|
async def stats(self) -> DockerStats:
|
|
"""Return stats of cli."""
|
|
try:
|
|
return await self.instance.stats()
|
|
except DockerError as err:
|
|
raise CliError() from err
|
|
|
|
def is_running(self) -> Awaitable[bool]:
|
|
"""Return True if Docker container is running.
|
|
|
|
Return a coroutine.
|
|
"""
|
|
return self.instance.is_running()
|
|
|
|
async def repair(self) -> None:
|
|
"""Repair cli container."""
|
|
if await self.instance.exists():
|
|
return
|
|
|
|
_LOGGER.info("Repairing HA cli %s", self.version)
|
|
try:
|
|
await self.instance.install(self.version)
|
|
except DockerError as err:
|
|
_LOGGER.error("Repair of HA cli failed")
|
|
self.sys_capture_exception(err)
|