Merge pull request #3 from pvizeli/addons

Addon support

Author: pvizeli

API.md

@@ -69,7 +69,15 @@ On success
     "version": "INSTALL_VERSION",
     "current": "CURRENT_VERSION",
     "beta": "true|false",
-    "addons": {}
+    "addons": [
+        {
+            "name": "xy bla",
+            "slug": "xy",
+            "version": "CURRENT_VERSION",
+            "installed": "none|INSTALL_VERSION",
+            "description": "description"
+        }
+    ]
 }
 ```
 

hassio/addons/__init__.py

@@ -0,0 +1,159 @@
+"""Init file for HassIO addons."""
+import asyncio
+import logging
+import os
+import shutil
+
+from .data import AddonsData
+from .git import AddonsRepo
+from ..const import STATE_STOPPED, STATE_STARTED
+from ..dock.addon import DockerAddon
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class AddonManager(AddonsData):
+    """Manage addons inside HassIO."""
+
+    def __init__(self, config, loop, dock):
+        """Initialize docker base wrapper."""
+        super().__init__(config)
+
+        self.loop = loop
+        self.dock = dock
+        self.repo = AddonsRepo(config, loop)
+        self.dockers = {}
+
+    async def prepare(self, arch):
+        """Startup addon management."""
+        self.arch = arch
+
+        # load addon repository
+        if await self.repo.load():
+            self.read_addons_repo()
+
+        # load installed addons
+        for addon in self.list_installed:
+            self.dockers[addon] = DockerAddon(
+                self.config, self.loop, self.dock, self, addon)
+
+    async def relaod(self):
+        """Update addons from repo and reload list."""
+        if not await self.repo.pull():
+            return
+        self.read_addons_repo()
+
+        # remove stalled addons
+        tasks = []
+        for addon in self.list_removed:
+            _LOGGER.info("Old addon %s found")
+            tasks.append(self.loop.create_task(self.dockers[addon].remove()))
+
+        if tasks:
+            await asyncio.wait(tasks, loop=self.loop)
+
+    async def auto_boot(self, start_type):
+        """Boot addons with mode auto."""
+        boot_list = self.list_startup(start_type)
+        tasks = []
+
+        for addon in boot_list:
+            tasks.append(self.loop.create_task(self.start(addon)))
+
+        _LOGGER.info("Startup %s run %d addons", start_type, len(tasks))
+        if tasks:
+            await asyncio.wait(tasks, loop=self.loop)
+
+    async def install(self, addon, version=None):
+        """Install a addon."""
+        if not self.exists_addon(addon):
+            _LOGGER.error("Addon %s not exists for install", addon)
+            return False
+
+        if self.is_installed(addon):
+            _LOGGER.error("Addon %s is already installed", addon)
+            return False
+
+        if not os.path.isdir(self.path_data(addon)):
+            _LOGGER.info("Create Home-Assistant addon data folder %s",
+                         self.path_data(addon))
+            os.mkdir(self.path_data(addon))
+
+        addon_docker = DockerAddon(
+            self.config, self.loop, self.dock, self, addon)
+
+        version = version or self.get_version(addon)
+        if not await addon_docker.install(version):
+            return False
+
+        self.dockers[addon] = addon_docker
+        self.set_install_addon(addon, version)
+        return True
+
+    async def uninstall(self, addon):
+        """Remove a addon."""
+        if not self.is_installed(addon):
+            _LOGGER.error("Addon %s is already uninstalled", addon)
+            return False
+
+        if addon not in self.dockers:
+            _LOGGER.error("No docker found for addon %s", addon)
+            return False
+
+        if not await self.dockers[addon].remove():
+            return False
+
+        if os.path.isdir(self.path_data(addon)):
+            _LOGGER.info("Remove Home-Assistant addon data folder %s",
+                         self.path_data(addon))
+            shutil.rmtree(self.path_data(addon))
+
+        self.dockers.pop(addon)
+        self.set_uninstall_addon(addon)
+        return True
+
+    async def state(self, addon):
+        """Return running state of addon."""
+        if addon not in self.dockers:
+            _LOGGER.error("No docker found for addon %s", addon)
+            return
+
+        if await self.dockers[addon].is_running():
+            return STATE_STARTED
+        return STATE_STOPPED
+
+    async def start(self, addon):
+        """Set options and start addon."""
+        if addon not in self.dockers:
+            _LOGGER.error("No docker found for addon %s", addon)
+            return False
+
+        if not self.write_addon_options(addon):
+            _LOGGER.error("Can't write options for addon %s", addon)
+            return False
+
+        return await self.dockers[addon].run()
+
+    async def stop(self, addon):
+        """Stop addon."""
+        if addon not in self.dockers:
+            _LOGGER.error("No docker found for addon %s", addon)
+            return False
+
+        return await self.dockers[addon].stop()
+
+    async def update(self, addon, version=None):
+        """Update addon."""
+        if not self.is_installed(addon):
+            _LOGGER.error("Addon %s is not installed", addon)
+            return False
+
+        if addon not in self.dockers:
+            _LOGGER.error("No docker found for addon %s", addon)
+            return False
+
+        version = version or self.get_version(addon)
+        if await self.dockers[addon].update(version):
+            self.set_version(addon, version)
+            return True
+        return False