feat(watch): Adding label-colums support

Signed-off-by: Vincent Boutour <bob@vibioh.fr>

README.md

@@ -131,6 +131,7 @@ Usage:
   kmux watch [flags]
 
 Flags:
+  -L, --label-columns strings     Labels that are going to be presented as columns
   -o, --output string             Output format. One of: (wide)
   -l, --selector stringToString   Labels to filter pods (default [])
       --show-annotations          Show all annotations as the last column (after labels if both asked)

cmd/watch.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"context"
 	"fmt"
+	"path"
 	"sort"
 	"strings"
 	"syscall"
@@ -31,6 +32,7 @@ var (
 	outputFormat    string
 	showLabels      bool
 	showAnnotations bool
+	labelColumns    []string
 )
 
 func initWatch() {
@@ -40,6 +42,7 @@ func initWatch() {
 	flags.StringToStringVarP(&labelsSelector, "selector", "l", nil, "Labels to filter pods")
 	flags.BoolVarP(&showLabels, "show-labels", "", false, "Show all labels as the last column")
 	flags.BoolVarP(&showAnnotations, "show-annotations", "", false, "Show all annotations as the last column (after labels if both asked)")
+	flags.StringSliceVarP(&labelColumns, "label-columns", "L", nil, "Labels that are going to be presented as columns")
 }
 
 var watchCmd = &cobra.Command{
@@ -115,6 +118,11 @@ func initWatchTable() *table.Table {
 		)
 	}
 
+	for _, label := range labelColumns {
+		defaultWidths = append(defaultWidths, 12)
+		content = append(content, table.NewCell(strings.ToUpper(path.Base(label))))
+	}
+
 	if showLabels {
 		defaultWidths = append(defaultWidths, 12)
 		content = append(content, table.NewCell("LABELS"))
@@ -245,6 +253,10 @@ func outputWatch(watchTable *table.Table, contextName string, pod v1.Pod) {
 		)
 	}
 
+	for _, label := range labelColumns {
+		content = append(content, table.NewCell(pod.GetLabels()[label]))
+	}
+
 	if showLabels {
 		content = append(content, table.NewCell(mapAsString(pod.GetLabels())))
 	}