Add domain handling

elro/auth.py

@@ -1,18 +1,26 @@
 """Elro Connects cloud authentication."""
 
 from __future__ import annotations
+import asyncio
 from datetime import datetime
-
 from json import dumps
 from dataclasses import dataclass
 from typing import TypedDict
 
+
+import socket
+import json
+import logging
 import aiohttp
 
+
 CLIENT_TYPE = "APP"
 PID = "01288154146"
-URL_LOGIN = "https://uaa-openapi.hekreu.me/login"
-URL_DEVICE_LIST = "https://user-openapi.hekreu.me/device"
+BASE_UAA_URL = "https://uaa-openapi."
+BASE_USER_URL = "https://user-openapi."
+DEFAULT_DOMAIN = "hekr.me"
+
+_LOGGER = logging.getLogger(__name__)
 
 
 @dataclass
@@ -48,6 +56,7 @@ class ElroConnectsSession:
     def __init__(self) -> None:
         """Initialize."""
         self._session_cache = None
+        self._domain = None
 
     async def async_login(
         self, username: str, password: str
@@ -62,9 +71,13 @@ async def async_login(
         headers = {
             "User-Agent": "lib-elro-connects",
         }
-
+        domain = await self._async_get_domain()
         async with aiohttp.ClientSession(json_serialize=dumps) as session:
-            async with session.post(URL_LOGIN, json=body, headers=headers) as resp:
+            async with session.post(
+                BASE_UAA_URL + domain + "/login",
+                json=body,
+                headers=headers,
+            ) as resp:
                 response = await resp.json()
 
         response["last_login"] = datetime.now()
@@ -76,6 +89,48 @@ def session(self) -> ElroConnectsCloudSessionCache | None:
         """Return the current session."""
         return self._session_cache
 
+    async def _async_get_domain(self) -> str:
+        """Return the API main domain name address."""
+
+        def _get_domain() -> str | None:
+            # domain has not been determined
+            sckt = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            host = "info.hekr.me"
+            port = 91
+            sckt.connect((host, port))
+            message = '{"action":"getAppDomain"}\n'
+            sckt.send(message.encode())
+            msg = sckt.recv(1024)
+
+            try:
+                parsed_data = json.loads(msg)
+                _LOGGER.debug(
+                    "%s result code: %s description: %s",
+                    host,
+                    parsed_data["code"],
+                    parsed_data["desc"],
+                )
+
+                if "dcInfo" in parsed_data:
+                    self._domain = parsed_data["dcInfo"]["domain"]
+            except json.decoder.JSONDecodeError:
+                return DEFAULT_DOMAIN
+
+            if not self._domain.startswith("hekr"):  # default
+                self._domain = None
+                return DEFAULT_DOMAIN
+
+            return self._domain
+
+        # Only run once
+        if self._domain:
+            return self._domain
+
+        loop = asyncio.get_event_loop()
+        if domain := await loop.run_in_executor(None, _get_domain):
+            self._domain = domain
+        return domain
+
     async def async_get_connectors(self) -> list[ElroConnectsConnector]:
         """Return as list of registered connectors."""
 
@@ -88,8 +143,12 @@ async def async_get_connectors(self) -> list[ElroConnectsConnector]:
             "Authorization": f"Bearer {self._session_cache['access_token']}",
         }
 
+        domain = await self._async_get_domain()
         async with aiohttp.ClientSession(json_serialize=dumps) as session:
-            async with session.get(URL_DEVICE_LIST, headers=headers) as resp:
+            async with session.get(
+                BASE_USER_URL + domain + "/device",
+                headers=headers,
+            ) as resp:
                 response = await resp.json()
         connector_list = [
             ElroConnectsConnector(