"""Install helper for repositories.""" import os import tempfile from custom_components.hacs.globals import get_hacs from custom_components.hacs.hacsbase.exceptions import HacsException from custom_components.hacs.hacsbase.backup import Backup, BackupNetDaemon from custom_components.hacs.helpers.download import download_content async def install_repository(repository): """Common installation steps of the repository.""" 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.status.installed and repository.data.category == "netdaemon": persistent_directory = BackupNetDaemon(repository) 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/", ) persistent_directory.create() if repository.status.installed and not repository.content.single: backup = Backup(repository.content.path.local) backup.create() if repository.data.zip_release and version != repository.data.default_branch: await repository.download_zip(repository) else: await download_content(repository) if repository.validate.errors: for error in repository.validate.errors: repository.logger.error(error) if repository.status.installed and not repository.content.single: backup.restore() if repository.status.installed and not repository.content.single: backup.cleanup() if persistent_directory is not None: persistent_directory.restore() 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.status.installed = True repository.versions.installed_commit = repository.versions.available_commit if version == repository.data.default_branch: repository.versions.installed = None else: repository.versions.installed = version await reload_after_install(repository) installation_complete(repository) async def reload_after_install(repository): """Reload action after installation success.""" if repository.data.category == "integration": if repository.config_flow: if repository.data.full_name != "hacs/integration": await repository.reload_custom_components() repository.pending_restart = True elif repository.data.category == "theme": try: await repository.hacs.hass.services.async_call( "frontend", "reload_themes", {} ) except Exception: # pylint: disable=broad-except pass elif repository.data.category == "netdaemon": try: await repository.hacs.hass.services.async_call( "hassio", "addon_restart", {"addon": "c6a2317c_netdaemon"} ) except Exception: # pylint: disable=broad-except pass def installation_complete(repository): """Action to run when the installation is complete.""" hacs = get_hacs() hacs.hass.bus.async_fire( "hacs/repository", {"id": 1337, "action": "install", "repository": repository.data.full_name}, ) def version_to_install(repository): """Determine which version to isntall.""" if repository.versions.available is not None: if repository.status.selected_tag is not None: if repository.status.selected_tag == repository.versions.available: repository.status.selected_tag = None return repository.versions.available return repository.status.selected_tag return repository.versions.available if repository.status.selected_tag is not None: if repository.status.selected_tag == repository.data.default_branch: return repository.data.default_branch if repository.status.selected_tag in repository.releases.published_tags: return repository.status.selected_tag if repository.data.default_branch is None: return "master" return repository.data.default_branch