From 44ef1faed867c77773c3ea3e10ef7bcaba346658 Mon Sep 17 00:00:00 2001
From: Kartik Joshi <kartikjoshi@microsoft.com>
Date: Wed, 31 Jan 2024 15:27:26 +0530
Subject: [PATCH] Provisioner: Add kbs provisioner in provisioner-cli

Fixes: #1471
Signed-off-by: Kartik Joshi <kartikjoshi@microsoft.com>
---
 test/tools/provisioner-cli/main.go | 39 +++++++++++++++++++++++++++++-
 1 file changed, 38 insertions(+), 1 deletion(-)

diff --git a/test/tools/provisioner-cli/main.go b/test/tools/provisioner-cli/main.go
index 3e3e85421f..3813c5e287 100644
--- a/test/tools/provisioner-cli/main.go
+++ b/test/tools/provisioner-cli/main.go
@@ -83,11 +83,38 @@ func main() {
 			}
 		}
 
+		props := provisioner.GetProperties(context.TODO(), cfg)
+		log.Info("Deploying kbs")
+		if props["KBS_IMAGE"] == "" || props["KBS_IMAGE_TAG"] == "" {
+			log.Fatal("kbs image not provided")
+		}
+
+		keyBrokerService, err := pv.NewKeyBrokerService(props["CLUSTER_NAME"])
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		if err = keyBrokerService.Deploy(context.TODO(), cfg, props); err != nil {
+			log.Fatal(err)
+		}
+
+		var kbsPodIP string
+		kbsPodIP, err = keyBrokerService.GetKbsSvcIP(context.TODO(), cfg)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		kbsparams := "cc_kbc::http:" + kbsPodIP + ":8080"
+		log.Infof("KBS PARAMS%s:", kbsparams)
+
+		props = provisioner.GetProperties(context.TODO(), cfg)
+		props["AA_KBC_PARAMS"] = kbsparams
+
 		cloudAPIAdaptor, err := pv.NewCloudAPIAdaptor(cloudProvider, installDirectory)
 		if err != nil {
 			log.Fatal(err)
 		}
-		if err := cloudAPIAdaptor.Deploy(context.TODO(), cfg, provisioner.GetProperties(context.TODO(), cfg)); err != nil {
+		if err := cloudAPIAdaptor.Deploy(context.TODO(), cfg, props); err != nil {
 			log.Fatal(err)
 		}
 	}
@@ -143,6 +170,16 @@ func main() {
 			log.Fatal(err)
 		}
 
+		props := provisioner.GetProperties(context.TODO(), cfg)
+		keyBrokerService, err := pv.NewKeyBrokerService(props["CLUSTER_NAME"])
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		if err = keyBrokerService.Delete(context.TODO(), cfg); err != nil {
+			log.Fatal(err)
+		}
+
 		kubeconfigPath := kconf.ResolveKubeConfigFile()
 		if kubeconfigPath == "" {
 			log.Fatal("Unabled to find a kubeconfig file")