feat: get metadata attributes and add to image entity (#4)

Author: Ivieb

custom_components/immich/hub.py

@@ -67,6 +67,26 @@ async def get_my_user_info(self) -> dict:
             _LOGGER.error("Error connecting to the API: %s", exception)
             raise CannotConnect from exception
 
+    async def get_asset_info(self, asset_id: str) -> dict | None:
+        """Get asset info."""
+        try:
+            async with aiohttp.ClientSession() as session:
+                url = urljoin(self.host, f"/api/assets/{asset_id}")
+                headers = {"Accept": "application/json", _HEADER_API_KEY: self.api_key}
+
+                async with session.get(url=url, headers=headers) as response:
+                    if response.status != 200:
+                        raw_result = await response.text()
+                        _LOGGER.error("Error from API: body=%s", raw_result)
+                        raise ApiError()
+
+                    asset_info: dict = await response.json()
+
+                    return asset_info
+        except aiohttp.ClientError as exception:
+            _LOGGER.error("Error connecting to the API: %s", exception)
+            raise CannotConnect from exception
+
     async def download_asset(self, asset_id: str) -> bytes | None:
         """Download the asset."""
         try:

custom_components/immich/image.py

@@ -75,6 +75,8 @@ def __init__(self, hass: HomeAssistant, hub: ImmichHub) -> None:
         self.hub = hub
         self.hass = hass
 
+        self._attr_extra_state_attributes = {}
+
     async def async_update(self) -> None:
         """Force a refresh of the image."""
         await self._load_and_cache_next_image()
@@ -128,8 +130,21 @@ async def _load_and_cache_next_image(self) -> None:
                 await asyncio.sleep(1)
                 continue
 
+            asset_info = await self.hub.get_asset_info(asset_id)
+
+            self._attr_extra_state_attributes["media_filename"] = (
+                asset_info.get("originalFileName") or ""
+            )
+            self._attr_extra_state_attributes["media_exif"] = (
+                asset_info.get("exifInfo") or ""
+            )
+            self._attr_extra_state_attributes["media_localdatetime"] = (
+                asset_info.get("localDateTime") or ""
+            )
+
             self._current_image_bytes = asset_bytes
             self._attr_image_last_updated = datetime.now()
+            self.async_write_ha_state()
 
 
 class ImmichImageFavorite(BaseImmichImage):