From 345f12530bf2268d58adadb538e3b90eab9e509f Mon Sep 17 00:00:00 2001
From: Carlos Martino <soy.carlosmartino@gmail.com>
Date: Fri, 27 Mar 2026 19:46:23 -0300
Subject: [PATCH] Add HA watchdog

---
 setup_homelogic.sh | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/setup_homelogic.sh b/setup_homelogic.sh
index 22f5259..10109d8 100755
--- a/setup_homelogic.sh
+++ b/setup_homelogic.sh
@@ -229,9 +229,10 @@ main() {
     printf "  1) Run ALL Steps (Recommended)\n"
     printf "  2) Run Main Steps Only\n"
     printf "  3) Run Optional Steps Only\n"
+    printf "  4) Install/Uninstall HA Watchdog\n"
     printf "  0) Exit\n"
     printf "${BLUE}=========================================${NC}\n"
-    printf "${YELLOW}Select an option [0-3]: ${NC}"
+    printf "${YELLOW}Select an option [0-4]: ${NC}"
     
     read opt
     case "$opt" in
@@ -239,7 +240,7 @@ main() {
             printf "\nExiting setup.\n"
             exit 0
             ;;
-        1|2|3)
+        1|2|3|4)
             setup_ui
             ;;
         *)
@@ -252,12 +253,12 @@ main() {
     
     case "$opt" in
         1)
+            update_status "$LINE_HA_WATCHDOG" "Skipped"
             setup_server
             setup_docker
             setup_zsh
             ask_optional_step "$LINE_TAILSCALE" "Tailscale" setup_tailscale
             ask_optional_step "$LINE_WIFITUI" "Wifitui" setup_wifitui
-            ask_optional_step "$LINE_HA_WATCHDOG" "HA Watchdog" setup_ha_watchdog
             ;;
         2) 
             update_status "$LINE_TAILSCALE" "Skipped"
@@ -271,9 +272,17 @@ main() {
             update_status "$LINE_SERVER" "Skipped"
             update_status "$LINE_DOCKER" "Skipped"
             update_status "$LINE_ZSH" "Skipped"
+            update_status "$LINE_HA_WATCHDOG" "Skipped"
             ask_optional_step "$LINE_TAILSCALE" "Tailscale" setup_tailscale
             ask_optional_step "$LINE_WIFITUI" "Wifitui" setup_wifitui
-            ask_optional_step "$LINE_HA_WATCHDOG" "HA Watchdog" setup_ha_watchdog
+            ;;
+        4)
+            update_status "$LINE_SERVER" "Skipped"
+            update_status "$LINE_DOCKER" "Skipped"
+            update_status "$LINE_ZSH" "Skipped"
+            update_status "$LINE_TAILSCALE" "Skipped"
+            update_status "$LINE_WIFITUI" "Skipped"
+            setup_ha_watchdog "$LINE_HA_WATCHDOG"
             ;;
     esac