database: raise exception on query failure

Raise an exception when querying the device key from the cloud fails.

Signed-off-by: Jordan Yates <jordan@embeint.com>

Author: JordanYates

src/infuse_iot/database.py

@@ -56,6 +56,18 @@ class DeviceKeyChangedError(KeyError):
     """Device key for the requested device has changed"""
 
 
+class DeviceKeyQueryFailed(Exception):
+    """Querying a device key from the cloud failed"""
+
+    def __init__(self, infuse_id: int, status_code: int, content: str):
+        self.infuse_id = infuse_id
+        self.status_code = status_code
+        self.content = content
+
+    def __str__(self):
+        return f"{self.infuse_id:016x}: <{self.status_code}> {self.content}"
+
+
 class DeviceDatabase:
     """Database of current device state"""
 
@@ -208,9 +220,11 @@ def observe_security_state(
                 base64.b64encode(challenge_resp).decode("utf-8"),
             )
             body = GetDeviceSharedSecretBody(f"{infuse_id:016x}", security_state)
-            response = get_device_shared_secret.sync(client=client, body=body)
-            if response is not None:
-                key = base64.b64decode(response.key)
+            response = get_device_shared_secret.sync_detailed(client=client, body=body)
+            if response.parsed is None:
+                raise DeviceKeyQueryFailed(infuse_id, response.status_code, response.content.decode("utf-8"))
+            else:
+                key = base64.b64decode(response.parsed.key)
                 self.devices[infuse_id].shared_key = key
                 self._update_cache(infuse_id, device_pub_key, key)
 

src/infuse_iot/tools/gateway.py

@@ -26,6 +26,7 @@
 from infuse_iot.common import InfuseID, InfuseType
 from infuse_iot.database import (
     DeviceDatabase,
+    DeviceKeyQueryFailed,
     NoKeyError,
 )
 from infuse_iot.epacket.packet import (
@@ -80,15 +81,19 @@ def query_device_key(self, infuse_id: int, cb_event: threading.Event | None = No
 
         def security_state_done(pkt: PacketReceived, _rc: int, response: bytes, challenge):
             decoded = defs.security_state.response.vla_from_buffer_copy(response)
-            self.ddb.observe_security_state(
-                infuse_id,
-                bytes(decoded.cloud_public_key),
-                bytes(decoded.device_public_key),
-                decoded.network_id,
-                challenge,
-                decoded.challenge_response_type,
-                bytes(decoded.challenge_response),
-            )
+            try:
+                self.ddb.observe_security_state(
+                    infuse_id,
+                    bytes(decoded.cloud_public_key),
+                    bytes(decoded.device_public_key),
+                    decoded.network_id,
+                    challenge,
+                    decoded.challenge_response_type,
+                    bytes(decoded.challenge_response),
+                )
+            except DeviceKeyQueryFailed as e:
+                Console.log_error(f"Failed to query key for {e.infuse_id:016x}: {e.content}")
+                return
             if cb_event is not None:
                 cb_event.set()