colorize apply result by the action

Author: Hidetatsu Yaginuma

printer/kubectl_apply.go

@@ -0,0 +1,52 @@
+package printer
+
+import (
+	"bufio"
+	"fmt"
+	"io"
+	"strings"
+
+	"github.com/dty1er/kubecolor/color"
+)
+
+type ApplyPrinter struct {
+	DarkBackground bool
+}
+
+// kubectl apply
+// deployment.apps/foo unchanged
+// deployment.apps/bar created
+// deployment.apps/quux configured
+func (ap *ApplyPrinter) Print(r io.Reader, w io.Writer) {
+	const (
+		applyActionCreated    = "created"
+		applyActionConfigured = "configured"
+		applyActionUnchanged  = "unchanged"
+	)
+
+	colors := map[string]color.Color{
+		applyActionCreated:    color.Green,
+		applyActionConfigured: color.Yellow,
+		applyActionUnchanged:  color.Magenta,
+	}
+
+	colorize := func(line, action string, wr io.Writer) {
+		arg := strings.TrimSuffix(line, " "+action)
+		fmt.Fprintf(w, "%s %s\n", arg, color.Apply(action, colors[action]))
+	}
+
+	scanner := bufio.NewScanner(r)
+	for scanner.Scan() {
+		line := scanner.Text()
+		switch {
+		case strings.HasSuffix(line, " "+applyActionCreated):
+			colorize(line, applyActionCreated, w)
+		case strings.HasSuffix(line, " "+applyActionConfigured):
+			colorize(line, applyActionConfigured, w)
+		case strings.HasSuffix(line, " "+applyActionUnchanged):
+			colorize(line, applyActionUnchanged, w)
+		default:
+			fmt.Fprintf(w, "%s\n", color.Apply(line, color.Green))
+		}
+	}
+}

printer/kubectl_apply_test.go

@@ -0,0 +1,67 @@
+package printer
+
+import (
+	"bytes"
+	"strings"
+	"testing"
+
+	"github.com/dty1er/kubecolor/testutil"
+)
+
+func Test_ApplyPrinter_Print(t *testing.T) {
+	tests := []struct {
+		name           string
+		darkBackground bool
+		tablePrinter   *TablePrinter
+		input          string
+		expected       string
+	}{
+		{
+			name:           "created",
+			darkBackground: true,
+			input: testutil.NewHereDoc(`
+				deployment.apps/foo created`),
+			expected: testutil.NewHereDoc(`
+				deployment.apps/foo created
+			`),
+		},
+		{
+			name:           "configured",
+			darkBackground: true,
+			input: testutil.NewHereDoc(`
+				deployment.apps/foo configured`),
+			expected: testutil.NewHereDoc(`
+				deployment.apps/foo configured
+			`),
+		},
+		{
+			name:           "unchanged",
+			darkBackground: true,
+			input: testutil.NewHereDoc(`
+				deployment.apps/foo unchanged`),
+			expected: testutil.NewHereDoc(`
+				deployment.apps/foo unchanged
+			`),
+		},
+		{
+			name:           "something else. This likely won't happen but fallbacks here just in case.",
+			darkBackground: true,
+			input: testutil.NewHereDoc(`
+				deployment.apps/foo bar`),
+			expected: testutil.NewHereDoc(`
+				deployment.apps/foo bar
+			`),
+		},
+	}
+	for _, tt := range tests {
+		tt := tt
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+			r := strings.NewReader(tt.input)
+			var w bytes.Buffer
+			printer := ApplyPrinter{DarkBackground: tt.darkBackground}
+			printer.Print(r, &w)
+			testutil.MustEqual(t, tt.expected, w.String())
+		})
+	}
+}

printer/kubectl_output_colored_printer.go

@@ -99,6 +99,7 @@ func (kp *KubectlOutputColoredPrinter) Print(r io.Reader, w io.Writer) {
 		case kp.SubcommandInfo.FormatOption == kubectl.Yaml:
 			printer = &YamlPrinter{DarkBackground: kp.DarkBackground}
 		default:
+			printer = &ApplyPrinter{DarkBackground: kp.DarkBackground}
 		}
 	}