updating to latest

2021-11-04 01:18:18 -04:00
parent f92b773514
commit 7eadb4c49c
153 changed files with 19015 additions and 5168 deletions
--- a/custom_components/hacs/helpers/functions/validate_repository.py
+++ b/custom_components/hacs/helpers/functions/validate_repository.py
@@ -0,0 +1,124 @@
+"""Helper to do common validation for repositories."""
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from aiogithubapi import AIOGitHubAPIException
+
+from custom_components.hacs.exceptions import (
+    HacsException,
+    HacsNotModifiedException,
+    HacsRepositoryArchivedException,
+    HacsRepositoryExistException,
+)
+from custom_components.hacs.helpers.functions.information import (
+    get_releases,
+    get_repository,
+    get_tree,
+)
+from custom_components.hacs.helpers.functions.version_to_install import (
+    version_to_install,
+)
+from custom_components.hacs.share import get_hacs, is_removed
+
+if TYPE_CHECKING:
+    from custom_components.hacs.helpers.classes.repository import HacsRepository
+
+
+async def common_validate(repository, ignore_issues=False):
+    """Common validation steps of the repository."""
+    repository.validate.errors = []
+
+    # Make sure the repository exist.
+    repository.logger.debug("%s Checking repository.", repository)
+    await common_update_data(repository, ignore_issues)
+
+    # Step 6: Get the content of hacs.json
+    await repository.get_repository_manifest_content()
+
+
+async def common_update_data(repository: HacsRepository, ignore_issues=False, force=False):
+    """Common update data."""
+    hacs = get_hacs()
+    releases = []
+    try:
+        repository_object, etag = await get_repository(
+            hacs.session,
+            hacs.configuration.token,
+            repository.data.full_name,
+            etag=None if force or repository.data.installed else repository.data.etag_repository,
+        )
+        repository.repository_object = repository_object
+        if repository.data.full_name.lower() != repository_object.full_name.lower():
+            hacs.common.renamed_repositories[
+                repository.data.full_name
+            ] = repository_object.full_name
+            if str(repository_object.id) not in hacs.common.default:
+                hacs.common.default.append(str(repository_object.id))
+            raise HacsRepositoryExistException
+        repository.data.update_data(repository_object.attributes)
+        repository.data.etag_repository = etag
+    except HacsNotModifiedException:
+        return
+    except HacsRepositoryExistException:
+        raise HacsRepositoryExistException from None
+    except (AIOGitHubAPIException, HacsException) as exception:
+        if not hacs.status.startup:
+            repository.logger.error("%s %s", repository, exception)
+        if not ignore_issues:
+            repository.validate.errors.append("Repository does not exist.")
+            raise HacsException(exception) from None
+
+    # Make sure the repository is not archived.
+    if repository.data.archived and not ignore_issues:
+        repository.validate.errors.append("Repository is archived.")
+        hacs.common.archived_repositories.append(repository.data.full_name)
+        raise HacsRepositoryArchivedException("Repository is archived.")
+
+    # Make sure the repository is not in the blacklist.
+    if is_removed(repository.data.full_name) and not ignore_issues:
+        repository.validate.errors.append("Repository is in the blacklist.")
+        raise HacsException("Repository is in the blacklist.")
+
+    # Get releases.
+    try:
+        releases = await get_releases(
+            repository.repository_object,
+            repository.data.show_beta,
+            hacs.configuration.release_limit,
+        )
+        if releases:
+            repository.data.releases = True
+            repository.releases.objects = [x for x in releases if not x.draft]
+            repository.data.published_tags = [x.tag_name for x in repository.releases.objects]
+            repository.data.last_version = next(iter(repository.data.published_tags))
+
+    except (AIOGitHubAPIException, HacsException):
+        repository.data.releases = False
+
+    if not repository.force_branch:
+        repository.ref = version_to_install(repository)
+    if repository.data.releases:
+        for release in repository.releases.objects or []:
+            if release.tag_name == repository.ref:
+                assets = release.assets
+                if assets:
+                    downloads = next(iter(assets)).attributes.get("download_count")
+                    repository.data.downloads = downloads
+
+    repository.logger.debug(
+        "%s Running checks against %s", repository, repository.ref.replace("tags/", "")
+    )
+
+    try:
+        repository.tree = await get_tree(repository.repository_object, repository.ref)
+        if not repository.tree:
+            raise HacsException("No files in tree")
+        repository.treefiles = []
+        for treefile in repository.tree:
+            repository.treefiles.append(treefile.full_path)
+    except (AIOGitHubAPIException, HacsException) as exception:
+        if not hacs.status.startup:
+            repository.logger.error("%s %s", repository, exception)
+        if not ignore_issues:
+            raise HacsException(exception) from None