updating to latest

custom_components/hacs/helpers/methods/installation.py

@@ -0,0 +1,113 @@
+# pylint: disable=missing-class-docstring,missing-module-docstring,missing-function-docstring,no-member
+from abc import ABC
+import os
+import tempfile
+
+from custom_components.hacs.exceptions import HacsException
+from custom_components.hacs.helpers.functions.download import download_content
+from custom_components.hacs.helpers.functions.version_to_install import (
+    version_to_install,
+)
+from custom_components.hacs.operational.backup import Backup, BackupNetDaemon
+from custom_components.hacs.share import get_hacs
+
+
+class RepositoryMethodPreInstall(ABC):
+    async def async_pre_install(self) -> None:
+        pass
+
+    async def _async_pre_install(self) -> None:
+        self.logger.info("Running pre installation steps")
+        await self.async_pre_install()
+        self.logger.info("Pre installation steps completed")
+
+
+class RepositoryMethodInstall(ABC):
+    async def async_install(self) -> None:
+        await self._async_pre_install()
+        self.logger.info("Running installation steps")
+        await async_install_repository(self)
+        self.logger.info("Installation steps completed")
+        await self._async_post_install()
+
+
+class RepositoryMethodPostInstall(ABC):
+    async def async_post_installation(self) -> None:
+        pass
+
+    async def _async_post_install(self) -> None:
+        self.logger.info("Running post installation steps")
+        await self.async_post_installation()
+        self.data.new = False
+        self.hacs.hass.bus.async_fire(
+            "hacs/repository",
+            {"id": 1337, "action": "install", "repository": self.data.full_name},
+        )
+        self.logger.info("Post installation steps completed")
+
+
+async def async_install_repository(repository):
+    """Common installation steps of the repository."""
+    hacs = get_hacs()
+    persistent_directory = None
+    await repository.update_repository()
+    if repository.content.path.local is None:
+        raise HacsException("repository.content.path.local is None")
+    repository.validate.errors = []
+
+    if not repository.can_install:
+        raise HacsException("The version of Home Assistant is not compatible with this version")
+
+    version = version_to_install(repository)
+    if version == repository.data.default_branch:
+        repository.ref = version
+    else:
+        repository.ref = f"tags/{version}"
+
+    if repository.data.installed and repository.data.category == "netdaemon":
+        persistent_directory = await hacs.hass.async_add_executor_job(BackupNetDaemon, repository)
+        await hacs.hass.async_add_executor_job(persistent_directory.create)
+
+    elif repository.data.persistent_directory:
+        if os.path.exists(
+            f"{repository.content.path.local}/{repository.data.persistent_directory}"
+        ):
+            persistent_directory = Backup(
+                f"{repository.content.path.local}/{repository.data.persistent_directory}",
+                tempfile.gettempdir() + "/hacs_persistent_directory/",
+            )
+            await hacs.hass.async_add_executor_job(persistent_directory.create)
+
+    if repository.data.installed and not repository.content.single:
+        backup = Backup(repository.content.path.local)
+        await hacs.hass.async_add_executor_job(backup.create)
+
+    if repository.data.zip_release and version != repository.data.default_branch:
+        await repository.download_zip_files(repository.validate)
+    else:
+        await download_content(repository)
+
+    if repository.validate.errors:
+        for error in repository.validate.errors:
+            repository.logger.error(error)
+        if repository.data.installed and not repository.content.single:
+            await hacs.hass.async_add_executor_job(backup.restore)
+
+    if repository.data.installed and not repository.content.single:
+        await hacs.hass.async_add_executor_job(backup.cleanup)
+
+    if persistent_directory is not None:
+        await hacs.hass.async_add_executor_job(persistent_directory.restore)
+        await hacs.hass.async_add_executor_job(persistent_directory.cleanup)
+
+    if repository.validate.success:
+        if repository.data.full_name not in repository.hacs.common.installed:
+            if repository.data.full_name == "hacs/integration":
+                repository.hacs.common.installed.append(repository.data.full_name)
+        repository.data.installed = True
+        repository.data.installed_commit = repository.data.last_commit
+
+        if version == repository.data.default_branch:
+            repository.data.installed_version = None
+        else:
+            repository.data.installed_version = version