exclude colorization if kubectl internal option is specified

Author: hidetatz

command/subcommand.go

@@ -2,6 +2,7 @@ package command
 
 import (
 	"os"
+	"strings"
 
 	"github.com/hidetatz/kubecolor/kubectl"
 	"github.com/mattn/go-isatty"
@@ -21,6 +22,15 @@ func ResolveSubcommand(args []string, config *KubecolorConfig) (bool, *kubectl.S
 		return false, subcommandInfo
 	}
 
+	// if there is an argument starting with __,
+	// the subcommand is probably an internal subcommand (like __completeNoDesc)
+	// and should probably not be colorized
+	for i := range args {
+		if strings.HasPrefix(args[i], "__") {
+			return false, subcommandInfo
+		}
+	}
+
 	// if subcommand is not found (e.g. kubecolor --help or just "kubecolor"),
 	// it is treated as help because kubectl shows help for such input
 	if !subcommandFound {

command/subcommand_test.go

@@ -81,6 +81,19 @@ func Test_ResolveSubcommand(t *testing.T) {
 			expectedShouldColorize: true,
 			expectedInfo:           &kubectl.SubcommandInfo{Help: true},
 		},
+		{
+			name:             "when the internal argument is found, it won't colorize",
+			args:             []string{"__completeNoDesc", "get", "pods"},
+			isOutputTerminal: func() bool { return true },
+			conf: &KubecolorConfig{
+				Plain:          false,
+				DarkBackground: true,
+				ForceColor:     false,
+				KubectlCmd:     "kubectl",
+			},
+			expectedShouldColorize: false,
+			expectedInfo:           &kubectl.SubcommandInfo{Subcommand: kubectl.Get},
+		},
 		{
 			name:             "when not tty, it won't colorize",
 			args:             []string{"get", "pods"},