working 2.0 clone

custom_components/remote_homeassistant/rest_api.py

@@ -0,0 +1,59 @@
+"""Simple implementation to call Home Assistant REST API."""
+
+from homeassistant import exceptions
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+
+API_URL = "{proto}://{host}:{port}/api/remote_homeassistant/discovery"
+
+
+class ApiProblem(exceptions.HomeAssistantError):
+    """Error to indicate problem reaching API."""
+
+
+class CannotConnect(exceptions.HomeAssistantError):
+    """Error to indicate we cannot connect."""
+
+
+class InvalidAuth(exceptions.HomeAssistantError):
+    """Error to indicate there is invalid auth."""
+
+
+class BadResponse(exceptions.HomeAssistantError):
+    """Error to indicate a bad response was received."""
+
+
+class UnsupportedVersion(exceptions.HomeAssistantError):
+    """Error to indicate an unsupported version of Home Assistant."""
+
+
+class EndpointMissing(exceptions.HomeAssistantError):
+    """Error to indicate there is invalid auth."""
+
+
+async def async_get_discovery_info(hass, host, port, secure, access_token, verify_ssl):
+    """Get discovery information from server."""
+    url = API_URL.format(
+        proto="https" if secure else "http",
+        host=host,
+        port=port,
+    )
+    headers = {
+        "Authorization": "Bearer " + access_token,
+        "Content-Type": "application/json",
+    }
+    session = async_get_clientsession(hass, verify_ssl)
+
+    # Fetch discovery info location for name and unique UUID
+    async with session.get(url, headers=headers) as resp:
+        if resp.status == 404:
+            raise EndpointMissing()
+        if 400 <= resp.status < 500:
+            raise InvalidAuth()
+        if resp.status != 200:
+            raise ApiProblem()
+        json = await resp.json()
+        if not isinstance(json, dict):
+            raise BadResponse(f"Bad response data: {json}")
+        if "uuid" not in json:
+            raise UnsupportedVersion()
+        return json