From 4338fae15f58d6a5b7bc3ffd6e0024c41f505ea5 Mon Sep 17 00:00:00 2001
From: Marcus Noble <github@marcusnoble.co.uk>
Date: Thu, 2 Feb 2023 15:33:07 +0000
Subject: [PATCH] Updated flux-refresh

Signed-off-by: Marcus Noble <github@marcusnoble.co.uk>
---
 home/.bin/{flux-refresh-all => flux-refresh} | 36 ++++++++++++++++----
 1 file changed, 29 insertions(+), 7 deletions(-)
 rename home/.bin/{flux-refresh-all => flux-refresh} (62%)

diff --git a/home/.bin/flux-refresh-all b/home/.bin/flux-refresh
similarity index 62%
rename from home/.bin/flux-refresh-all
rename to home/.bin/flux-refresh
index 24970ad..31b2362 100755
--- a/home/.bin/flux-refresh-all
+++ b/home/.bin/flux-refresh
@@ -5,14 +5,18 @@ source .utils
 set -e
 
 VERBOSE=""
+TARGET_RESOURCE="all"
+NAMESPACES="-A"
 
 print_usage() {
-  blue "flux-refresh-all - Refresh all flux resources"
+  blue "flux-refresh - Refresh all flux resources"
   echo " "
   underline "Usage:"
-  echo "flux-refresh-all [options]"
+  echo "flux-refresh [options]"
   echo " "
   underline "Options:"
+  echo "-t, --type            the resource type to target. Valid options: gitrepo, helmrepository, kustomization, helmrelease & all. Default: all"
+  echo "-n, --namespace       the namespace resources belong in. Default: all namespaces"
   echo "    --verbose         show full verbose output"
   echo "-h, --help            show this help text"
 }
@@ -23,20 +27,38 @@ while test $# -gt 0; do
       print_usage
       exit 0
       ;;
+    -t|--type)
+      shift
+      TARGET_RESOURCE=$1
+      shift
+      ;;
+    -n|--namespace)
+      shift
+      NAMESPACES="-n $1"
+      shift
+      ;;
     --verbose)
       VERBOSE="true"
       shift
       ;;
     *)
-      break
+      shift
       ;;
   esac
 done
 
-GITREPOS=$(kubectl get gitrepo -A -o json | jq -r '.items[] | "\(.metadata.namespace)/\( .kind)/\( .metadata.name)"')
-HELMREPOS=$(kubectl get helmrepository -A -o json | jq -r '.items[] | "\(.metadata.namespace)/\( .kind)/\( .metadata.name)"')
-KUSTOMIZATIONS=$(kubectl get kustomization -A -o json | jq -r '.items[] | "\(.metadata.namespace)/\( .kind)/\( .metadata.name)"')
-HEALMRELEASES=$(kubectl get helmrelease -A -o json | jq -r '.items[] | "\(.metadata.namespace)/\( .kind)/\( .metadata.name)"')
+if [[ "${TARGET_RESOURCE}" == "all" || "${TARGET_RESOURCE}" == "gitrepo" ]]; then
+  GITREPOS=$(kubectl get gitrepo ${NAMESPACES} -o json | jq -r '.items[] | "\(.metadata.namespace)/\( .kind)/\( .metadata.name)"')
+fi
+if [[ "${TARGET_RESOURCE}" == "all" || "${TARGET_RESOURCE}" == "helmrepository" ]]; then
+  HELMREPOS=$(kubectl get helmrepository ${NAMESPACES} -o json | jq -r '.items[] | "\(.metadata.namespace)/\( .kind)/\( .metadata.name)"')
+fi
+if [[ "${TARGET_RESOURCE}" == "all" || "${TARGET_RESOURCE}" == "kustomization" ]]; then
+  KUSTOMIZATIONS=$(kubectl get kustomization ${NAMESPACES} -o json | jq -r '.items[] | "\(.metadata.namespace)/\( .kind)/\( .metadata.name)"')
+fi
+if [[ "${TARGET_RESOURCE}" == "all" || "${TARGET_RESOURCE}" == "helmrelease" ]]; then
+  HEALMRELEASES=$(kubectl get helmrelease ${NAMESPACES} -o json | jq -r '.items[] | "\(.metadata.namespace)/\( .kind)/\( .metadata.name)"')
+fi
 
 if [[ "${GITREPOS}" != "" ]]; then
   blue "Refreshing GitRepositories"