made pipelines runtime non interactive (#31)

Author: ThatAmatoGuy

cmd/pipelines.go

@@ -32,52 +32,69 @@ import (
 	"github.com/spf13/cobra"
 )
 
+var pipelinesNamespace string
+var pipelinesRuntime string
+
 // pipelinesCmd represents the pipelines command
 var pipelinesCmd = &cobra.Command{
 	Use:   "pipelines",
 	Short: "Collect data for the Codefresh Pipelines Runtime",
 	Long:  `Collect data for the Codefresh Pipelines Runtime`,
 	Run: func(cmd *cobra.Command, args []string) {
 		const RuntimeType = "Codefresh Pipelines Runtime"
+		var runtimes []map[string]interface{}
+		var reSpec map[string]interface{}
+
 		dirPath := fmt.Sprintf("./codefresh-support-%d", time.Now().Unix())
+
 		cfConfig, err := codefresh.GetCodefreshCreds()
 		if err != nil {
 			cmd.PrintErrln("Error getting Codefresh credentials:", err)
-			return
 		}
 
-		runtimes, err := codefresh.AccountRuntimes(cfConfig)
-		if err != nil {
-			cmd.PrintErrln("Error getting Codefresh runtimes:", err)
-			return
-		}
+		if cfConfig != nil {
+			if pipelinesRuntime != "" {
+				reSpec, err = codefresh.SingleRuntime(cfConfig, pipelinesRuntime)
+				if err != nil {
+					cmd.PrintErrln("Error getting Codefresh runtimes:", err)
+				}
 
-		var pipelinesNamespace string
-		var reSpec map[string]interface{}
+			} else {
+				runtimes, err = codefresh.AccountRuntimes(cfConfig)
+				if err != nil {
+					cmd.PrintErrln("Error getting Codefresh runtimes:", err)
+				}
 
-		if len(runtimes) != 0 {
-			var selection int
-			for index, runtime := range runtimes {
-				cmd.Printf("%d. %s\n", index+1, runtime["metadata"].(map[string]interface{})["name"])
-			}
-			for {
-				cmd.Print("\nPlease select the runtime to gather data from (Number): ")
-				_, err := fmt.Scanf("%d", &selection)
-				if err != nil || selection < 1 || selection > len(runtimes) {
-					cmd.PrintErrln("Invalid selection. Please enter a number corresponding to one of the listed runtimes.")
-					continue
+				if len(runtimes) != 0 {
+					var selection int
+					for index, runtime := range runtimes {
+						cmd.Printf("%d. %s\n", index+1, runtime["metadata"].(map[string]interface{})["name"])
+					}
+					for {
+						cmd.Print("\nPlease select the runtime to gather data from (Number): ")
+						_, err := fmt.Scanf("%d", &selection)
+						if err != nil || selection < 1 || selection > len(runtimes) {
+							cmd.PrintErrln("Invalid selection. Please enter a number corresponding to one of the listed runtimes.")
+							continue
+						}
+						break
+
+					}
+					reSpec = runtimes[selection-1]
 				}
-				break
 
 			}
-			reSpec = runtimes[selection-1]
-			pipelinesNamespace = reSpec["runtimeScheduler"].(map[string]interface{})["cluster"].(map[string]interface{})["namespace"].(string)
-		} else {
-			cmd.Println("No runtimes found in Codefresh account.")
-			pipelinesNamespace, err = k8s.SelectNamespace(RuntimeType)
-			if err != nil {
-				cmd.PrintErrf("error getting Kubernetes namespace: %v", err)
-				return
+		}
+
+		if pipelinesNamespace == "" {
+			if reSpec != nil {
+				pipelinesNamespace = reSpec["runtimeScheduler"].(map[string]interface{})["cluster"].(map[string]interface{})["namespace"].(string)
+			} else {
+				pipelinesNamespace, err = k8s.SelectNamespace(RuntimeType)
+				if err != nil {
+					cmd.PrintErrf("error getting Kubernetes namespace: %v", err)
+					return
+				}
 			}
 		}
 
@@ -119,4 +136,6 @@ func init() {
 	// Cobra supports local flags which will only run when this command
 	// is called directly, e.g.:
 	// pipelinesCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
+	pipelinesCmd.Flags().StringVarP(&pipelinesNamespace, "namespace", "n", "", "The namespace where the Runtime is installed")
+	pipelinesCmd.Flags().StringVarP(&pipelinesRuntime, "runtime", "re", "", "The name of the Pipelines Runtime")
 }

internal/codefresh/account_runtimes.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"net/http"
+	"net/url"
 )
 
 func AccountRuntimes(cfConfig *CodefreshConfig) ([]map[string]interface{}, error) {
@@ -33,3 +34,32 @@ func AccountRuntimes(cfConfig *CodefreshConfig) ([]map[string]interface{}, error
 
 	return runtimes, nil
 }
+
+func SingleRuntime(cfConfig *CodefreshConfig, runtimeName string) (map[string]interface{}, error) {
+
+	req, err := http.NewRequest("GET", fmt.Sprintf("%s/runtime-environments/%s", cfConfig.BaseURL, url.PathEscape(runtimeName)), nil)
+	if err != nil {
+		return nil, err
+	}
+	for key, value := range cfConfig.Headers {
+		req.Header.Set(key, value)
+	}
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("failed to get runtime environment: %s", resp.Status)
+	}
+
+	var runtime map[string]interface{}
+	if err := json.NewDecoder(resp.Body).Decode(&runtime); err != nil {
+		return nil, err
+	}
+
+	return runtime, nil
+}