Use connection handler

pyhon/connection/api.py

@@ -0,0 +1,166 @@
+import asyncio
+import json
+import logging
+from datetime import datetime
+from typing import List
+
+from pyhon import const
+from pyhon.appliance import HonAppliance
+from pyhon.connection.connection import HonConnectionHandler, HonAnonymousConnectionHandler
+
+_LOGGER = logging.getLogger()
+
+
+class HonAPI:
+    def __init__(self, email="", password="") -> None:
+        super().__init__()
+        self._email = email
+        self._password = password
+        self._devices = []
+        self._hon = None
+        self._hon_anonymous = HonAnonymousConnectionHandler()
+
+    async def __aenter__(self):
+        self._hon = HonConnectionHandler(self._email, self._password)
+        await self._hon.create()
+        await self.setup()
+
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self._hon.close()
+
+    @property
+    def devices(self) -> List[HonAppliance]:
+        return self._devices
+
+    async def setup(self):
+        async with self._hon.get(f"{const.API_URL}/commands/v1/appliance") as resp:
+            try:
+                appliances = (await resp.json())["payload"]["appliances"]
+                for appliance in appliances:
+                    device = HonAppliance(self, appliance)
+                    if device.mac_address is None:
+                        continue
+                    await asyncio.gather(*[
+                        device.load_attributes(),
+                        device.load_commands(),
+                        device.load_statistics()])
+                    self._devices.append(device)
+            except json.JSONDecodeError:
+                _LOGGER.error("No JSON Data after GET: %s", await resp.text())
+                return False
+        return True
+
+    async def load_commands(self, device: HonAppliance):
+        params = {
+            "applianceType": device.appliance_type,
+            "code": device.appliance["code"],
+            "applianceModelId": device.appliance_model_id,
+            "firmwareId": device.appliance["eepromId"],
+            "macAddress": device.mac_address,
+            "fwVersion": device.appliance["fwVersion"],
+            "os": const.OS,
+            "appVersion": const.APP_VERSION,
+            "series": device.appliance["series"],
+        }
+        url = f"{const.API_URL}/commands/v1/retrieve"
+        async with self._hon.get(url, params=params) as response:
+            result = (await response.json()).get("payload", {})
+            if not result or result.pop("resultCode") != "0":
+                return {}
+            return result
+
+    async def command_history(self, device: HonAppliance):
+        url = f"{const.API_URL}/commands/v1/appliance/{device.mac_address}/history"
+        async with self._hon.get(url) as response:
+            result = await response.json()
+            if not result or not result.get("payload"):
+                return {}
+            return result["payload"]["history"]
+
+    async def last_activity(self, device: HonAppliance):
+        url = f"{const.API_URL}/commands/v1/retrieve-last-activity"
+        params = {"macAddress": device.mac_address}
+        async with self._hon.get(url, params=params) as response:
+            result = await response.json()
+            if result and (activity := result.get("attributes")):
+                return activity
+        return {}
+
+    async def load_attributes(self, device: HonAppliance):
+        params = {
+            "macAddress": device.mac_address,
+            "applianceType": device.appliance_type,
+            "category": "CYCLE"
+        }
+        url = f"{const.API_URL}/commands/v1/context"
+        async with self._hon.get(url, params=params) as response:
+            return (await response.json()).get("payload", {})
+
+    async def load_statistics(self, device: HonAppliance):
+        params = {
+            "macAddress": device.mac_address,
+            "applianceType": device.appliance_type
+        }
+        url = f"{const.API_URL}/commands/v1/statistics"
+        async with self._hon.get(url, params=params) as response:
+            return (await response.json()).get("payload", {})
+
+    async def send_command(self, device, command, parameters, ancillary_parameters):
+        now = datetime.utcnow().isoformat()
+        data = {
+            "macAddress": device.mac_address,
+            "timestamp": f"{now[:-3]}Z",
+            "commandName": command,
+            "transactionId": f"{device.mac_address}_{now[:-3]}Z",
+            "applianceOptions": device.commands_options,
+            "device": self._hon.device.get(),
+            "attributes": {
+                "channel": "mobileApp",
+                "origin": "standardProgram",
+                "energyLabel": "0"
+            },
+            "ancillaryParameters": ancillary_parameters,
+            "parameters": parameters,
+            "applianceType": device.appliance_type
+        }
+        url = f"{const.API_URL}/commands/v1/send"
+        async with self._hon.post(url, json=data) as resp:
+            try:
+                json_data = await resp.json()
+            except json.JSONDecodeError:
+                return False
+            if json_data["payload"]["resultCode"] == "0":
+                return True
+        return False
+
+    async def appliance_configuration(self):
+        url = f"{const.API_URL}/config/v1/appliance-configuration"
+        async with self._hon_anonymous.get(url) as response:
+            result = await response.json()
+            if result and (data := result.get("payload")):
+                return data
+        return {}
+
+    async def app_config(self, language="en", beta=True):
+        url = f"{const.API_URL}/app-config"
+        payload = {
+            "languageCode": language,
+            "beta": beta,
+            "appVersion": const.APP_VERSION,
+            "os": const.OS
+        }
+        payload = json.dumps(payload, separators=(',', ':'))
+        async with self._hon_anonymous.post(url, data=payload) as response:
+            if (result := await response.json()) and (data := result.get("payload")):
+                return data
+        return {}
+
+    async def translation_keys(self, language="en"):
+        config = await self.app_config(language=language)
+        if url := config.get("language", {}).get("jsonPath"):
+            async with self._hon_anonymous.get(url) as response:
+                if result := await response.json():
+                    return result
+        return {}

pyhon/connection/auth.py

@@ -0,0 +1,158 @@
+import json
+import logging
+import re
+import secrets
+import urllib
+from pprint import pprint
+from urllib import parse
+
+from yarl import URL
+
+from pyhon import const
+
+_LOGGER = logging.getLogger()
+
+
+class HonAuth:
+    def __init__(self, session, email, password, device) -> None:
+        self._session = session
+        self._email = email
+        self._password = password
+        self._access_token = ""
+        self._refresh_token = ""
+        self._cognito_token = ""
+        self._id_token = ""
+        self._device = device
+
+    @property
+    def cognito_token(self):
+        return self._cognito_token
+
+    @property
+    def id_token(self):
+        return self._id_token
+
+    @property
+    def access_token(self):
+        return self._access_token
+
+    @property
+    def refresh_token(self):
+        return self._refresh_token
+
+    async def _load_login(self):
+        nonce = secrets.token_hex(16)
+        nonce = f"{nonce[:8]}-{nonce[8:12]}-{nonce[12:16]}-{nonce[16:20]}-{nonce[20:]}"
+        params = {
+            "response_type": "token+id_token",
+            "client_id": const.CLIENT_ID,
+            "redirect_uri": urllib.parse.quote(f"{const.APP}://mobilesdk/detect/oauth/done"),
+            "display": "touch",
+            "scope": "api openid refresh_token web",
+            "nonce": nonce
+        }
+        params = "&".join([f"{k}={v}" for k, v in params.items()])
+        async with self._session.get(f"{const.AUTH_API}/services/oauth2/authorize/expid_Login?{params}") as resp:
+            if not (login_url := re.findall("url = '(.+?)'", await resp.text())):
+                return False
+        async with self._session.get(login_url[0], allow_redirects=False) as redirect1:
+            if not (url := redirect1.headers.get("Location")):
+                return False
+        async with self._session.get(url, allow_redirects=False) as redirect2:
+            if not (url := redirect2.headers.get("Location") + "&System=IoT_Mobile_App&RegistrationSubChannel=hOn"):
+                return False
+        async with self._session.get(URL(url, encoded=True)) as login_screen:
+            if context := re.findall('"fwuid":"(.*?)","loaded":(\\{.*?})', await login_screen.text()):
+                fw_uid, loaded_str = context[0]
+                loaded = json.loads(loaded_str)
+                login_url = login_url[0].replace("/".join(const.AUTH_API.split("/")[:-1]), "")
+                return fw_uid, loaded, login_url
+        return False
+
+    async def _login(self, fw_uid, loaded, login_url):
+        data = {
+            "message": {
+                "actions": [
+                    {
+                        "id": "79;a",
+                        "descriptor": "apex://LightningLoginCustomController/ACTION$login",
+                        "callingDescriptor": "markup://c:loginForm",
+                        "params": {
+                            "username": self._email,
+                            "password": self._password,
+                            "startUrl": parse.unquote(login_url.split("startURL=")[-1]).split("%3D")[0]
+                        }
+                    }
+                ]
+            },
+            "aura.context": {
+                "mode": "PROD",
+                "fwuid": fw_uid,
+                "app": "siteforce:loginApp2",
+                "loaded": loaded,
+                "dn": [],
+                "globals": {},
+                "uad": False},
+            "aura.pageURI": login_url,
+            "aura.token": None}
+
+        params = {"r": 3, "other.LightningLoginCustom.login": 1}
+        async with self._session.post(
+                const.AUTH_API + "/s/sfsites/aura",
+                headers={"Content-Type": "application/x-www-form-urlencoded"},
+                data="&".join(f"{k}={json.dumps(v)}" for k, v in data.items()),
+                params=params
+        ) as response:
+            if response.status == 200:
+                try:
+                    return (await response.json())["events"][0]["attributes"]["values"]["url"]
+                except json.JSONDecodeError:
+                    pass
+            _LOGGER.error("Unable to login: %s\n%s", response.status, await response.text())
+            return ""
+
+    async def _get_token(self, url):
+        async with self._session.get(url) as resp:
+            if resp.status != 200:
+                _LOGGER.error("Unable to get token: %s", resp.status)
+                return False
+            url = re.findall("href\\s*=\\s*[\"'](.*?)[\"']", await resp.text())
+        async with self._session.get(url[0]) as resp:
+            if resp.status != 200:
+                _LOGGER.error("Unable to get token: %s", resp.status)
+                return False
+            url = re.findall("href\\s*=\\s*[\"'](.*?)[\"']", await resp.text())
+            url = "/".join(const.AUTH_API.split("/")[:-1]) + url[0]
+        async with self._session.get(url) as resp:
+            if resp.status != 200:
+                _LOGGER.error("Unable to connect to the login service: %s", resp.status)
+                return False
+            text = await resp.text()
+        if access_token := re.findall("access_token=(.*?)&", text):
+            self._access_token = access_token[0]
+        if refresh_token := re.findall("refresh_token=(.*?)&", text):
+            self._refresh_token = refresh_token[0]
+        if id_token := re.findall("id_token=(.*?)&", text):
+            self._id_token = id_token[0]
+        return True
+
+    async def authorize(self):
+        if login_site := await self._load_login():
+            fw_uid, loaded, login_url = login_site
+        else:
+            return False
+        if not (url := await self._login(fw_uid, loaded, login_url)):
+            return False
+        if not await self._get_token(url):
+            return False
+
+        post_headers = {"Content-Type": "application/json", "id-token": self._id_token}
+        data = self._device.get()
+        async with self._session.post(f"{const.API_URL}/auth/v1/login", headers=post_headers, json=data) as resp:
+            try:
+                json_data = await resp.json()
+            except json.JSONDecodeError:
+                _LOGGER.error("No JSON Data after POST: %s", await resp.text())
+                return False
+            self._cognito_token = json_data["cognitoUser"]["Token"]
+        return True

pyhon/connection/connection.py

@@ -0,0 +1,100 @@
+from contextlib import asynccontextmanager
+
+import aiohttp
+
+from pyhon import const
+from pyhon.connection.auth import HonAuth, _LOGGER
+from pyhon.connection.device import HonDevice
+
+
+class HonBaseConnectionHandler:
+    _HEADERS = {"user-agent": const.USER_AGENT, "Content-Type": "application/json"}
+
+    def __init__(self):
+        self._session = None
+        self._auth = None
+
+    async def __aenter__(self):
+        await self.create()
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.close()
+
+    async def create(self):
+        self._session = aiohttp.ClientSession(headers=self._HEADERS)
+
+    @asynccontextmanager
+    async def get(self, *args, **kwargs):
+        raise NotImplemented
+
+    @asynccontextmanager
+    async def post(self, *args, **kwargs):
+        raise NotImplemented
+
+    async def close(self):
+        await self._session.close()
+
+
+class HonConnectionHandler(HonBaseConnectionHandler):
+    def __init__(self, email, password):
+        super().__init__()
+        self._device = HonDevice()
+        self._email = email
+        self._password = password
+        if not self._email:
+            raise PermissionError("Login-Error - An email address must be specified")
+        if not self._password:
+            raise PermissionError("Login-Error - A password address must be specified")
+        self._request_headers = {}
+
+    @property
+    def device(self):
+        return self._device
+
+    async def create(self):
+        await super().create()
+        self._auth = HonAuth(self._session, self._email, self._password, self._device)
+
+    async def _check_headers(self, headers):
+        if "cognito-token" not in self._request_headers or "id-token" not in self._request_headers:
+            if await self._auth.authorize():
+                self._request_headers["cognito-token"] = self._auth.cognito_token
+                self._request_headers["id-token"] = self._auth.id_token
+            else:
+                raise PermissionError("Can't Login")
+        return {h: v for h, v in self._request_headers.items() if h not in headers}
+
+    @asynccontextmanager
+    async def get(self, *args, loop=0, **kwargs):
+        kwargs["headers"] = await self._check_headers(kwargs.get("headers", {}))
+        async with self._session.get(*args, **kwargs) as response:
+            if response.status == 403 and not loop:
+                _LOGGER.warning("%s - Error %s - %s", response.request_info.url, response.status, await response.text())
+                await self.create()
+                yield await self.get(*args, loop=loop + 1, **kwargs)
+            elif loop >= 2:
+                _LOGGER.error("%s - Error %s - %s", response.request_info.url, response.status, await response.text())
+                raise PermissionError()
+            else:
+                yield response
+
+    @asynccontextmanager
+    async def post(self, *args, **kwargs):
+        kwargs["headers"] = await self._check_headers(kwargs.get("headers", {}))
+        async with self._session.post(*args, **kwargs) as response:
+            yield response
+
+
+class HonAnonymousConnectionHandler(HonBaseConnectionHandler):
+    _HEADERS = HonBaseConnectionHandler._HEADERS | {"x-api-key": const.API_KEY}
+
+    @asynccontextmanager
+    async def get(self, *args, **kwargs):
+        async with self._session.post(*args, **kwargs) as response:
+            yield response
+
+    @asynccontextmanager
+    async def post(self, *args, **kwargs):
+        async with self._session.post(*args, **kwargs) as response:
+            yield response

pyhon/connection/device.py

@@ -0,0 +1,36 @@
+import secrets
+
+from pyhon import const
+
+
+class HonDevice:
+    def __init__(self):
+        self._app_version = const.APP_VERSION
+        self._os_version = const.OS_VERSION
+        self._os = const.OS
+        self._device_model = const.DEVICE_MODEL
+        self._mobile_id = secrets.token_hex(8)
+
+    @property
+    def app_version(self):
+        return self._app_version
+
+    @property
+    def os_version(self):
+        return self._os_version
+
+    @property
+    def os(self):
+        return self._os
+
+    @property
+    def device_model(self):
+        return self._device_model
+
+    @property
+    def mobile_id(self):
+        return self._mobile_id
+
+    def get(self):
+        return {"appVersion": self.app_version, "mobileId": self.mobile_id, "osVersion": self.os_version,
+                "os": self.os, "deviceModel": self.device_model}