Add support for Nanoleaf push updates (#60708)

homeassistant/components/nanoleaf/__init__.py

@@ -1,17 +1,31 @@
 """The Nanoleaf integration."""
-from aionanoleaf import InvalidToken, Nanoleaf, Unavailable
+from __future__ import annotations
+
+import asyncio
+from dataclasses import dataclass
+
+from aionanoleaf import EffectsEvent, InvalidToken, Nanoleaf, StateEvent, Unavailable
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_HOST, CONF_TOKEN
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.dispatcher import async_dispatcher_send
 
 from .const import DOMAIN
 
 PLATFORMS = ["button", "light"]
 
 
+@dataclass
+class NanoleafEntryData:
+    """Class for sharing data within the Nanoleaf integration."""
+
+    device: Nanoleaf
+    event_listener: asyncio.Task
+
+
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up Nanoleaf from a config entry."""
     nanoleaf = Nanoleaf(
@@ -24,8 +38,29 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     except InvalidToken as err:
         raise ConfigEntryAuthFailed from err
 
-    hass.data.setdefault(DOMAIN, {})[entry.entry_id] = nanoleaf
+    async def _callback_update_light_state(event: StateEvent | EffectsEvent) -> None:
+        """Receive state and effect event."""
+        async_dispatcher_send(hass, f"{DOMAIN}_update_light_{nanoleaf.serial_no}")
+
+    event_listener = asyncio.create_task(
+        nanoleaf.listen_events(
+            state_callback=_callback_update_light_state,
+            effects_callback=_callback_update_light_state,
+        )
+    )
+
+    hass.data.setdefault(DOMAIN, {})[entry.entry_id] = NanoleafEntryData(
+        nanoleaf, event_listener
+    )
 
     hass.config_entries.async_setup_platforms(entry, PLATFORMS)
 
     return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    """Unload a config entry."""
+    await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
+    entry_data: NanoleafEntryData = hass.data[DOMAIN].pop(entry.entry_id)
+    entry_data.event_listener.cancel()
+    return True