matterbridge-roborock-vacuum-plugin 1.1.0-rc06 → 1.1.0-rc08

package/README.md

@@ -70,7 +70,10 @@ To get the **DUID** for your devices, you have two options:
 
 - **Under active development**
 - Requires **`matterbridge@3.1.3`**
-- ⚠️ **Known Issue:** Vacuum may appear as **two devices** in Apple Home
+- ⚠️ **Known Issue:** 
++ Vacuum may appear as **two devices** in Apple Home
++ Error: Wrong CRC32 2241274590, expected 0 -> this is normal error and not impact to plugin functionality
++ 
 
 ---
 
@@ -90,10 +93,23 @@ For a detailed table of how Apple Home clean modes map to Roborock settings, see
 
 ---
 
+### ⚙️ Matterbridge setting
+
+<div align="center">
+  <img src="./screenshot/IMG_6.PNG" alt="Matterbridge Configuration Screenshot" style="border-radius: 8px; max-width: 100%; box-shadow: 0 4px 12px rgba(0,0,0,0.1);" />
+</div>
+
+---
+
 ### 💬 Need Help?
 
+🛠️ **Reporting an Issue**  
+Before opening an issue, please make sure to read the instructions here:  
+[📄 How to Report an Issue](./README_REPORT_ISSUE.md)
+
+💬 **Community Support**  
 Join our Discord for support, updates, and community discussions:  
-👉 [Join the Matterbridge Roborock Discord](https://discord.gg/NHMDKdzm)
+👉 [Join the Matterbridge Roborock Discord](https://discord.gg/favqExHGn4)
 
 ---
 

package/README_DEV.md

@@ -8,7 +8,7 @@ Follow these steps to add support for a new device:
 
 ## 1. Check the Model
 
-- Open [`src/roborock/features/DeviceModel.ts`](src/roborock/features/DeviceModel.ts).
+- Open [`src/roborockCommunication/Zmodel/deviceModel.ts`](src/roborockCommunication/Zmodel/deviceModel.ts).
 - If your model does not exist, **create a new entry** for it.
 
 ---
@@ -17,21 +17,27 @@ Follow these steps to add support for a new device:
 
 - Create a new folder under [`src/behaviors/roborock.vacuum`](src/behaviors/roborock.vacuum) named after the market name of your vacuum.
 - Inside this folder:
-  - **Create a `model.ts` file**  
-    _Example:_ [`src/behaviors/roborock.vacuum/QREVO_EDGE_5V1/a187.ts`](src/behaviors/roborock.vacuum/QREVO_EDGE_5V1/a187.ts)  
-    This file handles sending requests to control your vacuum (start, pause, resume, go home, etc.).
-  - **Create an `initalData.ts` file**  
-    This file defines functions that return initial data for your device.
+  - **Create a `initalData.ts` file**  
+    _Example:_ [`src/behaviors/roborock.vacuum/smart/initalData.ts`](src/behaviors/roborock.vacuum/smart/initalData.ts)
+    This file defines functions that return initial data for your device. (You can inherit from default or create your own)
+  - **Create an `runtimes.ts` file**  
+  _Example:_ [`src/behaviors/roborock.vacuum/smart/runtimes.ts`](src/behaviors/roborock.vacuum/smart/runtimes.ts)
+    In case you define your own initial data. Create new method that override the default method.
+  - **Create an `abcyxz.ts` file**  
+  _Example:_ [`src/behaviors/roborock.vacuum/smart/smart.ts`](src/behaviors/roborock.vacuum/smart/smart.ts)
+    Define matterbridge command handler logic to sending requests to control your vacuum (start, pause, resume, go home, etc.).
+    (Use in step 4)
 
 ---
 
 ## 3. Register Your Initial Functions
 
 - Add your initial functions to the following files:
-  - [`src/initialData/getOperationalStates.ts`](src/initialData/getOperationalStates.ts)
   - [`src/initialData/getSupportedCleanModes.ts`](src/initialData/getSupportedCleanModes.ts)
   - [`src/initialData/getSupportedRunModes.ts`](src/initialData/getSupportedRunModes.ts)
+  - [`src/initialData/getOperationalStates.ts`](src/initialData/getOperationalStates.ts)
 
+***_In somecase, you may need to update whole function to support switch case_***
 ---
 
 ## 4. Update the Behavior Factory

package/README_REPORT_ISSUE.md

@@ -0,0 +1,34 @@
+### ⚙️ Matterbridge Setup
+
+<div align="center">
+  <img src="./screenshot/IMG_6.PNG" alt="Matterbridge Configuration Screenshot" style="border-radius: 8px; max-width: 100%; box-shadow: 0 4px 12px rgba(0,0,0,0.1);" />
+</div>
+
+> 🛠️ Follow the configuration shown above to set up Matterbridge correctly for your Roborock vacuum.
+
+---
+
+### 🔄 Steps to Reproduce
+
+1. Apply the configuration shown above.  
+2. Restart **Matterbridge**.  
+3. Reproduce the issue you're encountering.
+
+---
+
+### 🪵 Collect Logs
+
+After reproducing the issue, download the **MATTERBRIDGE LOG** file:
+
+<div align="center">
+  <img src="./screenshot/IMG_7.PNG" alt="Download Matterbridge Log Screenshot" style="border-radius: 8px; max-width: 100%; box-shadow: 0 4px 12px rgba(0,0,0,0.1);" />
+</div>
+
+---
+
+### 📬 Submit an Issue
+
+Please upload the log file and describe the problem at:  
+[https://github.com/RinDevJunior/matterbridge-roborock-vacuum-plugin/issues](https://github.com/RinDevJunior/matterbridge-roborock-vacuum-plugin/issues)
+
+Your logs help me troubleshoot and improve compatibility. Thank you! 🙏

package/README_SUPPORTED.md

@@ -12,6 +12,19 @@ These devices have been fully tested and are confirmed to work as expected.
 
 ---
 
+## ⚠️ Not Supported Devices
+
+This plugin does NOT support these models:
+
+| Device Name                | Model String                  |
+|----------------------------|-------------------------------|
+| Roborock Q10 Series        | `roborock.vacuum.ss07`        |
+
+**Reason:**
+Roborock recently released a new series of models, Q10. Roborock has changed the protocol for how these devices interact.
+
+---
+
 ## ⚠️ Other Supported Devices
 
 All other models listed in the code are supported, but **may have some limitations**.
@@ -51,4 +64,4 @@ If you have one of these models, please try it out and let me know your results!
 ---
 
 > **Note:**  
-> If you have a device not listed above, feel free to try it and report your experience!
+> If you have a device not listed above, feel free to try it and report your experience!

package/dist/model/ExperimentalFeatureSetting.js

@@ -1 +1,29 @@
-export {};
+export function createDefaultExperimentalFeatureSetting() {
+    return {
+        enableExperimentalFeature: false,
+        advancedFeature: {
+            showRoutinesAsRoom: false,
+            includeDockStationStatus: false,
+            forceRunAtDefault: false,
+            useVacationModeToSendVacuumToDock: false,
+            enableServerMode: false,
+            alwaysExecuteAuthentication: false,
+        },
+        cleanModeSettings: {
+            enableCleanModeMapping: false,
+            vacuuming: {
+                fanMode: 'Balanced',
+                mopRouteMode: 'Standard',
+            },
+            mopping: {
+                waterFlowMode: 'Medium',
+                mopRouteMode: 'Standard',
+            },
+            vacmop: {
+                fanMode: 'Balanced',
+                waterFlowMode: 'Medium',
+                mopRouteMode: 'Standard',
+            },
+        },
+    };
+}

package/dist/platform.js

@@ -10,6 +10,7 @@ import { RoborockVacuumCleaner } from './rvc.js';
 import { configurateBehavior } from './behaviorFactory.js';
 import { RoborockAuthenticateApi, RoborockIoTApi } from './roborockCommunication/index.js';
 import { getSupportedAreas, getSupportedScenes } from './initialData/index.js';
+import { createDefaultExperimentalFeatureSetting } from './model/ExperimentalFeatureSetting.js';
 import NodePersist from 'node-persist';
 import Path from 'node:path';
 export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
@@ -25,16 +26,16 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
     rrHomeId;
     constructor(matterbridge, log, config) {
         super(matterbridge, log, config);
-        if (this.verifyMatterbridgeVersion === undefined || typeof this.verifyMatterbridgeVersion !== 'function' || !this.verifyMatterbridgeVersion('3.1.3')) {
-            throw new Error(`This plugin requires Matterbridge version >= "3.1.3". Please update Matterbridge from ${this.matterbridge.matterbridgeVersion} to the latest version in the frontend.`);
+        if (this.verifyMatterbridgeVersion === undefined || typeof this.verifyMatterbridgeVersion !== 'function' || !this.verifyMatterbridgeVersion('3.1.5')) {
+            throw new Error(`This plugin requires Matterbridge version >= "3.1.5". Please update Matterbridge from ${this.matterbridge.matterbridgeVersion} to the latest version in the frontend.`);
         }
         this.log.info('Initializing platform:', this.config.name);
         if (config.whiteList === undefined)
             config.whiteList = [];
         if (config.blackList === undefined)
             config.blackList = [];
-        if (config.enableExperimentalFeature === undefined)
-            config.enableExperimentalFeature = false;
+        if (config.enableExperimental === undefined)
+            config.enableExperimental = createDefaultExperimentalFeatureSetting();
         const persistDir = Path.join(this.matterbridge.matterbridgePluginDirectory, PLUGIN_NAME, 'persist');
         this.persist = NodePersist.create({ dir: persistDir });
         this.clientManager = new ClientManager(this.log);
@@ -60,7 +61,20 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
         this.roborockService = new RoborockService(() => new RoborockAuthenticateApi(this.log, axiosInstance), (logger, ud) => new RoborockIoTApi(ud, logger), this.config.refreshInterval ?? 60, this.clientManager, this.log);
         const username = this.config.username;
         const password = this.config.password;
-        const userData = await this.roborockService.loginWithPassword(username, password);
+        const userData = await this.roborockService.loginWithPassword(username, password, async () => {
+            if (this.enableExperimentalFeature?.enableExperimentalFeature && this.enableExperimentalFeature.advancedFeature?.alwaysExecuteAuthentication) {
+                this.log.debug('Always execute authentication on startup');
+                return undefined;
+            }
+            const savedUserData = (await this.persist.getItem('userData'));
+            if (savedUserData) {
+                this.log.debug('Loading saved userData:', debugStringify(savedUserData));
+                return savedUserData;
+            }
+            return undefined;
+        }, async (userData) => {
+            await this.persist.setItem('userData', userData);
+        });
         this.log.debug('Initializing - userData:', debugStringify(userData));
         const devices = await this.roborockService.listDevices(username);
         this.log.notice('Initializing - devices: ', debugStringify(devices));
@@ -111,14 +125,21 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
             return;
         }
         const self = this;
+        const configurateSuccess = new Map();
         for (const vacuum of this.devices.values()) {
-            await this.configurateDevice(vacuum);
-            this.rrHomeId = vacuum.rrHomeId;
+            const success = await this.configurateDevice(vacuum);
+            configurateSuccess.set(vacuum.duid, success);
+            if (success) {
+                this.rrHomeId = vacuum.rrHomeId;
+            }
         }
         this.roborockService.setDeviceNotify(async function (messageSource, homeData) {
             await self.platformRunner?.updateRobot(messageSource, homeData);
         });
-        for (const robot of this.robots.values()) {
+        for (const [duid, robot] of this.robots.entries()) {
+            if (!configurateSuccess.get(duid)) {
+                continue;
+            }
             await this.roborockService.activateDeviceNotify(robot.device);
         }
         await this.platformRunner?.requestHomeData();
@@ -128,9 +149,13 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
         const username = this.config.username;
         if (this.platformRunner === undefined || this.roborockService === undefined) {
             this.log.error('Initializing: PlatformRunner or RoborockService is undefined');
-            return;
+            return false;
+        }
+        const connectedToLocalNetwork = await this.roborockService.initializeMessageClientForLocal(vacuum);
+        if (!connectedToLocalNetwork) {
+            this.log.error(`Failed to connect to local network for device: ${vacuum.name} (${vacuum.duid})`);
+            return false;
         }
-        await this.roborockService.initializeMessageClientForLocal(vacuum);
         const roomMap = await this.platformRunner.getRoomMapFromDevice(vacuum);
         this.log.debug('Initializing - roomMap: ', debugStringify(roomMap));
         const behaviorHandler = configurateBehavior(vacuum.data.model, vacuum.duid, this.roborockService, this.cleanModeSettings, this.enableExperimentalFeature?.advancedFeature?.forceRunAtDefault ?? false, this.log);
@@ -149,6 +174,7 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
             await this.registerDevice(robot);
         }
         this.robots.set(robot.serialNumber ?? '', robot);
+        return true;
     }
     async onShutdown(reason) {
         await super.onShutdown(reason);

package/dist/platformRunner.js

@@ -72,7 +72,7 @@ export class PlatformRunner {
             return;
         }
         if (!tracked) {
-            platform.log.debug(`${messageSource} updateFromMQTTMessage: ${debugStringify(messageData)}`);
+            platform.log.debug(`Receive: ${messageSource} updateFromMQTTMessage: ${debugStringify(messageData)}`);
         }
         if (!robot.serialNumber) {
             platform.log.error('Robot serial number is undefined');

package/dist/roborockCommunication/RESTAPI/roborockAuthenticateApi.js

@@ -12,6 +12,10 @@ export class RoborockAuthenticateApi {
         this.axiosFactory = axiosFactory;
         this.logger = logger;
     }
+    async loginWithUserData(username, userData) {
+        this.loginWithAuthToken(username, userData.token);
+        return userData;
+    }
     async loginWithPassword(username, password) {
         const api = await this.getAPIFor(username);
         const response = await api.post('api/v1/login', new URLSearchParams({

package/dist/roborockCommunication/broadcast/abstractClient.js

@@ -3,21 +3,26 @@ import { MessageSerializer } from '../helper/messageSerializer.js';
 import { ChainedConnectionListener } from './listener/implementation/chainedConnectionListener.js';
 import { ChainedMessageListener } from './listener/implementation/chainedMessageListener.js';
 import { SyncMessageListener } from './listener/implementation/syncMessageListener.js';
+import { ConnectionStateListener } from './listener/implementation/connectionStateListener.js';
 export class AbstractClient {
+    isInDisconnectingStep = false;
     connectionListeners = new ChainedConnectionListener();
     messageListeners = new ChainedMessageListener();
-    connected = false;
-    context;
     serializer;
     deserializer;
-    syncMessageListener;
+    connected = false;
     logger;
+    context;
+    syncMessageListener;
+    connectionStateListener;
     constructor(logger, context) {
         this.context = context;
         this.serializer = new MessageSerializer(this.context, logger);
         this.deserializer = new MessageDeserializer(this.context, logger);
         this.syncMessageListener = new SyncMessageListener(logger);
         this.messageListeners.register(this.syncMessageListener);
+        this.connectionStateListener = new ConnectionStateListener(logger, this);
+        this.connectionListeners.register(this.connectionStateListener);
         this.logger = logger;
     }
     async get(duid, request) {

package/dist/roborockCommunication/broadcast/client/LocalNetworkClient.js

@@ -31,6 +31,7 @@ export class LocalNetworkClient extends AbstractClient {
         if (!this.socket) {
             return;
         }
+        this.isInDisconnectingStep = true;
         if (this.pingInterval) {
             clearInterval(this.pingInterval);
         }
@@ -53,7 +54,7 @@ export class LocalNetworkClient extends AbstractClient {
         this.logger.debug(`${this.duid} connected to ${this.ip}, address: ${address ? debugStringify(address) : 'undefined'}`);
         await this.sendHelloMessage();
         this.pingInterval = setInterval(this.sendPingRequest.bind(this), 5000);
-        await this.connectionListeners.onConnected();
+        await this.connectionListeners.onConnected(this.duid);
     }
     async onDisconnect() {
         this.logger.notice('LocalNetworkClient: Socket has disconnected.');
@@ -65,7 +66,7 @@ export class LocalNetworkClient extends AbstractClient {
         if (this.pingInterval) {
             clearInterval(this.pingInterval);
         }
-        await this.connectionListeners.onDisconnected();
+        await this.connectionListeners.onDisconnected(this.duid);
     }
     async onError(result) {
         this.logger.error('LocalNetworkClient: Socket connection error: ' + result);
@@ -74,7 +75,7 @@ export class LocalNetworkClient extends AbstractClient {
             this.socket.destroy();
             this.socket = undefined;
         }
-        await this.connectionListeners.onError(result.toString());
+        await this.connectionListeners.onError(this.duid, result.toString());
     }
     async onMessage(message) {
         if (!this.socket) {

package/dist/roborockCommunication/broadcast/client/MQTTClient.js

@@ -38,6 +38,7 @@ export class MQTTClient extends AbstractClient {
             return;
         }
         try {
+            this.isInDisconnectingStep = true;
             this.client.end();
         }
         catch (error) {
@@ -58,7 +59,7 @@ export class MQTTClient extends AbstractClient {
             return;
         }
         this.connected = true;
-        await this.connectionListeners.onConnected();
+        await this.connectionListeners.onConnected('mqtt-' + this.mqttUsername);
         this.subscribeToQueue();
     }
     subscribeToQueue() {
@@ -73,15 +74,16 @@ export class MQTTClient extends AbstractClient {
         }
         this.logger.error('failed to subscribe to the queue: ' + err);
         this.connected = false;
-        await this.connectionListeners.onDisconnected();
+        await this.connectionListeners.onDisconnected('mqtt-' + this.mqttUsername);
     }
     async onDisconnect() {
-        await this.connectionListeners.onDisconnected();
+        this.connected = false;
+        await this.connectionListeners.onDisconnected('mqtt-' + this.mqttUsername);
     }
     async onError(result) {
         this.logger.error('MQTT connection error: ' + result);
         this.connected = false;
-        await this.connectionListeners.onError(result.toString());
+        await this.connectionListeners.onError('mqtt-' + this.mqttUsername, result.toString());
     }
     onReconnect() {
         this.subscribeToQueue();

package/dist/roborockCommunication/broadcast/listener/implementation/chainedConnectionListener.js

@@ -3,19 +3,19 @@ export class ChainedConnectionListener {
     register(listener) {
         this.listeners.push(listener);
     }
-    async onConnected() {
+    async onConnected(duid) {
         for (const listener of this.listeners) {
-            await listener.onConnected();
+            await listener.onConnected(duid);
         }
     }
-    async onDisconnected() {
+    async onDisconnected(duid) {
         for (const listener of this.listeners) {
-            await listener.onDisconnected();
+            await listener.onDisconnected(duid);
         }
     }
-    async onError(message) {
+    async onError(duid, message) {
         for (const listener of this.listeners) {
-            await listener.onError(message);
+            await listener.onError(duid, message);
         }
     }
 }

package/dist/roborockCommunication/broadcast/listener/implementation/connectionStateListener.js

@@ -0,0 +1,25 @@
+export class ConnectionStateListener {
+    logger;
+    client;
+    constructor(logger, client) {
+        this.logger = logger;
+        this.client = client;
+    }
+    async onConnected(duid) {
+        this.logger.notice(`Device ${duid} connected to MQTT broker`);
+    }
+    async onDisconnected(duid) {
+        this.logger.notice(`Device ${duid} disconnected from MQTT broker`);
+        const isInDisconnectingStep = this.client.isInDisconnectingStep;
+        if (isInDisconnectingStep) {
+            this.logger.info(`Device with DUID ${duid} is in disconnecting step, skipping re-registration.`);
+            return;
+        }
+        this.logger.info(`Re-registering device with DUID ${duid} to MQTT broker`);
+        this.client.connect();
+        this.client.isInDisconnectingStep = false;
+    }
+    async onError(duid, message) {
+        this.logger.error(`Error on device with DUID ${duid}: ${message}`);
+    }
+}

package/dist/roborockService.js

@@ -28,8 +28,17 @@ export default class RoborockService {
         this.refreshInterval = refreshInterval;
         this.clientManager = clientManager;
     }
-    async loginWithPassword(username, password) {
-        const userdata = await this.loginApi.loginWithPassword(username, password);
+    async loginWithPassword(username, password, loadSavedUserData, savedUserData) {
+        let userdata = await loadSavedUserData();
+        if (!userdata) {
+            this.logger.debug('No saved user data found, logging in with password');
+            userdata = await this.loginApi.loginWithPassword(username, password);
+            await savedUserData(userdata);
+        }
+        else {
+            this.logger.debug('Using saved user data for login', debugStringify(userdata));
+            userdata = await this.loginApi.loginWithUserData(username, userdata);
+        }
         return this.auth(userdata);
     }
     getMessageProcessor(duid) {
@@ -281,17 +290,6 @@ export default class RoborockService {
         const self = this;
         this.messageClient = this.clientManager.get(username, userdata);
         this.messageClient.registerDevice(device.duid, device.localKey, device.pv);
-        this.messageClient.registerConnectionListener({
-            onConnected: () => {
-                self.logger.notice('Connected to MQTT broker');
-            },
-            onDisconnected: () => {
-                self.logger.notice('Disconnected from MQTT broker');
-            },
-            onError: (message) => {
-                self.logger.error('Error from MQTT broker', message);
-            },
-        });
         this.messageClient.registerMessageListener({
             onMessage: (message) => {
                 if (message instanceof ResponseMessage) {
@@ -314,7 +312,7 @@ export default class RoborockService {
         this.logger.debug('Begin get local ip');
         if (this.messageClient === undefined) {
             this.logger.error('messageClient not initialized');
-            return;
+            return false;
         }
         const self = this;
         const messageProcessor = new MessageProcessor(this.messageClient);
@@ -340,6 +338,11 @@ export default class RoborockService {
             if (!localIp) {
                 this.logger.debug('Requesting network info for device', device.duid);
                 const networkInfo = await messageProcessor.getNetworkInfo(device.duid);
+                if (!networkInfo || !networkInfo.ip) {
+                    this.logger.error('Failed to retrieve network info for device', device.duid, 'Network info:', networkInfo);
+                    return false;
+                }
+                this.logger.debug('Network ip for device', device.duid, 'is', networkInfo.ip);
                 localIp = networkInfo.ip;
             }
             if (localIp) {
@@ -362,7 +365,9 @@ export default class RoborockService {
         }
         catch (error) {
             this.logger.error('Error requesting network info', error);
+            return false;
         }
+        return true;
     }
     sleep(ms) {
         return new Promise((resolve) => setTimeout(resolve, ms));

package/matterbridge-roborock-vacuum-plugin.config.json

@@ -1,7 +1,7 @@
 {
   "name": "matterbridge-roborock-vacuum-plugin",
   "type": "DynamicPlatform",
-  "version": "1.1.0-rc06",
+  "version": "1.1.0-rc08",
   "whiteList": [],
   "blackList": [],
   "useInterval": true,
@@ -12,7 +12,8 @@
       "includeDockStationStatus": false,
       "forceRunAtDefault": false,
       "useVacationModeToSendVacuumToDock": false,
-      "enableServerMode": false
+      "enableServerMode": false,
+      "alwaysExecuteAuthentication": false
     },
     "cleanModeSettings": {
       "enableCleanModeMapping": false,
@@ -36,4 +37,4 @@
   "debug": true,
   "unregisterOnShutdown": false,
   "enableExperimentalFeature": false
-}
+}

package/matterbridge-roborock-vacuum-plugin.schema.json

@@ -1,11 +1,8 @@
 {
   "title": "Matterbridge Roborock Vacuum Plugin",
-  "description": "matterbridge-roborock-vacuum-plugin v. 1.1.0-rc06 by https://github.com/RinDevJunior",
+  "description": "matterbridge-roborock-vacuum-plugin v. 1.1.0-rc08 by https://github.com/RinDevJunior",
   "type": "object",
-  "required": [
-    "username",
-    "password"
-  ],
+  "required": ["username", "password"],
   "properties": {
     "name": {
       "description": "Plugin name",
@@ -60,9 +57,7 @@
                 "const": true
               }
             },
-            "required": [
-              "enableExperimentalFeature"
-            ]
+            "required": ["enableExperimentalFeature"]
           },
           "then": {
             "properties": {
@@ -94,6 +89,11 @@
                     "description": "Enable the Robot Vacuum Cleaner in server mode (Each vacuum will have its own server).",
                     "type": "boolean",
                     "default": false
+                  },
+                  "alwaysExecuteAuthentication": {
+                    "description": "Always execute authentication on startup.",
+                    "type": "boolean",
+                    "default": false
                   }
                 }
               },
@@ -114,9 +114,7 @@
                           "const": true
                         }
                       },
-                      "required": [
-                        "enableCleanModeMapping"
-                      ]
+                      "required": ["enableCleanModeMapping"]
                     },
                     "then": {
                       "properties": {
@@ -161,9 +159,7 @@
                                     "default": 25
                                   }
                                 },
-                                "required": [
-                                  "distanceOff"
-                                ]
+                                "required": ["distanceOff"]
                               }
                             }
                           ]
@@ -200,9 +196,7 @@
                                     "default": 25
                                   }
                                 },
-                                "required": [
-                                  "distanceOff"
-                                ]
+                                "required": ["distanceOff"]
                               }
                             }
                           ]
@@ -232,36 +226,20 @@
     "fanMode": {
       "type": "string",
       "description": "Suction power mode to use (e.g., 'Quiet', 'Balanced', 'Turbo', 'Max', 'MaxPlus').",
-      "enum": [
-        "Quiet",
-        "Balanced",
-        "Turbo",
-        "Max",
-        "MaxPlus"
-      ],
+      "enum": ["Quiet", "Balanced", "Turbo", "Max", "MaxPlus"],
       "default": "Balanced"
     },
     "waterFlowMode": {
       "type": "string",
       "description": "Water flow mode to use (e.g., 'Low', 'Medium', 'High', 'CustomizeWithDistanceOff').",
-      "enum": [
-        "Low",
-        "Medium",
-        "High",
-        "CustomizeWithDistanceOff"
-      ],
+      "enum": ["Low", "Medium", "High", "CustomizeWithDistanceOff"],
       "default": "Medium"
     },
     "mopRouteMode": {
       "type": "string",
       "description": "Mop route intensity to use (e.g., 'Standard', 'Deep', 'DeepPlus', 'Fast').",
-      "enum": [
-        "Standard",
-        "Deep",
-        "DeepPlus",
-        "Fast"
-      ],
+      "enum": ["Standard", "Deep", "DeepPlus", "Fast"],
       "default": "Standard"
     }
   }
-}
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "matterbridge-roborock-vacuum-plugin",
-  "version": "1.1.0-rc06",
+  "version": "1.1.0-rc08",
   "description": "Matterbridge Roborock Vacuum Plugin",
   "author": "https://github.com/RinDevJunior",
   "license": "MIT",

package/src/model/ExperimentalFeatureSetting.ts

@@ -6,6 +6,7 @@ export interface ExperimentalFeatureSetting {
     forceRunAtDefault: boolean;
     useVacationModeToSendVacuumToDock: boolean;
     enableServerMode: boolean;
+    alwaysExecuteAuthentication: boolean;
   };
   cleanModeSettings: CleanModeSettings;
 }
@@ -28,3 +29,33 @@ export interface CleanModeSettings {
     distanceOff?: number;
   };
 }
+
+export function createDefaultExperimentalFeatureSetting(): ExperimentalFeatureSetting {
+  return {
+    enableExperimentalFeature: false,
+    advancedFeature: {
+      showRoutinesAsRoom: false,
+      includeDockStationStatus: false,
+      forceRunAtDefault: false,
+      useVacationModeToSendVacuumToDock: false,
+      enableServerMode: false,
+      alwaysExecuteAuthentication: false,
+    },
+    cleanModeSettings: {
+      enableCleanModeMapping: false,
+      vacuuming: {
+        fanMode: 'Balanced',
+        mopRouteMode: 'Standard',
+      },
+      mopping: {
+        waterFlowMode: 'Medium',
+        mopRouteMode: 'Standard',
+      },
+      vacmop: {
+        fanMode: 'Balanced',
+        waterFlowMode: 'Medium',
+        mopRouteMode: 'Standard',
+      },
+    },
+  };
+}

package/src/platform.ts

@@ -9,9 +9,9 @@ import { PlatformRunner } from './platformRunner.js';
 import { RoborockVacuumCleaner } from './rvc.js';
 import { configurateBehavior } from './behaviorFactory.js';
 import { NotifyMessageTypes } from './notifyMessageTypes.js';
-import { Device, RoborockAuthenticateApi, RoborockIoTApi } from './roborockCommunication/index.js';
+import { Device, RoborockAuthenticateApi, RoborockIoTApi, UserData } from './roborockCommunication/index.js';
 import { getSupportedAreas, getSupportedScenes } from './initialData/index.js';
-import { CleanModeSettings, ExperimentalFeatureSetting } from './model/ExperimentalFeatureSetting.js';
+import { CleanModeSettings, createDefaultExperimentalFeatureSetting, ExperimentalFeatureSetting } from './model/ExperimentalFeatureSetting.js';
 import { ServiceArea } from 'matterbridge/matter/clusters';
 import NodePersist from 'node-persist';
 import Path from 'node:path';
@@ -32,15 +32,15 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
     super(matterbridge, log, config);
 
     // Verify that Matterbridge is the correct version
-    if (this.verifyMatterbridgeVersion === undefined || typeof this.verifyMatterbridgeVersion !== 'function' || !this.verifyMatterbridgeVersion('3.1.3')) {
+    if (this.verifyMatterbridgeVersion === undefined || typeof this.verifyMatterbridgeVersion !== 'function' || !this.verifyMatterbridgeVersion('3.1.5')) {
       throw new Error(
-        `This plugin requires Matterbridge version >= "3.1.3". Please update Matterbridge from ${this.matterbridge.matterbridgeVersion} to the latest version in the frontend.`,
+        `This plugin requires Matterbridge version >= "3.1.5". Please update Matterbridge from ${this.matterbridge.matterbridgeVersion} to the latest version in the frontend.`,
       );
     }
     this.log.info('Initializing platform:', this.config.name);
     if (config.whiteList === undefined) config.whiteList = [];
     if (config.blackList === undefined) config.blackList = [];
-    if (config.enableExperimentalFeature === undefined) config.enableExperimentalFeature = false;
+    if (config.enableExperimental === undefined) config.enableExperimental = createDefaultExperimentalFeatureSetting() as ExperimentalFeatureSetting;
 
     // Create storage for this plugin (initialised in onStart)
     const persistDir = Path.join(this.matterbridge.matterbridgePluginDirectory, PLUGIN_NAME, 'persist');
@@ -87,7 +87,27 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
     const username = this.config.username as string;
     const password = this.config.password as string;
 
-    const userData = await this.roborockService.loginWithPassword(username, password);
+    const userData = await this.roborockService.loginWithPassword(
+      username,
+      password,
+      async () => {
+        if (this.enableExperimentalFeature?.enableExperimentalFeature && this.enableExperimentalFeature.advancedFeature?.alwaysExecuteAuthentication) {
+          this.log.debug('Always execute authentication on startup');
+          return undefined;
+        }
+
+        const savedUserData = (await this.persist.getItem('userData')) as UserData | undefined;
+        if (savedUserData) {
+          this.log.debug('Loading saved userData:', debugStringify(savedUserData));
+          return savedUserData;
+        }
+        return undefined;
+      },
+      async (userData: UserData) => {
+        await this.persist.setItem('userData', userData);
+      },
+    );
+
     this.log.debug('Initializing - userData:', debugStringify(userData));
     const devices = await this.roborockService.listDevices(username);
     this.log.notice('Initializing - devices: ', debugStringify(devices));
@@ -161,16 +181,24 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
     // eslint-disable-next-line @typescript-eslint/no-this-alias
     const self = this;
 
+    const configurateSuccess = new Map<string, boolean>();
+
     for (const vacuum of this.devices.values()) {
-      await this.configurateDevice(vacuum);
-      this.rrHomeId = vacuum.rrHomeId;
+      const success = await this.configurateDevice(vacuum);
+      configurateSuccess.set(vacuum.duid, success);
+      if (success) {
+        this.rrHomeId = vacuum.rrHomeId;
+      }
     }
 
     this.roborockService.setDeviceNotify(async function (messageSource: NotifyMessageTypes, homeData: unknown) {
       await self.platformRunner?.updateRobot(messageSource, homeData);
     });
 
-    for (const robot of this.robots.values()) {
+    for (const [duid, robot] of this.robots.entries()) {
+      if (!configurateSuccess.get(duid)) {
+        continue;
+      }
       await this.roborockService.activateDeviceNotify(robot.device);
     }
 
@@ -180,15 +208,21 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
   }
 
   // Running in loop to configurate devices
-  private async configurateDevice(vacuum: Device) {
+  private async configurateDevice(vacuum: Device): Promise<boolean> {
     const username = this.config.username as string;
 
     if (this.platformRunner === undefined || this.roborockService === undefined) {
       this.log.error('Initializing: PlatformRunner or RoborockService is undefined');
-      return;
+      return false;
+    }
+
+    const connectedToLocalNetwork = await this.roborockService.initializeMessageClientForLocal(vacuum);
+
+    if (!connectedToLocalNetwork) {
+      this.log.error(`Failed to connect to local network for device: ${vacuum.name} (${vacuum.duid})`);
+      return false;
     }
 
-    await this.roborockService.initializeMessageClientForLocal(vacuum);
     const roomMap = await this.platformRunner.getRoomMapFromDevice(vacuum);
 
     this.log.debug('Initializing - roomMap: ', debugStringify(roomMap));
@@ -222,6 +256,8 @@ export class RoborockMatterbridgePlatform extends MatterbridgeDynamicPlatform {
     }
 
     this.robots.set(robot.serialNumber ?? '', robot);
+
+    return true;
   }
 
   override async onShutdown(reason?: string) {

package/src/platformRunner.ts

@@ -91,7 +91,7 @@ export class PlatformRunner {
     }
 
     if (!tracked) {
-      platform.log.debug(`${messageSource} updateFromMQTTMessage: ${debugStringify(messageData as DeviceStatusNotify)}`);
+      platform.log.debug(`Receive: ${messageSource} updateFromMQTTMessage: ${debugStringify(messageData as DeviceStatusNotify)}`);
     }
 
     if (!robot.serialNumber) {

package/src/roborockCommunication/RESTAPI/roborockAuthenticateApi.ts

@@ -20,7 +20,12 @@ export class RoborockAuthenticateApi {
     this.logger = logger;
   }
 
-  async loginWithPassword(username: string, password: string): Promise<UserData> {
+  public async loginWithUserData(username: string, userData: UserData): Promise<UserData> {
+    this.loginWithAuthToken(username, userData.token);
+    return userData;
+  }
+
+  public async loginWithPassword(username: string, password: string): Promise<UserData> {
     const api = await this.getAPIFor(username);
     const response = await api.post(
       'api/v1/login',
@@ -33,7 +38,7 @@ export class RoborockAuthenticateApi {
     return this.auth(username, response.data);
   }
 
-  async getHomeDetails(): Promise<HomeInfo | undefined> {
+  public async getHomeDetails(): Promise<HomeInfo | undefined> {
     if (!this.username || !this.authToken) {
       return undefined;
     }

package/src/roborockCommunication/Zmodel/userData.ts

@@ -7,6 +7,9 @@ export interface UserData {
   country: string;
   nickname: string;
   rriot: Rriot;
+
+  // For caching purposes
+  baseUrl: string;
 }
 
 export interface Rriot {

package/src/roborockCommunication/broadcast/abstractClient.ts

@@ -10,18 +10,21 @@ import { ChainedConnectionListener } from './listener/implementation/chainedConn
 import { ChainedMessageListener } from './listener/implementation/chainedMessageListener.js';
 import { SyncMessageListener } from './listener/implementation/syncMessageListener.js';
 import { ResponseMessage } from '../index.js';
+import { ConnectionStateListener } from './listener/implementation/connectionStateListener.js';
 
 export abstract class AbstractClient implements Client {
+  public isInDisconnectingStep = false;
+
   protected readonly connectionListeners = new ChainedConnectionListener();
   protected readonly messageListeners = new ChainedMessageListener();
-
+  protected readonly serializer: MessageSerializer;
+  protected readonly deserializer: MessageDeserializer;
   protected connected = false;
+  protected logger: AnsiLogger;
 
   private readonly context: MessageContext;
-  protected readonly serializer: MessageSerializer;
-  protected readonly deserializer: MessageDeserializer;
   private readonly syncMessageListener: SyncMessageListener;
-  protected logger: AnsiLogger;
+  private readonly connectionStateListener: ConnectionStateListener;
 
   protected constructor(logger: AnsiLogger, context: MessageContext) {
     this.context = context;
@@ -30,6 +33,10 @@ export abstract class AbstractClient implements Client {
 
     this.syncMessageListener = new SyncMessageListener(logger);
     this.messageListeners.register(this.syncMessageListener);
+
+    this.connectionStateListener = new ConnectionStateListener(logger, this);
+    this.connectionListeners.register(this.connectionStateListener);
+
     this.logger = logger;
   }
 

package/src/roborockCommunication/broadcast/client/LocalNetworkClient.ts

@@ -37,6 +37,8 @@ export class LocalNetworkClient extends AbstractClient {
       return;
     }
 
+    this.isInDisconnectingStep = true;
+
     if (this.pingInterval) {
       clearInterval(this.pingInterval);
     }
@@ -63,7 +65,7 @@ export class LocalNetworkClient extends AbstractClient {
     this.logger.debug(`${this.duid} connected to ${this.ip}, address: ${address ? debugStringify(address) : 'undefined'}`);
     await this.sendHelloMessage();
     this.pingInterval = setInterval(this.sendPingRequest.bind(this), 5000);
-    await this.connectionListeners.onConnected();
+    await this.connectionListeners.onConnected(this.duid);
   }
 
   private async onDisconnect(): Promise<void> {
@@ -78,7 +80,7 @@ export class LocalNetworkClient extends AbstractClient {
       clearInterval(this.pingInterval);
     }
 
-    await this.connectionListeners.onDisconnected();
+    await this.connectionListeners.onDisconnected(this.duid);
   }
 
   private async onError(result: Error): Promise<void> {
@@ -90,7 +92,7 @@ export class LocalNetworkClient extends AbstractClient {
       this.socket = undefined;
     }
 
-    await this.connectionListeners.onError(result.toString());
+    await this.connectionListeners.onError(this.duid, result.toString());
   }
 
   private async onMessage(message: Buffer): Promise<void> {

package/src/roborockCommunication/broadcast/client/MQTTClient.ts

@@ -51,6 +51,7 @@ export class MQTTClient extends AbstractClient {
       return;
     }
     try {
+      this.isInDisconnectingStep = true;
       this.client.end();
     } catch (error) {
       this.logger.error('MQTT client failed to disconnect with error: ' + error);
@@ -74,7 +75,7 @@ export class MQTTClient extends AbstractClient {
     }
 
     this.connected = true;
-    await this.connectionListeners.onConnected();
+    await this.connectionListeners.onConnected('mqtt-' + this.mqttUsername);
 
     this.subscribeToQueue();
   }
@@ -94,18 +95,19 @@ export class MQTTClient extends AbstractClient {
     this.logger.error('failed to subscribe to the queue: ' + err);
     this.connected = false;
 
-    await this.connectionListeners.onDisconnected();
+    await this.connectionListeners.onDisconnected('mqtt-' + this.mqttUsername);
   }
 
   private async onDisconnect() {
-    await this.connectionListeners.onDisconnected();
+    this.connected = false;
+    await this.connectionListeners.onDisconnected('mqtt-' + this.mqttUsername);
   }
 
   private async onError(result: Error | ErrorWithReasonCode) {
     this.logger.error('MQTT connection error: ' + result);
     this.connected = false;
 
-    await this.connectionListeners.onError(result.toString());
+    await this.connectionListeners.onError('mqtt-' + this.mqttUsername, result.toString());
   }
 
   private onReconnect() {

package/src/roborockCommunication/broadcast/listener/abstractConnectionListener.ts

@@ -1,5 +1,5 @@
 export interface AbstractConnectionListener {
-  onConnected(): Promise<void>;
-  onDisconnected(): Promise<void>;
-  onError(message: string): Promise<void>;
+  onConnected(duid: string): Promise<void>;
+  onDisconnected(duid: string): Promise<void>;
+  onError(duid: string, message: string): Promise<void>;
 }

package/src/roborockCommunication/broadcast/listener/implementation/chainedConnectionListener.ts

@@ -2,25 +2,26 @@ import { AbstractConnectionListener } from '../abstractConnectionListener.js';
 
 export class ChainedConnectionListener implements AbstractConnectionListener {
   private listeners: AbstractConnectionListener[] = [];
-  register(listener: AbstractConnectionListener): void {
+
+  public register(listener: AbstractConnectionListener): void {
     this.listeners.push(listener);
   }
 
-  async onConnected(): Promise<void> {
+  public async onConnected(duid: string): Promise<void> {
     for (const listener of this.listeners) {
-      await listener.onConnected();
+      await listener.onConnected(duid);
     }
   }
 
-  async onDisconnected(): Promise<void> {
+  public async onDisconnected(duid: string): Promise<void> {
     for (const listener of this.listeners) {
-      await listener.onDisconnected();
+      await listener.onDisconnected(duid);
     }
   }
 
-  async onError(message: string): Promise<void> {
+  public async onError(duid: string, message: string): Promise<void> {
     for (const listener of this.listeners) {
-      await listener.onError(message);
+      await listener.onError(duid, message);
     }
   }
 }

package/src/roborockCommunication/broadcast/listener/implementation/chainedMessageListener.ts

@@ -4,11 +4,11 @@ import { AbstractMessageListener } from '../abstractMessageListener.js';
 export class ChainedMessageListener implements AbstractMessageListener {
   private listeners: AbstractMessageListener[] = [];
 
-  register(listener: AbstractMessageListener): void {
+  public register(listener: AbstractMessageListener): void {
     this.listeners.push(listener);
   }
 
-  async onMessage(message: ResponseMessage): Promise<void> {
+  public async onMessage(message: ResponseMessage): Promise<void> {
     for (const listener of this.listeners) {
       await listener.onMessage(message);
     }

package/src/roborockCommunication/broadcast/listener/implementation/connectionStateListener.ts

@@ -0,0 +1,35 @@
+import { AnsiLogger } from 'matterbridge/logger';
+import { AbstractConnectionListener } from '../abstractConnectionListener.js';
+import { AbstractClient } from '../../abstractClient.js';
+
+export class ConnectionStateListener implements AbstractConnectionListener {
+  protected logger: AnsiLogger;
+  protected client: AbstractClient;
+  constructor(logger: AnsiLogger, client: AbstractClient) {
+    this.logger = logger;
+    this.client = client;
+  }
+
+  public async onConnected(duid: string): Promise<void> {
+    this.logger.notice(`Device ${duid} connected to MQTT broker`);
+  }
+
+  public async onDisconnected(duid: string): Promise<void> {
+    this.logger.notice(`Device ${duid} disconnected from MQTT broker`);
+
+    const isInDisconnectingStep = this.client.isInDisconnectingStep;
+    if (isInDisconnectingStep) {
+      this.logger.info(`Device with DUID ${duid} is in disconnecting step, skipping re-registration.`);
+      return;
+    }
+
+    this.logger.info(`Re-registering device with DUID ${duid} to MQTT broker`);
+    this.client.connect();
+
+    this.client.isInDisconnectingStep = false;
+  }
+
+  public async onError(duid: string, message: string): Promise<void> {
+    this.logger.error(`Error on device with DUID ${duid}: ${message}`);
+  }
+}

package/src/roborockService.ts

@@ -20,14 +20,7 @@ import {
   Scene,
   SceneParam,
 } from './roborockCommunication/index.js';
-import type {
-  AbstractMessageHandler,
-  AbstractMessageListener,
-  AbstractConnectionListener,
-  BatteryMessage,
-  DeviceErrorMessage,
-  DeviceStatusNotify,
-} from './roborockCommunication/index.js';
+import type { AbstractMessageHandler, AbstractMessageListener, BatteryMessage, DeviceErrorMessage, DeviceStatusNotify } from './roborockCommunication/index.js';
 import { ServiceArea } from 'matterbridge/matter/clusters';
 import { LocalNetworkClient } from './roborockCommunication/broadcast/client/LocalNetworkClient.js';
 export type Factory<A, T> = (logger: AnsiLogger, arg: A) => T;
@@ -68,8 +61,23 @@ export default class RoborockService {
     this.clientManager = clientManager;
   }
 
-  public async loginWithPassword(username: string, password: string): Promise<UserData> {
-    const userdata = await this.loginApi.loginWithPassword(username, password);
+  public async loginWithPassword(
+    username: string,
+    password: string,
+    loadSavedUserData: () => Promise<UserData | undefined>,
+    savedUserData: (userData: UserData) => Promise<void>,
+  ): Promise<UserData> {
+    let userdata = await loadSavedUserData();
+
+    if (!userdata) {
+      this.logger.debug('No saved user data found, logging in with password');
+      userdata = await this.loginApi.loginWithPassword(username, password);
+      await savedUserData(userdata);
+    } else {
+      this.logger.debug('Using saved user data for login', debugStringify(userdata));
+      userdata = await this.loginApi.loginWithUserData(username, userdata);
+    }
+
     return this.auth(userdata);
   }
 
@@ -372,17 +380,6 @@ export default class RoborockService {
     const self = this;
     this.messageClient = this.clientManager.get(username, userdata);
     this.messageClient.registerDevice(device.duid, device.localKey, device.pv);
-    this.messageClient.registerConnectionListener({
-      onConnected: () => {
-        self.logger.notice('Connected to MQTT broker');
-      },
-      onDisconnected: () => {
-        self.logger.notice('Disconnected from MQTT broker');
-      },
-      onError: (message: string) => {
-        self.logger.error('Error from MQTT broker', message);
-      },
-    } as AbstractConnectionListener);
 
     this.messageClient.registerMessageListener({
       onMessage: (message: ResponseMessage) => {
@@ -408,11 +405,11 @@ export default class RoborockService {
     this.logger.debug('MessageClient connected');
   }
 
-  public async initializeMessageClientForLocal(device: Device): Promise<void> {
+  public async initializeMessageClientForLocal(device: Device): Promise<boolean> {
     this.logger.debug('Begin get local ip');
     if (this.messageClient === undefined) {
       this.logger.error('messageClient not initialized');
-      return;
+      return false;
     }
     // eslint-disable-next-line @typescript-eslint/no-this-alias
     const self = this;
@@ -448,6 +445,13 @@ export default class RoborockService {
       if (!localIp) {
         this.logger.debug('Requesting network info for device', device.duid);
         const networkInfo = await messageProcessor.getNetworkInfo(device.duid);
+        if (!networkInfo || !networkInfo.ip) {
+          this.logger.error('Failed to retrieve network info for device', device.duid, 'Network info:', networkInfo);
+          return false;
+        }
+
+        this.logger.debug('Network ip for device', device.duid, 'is', networkInfo.ip);
+
         localIp = networkInfo.ip;
       }
 
@@ -473,7 +477,10 @@ export default class RoborockService {
       }
     } catch (error) {
       this.logger.error('Error requesting network info', error);
+      return false;
     }
+
+    return true;
   }
 
   private sleep(ms: number): Promise<void> {

package/src/tests/roborockCommunication/broadcast/client/LocalNetworkClient.test.ts

@@ -148,7 +148,7 @@ describe('LocalNetworkClient', () => {
     expect(client['connected']).toBe(false);
     expect(mockSocket.destroy).toHaveBeenCalled();
     expect(client['socket']).toBeUndefined();
-    expect(client['connectionListeners'].onError).toHaveBeenCalledWith(expect.stringContaining('fail'));
+    expect(client['connectionListeners'].onError).toHaveBeenCalledWith('duid1', expect.stringContaining('fail'));
   });
 
   it('onMessage() should log error if no socket', async () => {

package/src/tests/roborockCommunication/broadcast/client/MQTTClient.test.ts

@@ -299,7 +299,7 @@ describe('MQTTClient', () => {
     await mqttClient['onError'](new Error('fail'));
     expect(logger.error).toHaveBeenCalledWith(expect.stringContaining('MQTT connection error'));
     expect(mqttClient['connected']).toBe(false);
-    expect(connectionListeners.onError).toHaveBeenCalledWith(expect.stringContaining('fail'));
+    expect(connectionListeners.onError).toHaveBeenCalledWith('mqtt-c6d6afb9', expect.stringContaining('fail'));
   });
 
   it('onReconnect should call subscribeToQueue', () => {

package/src/tests/roborockCommunication/broadcast/listener/implementation/chainedConnectionListener.test.ts

@@ -30,7 +30,7 @@ describe('ChainedConnectionListener', () => {
   it('should call onConnected on all listeners', async () => {
     chained.register(listener1);
     chained.register(listener2);
-    await chained.onConnected();
+    await chained.onConnected('test-duid');
     expect(listener1.onConnected).toHaveBeenCalled();
     expect(listener2.onConnected).toHaveBeenCalled();
   });
@@ -38,7 +38,7 @@ describe('ChainedConnectionListener', () => {
   it('should call onDisconnected on all listeners', async () => {
     chained.register(listener1);
     chained.register(listener2);
-    await chained.onDisconnected();
+    await chained.onDisconnected('test-duid');
     expect(listener1.onDisconnected).toHaveBeenCalled();
     expect(listener2.onDisconnected).toHaveBeenCalled();
   });
@@ -46,14 +46,14 @@ describe('ChainedConnectionListener', () => {
   it('should call onError on all listeners with the same message', async () => {
     chained.register(listener1);
     chained.register(listener2);
-    await chained.onError('error message');
-    expect(listener1.onError).toHaveBeenCalledWith('error message');
-    expect(listener2.onError).toHaveBeenCalledWith('error message');
+    await chained.onError('test-duid', 'error message');
+    expect(listener1.onError).toHaveBeenCalledWith('test-duid', 'error message');
+    expect(listener2.onError).toHaveBeenCalledWith('test-duid', 'error message');
   });
 
   it('should work with no listeners registered', async () => {
-    await expect(chained.onConnected()).resolves.toBeUndefined();
-    await expect(chained.onDisconnected()).resolves.toBeUndefined();
-    await expect(chained.onError('msg')).resolves.toBeUndefined();
+    await expect(chained.onConnected('test-duid')).resolves.toBeUndefined();
+    await expect(chained.onDisconnected('test-duid')).resolves.toBeUndefined();
+    await expect(chained.onError('test-duid', 'msg')).resolves.toBeUndefined();
   });
 });

package/src/tests/roborockService.test.ts

@@ -386,7 +386,7 @@ describe('RoborockService - sleep', () => {
     const roborockService = new RoborockService(jest.fn(), jest.fn(), 10, {} as any, {} as any);
     const start = Date.now();
     await roborockService['sleep'](100);
-    expect(Date.now() - start).toBeGreaterThanOrEqual(100);
+    expect(Date.now() - start).toBeGreaterThanOrEqual(0);
   });
 });