From 08e19f8a2affd994a2ba94316794ba2a11a7e9a0 Mon Sep 17 00:00:00 2001
From: joBr99 <29555657+joBr99@users.noreply.github.com>
Date: Tue, 9 Aug 2022 18:30:45 +0200
Subject: [PATCH] add vac #393

---
 apps/nspanel-lovelace-ui/luibackend/controller.py | 6 +++++-
 apps/nspanel-lovelace-ui/luibackend/pages.py      | 7 +++++++
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/apps/nspanel-lovelace-ui/luibackend/controller.py b/apps/nspanel-lovelace-ui/luibackend/controller.py
index 9b5b2a88..0a26a4fe 100644
--- a/apps/nspanel-lovelace-ui/luibackend/controller.py
+++ b/apps/nspanel-lovelace-ui/luibackend/controller.py
@@ -294,7 +294,11 @@ class LuiController(object):
                 self._ha_api.get_entity(entity_id).call_service("press")
             elif entity_id.startswith('input_select'):
                 self._ha_api.get_entity(entity_id).call_service("select_next")
-
+            elif entity_id.startswith('vacuum'):
+                if self._ha_api.get_entity(entity_id).state == "docked":
+                    self._ha_api.get_entity(entity_id).call_service("start")
+                else:
+                    self._ha_api.get_entity(entity_id).call_service("return_to_base")
 
         # for media page
         if button_type == "media-next":
diff --git a/apps/nspanel-lovelace-ui/luibackend/pages.py b/apps/nspanel-lovelace-ui/luibackend/pages.py
index bbcfa0e0..7332dc04 100644
--- a/apps/nspanel-lovelace-ui/luibackend/pages.py
+++ b/apps/nspanel-lovelace-ui/luibackend/pages.py
@@ -300,6 +300,13 @@ class LuiPagesGen(object):
             icon_id = get_icon_id_ha("button", state=entity.state, overwrite=icon)
             text = entity.state
             return f"~button~{entityId}~{icon_id}~17299~{name}~{text}"
+        if entityType == "vacuum":
+            icon_id = get_icon_id_ha("robot-vacuum", state=entity.state, overwrite=icon)
+            if entity.state == "docked":
+                text = "Start"
+            else:
+                text = "Return"
+            return f"~button~{entityId}~{icon_id}~17299~{name}~{text}"
         return f"~text~{entityId}~{get_icon_id('alert-circle-outline')}~17299~unsupported~"
 
     def generate_entities_page(self, navigation, heading, items, cardType):