Initial Home Assistant commit

2026-06-05 22:34:31 -04:00
commit 6a58b10e6c
4494 changed files with 297833 additions and 0 deletions
@@ -0,0 +1,52 @@
+import logging
+from inspect import iscoroutinefunction
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_USERNAME, CONF_PASSWORD
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryAuthFailed
+from wyzeapy.exceptions import AccessTokenError, LoginError
+from wyzeapy.wyze_auth_lib import Token
+
+from .const import DOMAIN, ACCESS_TOKEN, REFRESH_TOKEN, REFRESH_TIME
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class TokenManager:
+    hass: HomeAssistant = None
+    config_entry: ConfigEntry = None
+
+    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry):
+        TokenManager.hass = hass
+        TokenManager.config_entry = config_entry
+
+    @staticmethod
+    async def token_callback(token: Token = None):
+        _LOGGER.debug("TokenManager: Received new token, updating config entry.")
+        if TokenManager.hass.config_entries.async_entries(DOMAIN):
+            for entry in TokenManager.hass.config_entries.async_entries(DOMAIN):
+                TokenManager.hass.config_entries.async_update_entry(
+                    entry,
+                    data={
+                        CONF_USERNAME: entry.data.get(CONF_USERNAME),
+                        CONF_PASSWORD: entry.data.get(CONF_PASSWORD),
+                        ACCESS_TOKEN: token.access_token,
+                        REFRESH_TOKEN: token.refresh_token,
+                        REFRESH_TIME: str(token.refresh_time),
+                    },
+                )
+
+
+def token_exception_handler(func):
+    async def inner_function(*args, **kwargs):
+        try:
+            if iscoroutinefunction(func):
+                await func(*args, **kwargs)
+            else:
+                func(*args, **kwargs)
+        except (AccessTokenError, LoginError) as err:
+            _LOGGER.error("TokenManager detected a login issue please re-login.")
+            raise ConfigEntryAuthFailed("Unable to login, please re-login.") from err
+
+    return inner_function