Update the UX for "context use" and "context list" commands to show P…

…rojectID details Signed-off-by: Prem Kumar Kalle <prem.kalle@broadcom.com>
vmware-tanzu · Apr 2, 2024 · 7637a3f · 7637a3f
1 parent 50befed
commit 7637a3f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 6 deletions.
diff --git a/pkg/command/context.go b/pkg/command/context.go
@@ -13,6 +13,7 @@ import (
 	"os"
 	"reflect"
 	"regexp"
+	"slices"
 	"sort"
 	"strconv"
 	"strings"
@@ -1182,10 +1183,13 @@ func useCtx(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
-	suffixString := ""
+	suffixString := fmt.Sprintf("Type: %s", ctx.ContextType)
 	if ctx.ContextType == configtypes.ContextTypeTanzu {
 		if project, exists := ctx.AdditionalMetadata[config.ProjectNameKey]; exists && project != "" {
-			suffixString += fmt.Sprintf("Project: %s", project)
+			suffixString += fmt.Sprintf(", Project: %s", project)
+		}
+		if projectID, exists := ctx.AdditionalMetadata[config.ProjectIDKey]; exists && projectID != "" {
+			suffixString += fmt.Sprintf(", ProjectID: %s", projectID)
 		}
 		if space, exists := ctx.AdditionalMetadata[config.SpaceNameKey]; exists && space != "" {
 			suffixString += fmt.Sprintf(", Space: %s", space)
@@ -1195,10 +1199,10 @@ func useCtx(cmd *cobra.Command, args []string) error {
 		}
 	}
 	if suffixString != "" {
-		suffixString = " (" + suffixString + ")"
+		suffixString = "(" + suffixString + ")"
 	}
 
-	log.Infof("Successfully activated context '%s' of type '%s'%s.", ctxName, ctx.ContextType, suffixString)
+	log.Infof("Activated context '%s' %s successfully ", ctxName, suffixString)
 
 	// Sync all required plugins
 	_ = syncContextPlugins(cmd, ctx.ContextType, ctxName, true)
@@ -1366,6 +1370,7 @@ type ContextListOutputRow struct {
 	IsActive       string
 	Type           string
 	Project        string
+	ProjectID      string
 	Space          string
 	ClusterGroup   string
 	Endpoint       string
@@ -1388,6 +1393,7 @@ func displayContextListOutputWithDynamicColumns(cfg *configtypes.ClientConfig, w
 		path := NA
 		context := NA
 		project := NA
+		projectID := NA
 		space := NA
 		clustergroup := NA
 
@@ -1401,6 +1407,7 @@ func displayContextListOutputWithDynamicColumns(cfg *configtypes.ClientConfig, w
 		case configtypes.ContextTypeTanzu:
 			tanzuContextExists = true
 			project = ""
+			projectID = ""
 			space = ""
 			clustergroup = ""
 			ep = ""
@@ -1414,6 +1421,9 @@ func displayContextListOutputWithDynamicColumns(cfg *configtypes.ClientConfig, w
 			if ctx.AdditionalMetadata[config.ProjectNameKey] != nil {
 				project = ctx.AdditionalMetadata[config.ProjectNameKey].(string)
 			}
+			if ctx.AdditionalMetadata[config.ProjectIDKey] != nil {
+				projectID = ctx.AdditionalMetadata[config.ProjectIDKey].(string)
+			}
 			if ctx.AdditionalMetadata[config.SpaceNameKey] != nil {
 				space = ctx.AdditionalMetadata[config.SpaceNameKey].(string)
 			}
@@ -1427,7 +1437,7 @@ func displayContextListOutputWithDynamicColumns(cfg *configtypes.ClientConfig, w
 				context = ctx.ClusterOpts.Context
 			}
 		}
-		row := ContextListOutputRow{ctx.Name, strconv.FormatBool(isCurrent), string(ctx.ContextType), project, space, clustergroup, ep, path, context}
+		row := ContextListOutputRow{ctx.Name, strconv.FormatBool(isCurrent), string(ctx.ContextType), project, projectID, space, clustergroup, ep, path, context}
 		rows = append(rows, row)
 	}
 
@@ -1438,6 +1448,11 @@ func displayContextListOutputWithDynamicColumns(cfg *configtypes.ClientConfig, w
 		dynamicColumns = append(dynamicColumns, "ClusterGroup")
 	}
 	if showAllColumns {
+		if tanzuContextExists {
+			// add "ProjectID" column after "Project"
+			projectIndex := slices.Index(requiredColumns, "Project")
+			requiredColumns = slices.Insert(requiredColumns, projectIndex+1, "ProjectID")
+		}
 		requiredColumns = append(requiredColumns, "Endpoint", "KubeconfigPath", "KubeContext")
 		requiredColumns = append(requiredColumns, dynamicColumns...)
 	}

diff --git a/pkg/command/context_test.go b/pkg/command/context_test.go
@@ -233,7 +233,7 @@ var _ = Describe("Test tanzu context command", func() {
 			columnsString := strings.Join(strings.Fields(lines[0]), " ")
 
 			Expect(err).To(BeNil())
-			Expect(columnsString).To(Equal("NAME ISACTIVE TYPE PROJECT SPACE CLUSTERGROUP ENDPOINT KUBECONFIGPATH KUBECONTEXT"))
+			Expect(columnsString).To(Equal("NAME ISACTIVE TYPE PROJECT PROJECTID SPACE CLUSTERGROUP ENDPOINT KUBECONFIGPATH KUBECONTEXT"))
 		})
 
 		It("should not return tanzu related columns when not listing tanzu contexts without --wide", func() {