roborock-cli 0.1.1__py3-none-any.whl

roborock_cli/_vendor/roborock/roborock_message.py

@@ -0,0 +1,246 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from enum import StrEnum
+
+from roborock_cli._vendor.roborock import RoborockEnum
+from roborock_cli._vendor.roborock.util import get_next_int, get_timestamp
+
+
+class RoborockMessageProtocol(RoborockEnum):
+    HELLO_REQUEST = 0
+    HELLO_RESPONSE = 1
+    PING_REQUEST = 2
+    PING_RESPONSE = 3
+    GENERAL_REQUEST = 4
+    GENERAL_RESPONSE = 5
+    RPC_REQUEST = 101
+    RPC_RESPONSE = 102
+    MAP_RESPONSE = 301
+
+
+class RoborockDataProtocol(RoborockEnum):
+    ERROR_CODE = 120
+    STATE = 121
+    BATTERY = 122
+    FAN_POWER = 123
+    WATER_BOX_MODE = 124
+    MAIN_BRUSH_WORK_TIME = 125
+    SIDE_BRUSH_WORK_TIME = 126
+    FILTER_WORK_TIME = 127
+    ADDITIONAL_PROPS = 128
+    TASK_COMPLETE = 130
+    TASK_CANCEL_LOW_POWER = 131
+    TASK_CANCEL_IN_MOTION = 132
+    CHARGE_STATUS = 133
+    DRYING_STATUS = 134
+    OFFLINE_STATUS = 135
+
+    @classmethod
+    def _missing_(cls: type[RoborockEnum], key) -> RoborockEnum:
+        raise ValueError("%s not a valid key for Data Protocol", key)
+
+
+class RoborockDyadDataProtocol(RoborockEnum):
+    DRYING_STATUS = 134
+    START = 200
+    STATUS = 201
+    SELF_CLEAN_MODE = 202
+    SELF_CLEAN_LEVEL = 203
+    WARM_LEVEL = 204
+    CLEAN_MODE = 205
+    SUCTION = 206
+    WATER_LEVEL = 207
+    BRUSH_SPEED = 208
+    POWER = 209
+    COUNTDOWN_TIME = 210
+    AUTO_SELF_CLEAN_SET = 212
+    AUTO_DRY = 213
+    MESH_LEFT = 214
+    BRUSH_LEFT = 215
+    ERROR = 216
+    MESH_RESET = 218
+    BRUSH_RESET = 219
+    VOLUME_SET = 221
+    STAND_LOCK_AUTO_RUN = 222
+    AUTO_SELF_CLEAN_SET_MODE = 223
+    AUTO_DRY_MODE = 224
+    SILENT_DRY_DURATION = 225
+    SILENT_MODE = 226
+    SILENT_MODE_START_TIME = 227
+    SILENT_MODE_END_TIME = 228
+    RECENT_RUN_TIME = 229
+    TOTAL_RUN_TIME = 230
+    FEATURE_INFO = 235
+    RECOVER_SETTINGS = 236
+    DRY_COUNTDOWN = 237
+    ID_QUERY = 10000
+    F_C = 10001
+    SCHEDULE_TASK = 10002
+    SND_SWITCH = 10003
+    SND_STATE = 10004
+    PRODUCT_INFO = 10005
+    PRIVACY_INFO = 10006
+    OTA_NFO = 10007
+    RPC_REQUEST = 10101
+    RPC_RESPONSE = 10102
+
+
+class RoborockZeoProtocol(RoborockEnum):
+    START = 200  # rw
+    PAUSE = 201  # rw
+    SHUTDOWN = 202  # rw
+    STATE = 203  # ro
+    MODE = 204  # rw
+    PROGRAM = 205  # rw
+    CHILD_LOCK = 206  # rw
+    TEMP = 207  # rw
+    RINSE_TIMES = 208  # rw
+    SPIN_LEVEL = 209  # rw
+    DRYING_MODE = 210  # rw
+    DETERGENT_SET = 211  # rw
+    SOFTENER_SET = 212  # rw
+    DETERGENT_TYPE = 213  # rw
+    SOFTENER_TYPE = 214  # rw
+    COUNTDOWN = 217  # rw
+    WASHING_LEFT = 218  # ro
+    DOORLOCK_STATE = 219  # ro
+    ERROR = 220  # ro
+    CUSTOM_PARAM_SAVE = 221  # rw
+    CUSTOM_PARAM_GET = 222  # ro
+    SOUND_SET = 223  # rw
+    TIMES_AFTER_CLEAN = 224  # ro
+    DEFAULT_SETTING = 225  # rw
+    DETERGENT_EMPTY = 226  # ro
+    SOFTENER_EMPTY = 227  # ro
+    LIGHT_SETTING = 229  # rw
+    DETERGENT_VOLUME = 230  # rw
+    SOFTENER_VOLUME = 231  # rw
+    APP_AUTHORIZATION = 232  # rw
+    ID_QUERY = 10000
+    F_C = 10001
+    SND_STATE = 10004
+    PRODUCT_INFO = 10005
+    PRIVACY_INFO = 10006
+    OTA_NFO = 10007
+    WASHING_LOG = 10008
+    RPC_REQ = 10101
+    RPC_RESp = 10102
+
+
+class RoborockB01Protocol(RoborockEnum):
+    RPC_REQUEST = 101
+    RPC_RESPONSE = 102
+    ERROR_CODE = 120
+    STATE = 121
+    BATTERY = 122
+    FAN_POWER = 123
+    WATER_BOX_MODE = 124
+    MAIN_BRUSH_LIFE = 125
+    SIDE_BRUSH_LIFE = 126
+    FILTER_LIFE = 127
+    OFFLINE_STATUS = 135
+    CLEAN_TIMES = 136
+    CLEANING_PREFERENCE = 137
+    CLEAN_TASK_TYPE = 138
+    BACK_TYPE = 139
+    DOCK_TASK_TYPE = 140
+    CLEANING_PROGRESS = 141
+    FC_STATE = 142
+    START_CLEAN_TASK = 201
+    START_BACK_DOCK_TASK = 202
+    START_DOCK_TASK = 203
+    PAUSE = 204
+    RESUME = 205
+    STOP = 206
+    CEIP = 207
+
+
+class RoborockB01Props(StrEnum):
+    """Properties requested by the Roborock B01 model."""
+
+    STATUS = "status"
+    FAULT = "fault"
+    WIND = "wind"
+    WATER = "water"
+    MODE = "mode"
+    QUANTITY = "quantity"
+    ALARM = "alarm"
+    VOLUME = "volume"
+    HYPA = "hypa"
+    MAIN_BRUSH = "main_brush"
+    SIDE_BRUSH = "side_brush"
+    MOP_LIFE = "mop_life"
+    MAIN_SENSOR = "main_sensor"
+    NET_STATUS = "net_status"
+    REPEAT_STATE = "repeat_state"
+    TANK_STATE = "tank_state"
+    SWEEP_TYPE = "sweep_type"
+    CLEAN_PATH_PREFERENCE = "clean_path_preference"
+    CLOTH_STATE = "cloth_state"
+    TIME_ZONE = "time_zone"
+    TIME_ZONE_INFO = "time_zone_info"
+    LANGUAGE = "language"
+    CLEANING_TIME = "cleaning_time"
+    REAL_CLEAN_TIME = "real_clean_time"
+    CLEANING_AREA = "cleaning_area"
+    CUSTOM_TYPE = "custom_type"
+    SOUND = "sound"
+    WORK_MODE = "work_mode"
+    STATION_ACT = "station_act"
+    CHARGE_STATE = "charge_state"
+    CURRENT_MAP_ID = "current_map_id"
+    MAP_NUM = "map_num"
+    DUST_ACTION = "dust_action"
+    QUIET_IS_OPEN = "quiet_is_open"
+    QUIET_BEGIN_TIME = "quiet_begin_time"
+    QUIET_END_TIME = "quiet_end_time"
+    CLEAN_FINISH = "clean_finish"
+    VOICE_TYPE = "voice_type"
+    VOICE_TYPE_VERSION = "voice_type_version"
+    ORDER_TOTAL = "order_total"
+    BUILD_MAP = "build_map"
+    PRIVACY = "privacy"
+    DUST_AUTO_STATE = "dust_auto_state"
+    DUST_FREQUENCY = "dust_frequency"
+    CHILD_LOCK = "child_lock"
+    MULTI_FLOOR = "multi_floor"
+    MAP_SAVE = "map_save"
+    LIGHT_MODE = "light_mode"
+    GREEN_LASER = "green_laser"
+    DUST_BAG_USED = "dust_bag_used"
+    ORDER_SAVE_MODE = "order_save_mode"
+    MANUFACTURER = "manufacturer"
+    BACK_TO_WASH = "back_to_wash"
+    CHARGE_STATION_TYPE = "charge_station_type"
+    PV_CUT_CHARGE = "pv_cut_charge"
+    PV_CHARGING = "pv_charging"
+    SERIAL_NUMBER = "serial_number"
+    RECOMMEND = "recommend"
+    ADD_SWEEP_STATUS = "add_sweep_status"
+
+
+ROBOROCK_DATA_STATUS_PROTOCOL = [
+    RoborockDataProtocol.ERROR_CODE,
+    RoborockDataProtocol.STATE,
+    RoborockDataProtocol.BATTERY,
+    RoborockDataProtocol.FAN_POWER,
+    RoborockDataProtocol.WATER_BOX_MODE,
+    RoborockDataProtocol.CHARGE_STATUS,
+]
+
+ROBOROCK_DATA_CONSUMABLE_PROTOCOL = [
+    RoborockDataProtocol.MAIN_BRUSH_WORK_TIME,
+    RoborockDataProtocol.SIDE_BRUSH_WORK_TIME,
+    RoborockDataProtocol.FILTER_WORK_TIME,
+]
+
+
+@dataclass
+class RoborockMessage:
+    protocol: RoborockMessageProtocol
+    payload: bytes | None = None
+    seq: int = field(default_factory=lambda: get_next_int(100000, 999999))
+    version: bytes = b"1.0"
+    random: int = field(default_factory=lambda: get_next_int(10000, 99999))
+    timestamp: int = field(default_factory=lambda: get_timestamp())

roborock_cli/_vendor/roborock/roborock_typing.py

@@ -0,0 +1,382 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from enum import Enum, StrEnum
+
+from .data import (
+    CleanRecord,
+    CleanSummary,
+    Consumable,
+    DustCollectionMode,
+    RoborockBase,
+    SmartWashParams,
+    Status,
+    WashTowelMode,
+)
+
+
+class RoborockCommand(str, Enum):
+    """Enum of all known Roborock V1 protocol commands.
+
+    These commands can be sent to a device using the `CommandTrait`.
+    For example: `device.v1_properties.command.send(RoborockCommand.APP_START)`.
+    """
+
+    ADD_MOP_TEMPLATE_PARAMS = "add_mop_template_params"
+    APP_AMETHYST_SELF_CHECK = "app_amethyst_self_check"
+    APP_CHARGE = "app_charge"
+    APP_DELETE_WIFI = "app_delete_wifi"
+    APP_GET_AMETHYST_STATUS = "app_get_amethyst_status"
+    APP_GET_CARPET_DEEP_CLEAN_STATUS = "app_get_carpet_deep_clean_status"
+    APP_GET_CLEAN_ESTIMATE_INFO = "app_get_clean_estimate_info"
+    APP_GET_DRYER_SETTING = "app_get_dryer_setting"
+    APP_GET_INIT_STATUS = "app_get_init_status"
+    APP_GET_LOCALE = "app_get_locale"
+    APP_GET_WIFI_LIST = "app_get_wifi_list"
+    APP_GOTO_TARGET = "app_goto_target"
+    APP_KEEP_EASTER_EGG = "app_keep_easter_egg"
+    APP_PAUSE = "app_pause"
+    APP_RC_END = "app_rc_end"
+    APP_RC_MOVE = "app_rc_move"
+    APP_RC_START = "app_rc_start"
+    APP_RC_STOP = "app_rc_stop"
+    APP_RESUME_BUILD_MAP = "app_resume_build_map"
+    APP_RESUME_PATROL = "app_resume_patrol"
+    APP_SEGMENT_CLEAN = "app_segment_clean"
+    APP_SET_AMETHYST_STATUS = "app_set_amethyst_status"
+    APP_SET_CARPET_DEEP_CLEAN_STATUS = "app_set_carpet_deep_clean_status"
+    APP_SET_CROSS_CARPET_CLEANING_STATUS = "app_set_cross_carpet_cleaning_status"
+    APP_SET_DOOR_SILL_BLOCKS = "app_set_door_sill_blocks"
+    APP_SET_DIRTY_REPLENISH_CLEAN_STATUS = "app_set_dirty_replenish_clean_status"
+    APP_SET_DRYER_SETTING = "app_set_dryer_setting"
+    APP_SET_DRYER_STATUS = "app_set_dryer_status"
+    APP_SET_DYNAMIC_CONFIG = "app_set_dynamic_config"
+    APP_SET_IGNORE_STUCK_POINT = "app_set_ignore_stuck_point"
+    APP_SET_SMART_CLIFF_FORBIDDEN = "app_set_smart_cliff_forbidden"
+    APP_SET_SMART_DOOR_SILL = "app_set_smart_door_sill"
+    APP_SPOT = "app_spot"
+    APP_START = "app_start"
+    APP_START_BUILD_MAP = "app_start_build_map"
+    APP_START_COLLECT_DUST = "app_start_collect_dust"
+    APP_START_EASTER_EGG = "app_start_easter_egg"
+    APP_START_PATROL = "app_start_patrol"
+    APP_START_PET_PATROL = "app_start_pet_patrol"
+    APP_START_WASH = "app_start_wash"
+    APP_STAT = "app_stat"
+    APP_STOP = "app_stop"
+    APP_STOP_COLLECT_DUST = "app_stop_collect_dust"
+    APP_STOP_WASH = "app_stop_wash"
+    APP_UPDATE_UNSAVE_MAP = "app_update_unsave_map"
+    APP_WAKEUP_ROBOT = "app_wakeup_robot"
+    APP_ZONED_CLEAN = "app_zoned_clean"
+    CHANGE_SOUND_VOLUME = "change_sound_volume"
+    CHECK_HOMESEC_PASSWORD = "check_homesec_password"
+    CLOSE_DND_TIMER = "close_dnd_timer"
+    CLOSE_VALLEY_ELECTRICITY_TIMER = "close_valley_electricity_timer"
+    DEL_CLEAN_RECORD = "del_clean_record"
+    DEL_CLEAN_RECORD_MAP_V2 = "del_clean_record_map_v2"
+    DEL_MAP = "del_map"
+    DEL_MOP_TEMPLATE_PARAMS = "del_mop_template_params"
+    DEL_SERVER_TIMER = "del_server_timer"
+    DEL_TIMER = "del_timer"
+    DNLD_INSTALL_SOUND = "dnld_install_sound"
+    ENABLE_HOMESEC_VOICE = "enable_homesec_voice"
+    ENABLE_LOG_UPLOAD = "enable_log_upload"
+    END_EDIT_MAP = "end_edit_map"
+    FIND_ME = "find_me"
+    GET_AUTO_DELIVERY_CLEANING_FLUID = "get_auto_delivery_cleaning_fluid"
+    GET_CAMERA_STATUS = "get_camera_status"
+    GET_CARPET_CLEAN_MODE = "get_carpet_clean_mode"
+    GET_CARPET_MODE = "get_carpet_mode"
+    GET_CHILD_LOCK_STATUS = "get_child_lock_status"
+    GET_CLEAN_FOLLOW_GROUND_MATERIAL_STATUS = "get_clean_follow_ground_material_status"
+    GET_CLEAN_MOTOR_MODE = "get_clean_motor_mode"
+    GET_CLEAN_RECORD = "get_clean_record"
+    GET_CLEAN_RECORD_MAP = "get_clean_record_map"
+    GET_CLEAN_SEQUENCE = "get_clean_sequence"
+    GET_CLEAN_SUMMARY = "get_clean_summary"
+    GET_COLLISION_AVOID_STATUS = "get_collision_avoid_status"
+    GET_CONSUMABLE = "get_consumable"
+    GET_CURRENT_SOUND = "get_current_sound"
+    GET_CUSTOM_MODE = "get_custom_mode"
+    GET_CUSTOMIZE_CLEAN_MODE = "get_customize_clean_mode"
+    GET_DEVICE_ICE = "get_device_ice"
+    GET_DEVICE_SDP = "get_device_sdp"
+    GET_DND_TIMER = "get_dnd_timer"
+    GET_DOCK_INFO = "get_dock_info"
+    GET_DUST_COLLECTION_MODE = "get_dust_collection_mode"
+    GET_DUST_COLLECTION_SWITCH_STATUS = "get_dust_collection_switch_status"
+    GET_DYNAMIC_DATA = "get_dynamic_data"
+    GET_DYNAMIC_MAP_DIFF = "get_dynamic_map_diff"
+    GET_FAN_MOTOR_WORK_TIMEOUT = "get_fan_motor_work_timeout"
+    GET_FLOW_LED_STATUS = "get_flow_led_status"
+    GET_FRESH_MAP = "get_fresh_map"
+    GET_FW_FEATURES = "get_fw_features"
+    GET_HOMESEC_CONNECT_STATUS = "get_homesec_connect_status"
+    GET_IDENTIFY_FURNITURE_STATUS = "get_identify_furniture_status"
+    GET_IDENTIFY_GROUND_MATERIAL_STATUS = "get_identify_ground_material_status"
+    GET_LED_STATUS = "get_led_status"
+    GET_LOG_UPLOAD_STATUS = "get_log_upload_status"
+    GET_MAP = "get_map"
+    GET_MAP_BEAUTIFICATION_STATUS = "get_map_beautification_status"
+    GET_MAP_STATUS = "get_map_status"
+    GET_MAP_V1 = "get_map_v1"
+    GET_MAP_V2 = "get_map_v2"
+    GET_MAP_CALIBRATION = "get_map_calibration"  # Custom command
+    GET_MOP_MOTOR_STATUS = "get_mop_motor_status"
+    GET_MOP_TEMPLATE_PARAMS_BY_ID = "get_mop_template_params_by_id"
+    GET_MOP_TEMPLATE_PARAMS_SUMMARY = "get_mop_template_params_summary"
+    GET_MULTI_MAP = "get_multi_map"
+    GET_MULTI_MAPS_LIST = "get_multi_maps_list"
+    GET_NETWORK_INFO = "get_network_info"
+    GET_OFFLINE_MAP_STATUS = "get_offline_map_status"
+    GET_PERSIST = "get_persist_map"
+    GET_PROP = "get_prop"
+    GET_RANDOM_PKEY = "get_random_pkey"
+    GET_RECOVER_MAP = "get_recover_map"
+    GET_RECOVER_MAPS = "get_recover_maps"
+    GET_ROOM_MAPPING = "get_room_mapping"
+    GET_SCENES_VALID_TIDS = "get_scenes_valid_tids"
+    GET_SEGMENT_STATUS = "get_segment_status"
+    GET_SERIAL_NUMBER = "get_serial_number"
+    GET_SERVER_TIMER = "get_server_timer"
+    GET_SMART_WASH_PARAMS = "get_smart_wash_params"
+    GET_SOUND_PROGRESS = "get_sound_progress"
+    GET_SOUND_VOLUME = "get_sound_volume"
+    GET_STATUS = "get_status"
+    GET_TESTID = "get_testid"
+    GET_TIMER = "get_timer"
+    GET_TIMER_DETAIL = "get_timer_detail"
+    GET_TIMER_SUMMARY = "get_timer_summary"
+    GET_TIMEZONE = "get_timezone"
+    GET_TURN_SERVER = "get_turn_server"
+    GET_VALLEY_ELECTRICITY_TIMER = "get_valley_electricity_timer"
+    GET_WASH_DEBUG_PARAMS = "get_wash_debug_params"
+    GET_WASH_TOWEL_MODE = "get_wash_towel_mode"
+    GET_WASH_TOWEL_PARAMS = "get_wash_towel_params"
+    GET_WATER_BOX_CUSTOM_MODE = "get_water_box_custom_mode"
+    LOAD_MULTI_MAP = "load_multi_map"
+    MANUAL_BAK_MAP = "manual_bak_map"
+    MANUAL_SEGMENT_MAP = "manual_segment_map"
+    MERGE_SEGMENT = "merge_segment"
+    MOP_MODE = "mop_mode"
+    MOP_TEMPLATE_ID = "mop_template_id"
+    NAME_MULTI_MAP = "name_multi_map"
+    NAME_SEGMENT = "name_segment"
+    PLAY_AUDIO = "play_audio"
+    RECOVER_MAP = "recover_map"
+    RECOVER_MULTI_MAP = "recover_multi_map"
+    RESET_CONSUMABLE = "reset_consumable"
+    RESET_HOMESEC_PASSWORD = "reset_homesec_password"
+    RESET_MAP = "reset_map"
+    RESOLVE_ERROR = "resolve_error"
+    RESUME_SEGMENT_CLEAN = "resume_segment_clean"
+    RESUME_ZONED_CLEAN = "resume_zoned_clean"
+    RETRY_REQUEST = "retry_request"
+    REUNION_SCENES = "reunion_scenes"
+    SAVE_FURNITURES = "save_furnitures"
+    SAVE_MAP = "save_map"
+    SEND_ICE_TO_ROBOT = "send_ice_to_robot"
+    SEND_SDP_TO_ROBOT = "send_sdp_to_robot"
+    SET_AIRDRY_HOURS = "set_airdry_hours"
+    SET_APP_TIMEZONE = "set_app_timezone"
+    SET_AUTO_DELIVERY_CLEANING_FLUID = "set_auto_delivery_cleaning_fluid"
+    SET_CAMERA_STATUS = "set_camera_status"
+    SET_CARPET_AREA = "set_carpet_area"
+    SET_CARPET_CLEAN_MODE = "set_carpet_clean_mode"
+    SET_CARPET_MODE = "set_carpet_mode"
+    SET_CHILD_LOCK_STATUS = "set_child_lock_status"
+    SET_CLEAN_FOLLOW_GROUND_MATERIAL_STATUS = "set_clean_follow_ground_material_status"
+    SET_CLEAN_MOTOR_MODE = "set_clean_motor_mode"
+    SET_CLEAN_SEQUENCE = "set_clean_sequence"
+    SET_CLEAN_REPEAT_TIMES = "set_clean_repeat_times"
+    SET_COLLISION_AVOID_STATUS = "set_collision_avoid_status"
+    SET_CUSTOM_MODE = "set_custom_mode"
+    SET_CUSTOMIZE_CLEAN_MODE = "set_customize_clean_mode"
+    SET_DND_TIMER = "set_dnd_timer"
+    SET_DND_TIMER_ACTIONS = "set_dnd_timer_actions"
+    SET_DUST_COLLECTION_MODE = "set_dust_collection_mode"
+    SET_DUST_COLLECTION_SWITCH_STATUS = "set_dust_collection_switch_status"
+    SET_FAN_MOTOR_WORK_TIMEOUT = "set_fan_motor_work_timeout"
+    SET_FDS_ENDPOINT = "set_fds_endpoint"
+    SET_FLOW_LED_STATUS = "set_flow_led_status"
+    SET_HOMESEC_PASSWORD = "set_homesec_password"
+    SET_IDENTIFY_FURNITURE_STATUS = "set_identify_furniture_status"
+    SET_IDENTIFY_GROUND_MATERIAL_STATUS = "set_identify_ground_material_status"
+    SET_IGNORE_CARPET_ZONE = "set_ignore_carpet_zone"
+    SET_IGNORE_IDENTIFY_AREA = "set_ignore_identify_area"
+    SET_LAB_STATUS = "set_lab_status"
+    SET_LED_STATUS = "set_led_status"
+    SET_MAP_BEAUTIFICATION_STATUS = "set_map_beautification_status"
+    SET_MOP_MODE = "set_mop_mode"
+    SET_MOP_MOTOR_STATUS = "set_mop_motor_status"
+    SET_MOP_TEMPLATE_ID = "set_mop_template_id"
+    SET_OFFLINE_MAP_STATUS = "set_offline_map_status"
+    SET_SCENES_SEGMENTS = "set_scenes_segments"
+    SET_SCENES_ZONES = "set_scenes_zones"
+    SET_SEGMENT_GROUND_MATERIAL = "set_segment_ground_material"
+    SET_SERVER_TIMER = "set_server_timer"
+    SET_SMART_WASH_PARAMS = "set_smart_wash_params"
+    SET_SWITCH_MOP_MODE = "set_switch_map_mode"
+    SET_TIMER = "set_timer"
+    SET_TIMEZONE = "set_timezone"
+    SET_VALLEY_ELECTRICITY_TIMER = "set_valley_electricity_timer"
+    SET_VOICE_CHAT_VOLUME = "set_voice_chat_volume"
+    SET_WASH_DEBUG_PARAMS = "set_wash_debug_params"
+    SET_WASH_TOWEL_MODE = "set_wash_towel_mode"
+    SET_WASH_TOWEL_PARAMS = "set_wash_towel_params"
+    SET_WATER_BOX_CUSTOM_MODE = "set_water_box_custom_mode"
+    SET_WATER_BOX_DISTANCE_OFF = "set_water_box_distance_off"
+    SORT_MOP_TEMPLATE_PARAMS = "sort_mop_template_params"
+    SPLIT_SEGMENT = "split_segment"
+    START_CAMERA_PREVIEW = "start_camera_preview"
+    START_CLEAN = "start_clean"
+    START_EDIT_MAP = "start_edit_map"
+    START_VOICE_CHAT = "start_voice_chat"
+    START_WASH_THEN_CHARGE = "start_wash_then_charge"
+    STOP_CAMERA_PREVIEW = "stop_camera_preview"
+    STOP_FAN_MOTOR_WORK = "stop_fan_motor_work"
+    STOP_GOTO_TARGET = "stop_goto_target"
+    STOP_SEGMENT_CLEAN = "stop_segment_clean"
+    STOP_VOICE_CHAT = "stop_voice_chat"
+    STOP_ZONED_CLEAN = "stop_zoned_clean"
+    SWITCH_VIDEO_QUALITY = "switch_video_quality"
+    SWITCH_WATER_MARK = "switch_water_mark"
+    TEST_SOUND_VOLUME = "test_sound_volume"
+    UPD_SERVER_TIMER = "upd_server_timer"
+    UPD_TIMER = "upd_timer"
+    UPDATE_DOCK = "update_dock"
+    UPDATE_MOP_TEMPLATE_PARAMS = "update_mop_template_params"
+    UPLOAD_DATA_FOR_DEBUG_MODE = "upload_data_for_debug_mode"
+    UPLOAD_PHOTO = "upload_photo"
+    USE_NEW_MAP = "use_new_map"
+    USE_OLD_MAP = "use_old_map"
+    USER_UPLOAD_LOG = "user_upload_log"
+    SET_STRETCH_TAG_STATUS = "set_stretch_tag_status"
+    GET_STRETCH_TAG_STATUS = "get_stretch_tag_status"
+    SET_RIGHT_BRUSH_STRETCH_STATUS = "set_right_brush_stretch_status"
+    GET_RIGHT_BRUSH_STRETCH_STATUS = "get_right_brush_stretch_status"
+    SET_DIRTY_OBJECT_DETECT_STATUS = "set_dirty_object_detect_status"
+    GET_DIRTY_OBJECT_DETECT_STATUS = "get_dirty_object_detect_status"
+    SET_WASH_WATER_TEMPERATURE = "set_wash_water_temperature"
+    GET_WASH_WATER_TEMPERATURE = "get_wash_water_temperature"
+    APP_EMPTY_RINSE_TANK_WATER = "app_empty_rinse_tank_water"
+    SET_PET_SUPPLIES_DEEP_CLEAN_STATUS = "set_pet_supplies_deep_clean_status"
+    GET_PET_SUPPLIES_DEEP_CLEAN_STATUS = "get_pet_supplies_deep_clean_status"
+    SET_AP_MIC_LED_STATUS = "set_ap_mic_led_status"
+    GET_AP_MIC_LED_STATUS = "get_ap_mic_led_status"
+    SET_HANDLE_LEAK_WATER_STATUS = "set_handle_leak_water_status"
+    GET_HANDLE_LEAK_WATER_STATUS = "get_handle_leak_water_status"
+    APP_IGNORE_DIRTY_OBJECTS = "app_ignore_dirty_objects"
+    MATTER_GET_STATUS = "matter.get_status"
+    MATTER_DNLD_KEY = "matter.dnld_key"
+    MATTER_RESET = "matter.reset"
+    SET_GAP_DEEP_CLEAN_STATUS = "set_gap_deep_clean_status"
+    GET_GAP_DEEP_CLEAN_STATUS = "get_gap_deep_clean_status"
+    APP_SET_ROBOT_SETTING = "app_set_robot_setting"
+    APP_GET_ROBOT_SETTING = "app_get_robot_setting"
+
+
+class RoborockB01Q7Methods(StrEnum):
+    """Methods used by the Roborock Q7 model."""
+
+    # NOTE: In the Q7 Hermes dump these appear as suffixes and are also used
+    # with an "event." prefix at runtime (see `hermes/.../module_524.js`).
+    ADD_CLEAN_FAILED_POST = "add_clean_failed.post"
+    EVENT_ADD_CLEAN_FAILED_POST = "event.add_clean_failed.post"
+    CLEAN_FINISH_POST = "clean_finish.post"
+    EVENT_CLEAN_FINISH_POST = "event.clean_finish.post"
+    EVENT_BUILD_MAP_FINISH_POST = "event.BuildMapFinish.post"
+    EVENT_MAP_CHANGE_POST = "event.map_change.post"
+    EVENT_WORK_APPOINT_CLEAN_FAILED_POST = "event.work_appoint_clean_failed.post"
+    START_CLEAN_POST = "startClean.post"
+    ADD_ORDER = "service.add_order"
+    ADD_SWEEP_CLEAN = "service.add_sweep_clean"
+    ARRANGE_ROOM = "service.arrange_room"
+    DEL_MAP = "service.del_map"
+    DEL_ORDER = "service.del_order"
+    DEL_ORDERS = "service.del_orders"
+    DELETE_RECORD_BY_URL = "service.delete_record_by_url"
+    DOWNLOAD_VOICE_TYPE = "service.download_voice_type"
+    ERASE_PREFERENCE = "service.erase_preference"
+    FIND_DEVICE = "service.find_device"
+    GET_ROOM_ORDER = "service.get_room_order"
+    GET_VOICE_DOWNLOAD = "service.get_voice_download"
+    HELLO_WIKKA = "service.hello_wikka"
+    RENAME_MAP = "service.rename_map"
+    RENAME_ROOM = "service.rename_room"
+    RENAME_ROOMS = "service.rename_rooms"
+    REPLACE_MAP = "service.replace_map"
+    RESET_CONSUMABLE = "service.reset_consumable"
+    SAVE_CARPET = "service.save_carpet"
+    SAVE_RECOMMEND_FB = "service.save_recommend_fb"
+    SAVE_SILL = "service.save_sill"
+    SET_AREA_START = "service.set_area_start"
+    SET_AREAS_START = "service.set_areas_start"
+    SET_CUR_MAP = "service.set_cur_map"
+    SET_DIRECTION = "service.set_direction"
+    SET_GLOBAL_SORT = "service.set_global_sort"
+    SET_MAP_HIDE = "service.set_map_hide"
+    SET_MULTI_ROOM_MATERIAL = "service.set_multi_room_material"
+    SET_POINT_CLEAN = "service.set_point_clean"
+    SET_PREFERENCE = "service.set_preference"
+    SET_PREFERENCE_TYPE = "service.set_preference_type"
+    SET_QUIET_TIME = "service.set_quiet_time"
+    SET_ROOM_CLEAN = "service.set_room_clean"
+    SET_ROOM_ORDER = "service.set_room_order"
+    SET_VIRTUAL_WALL = "service.set_virtual_wall"
+    SET_ZONE_CLEAN = "service.set_zone_clean"
+    SET_ZONE_POINTS = "service.set_zone_points"
+    SPLIT_ROOM = "service.split_room"
+    START_EXPLORE = "service.start_explore"
+    START_POINT_CLEAN = "service.start_point_clean"
+    START_RECHARGE = "service.start_recharge"
+    STOP_RECHARGE = "service.stop_recharge"
+    UPLOAD_BY_MAPID = "service.upload_by_mapid"
+    UPLOAD_RECORD_BY_URL = "service.upload_record_by_url"
+    GET_PROP = "prop.get"
+    GET_MAP_LIST = "service.get_map_list"
+    UPLOAD_BY_MAPTYPE = "service.upload_by_maptype"
+    SET_PROP = "prop.set"
+    GET_PREFERENCE = "service.get_preference"
+    GET_RECORD_LIST = "service.get_record_list"
+    GET_ORDER = "service.get_order"
+    POST_PROP = "prop.post"
+
+
+@dataclass
+class DockSummary(RoborockBase):
+    dust_collection_mode: DustCollectionMode | None = None
+    wash_towel_mode: WashTowelMode | None = None
+    smart_wash_params: SmartWashParams | None = None
+
+
+@dataclass
+class DeviceProp(RoborockBase):
+    status: Status = field(default_factory=Status)
+    clean_summary: CleanSummary = field(default_factory=CleanSummary)
+    consumable: Consumable = field(default_factory=Consumable)
+    last_clean_record: CleanRecord | None = None
+    dock_summary: DockSummary | None = None
+    dust_collection_mode_name: str | None = None
+
+    def __post_init__(self) -> None:
+        if (
+            self.dock_summary
+            and self.dock_summary.dust_collection_mode is not None
+            and self.dock_summary.dust_collection_mode.mode is not None
+        ):
+            self.dust_collection_mode_name = self.dock_summary.dust_collection_mode.mode.name
+
+    def update(self, device_prop: DeviceProp) -> None:
+        if device_prop.status:
+            self.status = device_prop.status
+        if device_prop.clean_summary:
+            self.clean_summary = device_prop.clean_summary
+        if device_prop.consumable:
+            self.consumable = device_prop.consumable
+        if device_prop.last_clean_record:
+            self.last_clean_record = device_prop.last_clean_record
+        if device_prop.dock_summary:
+            self.dock_summary = device_prop.dock_summary
+        self.__post_init__()

roborock_cli/_vendor/roborock/util.py

@@ -0,0 +1,54 @@
+from __future__ import annotations
+
+import logging
+import math
+import time
+from collections.abc import MutableMapping
+from typing import Any, TypeVar
+
+from roborock_cli._vendor.roborock.diagnostics import redact_device_uid
+
+T = TypeVar("T")
+
+
+def unpack_list(value: list[T], size: int) -> list[T | None]:
+    return (value + [None] * size)[:size]  # type: ignore
+
+
+class RoborockLoggerAdapter(logging.LoggerAdapter):
+    def __init__(
+        self,
+        duid: str | None = None,
+        name: str | None = None,
+        logger: logging.Logger | None = None,
+    ) -> None:
+        super().__init__(logger or logging.getLogger(__name__), {})
+        if name is not None:
+            self.prefix = name
+        elif duid is not None:
+            self.prefix = redact_device_uid(duid)
+        else:
+            raise ValueError("Either duid or name must be provided")
+
+    def process(self, msg: str, kwargs: MutableMapping[str, Any]) -> tuple[str, MutableMapping[str, Any]]:
+        return f"[{self.prefix}] {msg}", kwargs
+
+
+counter_map: dict[tuple[int, int], int] = {}
+
+
+def get_next_int(min_val: int, max_val: int) -> int:
+    """Gets a random int in the range, precached to help keep it fast."""
+    if (min_val, max_val) not in counter_map:
+        # If we have never seen this range, or if the cache is getting low, make a bunch of preshuffled values.
+        counter_map[(min_val, max_val)] = min_val
+    counter_map[(min_val, max_val)] += 1
+    return counter_map[(min_val, max_val)] % max_val + min_val
+
+
+def get_timestamp() -> int:
+    """Get the current timestamp in seconds since epoch.
+
+    This is separated out to allow for easier mocking in tests.
+    """
+    return math.floor(time.time())