From 2c79888a1cadd7c465b47d16f672ab9a83e46d62 Mon Sep 17 00:00:00 2001
From: Stefan Reimer <stefan@zero-downtime.net>
Date: Wed, 5 Jan 2022 13:34:48 +0000
Subject: [PATCH] feat: inject default namespaces during bootstrap

---
 charts/kubezero/bootstrap.sh | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/charts/kubezero/bootstrap.sh b/charts/kubezero/bootstrap.sh
index dfcf4e7d..2b020dc9 100755
--- a/charts/kubezero/bootstrap.sh
+++ b/charts/kubezero/bootstrap.sh
@@ -33,7 +33,7 @@ function wait_for() {
 
 function chart_location() {
   if [ -z "$LOCATION" ]; then
-    echo "$1 --repo https://zero-down-time.github.io/kubezero"
+    echo "$1 --repo https://cdn.zero-downtime.net/charts"
   else
     echo "$LOCATION/$1"
   fi
@@ -72,15 +72,20 @@ function _crds() {
 # helm template | kubectl apply -f -
 # confine to one namespace if possible
 function apply(){
-  helm template $(chart_location $chart) -n $namespace --name-template $module $targetRevision --skip-crds -f $TMPDIR/values.yaml $API_VERSIONS $KUBE_VERSION $@ > $TMPDIR/helm.yaml
+  helm template $(chart_location $chart) -n $namespace --name-template $module $targetRevision --skip-crds -f $TMPDIR/values.yaml $API_VERSIONS $KUBE_VERSION $@ \
+    | python3 -c '
+#!/usr/bin/python3
+import yaml
+import sys
 
-  # If resources are in more than ONE $namespace, apply without restrictions
-  nr_ns=$(grep -e '^  namespace:' $TMPDIR/helm.yaml  | sed "s/\"//g" | sort | uniq | wc -l)
-  if [ $nr_ns -gt 1 ]; then
-    kubectl $action -f $TMPDIR/helm.yaml && rc=$? || rc=$?
-  else
-    kubectl $action --namespace $namespace -f $TMPDIR/helm.yaml && rc=$? || rc=$?
-  fi
+for manifest in yaml.safe_load_all(sys.stdin):
+    if manifest:
+        if "metadata" in manifest and "namespace" not in manifest["metadata"]:
+            manifest["metadata"]["namespace"] = sys.argv[1]
+        print("---")
+        print(yaml.dump(manifest))' $namespace > $TMPDIR/helm.yaml
+
+  kubectl $action -f $TMPDIR/helm.yaml && rc=$? || rc=$?
 }