From 54f341b1ff0a1571804a6e2f515991aa611896b1 Mon Sep 17 00:00:00 2001
From: David Schroeder <david@schroedercity.com>
Date: Wed, 24 Jan 2024 19:16:39 -0600
Subject: [PATCH] Update powerwall.sh

---
 powerwall.sh | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/powerwall.sh b/powerwall.sh
index a61a4fcd..cacb998c 100755
--- a/powerwall.sh
+++ b/powerwall.sh
@@ -584,8 +584,13 @@ HOSTMGMT(){
 	
 	([ "${HOST}" == "" ] || ([ "${POWER}" == "" ] && [ "${ESXI}" == "" ])) && echo -e "Invalid command arguments\n" && exit 1
 	
-	if [ "${POWER^^}" == "OFF" ] || [ "${ESXI^^}" == "OFF" ]; then
-		read -n 1 -p "Are you sure you wish to power down '${ESXI_HOST}' ? (y/N): " choice; echo
+	if [ "${POWER^^}" == "OFF" ] || [ "${ESXI^^}" == "OFF" ] || [ ${MM} -eq 1 ]; then
+		if [ ${MM} -eq 1 ]; then
+			read -n 1 -p "Are you sure you wish to put '${ESXI_HOST}' into maintenance mode ? (y/N): " choice
+		else
+			read -n 1 -p "Are you sure you wish to power down '${ESXI_HOST}' ? (y/N): " choice
+		fi
+		echo
 		[ "${choice^^}" != "Y" ] && exit 0
 	fi