From 608d38845069e17b6d22ee655f4d8214edfdb975 Mon Sep 17 00:00:00 2001
From: Austin Zielman <austin@abacus.ai>
Date: Sun, 30 Jun 2024 14:07:37 -0700
Subject: [PATCH] Add R1 series vacuum

---
 custom_components/robovac/robovac.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/custom_components/robovac/robovac.py b/custom_components/robovac/robovac.py
index 51e0778..2070556 100644
--- a/custom_components/robovac/robovac.py
+++ b/custom_components/robovac/robovac.py
@@ -49,10 +49,11 @@ ROBOVAC_SERIES = {
         "T2273",
     ],
     "L": ["T2181", "T2182", "T2190", "T2192", "T2193", "T2194"],
+    "R": ["T2080"],
     "X": ["T2261", "T2262", "T2320"],
 }
 
-HAS_MAP_FEATURE = ["T2253", *ROBOVAC_SERIES["L"], *ROBOVAC_SERIES["X"]]
+HAS_MAP_FEATURE = ["T2253", *ROBOVAC_SERIES["L"], *ROBOVAC_SERIES["R"], *ROBOVAC_SERIES["X"]]
 
 HAS_CONSUMABLES = [
     "T1250",
@@ -82,6 +83,13 @@ ROBOVAC_SERIES_FEATURES = {
     | RoboVacEntityFeature.ROOM
     | RoboVacEntityFeature.ZONE
     | RoboVacEntityFeature.BOOST_IQ,
+    "R": RoboVacEntityFeature.CLEANING_TIME
+    | RoboVacEntityFeature.CLEANING_AREA
+    | RoboVacEntityFeature.DO_NOT_DISTURB
+    | RoboVacEntityFeature.AUTO_RETURN
+    | RoboVacEntityFeature.ROOM
+    | RoboVacEntityFeature.ZONE
+    | RoboVacEntityFeature.BOOST_IQ,
     "X": RoboVacEntityFeature.CLEANING_TIME
     | RoboVacEntityFeature.CLEANING_AREA
     | RoboVacEntityFeature.DO_NOT_DISTURB
@@ -95,6 +103,7 @@ ROBOVAC_SERIES_FAN_SPEEDS = {
     "C": ["No Suction", "Standard", "Boost IQ", "Max"],
     "G": ["Standard", "Turbo", "Max", "Boost IQ"],
     "L": ["Quiet", "Standard", "Turbo", "Max"],
+    "R": ["Quiet", "Standard", "Turbo", "Max"],
     "X": ["Pure", "Standard", "Turbo", "Max"],
 }