From c6ebc004c600b9f35b4b3e4bece86b10395d99e6 Mon Sep 17 00:00:00 2001
From: abhishek9686 <abhi281342@gmail.com>
Date: Tue, 27 Aug 2024 09:19:16 +0530
Subject: [PATCH] collect tenant id and license on pro install

---
 scripts/nm-quick.sh | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/scripts/nm-quick.sh b/scripts/nm-quick.sh
index e811af46..6fe6da10 100755
--- a/scripts/nm-quick.sh
+++ b/scripts/nm-quick.sh
@@ -19,6 +19,7 @@ unset BUILD_TAG
 unset IMAGE_TAG
 unset NETMAKER_BASE_DOMAIN
 unset UPGRADE_FLAG
+unset COLLECT_PRO_VARS
 # usage - displays usage instructions
 usage() {
 	echo "nm-quick.sh v$NM_QUICK_VERSION"
@@ -568,7 +569,17 @@ set_install_vars() {
 	EMAIL="$GET_EMAIL"
 
 	wait_seconds 1
-
+	if [ "$COLLECT_PRO_VARS" = "true" ]; then
+		unset LICENSE_KEY
+		while [ -z "$LICENSE_KEY" ]; do
+			read -p "License Key: " LICENSE_KEY
+		done
+		unset TENANT_ID
+		while [ -z ${TENANT_ID} ]; do
+			read -p "Tenant ID: " TENANT_ID
+		done
+	fi
+	wait_seconds 1
 	unset GET_MQ_USERNAME
 	unset GET_MQ_PASSWORD
 	unset CONFIRM_MQ_PASSWORD
@@ -891,6 +902,7 @@ main (){
 		;;
 	p)
 		INSTALL_TYPE="pro"
+		COLLECT_PRO_VARS="true
 		;;
 	v)
 		usage