From 6bf9e68f89aca2a64687be82802099de365988f8 Mon Sep 17 00:00:00 2001
From: tteckster <tteckster@gmail.com>
Date: Mon, 27 Mar 2023 20:24:49 -0400
Subject: [PATCH] Update haos-vm.sh

add SSH_CHECK
---
 vm/haos-vm.sh | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/vm/haos-vm.sh b/vm/haos-vm.sh
index 5160d8be..87cf3b3e 100644
--- a/vm/haos-vm.sh
+++ b/vm/haos-vm.sh
@@ -95,6 +95,18 @@ function ARCH_CHECK() {
     exit
   fi
 }
+function SSH_CHECK() {
+  if command -v pveversion >/dev/null 2>&1; then
+    if [ -n "$SSH_CLIENT" ]; then
+      if whiptail --defaultno --title "SSH DETECTED" --yesno "It's suggested to use the Proxmox shell instead of SSH, since SSH can create issues while gathering variables. Would you like to proceed with using SSH?" 10 62; then
+        echo "you've been warned"
+      else
+        clear
+        exit
+      fi
+    fi
+  fi
+}
 function default_settings() {
   BRANCH="$stable"
   VMID="$NEXTID"
@@ -257,6 +269,7 @@ function START_SCRIPT() {
 }
 ARCH_CHECK
 PVE_CHECK
+SSH_CHECK
 START_SCRIPT
 msg_info "Validating Storage"
 while read -r line; do