Handle added and removed monitors (#54228)

homeassistant/components/uptimerobot/__init__.py

@@ -7,6 +7,7 @@ from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_API_KEY
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.device_registry import DeviceRegistry, async_get_registry
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
 from .const import API_ATTR_OK, COORDINATOR_UPDATE_INTERVAL, DOMAIN, LOGGER, PLATFORMS
@@ -18,25 +19,13 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     uptime_robot_api = UptimeRobot(
         entry.data[CONF_API_KEY], async_get_clientsession(hass)
     )
+    dev_reg = await async_get_registry(hass)
 
-    async def async_update_data() -> list[UptimeRobotMonitor]:
+    hass.data[DOMAIN][entry.entry_id] = coordinator = UptimeRobotDataUpdateCoordinator(
-        """Fetch data from API UptimeRobot API."""
-        try:
-            response = await uptime_robot_api.async_get_monitors()
-        except UptimeRobotException as exception:
-            raise UpdateFailed(exception) from exception
-        else:
-            if response.status == API_ATTR_OK:
-                monitors: list[UptimeRobotMonitor] = response.data
-                return monitors
-            raise UpdateFailed(response.error.message)
-
-    hass.data[DOMAIN][entry.entry_id] = coordinator = DataUpdateCoordinator(
         hass,
-        LOGGER,
+        config_entry_id=entry.entry_id,
-        name=DOMAIN,
+        dev_reg=dev_reg,
-        update_method=async_update_data,
+        api=uptime_robot_api,
-        update_interval=COORDINATOR_UPDATE_INTERVAL,
     )
 
     await coordinator.async_config_entry_first_refresh()
@@ -53,3 +42,62 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
         hass.data[DOMAIN].pop(entry.entry_id)
 
     return unload_ok
+
+
+class UptimeRobotDataUpdateCoordinator(DataUpdateCoordinator):
+    """Data update coordinator for Uptime Robot."""
+
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        config_entry_id: str,
+        dev_reg: DeviceRegistry,
+        api: UptimeRobot,
+    ) -> None:
+        """Initialize coordinator."""
+        super().__init__(
+            hass,
+            LOGGER,
+            name=DOMAIN,
+            update_method=self._async_update_data,
+            update_interval=COORDINATOR_UPDATE_INTERVAL,
+        )
+        self._config_entry_id = config_entry_id
+        self._device_registry = dev_reg
+        self._api = api
+
+    async def _async_update_data(self) -> list[UptimeRobotMonitor] | None:
+        """Update data."""
+        try:
+            response = await self._api.async_get_monitors()
+        except UptimeRobotException as exception:
+            raise UpdateFailed(exception) from exception
+        else:
+            if response.status != API_ATTR_OK:
+                raise UpdateFailed(response.error.message)
+
+        monitors: list[UptimeRobotMonitor] = response.data
+
+        current_monitors = {
+            list(device.identifiers)[0][1]
+            for device in self._device_registry.devices.values()
+            if self._config_entry_id in device.config_entries
+            and list(device.identifiers)[0][0] == DOMAIN
+        }
+        new_monitors = {str(monitor.id) for monitor in monitors}
+        if stale_monitors := current_monitors - new_monitors:
+            for monitor_id in stale_monitors:
+                if device := self._device_registry.async_get_device(
+                    {(DOMAIN, monitor_id)}
+                ):
+                    self._device_registry.async_remove_device(device.id)
+
+        # If there are new monitors, we should reload the config entry so we can
+        # create new devices and entities.
+        if self.data and new_monitors - {str(monitor.id) for monitor in self.data}:
+            self.hass.async_create_task(
+                self.hass.config_entries.async_reload(self._config_entry_id)
+            )
+            return None
+
+        return monitors