From 1c77cb7073f5d8df3240f17028cfa9b3d48d5695 Mon Sep 17 00:00:00 2001
From: randomshell <randshell@protonmail.com>
Date: Thu, 2 Jul 2020 09:18:43 +0000
Subject: [PATCH] Change public IP detection

The sed command is easier to understand than multiple grep

If IPv4 isn't available use IPv6
---
 wireguard-install.sh | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/wireguard-install.sh b/wireguard-install.sh
index de52c84..401df86 100644
--- a/wireguard-install.sh
+++ b/wireguard-install.sh
@@ -54,9 +54,13 @@ function installWireGuard() {
 	echo "I need to ask you a few questions before starting the setup."
 	echo "You can leave the default options and just press enter if you are ok with them."
 
-	# Detect public IPv4 address and pre-fill for the user
-	SERVER_PUB_IPV4=$(ip addr | grep 'inet' | grep -v inet6 | grep -vE '127\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}' | grep -oE '[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}' | head -1)
-	read -rp "IPv4 or IPv6 public address: " -e -i "$SERVER_PUB_IPV4" SERVER_PUB_IP
+	# Detect public IPv4 or IPv6 address and pre-fill for the user
+	SERVER_PUB_IP=$(ip -4 addr | sed -ne 's|^.* inet \([^/]*\)/.* scope global.*$|\1|p' | head -1)
+	if [[ -z $SERVER_PUB_IP ]]; then
+		# Detect public IPv6 address
+		SERVER_PUB_IP=$(ip -6 addr | sed -ne 's|^.* inet6 \([^/]*\)/.* scope global.*$|\1|p' | head -1)
+	fi
+	read -rp "IPv4 or IPv6 public address: " -e -i "$SERVER_PUB_IP" SERVER_PUB_IP
 
 	# Detect public interface and pre-fill for the user
 	SERVER_PUB_NIC="$(ip -4 route ls | grep default | grep -Po '(?<=dev )(\S+)' | head -1)"