From 67b2e505f97cfad32c08de53a6e878e573128b84 Mon Sep 17 00:00:00 2001
From: Jelle Bleyaert <9694910+jbleyaert@users.noreply.github.com>
Date: Fri, 22 Mar 2024 17:14:09 +0100
Subject: [PATCH] feat: Adding L60 SES support (UNTESTED) (#67)

* Adding L60 SES support

* Update custom_components/robovac/vacuums/T2267.py

Co-authored-by: Luke Morrigan <luke@lukeb.co.uk>

---------

Co-authored-by: Luke Morrigan <luke@lukeb.co.uk>
---
 custom_components/robovac/vacuums/T2267.py    | 56 +++++++++++++++++++
 custom_components/robovac/vacuums/__init__.py |  2 +
 2 files changed, 58 insertions(+)
 create mode 100644 custom_components/robovac/vacuums/T2267.py

diff --git a/custom_components/robovac/vacuums/T2267.py b/custom_components/robovac/vacuums/T2267.py
new file mode 100644
index 0000000..d3963c9
--- /dev/null
+++ b/custom_components/robovac/vacuums/T2267.py
@@ -0,0 +1,56 @@
+from homeassistant.components.vacuum import VacuumEntityFeature
+from .base import RoboVacEntityFeature, RobovacCommand
+
+
+class T2267:
+    homeassistant_features = (
+        VacuumEntityFeature.BATTERY
+        | VacuumEntityFeature.CLEAN_SPOT
+        | VacuumEntityFeature.FAN_SPEED
+        | VacuumEntityFeature.LOCATE
+        | VacuumEntityFeature.PAUSE
+        | VacuumEntityFeature.RETURN_HOME
+        | VacuumEntityFeature.SEND_COMMAND
+        | VacuumEntityFeature.START
+        | VacuumEntityFeature.STATE
+        | VacuumEntityFeature.STOP
+        | VacuumEntityFeature.MAP
+    )
+    robovac_features = (
+        RoboVacEntityFeature.CLEANING_TIME
+        | RoboVacEntityFeature.CLEANING_AREA
+        | RoboVacEntityFeature.DO_NOT_DISTURB
+        | RoboVacEntityFeature.AUTO_RETURN
+        | RoboVacEntityFeature.ROOM
+        | RoboVacEntityFeature.ZONE
+        | RoboVacEntityFeature.BOOST_IQ
+        | RoboVacEntityFeature.MAP
+        | RoboVacEntityFeature.CONSUMABLES
+    )
+    commands = {
+        RobovacCommand.START_PAUSE: 156,
+        RobovacCommand.DIRECTION: {
+            "code": 155,
+            "values": ["Brake", "Forward", "Back", "Left", "Right"],
+        },
+        RobovacCommand.MODE: {
+            "code": 152,
+            # "values": ["auto", "SmallRoom", "Spot", "Edge", "Nosweep"],
+        },
+        RobovacCommand.STATUS: 153,
+        RobovacCommand.RETURN_HOME: 173,
+        RobovacCommand.FAN_SPEED: {
+            "code": 158,
+            "values": ["Quiet", "Standard", "Turbo", "Max"],
+        },
+        RobovacCommand.LOCATE: 160,
+        RobovacCommand.BATTERY: 163,
+        RobovacCommand.ERROR: 177,
+        RobovacCommand.DO_NOT_DISTURB: 157,
+        RobovacCommand.BOOST_IQ: 159,
+        RobovacCommand.CONSUMABLES: 168,
+        # These commands need codes adding
+        # RobovacCommand.CLEANING_AREA: 0,
+        # RobovacCommand.CLEANING_TIME: 0,
+        # RobovacCommand.AUTO_RETURN: 0,
+    }
diff --git a/custom_components/robovac/vacuums/__init__.py b/custom_components/robovac/vacuums/__init__.py
index 979a5cb..880b536 100644
--- a/custom_components/robovac/vacuums/__init__.py
+++ b/custom_components/robovac/vacuums/__init__.py
@@ -28,6 +28,7 @@ from .T2190 import T2190
 from .T2192 import T2192
 from .T2193 import T2193
 from .T2194 import T2194
+from .T2267 import T2267
 from .T2261 import T2261
 from .T2262 import T2262
 from .T2320 import T2320
@@ -64,6 +65,7 @@ ROBOVAC_MODELS = {
     "T2192": T2192,
     "T2193": T2193,
     "T2194": T2194,
+    "T2267": T2267,
     "T2261": T2261,
     "T2262": T2262,
     "T2320": T2320