init

custom_components/govee/__init__.py

@@ -0,0 +1,114 @@
+"""The Govee integration."""
+import asyncio
+import logging
+
+from govee_api_laggat import Govee
+import voluptuous as vol
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_API_KEY
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import PlatformNotReady
+
+from .const import DOMAIN
+from .learning_storage import GoveeLearningStorage
+
+_LOGGER = logging.getLogger(__name__)
+CONFIG_SCHEMA = vol.Schema({DOMAIN: vol.Schema({})}, extra=vol.ALLOW_EXTRA)
+
+# supported platforms
+PLATFORMS = ["light"]
+
+
+def setup(hass, config):
+    """This setup does nothing, we use the async setup."""
+    hass.states.set("govee.state", "setup called")
+    return True
+
+
+async def async_setup(hass: HomeAssistant, config: dict):
+    """Set up the Govee component."""
+    hass.states.async_set("govee.state", "async_setup called")
+    hass.data[DOMAIN] = {}
+    return True
+
+
+def is_online(online: bool):
+    """Log online/offline change."""
+    msg = "API is offline."
+    if online:
+        msg = "API is back online."
+    _LOGGER.warning(msg)
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
+    """Set up Govee from a config entry."""
+
+    # get vars from ConfigFlow/OptionsFlow
+    config = entry.data
+    options = entry.options
+    api_key = options.get(CONF_API_KEY, config.get(CONF_API_KEY, ""))
+
+    # Setup connection with devices/cloud
+    hub = await Govee.create(
+        api_key, learning_storage=GoveeLearningStorage(hass.config.config_dir)
+    )
+    # keep reference for disposing
+    hass.data[DOMAIN] = {}
+    hass.data[DOMAIN]["hub"] = hub
+
+    # inform when api is offline/online
+    hub.events.online += is_online
+
+    # Verify that passed in configuration works
+    _, err = await hub.get_devices()
+    if err:
+        _LOGGER.warning("Could not connect to Govee API: %s", err)
+        await hub.rate_limit_delay()
+        await async_unload_entry(hass, entry)
+        raise PlatformNotReady()
+
+    for component in PLATFORMS:
+        hass.async_create_task(
+            hass.config_entries.async_forward_entry_setup(entry, component)
+        )
+
+    return True
+
+
+async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
+    """Unload a config entry."""
+
+    unload_ok = all(
+        await asyncio.gather(
+            *[
+                _unload_component_entry(hass, entry, component)
+                for component in PLATFORMS
+            ]
+        )
+    )
+
+    if unload_ok:
+        hub = hass.data[DOMAIN].pop("hub")
+        await hub.close()
+
+    return unload_ok
+
+
+def _unload_component_entry(
+    hass: HomeAssistant, entry: ConfigEntry, component: str
+) -> bool:
+    """Unload an entry for a specific component."""
+    success = False
+    try:
+        success = hass.config_entries.async_forward_entry_unload(entry, component)
+    except ValueError:
+        # probably ValueError: Config entry was never loaded!
+        return success
+    except Exception as ex:
+        _LOGGER.warning(
+            "Continuing on exception when unloading %s component's entry: %s",
+            component,
+            ex,
+        )
+        return success