refactor command get/list environment

Signed-off-by: Shivam Purohit <[email protected]>

Author: shivam-Purohit

pkg/apis/environment/environment.go

@@ -5,7 +5,6 @@ import (
 	"errors"
 	"io/ioutil"
 	"net/http"
-
 	models "github.com/litmuschaos/litmus/chaoscenter/graphql/server/graph/model"
 	"github.com/litmuschaos/litmusctl/pkg/apis"
 	"github.com/litmuschaos/litmusctl/pkg/types"
@@ -58,7 +57,6 @@ func GetEnvironmentList(pid string, cred types.Credentials) (ListEnvironmentData
 	if err != nil {
 		return ListEnvironmentData{}, err
 	}
-
 	resp, err := apis.SendRequest(
 		apis.SendRequestParams{
 			Endpoint: cred.ServerEndpoint + utils.GQLAPIPath,
@@ -67,11 +65,9 @@ func GetEnvironmentList(pid string, cred types.Credentials) (ListEnvironmentData
 		query,
 		string(types.Post),
 	)
-
 	if err != nil {
 		return ListEnvironmentData{}, errors.New("Error in Getting Chaos Environment List: " + err.Error())
 	}
-
 	bodyBytes, err := ioutil.ReadAll(resp.Body)
 	defer resp.Body.Close()
 	if err != nil {

pkg/apis/environment/query.go

@@ -15,6 +15,12 @@ const (
 	                 listEnvironments(projectID: $projectID,request: $request){
 						environments {
 							environmentID
+							name
+							createdAt
+							updatedAt
+							createdBy{
+								username
+							  }
 						}
 					}
 	               }`

pkg/cmd/get/environment.go

@@ -20,6 +20,9 @@ import (
 	"fmt"
 	"os"
 	"strings"
+	"time"
+	"text/tabwriter"
+	"strconv"
 	"github.com/litmuschaos/litmusctl/pkg/apis/environment"
 	"github.com/litmuschaos/litmusctl/pkg/utils"
 	"github.com/spf13/cobra"
@@ -70,18 +73,33 @@ var ChaosEnvironmentCmd = &cobra.Command{
 				os.Exit(1)
 			}
 		}
-		environmentListData := environmentList.Data.ListEnvironmentDetails.Environments;
-
-		for i :=0; i <= len(environmentListData); i++ {
+		environmentListData := environmentList.Data.ListEnvironmentDetails.Environments
+		writer := tabwriter.NewWriter(os.Stdout, 30, 8, 0,'\t', tabwriter.AlignRight)
+		writer.Flush()
+		utils.White_B.Fprintln(writer, "CHAOS ENVIRONMENT ID\tCHAOS ENVIRONMENT NAME\tCREATED AT\tCREATED BY")
+		for i :=0; i < len(environmentListData); i++ {
 			if(environmentListData[i].EnvironmentID == environmentID){
-				utils.PrintInJsonFormat(environmentListData[i]);
-				os.Exit(1)
+				intTime, err := strconv.ParseInt(environmentListData[i].CreatedAt, 10, 64)
+				if err != nil {
+					fmt.Println("Error converting CreatedAt to int64:", err)
+					continue
+				}
+				humanTime := time.Unix(intTime, 0)
+				writer.Flush()
+				utils.White.Fprintln(
+					writer,
+					environmentListData[i].EnvironmentID+"\t"+environmentListData[i].Name+"\t"+humanTime.String()+"\t"+environmentListData[i].CreatedBy.Username+"\t")
+				break
 			}
 		}
+		writer.Flush()
+		
+
 	},
 
 
 }
+
 func init() { 
 	GetCmd.AddCommand(ChaosEnvironmentCmd);
 	ChaosEnvironmentCmd.Flags().String("project-id", "", "Set the project-id to get Chaos Environment from a particular project.")

pkg/cmd/get/projects.go

@@ -19,7 +19,6 @@ import (
 	"os"
 	"text/tabwriter"
 	"time"
-
 	"github.com/litmuschaos/litmusctl/pkg/apis"
 	"github.com/litmuschaos/litmusctl/pkg/utils"
 	"github.com/manifoldco/promptui"
@@ -49,7 +48,6 @@ var projectsCmd = &cobra.Command{
 			utils.PrintInYamlFormat(projects.Data)
 
 		case "":
-
 			itemsPerPage := 5
 			page := 1
 			totalProjects := len(projects.Data)

pkg/cmd/list/environment.go

@@ -20,10 +20,14 @@ import (
 	"fmt"
 	"os"
 	"strings"
-
+	"text/tabwriter"
+	"time"
+	"strconv"
 	"github.com/litmuschaos/litmusctl/pkg/apis/environment"
 	"github.com/litmuschaos/litmusctl/pkg/utils"
+	"github.com/manifoldco/promptui"
 	"github.com/spf13/cobra"
+	
 )
 
 
@@ -58,7 +62,57 @@ var ListChaosEnvironmentCmd = &cobra.Command{
 				os.Exit(1)
 			}
 		}
-		utils.PrintInJsonFormat(environmentList.Data)
+		environmentListData := environmentList.Data.ListEnvironmentDetails.Environments
+
+		itemsPerPage := 5
+		page := 1
+		totalEnvironments := len(environmentListData)
+
+		writer := tabwriter.NewWriter(os.Stdout, 30, 8, 0,'\t', tabwriter.AlignRight)
+		utils.White_B.Fprintln(writer, "CHAOS ENVIRONMENT ID\tCHAOS ENVIRONMENT NAME\tCREATED AT\tCREATED BY")
+		for {
+			writer.Flush()	
+			// calculating the start and end indices for the current page
+			start := (page - 1) * itemsPerPage
+			if start >= totalEnvironments {
+				writer.Flush()
+				utils.Red.Println("No more environments to display")
+				break
+			}
+			end := start + itemsPerPage
+			if end > totalEnvironments {
+				end = totalEnvironments
+
+			}
+			for _, environment := range environmentListData[start:end] {
+				intTime, err := strconv.ParseInt(environment.CreatedAt, 10, 64)
+				if err != nil {
+					fmt.Println("Error converting CreatedAt to int64:", err)
+					continue
+				}
+				humanTime := time.Unix(intTime, 0)
+				utils.White.Fprintln(
+					writer,
+					environment.EnvironmentID+"\t"+environment.Name+"\t"+humanTime.String()+"\t"+environment.CreatedBy.Username,
+				)
+				}
+				writer.Flush()		
+				// Check if it's the last item or if user wants to see more
+				paginationPrompt := promptui.Prompt{
+					Label:     "Press Enter to show more environments (or type 'q' to quit)",
+					AllowEdit: true,
+					Default:   "",
+				}
+		
+				userInput, err := paginationPrompt.Run()
+				utils.PrintError(err)
+		
+				if userInput == "q" {
+					break
+				}
+			    // Move to the next page
+				page++
+		}
 	},