From efab18fa50d08599f396951c3f12c82db41bd2f5 Mon Sep 17 00:00:00 2001
From: divolgin <dmitriy@replicated.com>
Date: Wed, 25 Nov 2020 00:55:36 +0000
Subject: [PATCH] Missing arguments for upgrade.sh script

---
 scripts/common/addon.sh   |  6 +++++-
 scripts/common/upgrade.sh | 22 ++++++++++++++++++----
 2 files changed, 23 insertions(+), 5 deletions(-)

diff --git a/scripts/common/addon.sh b/scripts/common/addon.sh
index 2ed6508ad3..e0805bb9d3 100644
--- a/scripts/common/addon.sh
+++ b/scripts/common/addon.sh
@@ -125,7 +125,11 @@ function addon_outro() {
         fi
 
         printf "\n${YELLOW}Run this script on all remote nodes to apply changes${NC}\n"
-        printf "\n\t${GREEN}${prefix}upgrade.sh | sudo bash -s${dockerRegistryIP}${noProxyAddrs}${NC}\n\n"
+        if [ "$AIRGAP" = "1" ]; then
+            printf "\n\t${GREEN}${prefix}upgrade.sh | sudo bash -s airgap ${dockerRegistryIP}${noProxyAddrs}${NC}\n\n"
+        else
+            printf "\n\t${GREEN}${prefix}upgrade.sh | sudo bash -s${dockerRegistryIP}${noProxyAddrs}${NC}\n\n"
+        fi
         printf "Press enter to proceed\n"
         prompt
 
diff --git a/scripts/common/upgrade.sh b/scripts/common/upgrade.sh
index 2189d64217..51b977229f 100644
--- a/scripts/common/upgrade.sh
+++ b/scripts/common/upgrade.sh
@@ -126,10 +126,17 @@ function upgrade_kubernetes_remote_node_patch() {
         continue
     fi
 
+    DOCKER_REGISTRY_IP=$(kubectl -n kurl get service registry -o=jsonpath='{@.spec.clusterIP}' 2>/dev/null || echo "")
+
     printf "${YELLOW}Drain node $nodeName to prepare for upgrade? ${NC}"
     confirmY " "
     kubernetes_drain "$nodeName"
 
+    local dockerRegistryIP=""
+    if [ -n "$DOCKER_REGISTRY_IP" ]; then
+        dockerRegistryIP=" docker-registry-ip=$DOCKER_REGISTRY_IP"
+    fi
+
     local noProxyAddrs=""
     if [ -n "$NO_PROXY_ADDRESSES" ]; then
         noProxyAddrs=" additional-no-proxy-addresses=${NO_PROXY_ADDRESSES}"
@@ -138,9 +145,9 @@ function upgrade_kubernetes_remote_node_patch() {
     printf "\n\n\tRun the upgrade script on remote node to proceed: ${GREEN}$nodeName${NC}\n\n"
 
     if [ "$AIRGAP" = "1" ]; then
-        printf "\t${GREEN}cat upgrade.sh | sudo bash -s airgap kubernetes-version=${KUBERNETES_VERSION}${noProxyAddrs}${NC}\n\n"
+        printf "\t${GREEN}cat upgrade.sh | sudo bash -s airgap kubernetes-version=${KUBERNETES_VERSION}${noProxyAddrs}${dockerRegistryIP}${NC}\n\n"
     elif [ -z "$KURL_URL" ]; then
-        printf "\t${GREEN}cat upgrade.sh | sudo bash -s kubernetes-version=${KUBERNETES_VERSION}${noProxyAddrs}${NC}\n\n"
+        printf "\t${GREEN}cat upgrade.sh | sudo bash -s kubernetes-version=${KUBERNETES_VERSION}${noProxyAddrs}${dockerRegistryIP}${NC}\n\n"
     else
         local prefix="curl $KURL_URL/$INSTALLER_ID/"
         if [ -z "$KURL_URL" ]; then
@@ -231,10 +238,17 @@ function upgrade_kubernetes_remote_node_minor() {
         continue
     fi
 
+    DOCKER_REGISTRY_IP=$(kubectl -n kurl get service registry -o=jsonpath='{@.spec.clusterIP}' 2>/dev/null || echo "")
+
     printf "${YELLOW}Drain node $nodeName to prepare for upgrade? ${NC}"
     confirmY " "
     kubernetes_drain "$nodeName"
 
+    local dockerRegistryIP=""
+    if [ -n "$DOCKER_REGISTRY_IP" ]; then
+        dockerRegistryIP=" docker-registry-ip=$DOCKER_REGISTRY_IP"
+    fi
+
     local noProxyAddrs=""
     if [ -n "$NO_PROXY_ADDRESSES" ]; then
         noProxyAddrs=" additional-no-proxy-addresses=${NO_PROXY_ADDRESSES}"
@@ -243,9 +257,9 @@ function upgrade_kubernetes_remote_node_minor() {
     printf "\n\n\tRun the upgrade script on remote node to proceed: ${GREEN}$nodeName${NC}\n\n"
 
     if [ "$AIRGAP" = "1" ]; then
-        printf "\t${GREEN}cat upgrade.sh | sudo bash -s airgap kubernetes-version=${targetK8sVersion}${noProxyAddrs}${NC}\n\n"
+        printf "\t${GREEN}cat upgrade.sh | sudo bash -s airgap kubernetes-version=${targetK8sVersion}${noProxyAddrs}${dockerRegistryIP}${NC}\n\n"
     elif [ -z "$KURL_URL" ]; then
-        printf "\t${GREEN}cat upgrade.sh | sudo bash -s kubernetes-version=${targetK8sVersion}${noProxyAddrs}${NC}\n\n"
+        printf "\t${GREEN}cat upgrade.sh | sudo bash -s kubernetes-version=${targetK8sVersion}${noProxyAddrs}${dockerRegistryIP}${NC}\n\n"
     else
         local prefix="curl $KURL_URL/$INSTALLER_ID/"
         if [ -z "$KURL_URL" ]; then