Finally got around to installing This amazing component by @ludeeus

config/custom_components/hacs/repositories/integration.py

@@ -0,0 +1,165 @@
+"""Class for integrations in HACS."""
+import json
+from aiogithubapi import AIOGitHubException
+from homeassistant.loader import async_get_custom_components
+from .repository import HacsRepository, register_repository_class
+from ..hacsbase.exceptions import HacsException
+
+
+@register_repository_class
+class HacsIntegration(HacsRepository):
+    """Integrations in HACS."""
+
+    category = "integration"
+
+    def __init__(self, full_name):
+        """Initialize."""
+        super().__init__()
+        self.information.full_name = full_name
+        self.information.category = self.category
+        self.domain = None
+        self.content.path.remote = "custom_components"
+        self.content.path.local = self.localpath
+
+    @property
+    def localpath(self):
+        """Return localpath."""
+        return f"{self.system.config_path}/custom_components/{self.domain}"
+
+    async def validate_repository(self):
+        """Validate."""
+        await self.common_validate()
+
+        # Attach repository
+        if self.repository_object is None:
+            self.repository_object = await self.github.get_repo(
+                self.information.full_name
+            )
+
+        # Custom step 1: Validate content.
+        if self.repository_manifest:
+            if self.repository_manifest.content_in_root:
+                self.content.path.remote = ""
+
+        if self.content.path.remote == "custom_components":
+            try:
+                ccdir = await self.repository_object.get_contents(
+                    self.content.path.remote, self.ref
+                )
+            except AIOGitHubException:
+                raise HacsException(
+                    f"Repostitory structure for {self.ref.replace('tags/','')} is not compliant"
+                )
+
+            for item in ccdir or []:
+                if item.type == "dir":
+                    self.content.path.remote = item.path
+                    break
+
+        if self.repository_manifest.zip_release:
+            self.content.objects = self.releases.last_release_object.assets
+
+        else:
+            self.content.objects = await self.repository_object.get_contents(
+                self.content.path.remote, self.ref
+            )
+
+        self.content.files = []
+        for filename in self.content.objects or []:
+            self.content.files.append(filename.name)
+
+        if not await self.get_manifest():
+            self.validate.errors.append("Missing manifest file.")
+
+        # Handle potential errors
+        if self.validate.errors:
+            for error in self.validate.errors:
+                if not self.system.status.startup:
+                    self.logger.error(error)
+        return self.validate.success
+
+    async def registration(self):
+        """Registration."""
+        if not await self.validate_repository():
+            return False
+
+        # Run common registration steps.
+        await self.common_registration()
+
+        # Get the content of the manifest file.
+        await self.get_manifest()
+
+        # Set local path
+        self.content.path.local = self.localpath
+
+    async def update_repository(self):
+        """Update."""
+        if self.github.ratelimits.remaining == 0:
+            return
+        await self.common_update()
+
+        # Get integration objects.
+
+        if self.repository_manifest:
+            if self.repository_manifest.content_in_root:
+                self.content.path.remote = ""
+
+        if self.content.path.remote == "custom_components":
+            ccdir = await self.repository_object.get_contents(
+                self.content.path.remote, self.ref
+            )
+            if not isinstance(ccdir, list):
+                self.validate.errors.append("Repostitory structure not compliant")
+
+            self.content.path.remote = ccdir[0].path
+
+        try:
+            self.content.objects = await self.repository_object.get_contents(
+                self.content.path.remote, self.ref
+            )
+        except AIOGitHubException:
+            return
+
+        self.content.files = []
+        if isinstance(self.content.objects, list):
+            for filename in self.content.objects or []:
+                self.content.files.append(filename.name)
+
+        await self.get_manifest()
+
+        # Set local path
+        self.content.path.local = self.localpath
+
+    async def reload_custom_components(self):
+        """Reload custom_components (and config flows)in HA."""
+        self.logger.info("Reloading custom_component cache")
+        del self.hass.data["custom_components"]
+        await async_get_custom_components(self.hass)
+
+    async def get_manifest(self):
+        """Get info from the manifest file."""
+        manifest_path = f"{self.content.path.remote}/manifest.json"
+        try:
+            manifest = await self.repository_object.get_contents(
+                manifest_path, self.ref
+            )
+            manifest = json.loads(manifest.content)
+        except Exception:  # pylint: disable=broad-except
+            return False
+
+        if manifest:
+            try:
+                self.manifest = manifest
+                self.information.authors = manifest["codeowners"]
+                self.domain = manifest["domain"]
+                self.information.name = manifest["name"]
+                self.information.homeassistant_version = manifest.get("homeassistant")
+
+                # Set local path
+                self.content.path.local = self.localpath
+                return True
+            except KeyError as exception:
+                raise HacsException(
+                    f"Missing expected key {exception} in 'manifest.json'"
+                )
+        return False