From 4133d38062af6ac39c8b4b0c2f212bc27107fc7d Mon Sep 17 00:00:00 2001
From: tteckster <tteckster@gmail.com>
Date: Sun, 2 Jul 2023 02:54:37 -0400
Subject: [PATCH] Update post-pve-install.sh

add Enable high availability
---
 misc/post-pve-install.sh | 27 +++++++++++++++++++++------
 1 file changed, 21 insertions(+), 6 deletions(-)

diff --git a/misc/post-pve-install.sh b/misc/post-pve-install.sh
index fcf634df..b3fb1bf6 100644
--- a/misc/post-pve-install.sh
+++ b/misc/post-pve-install.sh
@@ -162,12 +162,9 @@ EOF
     case $CHOICE in
     yes)
       msg_info "Disabling high availability"
-      systemctl stop pve-ha-lrm
-      systemctl disable pve-ha-lrm &>/dev/null
-      systemctl stop pve-ha-crm
-      systemctl disable pve-ha-crm &>/dev/null
-      systemctl stop corosync
-      systemctl disable corosync &>/dev/null
+      systemctl disable -q --now pve-ha-lrm
+      systemctl disable -q --now pve-ha-crm
+      systemctl disable -q --now corosync
       msg_ok "Disabled high availability"
       ;;
     no)
@@ -176,6 +173,24 @@ EOF
     esac
   fi
 
+  if ! systemctl is-active --quiet pve-ha-lrm; then
+    CHOICE=$(whiptail --title "HIGH AVAILABILITY" --menu "Enable high availability?" 10 58 2 \
+      "yes" " " \
+      "no" " " 3>&2 2>&1 1>&3)
+    case $CHOICE in
+    yes)
+      msg_info "Enabling high availability"
+      systemctl enable -q --now pve-ha-lrm
+      systemctl enable -q --now pve-ha-crm
+      systemctl enable -q --now corosync
+      msg_ok "Enabled high availability"
+      ;;
+    no)
+      msg_error "Selected no to Enabling high availability"
+      ;;
+    esac
+  fi
+  
   CHOICE=$(whiptail --title "UPDATE" --menu "\nUpdate Proxmox VE now?" 11 58 2 \
     "yes" " " \
     "no" " " 3>&2 2>&1 1>&3)