From 0ce94a81db5a578818571c6ec1d01321df1f2489 Mon Sep 17 00:00:00 2001
From: Zoran Zorica <zoran.zorica@gmail.com>
Date: Sat, 20 Jan 2024 00:01:13 +0100
Subject: [PATCH] fix: dont try helm with --remove arg

---
 setup.sh | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/setup.sh b/setup.sh
index 05cd006..f658b5b 100755
--- a/setup.sh
+++ b/setup.sh
@@ -390,13 +390,16 @@ EOF'
 cleanup() {
     calling_function="cleanup"
     echo -e "${INFO} Cleaning up..."
-    check_helm_release watkins watkins
+
+    if [[ $1 != "--remove" ]]; then
+        check_helm_release watkins watkins
+    fi
     rm -rf ingress-values.yaml \
         longhorn-values.yaml \
         watkins-values.yaml
 }
 
-trap '[[ $1 != "--version" && $1 != "--help" ]] && cleanup' EXIT
+trap '[[ $1 != "--version" && $1 != "--help" ]] && cleanup "$1"' EXIT
 
 # Install k3s and setup kubeconfig
 install_k3s() {