Add PRO flag/switch to k3sup installer

Signed-off-by: Alex Ellis (OpenFaaS Ltd) <alexellis2@gmail.com>

Author: alexellis

README.md

@@ -175,9 +175,12 @@ The recommended option is to use K3sup CE to obtain K3sup Pro:
 
 ```bash
 # Get the latest K3sup CE version
-curl -sLS https://get.k3sup.dev | sudo sh
+PRO=1 curl -sLS https://get.k3sup.dev | sudo sh
+```
+
+Or, if you already have the latest K3sup CE version, it can replace itself:
 
-# Use K3sup CE to replace itself with the Pro edition
+```bash
 sudo k3sup get pro
 ```
 

get.sh

@@ -12,6 +12,23 @@ export REPO=k3sup
 export BINLOCATION="/usr/local/bin"
 export SUCCESS_CMD="$BINLOCATION/$REPO version"
 
+GET_PRO=false
+
+if [ "$PRO" = "true" ]; then
+    GET_PRO=true
+fi
+
+if [ "$PRO" = "1" ]; then
+   GET_PRO=true
+fi
+
+if [ "$GET_PRO" = "true" ]; then
+    echo "PRO=true"
+    echo ""
+    echo "Plan: download K3sup CE then upgrade to K3sup Pro"
+    echo ""
+fi
+
 ###############################
 # Content common across repos #
 ###############################
@@ -185,6 +202,7 @@ getPackage() {
     fi
 }
 
+
 thanks() {
     echo
     echo "================================================================"
@@ -197,4 +215,11 @@ thanks() {
 
 hasCli
 getPackage
-thanks
+
+if [ "$GET_PRO" = "false" ]; then
+    thanks
+else
+    echo "Upgrading to K3sup Pro"
+
+    $BINLOCATION/$REPO get pro
+fi