fix: handle auth expiring

Author: Lash-L

roborock/cloud_api.py

@@ -8,10 +8,11 @@
 from typing import Any
 
 import paho.mqtt.client as mqtt
+from paho.mqtt.reasoncodes import ReasonCodes
 
 from .api import KEEPALIVE, RoborockClient
 from .containers import DeviceData, UserData
-from .exceptions import RoborockException, VacuumError
+from .exceptions import RoborockException, RoborockInvalidUserData, VacuumError
 from .protocol import (
     Decoder,
     Encoder,
@@ -78,21 +79,27 @@ def __init__(self, user_data: UserData, device_info: DeviceData) -> None:
         self._encoder: Encoder = create_mqtt_encoder(device_info.device.local_key)
 
     def _mqtt_on_connect(self, *args, **kwargs):
-        _, __, ___, rc, ____ = args
+        rc: ReasonCodes = args[3]
         connection_queue = self._waiting_queue.get(CONNECT_REQUEST_ID)
-        if rc != mqtt.MQTT_ERR_SUCCESS:
-            message = f"Failed to connect ({mqtt.error_string(rc)})"
+        if rc != 0:
+            message = f"Failed to connect ({str(rc)})"
             self._logger.error(message)
             if connection_queue:
-                connection_queue.set_exception(VacuumError(message))
+                # These are the ReasonCodes relating to authorization issues.
+                if rc.value in {24, 25, 133, 134, 135, 144}:
+                    connection_queue.set_exception(
+                        RoborockInvalidUserData("Failed to connect to mqtt. Invalid user data. Re-auth is needed.")
+                    )
+                else:
+                    connection_queue.set_exception(VacuumError(message))
             else:
                 self._logger.debug("Failed to notify connect future, not in queue")
             return
         self._logger.info(f"Connected to mqtt {self._mqtt_host}:{self._mqtt_port}")
         topic = f"rr/m/o/{self._mqtt_user}/{self._hashed_user}/{self.device_info.device.duid}"
         (result, mid) = self._mqtt_client.subscribe(topic)
         if result != 0:
-            message = f"Failed to subscribe ({mqtt.error_string(rc)})"
+            message = f"Failed to subscribe ({str(rc)})"
             self._logger.error(message)
             if connection_queue:
                 connection_queue.set_exception(VacuumError(message))

roborock/exceptions.py

@@ -1,4 +1,5 @@
 """Roborock exceptions."""
+
 from __future__ import annotations
 
 
@@ -76,3 +77,7 @@ class RoborockTooManyRequest(RoborockException):
 
 class RoborockRateLimit(RoborockException):
     """Class for our rate limits exceptions."""
+
+
+class RoborockInvalidUserData(RoborockException):
+    """Class to state the user data is invalid (expired or manipulated)."""

roborock/roborock_future.py

@@ -5,7 +5,7 @@
 
 import async_timeout
 
-from .exceptions import VacuumError
+from .exceptions import RoborockInvalidUserData, VacuumError
 
 
 class RoborockFuture:
@@ -21,11 +21,11 @@ def _set_result(self, item: Any) -> None:
     def set_result(self, item: Any) -> None:
         self.loop.call_soon_threadsafe(self._set_result, item)
 
-    def _set_exception(self, exc: VacuumError) -> None:
+    def _set_exception(self, exc: VacuumError | RoborockInvalidUserData) -> None:
         if not self.fut.cancelled():
             self.fut.set_exception(exc)
 
-    def set_exception(self, exc: VacuumError) -> None:
+    def set_exception(self, exc: VacuumError | RoborockInvalidUserData) -> None:
         self.loop.call_soon_threadsafe(self._set_exception, exc)
 
     async def async_get(self, timeout: float | int) -> tuple[Any, VacuumError | None]: