Upgrade HACS

custom_components/hacs/update.py

@@ -3,14 +3,16 @@ from __future__ import annotations
 
 from typing import Any
 
-from homeassistant.components.update import UpdateEntity
-from homeassistant.core import callback
+from homeassistant.components.update import UpdateEntity, UpdateEntityFeature
+from homeassistant.core import HomeAssistantError, callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
 
 from .base import HacsBase
 from .const import DOMAIN
 from .entity import HacsRepositoryEntity
 from .enums import HacsCategory, HacsDispatchEvent
+from .exceptions import HacsException
+from .repositories.base import HacsManifest
 
 
 async def async_setup_entry(hass, _config_entry, async_add_devices):
@@ -25,13 +27,12 @@ async def async_setup_entry(hass, _config_entry, async_add_devices):
 class HacsRepositoryUpdateEntity(HacsRepositoryEntity, UpdateEntity):
     """Update entities for repositories downloaded with HACS."""
 
-    @property
-    def supported_features(self) -> int | None:
-        """Return the supported features of the entity."""
-        features = 4 | 16
-        if self.repository.can_download:
-            features = features | 1
-        return features
+    _attr_supported_features = (
+        UpdateEntityFeature.INSTALL
+        | UpdateEntityFeature.SPECIFIC_VERSION
+        | UpdateEntityFeature.PROGRESS
+        | UpdateEntityFeature.RELEASE_NOTES
+    )
 
     @property
     def name(self) -> str | None:
@@ -75,15 +76,60 @@ class HacsRepositoryUpdateEntity(HacsRepositoryEntity, UpdateEntity):
 
         return f"https://brands.home-assistant.io/_/{self.repository.data.domain}/icon.png"
 
+    async def _ensure_capabilities(self, version: str | None, **kwargs: Any) -> None:
+        """Ensure that the entity has capabilities."""
+        target_manifest: HacsManifest | None = None
+        if version is None:
+            if not self.repository.can_download:
+                raise HomeAssistantError(
+                    f"This {self.repository.data.category.value} is not available for download."
+                )
+            return
+
+        if version == self.repository.data.last_version:
+            target_manifest = self.repository.repository_manifest
+        else:
+            target_manifest = await self.repository.get_hacs_json(version=version)
+
+        if target_manifest is None:
+            raise HomeAssistantError(
+                f"The version {version} for this {self.repository.data.category.value} can not be used with HACS."
+            )
+
+        if (
+            target_manifest.homeassistant is not None
+            and self.hacs.core.ha_version < target_manifest.homeassistant
+        ):
+            raise HomeAssistantError(
+                f"This version requires Home Assistant {target_manifest.homeassistant} or newer."
+            )
+        if target_manifest.hacs is not None and self.hacs.version < target_manifest.hacs:
+            raise HomeAssistantError(f"This version requires HACS {target_manifest.hacs} or newer.")
+
     async def async_install(self, version: str | None, backup: bool, **kwargs: Any) -> None:
         """Install an update."""
+        await self._ensure_capabilities(version)
+        self.repository.logger.info("Starting update, %s", version)
         if self.repository.display_version_or_commit == "version":
             self._update_in_progress(progress=10)
-            self.repository.data.selected_tag = self.latest_version
-            await self.repository.update_repository(force=True)
+            if not version:
+                await self.repository.update_repository(force=True)
+            else:
+                self.repository.ref = version
+            self.repository.data.selected_tag = version
+            self.repository.force_branch = version is not None
             self._update_in_progress(progress=20)
-        await self.repository.async_install()
-        self._update_in_progress(progress=False)
+
+        try:
+            await self.repository.async_install(version=version)
+        except HacsException as exception:
+            raise HomeAssistantError(
+                f"Downloading {self.repository.data.full_name} with version {version or self.repository.data.last_version or self.repository.data.last_commit} failed with ({exception})"
+            ) from exception
+        finally:
+            self.repository.data.selected_tag = None
+            self.repository.force_branch = False
+            self._update_in_progress(progress=False)
 
     async def async_release_notes(self) -> str | None:
         """Return the release notes."""