feat: Update cli to use new command interface

Author: allenporter

roborock/cli.py

@@ -596,21 +596,14 @@ async def home(ctx, device_id: str, refresh: bool):
 @async_command
 async def command(ctx, cmd, device_id, params):
     context: RoborockContext = ctx.obj
-    cache_data = await context.get_devices()
-
-    home_data = cache_data.home_data
-    devices = home_data.get_all_devices()
-    device = next(device for device in devices if device.duid == device_id)
-    model = next(
-        (product.model for product in home_data.products if device is not None and product.id == device.product_id),
-        None,
-    )
-    if model is None:
-        raise RoborockException(f"Could not find model for device {device.name}")
-    device_info = DeviceData(device=device, model=model)
-    mqtt_client = RoborockMqttClientV1(cache_data.user_data, device_info)
-    await mqtt_client.send_command(cmd, json.loads(params) if params is not None else None)
-    await mqtt_client.async_release()
+    device_manager = await context.get_device_manager()
+    device = await device_manager.get_device(device_id)
+    if device.v1_properties is None:
+        raise RoborockException(f"Device {device.name} does not support V1 protocol")
+    command_trait: Trait = device.v1_properties.command
+    result = await command_trait.send(cmd, json.loads(params) if params is not None else None)
+    if result:
+        click.echo(dump_json(result))
 
 
 @click.command()

roborock/devices/traits/v1/__init__.py

@@ -11,6 +11,7 @@
 
 from .clean_summary import CleanSummaryTrait
 from .common import V1TraitMixin
+from .command import CommandTrait
 from .consumeable import ConsumableTrait
 from .device_features import DeviceFeaturesTrait
 from .do_not_disturb import DoNotDisturbTrait
@@ -35,6 +36,7 @@
     "ConsumableTrait",
     "HomeTrait",
     "DeviceFeaturesTrait",
+    "CommandTrait",
 ]
 
 
@@ -47,6 +49,7 @@ class PropertiesApi(Trait):
 
     # All v1 devices have these traits
     status: StatusTrait
+    command: CommandTrait
     dnd: DoNotDisturbTrait
     clean_summary: CleanSummaryTrait
     sound_volume: SoundVolumeTrait

roborock/devices/traits/v1/command.py

@@ -14,8 +14,8 @@ def __post_init__(self) -> None:
         """
         self._rpc_channel = None
 
-    async def send(self, command: RoborockCommand, params: dict[str, Any] | None = None) -> None:
+    async def send(self, command: RoborockCommand, params: dict[str, Any] | None = None) -> Any:
         """Send a command to the device."""
         if not self._rpc_channel:
             raise ValueError("Device trait in invalid state")
-        await self._rpc_channel.send_command(command, params=params)
+        return await self._rpc_channel.send_command(command, params=params)