From bfd977f5731617287e5bb3f9b17b9b0417265623 Mon Sep 17 00:00:00 2001
From: dinger1986 <daniel.lamb@flonix.co.uk>
Date: Wed, 10 Aug 2022 18:19:46 +0100
Subject: [PATCH] Update install.sh

---
 install.sh | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/install.sh b/install.sh
index d2bb2bc..c372506 100644
--- a/install.sh
+++ b/install.sh
@@ -4,6 +4,26 @@
 uname=$(whoami)
 admintoken=$(head /dev/urandom | tr -dc A-Za-z0-9 | head -c16)
 
+# Choice for DNS or IP
+PS3='Choose your preferred option, IP or DNS/Domain: '
+WAN=("IP" "DNS/Domain")
+select WANOPT in "${WAN[@]}"; do
+    case $WANOPT in
+        "IP")
+	   wanip=$(dig @resolver4.opendns.com myip.opendns.com +short)
+	   echo $wanip
+		break
+            ;;
+
+        "DNS/Domain")
+	    echo -ne "Enter your preferred domain/dns address ${NC}: "
+        read wanip
+		echo $wanip
+		break
+		;;
+        *) echo "invalid option $REPLY";;
+    esac
+done
 
 # Setup prereqs for server
 if [[ $(which yum) ]]; then
@@ -94,9 +114,6 @@ while ! [[ $CHECK_RUSTDESK_READY ]]; do
   sleep 3
 done
 
-#Get WAN IP
-wanip=$(dig @resolver4.opendns.com myip.opendns.com +short)
-
 pubname=$(find /opt/rustdesk -name *.pub)
 key=$(cat "${pubname}")