Initial Commit

2026-06-11 11:50:50 -04:00
commit d4a69c41be
2748 changed files with 80489 additions and 0 deletions
@@ -0,0 +1,175 @@
+"""Config flow to configure the Hilo component."""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from awesomeversion import AwesomeVersion
+from homeassistant import config_entries
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_SCAN_INTERVAL, __version__ as HAVERSION
+from homeassistant.core import callback
+from homeassistant.data_entry_flow import FlowResult
+from homeassistant.helpers import config_validation as cv, selector
+from homeassistant.helpers.config_entry_oauth2_flow import AbstractOAuth2FlowHandler
+import jwt
+import voluptuous as vol
+
+from .const import (
+    CONF_APPRECIATION_PHASE,
+    CONF_CHALLENGE_LOCK,
+    CONF_GENERATE_ENERGY_METERS,
+    CONF_HQ_PLAN_NAME,
+    CONF_LOG_TRACES,
+    CONF_PRE_COLD_PHASE,
+    CONF_TARIFF,
+    CONF_TRACK_UNKNOWN_SOURCES,
+    CONF_UNTARIFICATED_DEVICES,
+    DEFAULT_APPRECIATION_PHASE,
+    DEFAULT_CHALLENGE_LOCK,
+    DEFAULT_GENERATE_ENERGY_METERS,
+    DEFAULT_HQ_PLAN_NAME,
+    DEFAULT_LOG_TRACES,
+    DEFAULT_PRE_COLD_PHASE,
+    DEFAULT_TRACK_UNKNOWN_SOURCES,
+    DEFAULT_UNTARIFICATED_DEVICES,
+    DOMAIN,
+    LOG,
+    MIN_SCAN_INTERVAL,
+)
+from .oauth2 import AuthCodeWithPKCEImplementation
+
+STEP_OPTION_SCHEMA = vol.Schema(
+    {
+        vol.Optional(
+            CONF_GENERATE_ENERGY_METERS, default=DEFAULT_GENERATE_ENERGY_METERS
+        ): cv.boolean,
+        vol.Optional(
+            CONF_UNTARIFICATED_DEVICES,
+            default=DEFAULT_UNTARIFICATED_DEVICES,
+        ): cv.boolean,
+        vol.Optional(
+            CONF_LOG_TRACES,
+            default=DEFAULT_LOG_TRACES,
+        ): cv.boolean,
+        vol.Optional(
+            CONF_CHALLENGE_LOCK,
+            default=DEFAULT_CHALLENGE_LOCK,
+        ): cv.boolean,
+        vol.Optional(
+            CONF_TRACK_UNKNOWN_SOURCES,
+            default=DEFAULT_TRACK_UNKNOWN_SOURCES,
+        ): cv.boolean,
+        vol.Optional(
+            CONF_HQ_PLAN_NAME, default=DEFAULT_HQ_PLAN_NAME
+        ): selector.SelectSelector(
+            selector.SelectSelectorConfig(options=list(CONF_TARIFF.keys()), mode="list")
+        ),
+        vol.Optional(
+            CONF_APPRECIATION_PHASE,
+            default=DEFAULT_APPRECIATION_PHASE,
+        ): cv.positive_int,
+        vol.Optional(
+            CONF_PRE_COLD_PHASE,
+            default=DEFAULT_PRE_COLD_PHASE,
+        ): cv.positive_int,
+        vol.Optional(CONF_SCAN_INTERVAL): (
+            vol.All(cv.positive_int, vol.Range(min=MIN_SCAN_INTERVAL))
+        ),
+    }
+)
+
+
+class HiloFlowHandler(AbstractOAuth2FlowHandler, domain=DOMAIN):
+    """Handle a Hilo config flow."""
+
+    DOMAIN = DOMAIN
+    VERSION = 2
+
+    _reauth_entry: ConfigEntry | None = None
+
+    async def async_step_user(self, user_input=None) -> FlowResult:
+        """Handle a flow initialized by the user."""
+        await self.async_set_unique_id(DOMAIN)
+
+        if self._async_current_entries():
+            return self.async_abort(reason="single_instance_allowed")
+
+        self.async_register_implementation(
+            self.hass,
+            AuthCodeWithPKCEImplementation(self.hass),
+        )
+
+        return await super().async_step_user(user_input)
+
+    @property
+    def logger(self) -> logging.Logger:
+        """Return logger."""
+        return LOG
+
+    @staticmethod
+    @callback
+    def async_get_options_flow(
+        config_entry: ConfigEntry,
+    ) -> HiloOptionsFlowHandler:
+        """Define the config flow to handle options."""
+        return HiloOptionsFlowHandler(config_entry)
+
+    async def async_step_reauth(self, user_input=None) -> FlowResult:
+        """Perform reauth upon an API authentication error."""
+        LOG.debug("async_step_reauth")
+        self._reauth_entry = self.hass.config_entries.async_get_entry(
+            self.context["entry_id"]
+        )
+        return await self.async_step_reauth_confirm()
+
+    async def async_step_reauth_confirm(self, user_input=None) -> FlowResult:
+        """Dialog that informs the user that reauth is required."""
+        if user_input is None:
+            return self.async_show_form(
+                step_id="reauth_confirm",
+                data_schema=vol.Schema({}),
+            )
+        user_input["implementation"] = DOMAIN
+        return await super().async_step_user(user_input)
+
+    async def async_oauth_create_entry(self, data: dict) -> FlowResult:
+        """Create an oauth config entry or update existing entry for reauth."""
+        if self._reauth_entry:
+            self.hass.config_entries.async_update_entry(self._reauth_entry, data=data)
+            await self.hass.config_entries.async_reload(self._reauth_entry.entry_id)
+            return self.async_abort(reason="reauth_successful")
+
+        LOG.debug("Creating entry: %s", data)
+
+        token = data["token"]["access_token"]
+        decoded_token = jwt.decode(token, options={"verify_signature": False})
+        email = decoded_token["email"]
+
+        return self.async_create_entry(title=email, data=data)
+
+
+class HiloOptionsFlowHandler(config_entries.OptionsFlow):
+    """Handle a Hilo options flow."""
+
+    def __init__(self, config_entry: ConfigEntry) -> None:
+        """Initialize."""
+        if AwesomeVersion(HAVERSION) < "2024.11.99":
+            self.config_entry = config_entry
+        else:
+            self._config_entry = config_entry
+
+    async def async_step_init(
+        self, user_input: dict[str, Any] | None = None
+    ) -> FlowResult:
+        """Manage the options."""
+        if user_input is not None:
+            return self.async_create_entry(title="", data=user_input)
+
+        return self.async_show_form(
+            step_id="init",
+            data_schema=self.add_suggested_values_to_schema(
+                STEP_OPTION_SCHEMA, self.config_entry.options
+            ),
+        )