Initial Commit

custom_components/asusrouter/__init__.py

@@ -0,0 +1,120 @@
+"""Support for AsusRouter devices."""
+
+from __future__ import annotations
+
+import logging
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import EVENT_HOMEASSISTANT_STOP
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import DeviceEntry
+
+from .const import ASUSROUTER, DOMAIN, PLATFORMS, STOP_LISTENER
+from .router import ARDevice
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+) -> bool:
+    """Set up AsusRouter platform."""
+
+    _LOGGER.debug("Setting up entry")
+
+    router = ARDevice(hass, config_entry)
+    await router.setup()
+
+    router.async_on_close(config_entry.add_update_listener(update_listener))
+
+    async def async_close_connection(event):
+        """Close router connection on HA stop."""
+
+        await router.close()
+
+    stop_listener = hass.bus.async_listen_once(
+        EVENT_HOMEASSISTANT_STOP, async_close_connection
+    )
+
+    hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = {
+        ASUSROUTER: router,
+        STOP_LISTENER: stop_listener,
+    }
+
+    await hass.config_entries.async_forward_entry_setups(
+        config_entry, PLATFORMS
+    )
+
+    return True
+
+
+async def async_unload_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+) -> bool:
+    """Unload AsusRouter config entry."""
+
+    _LOGGER.debug("Unloading entry")
+
+    unload = await hass.config_entries.async_unload_platforms(
+        config_entry, PLATFORMS
+    )
+
+    if unload:
+        # Close connection
+        hass.data[DOMAIN][config_entry.entry_id][STOP_LISTENER]()
+        await hass.data[DOMAIN][config_entry.entry_id][ASUSROUTER].close()
+        hass.data[DOMAIN].pop(config_entry.entry_id)
+
+    return unload
+
+
+async def update_listener(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+) -> None:
+    """Reload on config entry update."""
+
+    _LOGGER.debug("Update listener activated")
+
+    router = hass.data[DOMAIN][config_entry.entry_id][ASUSROUTER]
+
+    if router.update_options(config_entry.options):
+        await hass.config_entries.async_reload(config_entry.entry_id)
+
+
+# Example migration function
+async def async_migrate_entry(
+    hass: HomeAssistant, config_entry: ConfigEntry
+) -> bool:
+    """Migrate old entry."""
+
+    _LOGGER.debug("Migrating from version %s", config_entry.version)
+
+    if config_entry.version == 4:  # noqa: PLR2004
+        new_options = {**config_entry.options}
+        new_options["interval_network"] = new_options.pop(
+            "interval_network_stat", 30
+        )
+
+        config_entry.version = 5
+        hass.config_entries.async_update_entry(
+            config_entry, options=new_options
+        )
+
+    _LOGGER.debug("Migration to version %s successful", config_entry.version)
+
+    return True
+
+
+async def async_remove_config_entry_device(
+    hass: HomeAssistant, config_entry: ConfigEntry, device_entry: DeviceEntry
+) -> bool:
+    """Remove a device."""
+
+    # This would actually work and should not provide any issues
+
+    _LOGGER.debug("Removing device")
+
+    return True