From 8694882c4599a4d2778e566ce20aacb4802d65e4 Mon Sep 17 00:00:00 2001
From: Andre Basche <andre.basche@gmail.com>
Date: Sat, 10 Feb 2024 01:02:26 +0100
Subject: [PATCH] Set refresh token and mobile id

---
 custom_components/hon/__init__.py | 9 ++++++---
 custom_components/hon/const.py    | 2 ++
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/custom_components/hon/__init__.py b/custom_components/hon/__init__.py
index a970fc1..24b97a3 100644
--- a/custom_components/hon/__init__.py
+++ b/custom_components/hon/__init__.py
@@ -8,7 +8,7 @@ from homeassistant.helpers import config_validation as cv, aiohttp_client
 from homeassistant.helpers.typing import HomeAssistantType
 from pyhon import Hon
 
-from .const import DOMAIN, PLATFORMS
+from .const import DOMAIN, PLATFORMS, MOBILE_ID, CONF_REFRESH_TOKEN
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -30,9 +30,11 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool
     if (config_dir := hass.config.config_dir) is None:
         raise ValueError("Missing Config Dir")
     hon = await Hon(
-        entry.data["email"],
-        entry.data["password"],
+        entry.data[CONF_EMAIL],
+        entry.data[CONF_PASSWORD],
+        mobile_id=MOBILE_ID,
         session=session,
+        refresh_token=entry.data[CONF_REFRESH_TOKEN],
         test_data_path=Path(config_dir),
     ).create()
     hass.data.setdefault(DOMAIN, {})
@@ -47,6 +49,7 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool
 
 
 async def async_unload_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool:
+    entry.data[CONF_REFRESH_TOKEN] = hass.data[DOMAIN][entry.unique_id].api.auth.refresh_token
     unload = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
     if unload:
         if not hass.data[DOMAIN]:
diff --git a/custom_components/hon/const.py b/custom_components/hon/const.py
index 5fbb792..37ba055 100644
--- a/custom_components/hon/const.py
+++ b/custom_components/hon/const.py
@@ -8,6 +8,8 @@ from homeassistant.components.climate import (
 
 DOMAIN: str = "hon"
 UPDATE_INTERVAL: int = 60
+MOBILE_ID: str = "homassistant"
+CONF_REFRESH_TOKEN = "refresh_token"
 
 PLATFORMS: list[str] = [
     "sensor",