1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
|
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()))
|
