feat: add room mapping (#41)

* feat: add room mapping

* fix: lint

* chore: move room mapping to super class client

* chore: linting

* Update roborock/api.py

Co-authored-by: Humberto Gontijo <[email protected]>

---------

Co-authored-by: Humberto Gontijo <[email protected]>

Author: Lash-L

roborock/api.py

@@ -31,6 +31,7 @@
     MultiMapsList,
     NetworkInfo,
     RoborockDeviceInfo,
+    RoomMapping,
     SmartWashParams,
     Status,
     UserData,
@@ -318,6 +319,16 @@ async def get_networking(self, device_id) -> NetworkInfo | None:
             _LOGGER.error(e)
         return None
 
+    async def get_room_mapping(self, device_id: str) -> list[RoomMapping]:
+        """Gets the mapping from segment id -> iot id. Only works on local api."""
+        mapping = await self.send_command(device_id, RoborockCommand.GET_ROOM_MAPPING)
+        if isinstance(mapping, list):
+            return [
+                RoomMapping(segment_id, iot_id)
+                for segment_id, iot_id in [unpack_list(room, 2) for room in mapping]
+            ]
+        return []
+
 
 class RoborockApiClient:
     def __init__(self, username: str, base_url=None) -> None:

roborock/containers.py

@@ -341,3 +341,9 @@ class RoborockDeviceInfo(RoborockBase):
 @dataclass
 class RoborockLocalDeviceInfo(RoborockDeviceInfo):
     network_info: NetworkInfo
+
+
+@dataclass
+class RoomMapping(RoborockBase):
+    segment_id: int
+    iot_id: str

roborock/local_api.py

@@ -8,7 +8,7 @@
 
 import async_timeout
 
-from .api import SPECIAL_COMMANDS, RoborockClient, QUEUE_TIMEOUT
+from .api import QUEUE_TIMEOUT, SPECIAL_COMMANDS, RoborockClient
 from .containers import RoborockLocalDeviceInfo
 from .exceptions import CommandVacuumError, RoborockConnectionException, RoborockException
 from .roborock_message import RoborockMessage, RoborockParser