From 5c64b395ddc247300a79541984e41cc037a4b7bd Mon Sep 17 00:00:00 2001
From: Jake Nabasny <jake.nabasny@canonical.com>
Date: Thu, 31 Oct 2024 13:35:22 -0500
Subject: [PATCH] add k8s bootstrap script

---
 tools/bootstrap-k8s-cloud.sh | 37 ++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100755 tools/bootstrap-k8s-cloud.sh

diff --git a/tools/bootstrap-k8s-cloud.sh b/tools/bootstrap-k8s-cloud.sh
new file mode 100755
index 00000000..232f7693
--- /dev/null
+++ b/tools/bootstrap-k8s-cloud.sh
@@ -0,0 +1,37 @@
+#!/bin/bash -x
+check_kubeconfig() {
+	if [[ -z "$KUBECONFIG" && ! -f "${HOME}/.kube/config" ]]; then
+		echo 'ERROR: Cannot find KUBECONFIG! Set as environment variable.'
+		exit 1
+	fi
+}
+
+get_storageclass() {
+	default_storageclass="$(kubectl get sc | grep default | awk '{print $1}')"
+	if [[ -z "$default_storageclass" ]]; then
+		enable_storage
+	fi
+}
+
+enable_storage() {
+	if leader="$(juju status --format json| jq -r '.applications["microk8s"].units|to_entries[]|select(.value["leader"])|.key' 2> /dev/null)"; then
+		echo 'Enabling hostpath-storage on Microk8s.'
+		juju ssh "$leader" sudo microk8s enable hostpath-storage
+	elif leader="$(juju status --format json| jq -r '.applications["k8s"].units|to_entries[]|select(.value["leader"])|.key' 2> /dev/null)"; then
+		echo 'Enabling local-storage on Canonical K8s.'
+		juju ssh "$leader" sudo k8s enable local-storage
+	else
+		echo 'ERROR: No default storage class in Kubernetes. Try deploying a k8s bundle with --ceph.'
+		exit 1
+	fi
+}
+
+bootstrap_k8s() { 
+	juju add-k8s k8s-cloud --client
+	juju bootstrap k8s-cloud
+	juju add-model secloud
+}
+
+check_kubeconfig
+get_storageclass
+bootstrap_k8s