from __future__ import annotations import math import time from dataclasses import dataclass, field from enum import StrEnum from roborock import RoborockEnum from roborock.util import get_next_int 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: math.floor(time.time()))