From 758825ed8ed05c721b8dc23aaa437355093fb230 Mon Sep 17 00:00:00 2001
From: Yolanda Robla <yroblamo@redhat.com>
Date: Sat, 25 May 2019 09:29:09 +0200
Subject: [PATCH] Add new endpoint to generate workloads

Change-Id: Iadce1486585781c68150943ba35f8fac90b90d37
---
 Makefile         |   6 +++-
 cmd/workloads.go | 108 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 113 insertions(+), 1 deletion(-)
 create mode 100644 cmd/workloads.go

diff --git a/Makefile b/Makefile
index 7b31542..aff892a 100644
--- a/Makefile
+++ b/Makefile
@@ -28,6 +28,10 @@ ifndef SSH_KEY_PATH
 override SSH_KEY_PATH = "${HOME}/.ssh/id_rsa"
 endif
 
+ifndef CLUSTER_CREDENTIALS
+override CLUSTER_CREDENTIALS="$(shell pwd)/build/auth/kubeconfig"
+endif
+
 all: watch
 
 binary:
@@ -53,7 +57,7 @@ deploy: dependencies
 	$(MAKE) workloads
 
 workloads:
-	#@./bin/$(GONAME) workloads --site_repository $(SITE_REPO)
+	@./bin/$(GONAME) workloads --site_repository $(SITE_REPO) --cluster_credentials $(CLUSTER_CREDENTIALS)
 
 images:
 	@echo "Launching image generation"
diff --git a/cmd/workloads.go b/cmd/workloads.go
new file mode 100644
index 0000000..18266d8
--- /dev/null
+++ b/cmd/workloads.go
@@ -0,0 +1,108 @@
+// Copyright © 2019 Red Hat <yroblamo@redhat.com>
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package cmd
+
+import (
+	"fmt"
+	"io/ioutil"
+	"log"
+	"os"
+	"os/exec"
+
+	getter "github.com/hashicorp/go-getter"
+	"github.com/spf13/cobra"
+)
+
+// generateWorkloads will run kustomize and apply the generated workloads on the cluster
+func generateWorkloads(siteRepository string, buildPath string, clusterCredentials string) {
+	// Clone the site repository
+	log.Println("Cloning the site repository")
+	siteBuildPath := fmt.Sprintf("%s/site", buildPath)
+	client := &getter.Client{Src: siteRepository, Dst: siteBuildPath, Mode: getter.ClientModeAny}
+	err := client.Get()
+	if err != nil {
+		log.Fatal(fmt.Sprintf("Error cloning site repository: %s", err))
+	}
+
+	// apply kustomize on the given path
+	log.Println("Generating workloads")
+	workloadsPath := fmt.Sprintf("%s/workloads", siteBuildPath)
+	cmd := exec.Command("kustomize", "build", workloadsPath)
+	out, err := cmd.CombinedOutput()
+	if err != nil {
+		log.Fatal(fmt.Sprintf("Error calling kustomize: %s - %s", err, string(out)))
+		os.Exit(1)
+	}
+	outfile, err := os.Create(fmt.Sprintf("%s/content.yaml", workloadsPath))
+	if err != nil {
+		log.Fatal(fmt.Sprintf("Error storing output from kustomize: %s", err))
+		os.Exit(1)
+	}
+	_, err = outfile.WriteString(fmt.Sprintf("%s", out))
+	if err != nil {
+		log.Fatal(fmt.Sprintf("Error writing kustomize file: %s", err))
+	}
+	defer outfile.Close()
+
+	// and now apply the content of the file
+	log.Println("Applying the generated workloads")
+	cmd = exec.Command("oc", "apply", "-f", fmt.Sprintf("%s/content.yaml", workloadsPath))
+	cmd.Env = os.Environ()
+	cmd.Env = append(cmd.Env, fmt.Sprintf("KUBECONFIG=%s", clusterCredentials))
+	out, err = cmd.CombinedOutput()
+	if err != nil {
+		log.Fatal(fmt.Sprintf("Error applying kustomize workloads: %s - %s", err, out))
+		os.Exit(1)
+	}
+	log.Println(fmt.Sprintf("%s", out))
+}
+
+// workloadsCmd represents the workloads command
+var workloadsCmd = &cobra.Command{
+	Use:              "workloads",
+	Short:            "Command to apply workloads on a working cluster",
+	Long:             ``,
+	TraverseChildren: true,
+	Run: func(cmd *cobra.Command, args []string) {
+		// retrieve config values and start generation
+		siteRepository, _ := cmd.Flags().GetString("site_repository")
+
+		// Check if build path exists, create if not
+		buildPath, _ := cmd.Flags().GetString("build_path")
+		if len(buildPath) == 0 {
+			// will generate a temporary directory
+			buildPath, _ = ioutil.TempDir("/tmp", "kni")
+		} else {
+			// remove if exists, recreate
+			os.RemoveAll(buildPath)
+			os.MkdirAll(buildPath, 0775)
+		}
+
+		clusterCredentials, _ := cmd.Flags().GetString("cluster_credentials")
+
+		// start generation process
+		generateWorkloads(siteRepository, buildPath, clusterCredentials)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(workloadsCmd)
+
+	workloadsCmd.Flags().StringP("site_repository", "", "", "Url for the specific site workloads folder")
+	workloadsCmd.MarkFlagRequired("site_repository")
+	workloadsCmd.Flags().StringP("cluster_credentials", "", "", "The credentials to use to access the cluster")
+	workloadsCmd.MarkFlagRequired("cluster_credentials")
+
+}
-- 
2.16.6