Fix color output on Windows (#220)

ahmetb · May 15, 2020 · 401188f · 401188f
1 parent 01bd237
commit 401188f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/cmd/kubectx/main.go b/cmd/kubectx/main.go
@@ -8,22 +8,23 @@ import (
 	"github.com/ahmetb/kubectx/internal/cmdutil"
 	"github.com/ahmetb/kubectx/internal/env"
 	"github.com/ahmetb/kubectx/internal/printer"
+	"github.com/fatih/color"
 )
 
 type Op interface {
 	Run(stdout, stderr io.Writer) error
 }
 
 func main() {
-	cmdutil.PrintDeprecatedEnvWarnings(os.Stderr, os.Environ())
+	cmdutil.PrintDeprecatedEnvWarnings(color.Error, os.Environ())
 
 	op := parseArgs(os.Args[1:])
-	if err := op.Run(os.Stdout, os.Stderr); err != nil {
-		printer.Error(os.Stderr, err.Error())
+	if err := op.Run(color.Output, color.Error); err != nil {
+		printer.Error(color.Error, err.Error())
 
 		if _, ok := os.LookupEnv(env.EnvDebug); ok {
 			// print stack trace in verbose mode
-			fmt.Fprintf(os.Stderr, "[DEBUG] error: %+v\n", err)
+			fmt.Fprintf(color.Error, "[DEBUG] error: %+v\n", err)
 		}
 		defer os.Exit(1)
 	}

diff --git a/cmd/kubens/main.go b/cmd/kubens/main.go
@@ -8,21 +8,22 @@ import (
 	"github.com/ahmetb/kubectx/internal/cmdutil"
 	"github.com/ahmetb/kubectx/internal/env"
 	"github.com/ahmetb/kubectx/internal/printer"
+	"github.com/fatih/color"
 )
 
 type Op interface {
 	Run(stdout, stderr io.Writer) error
 }
 
 func main() {
-	cmdutil.PrintDeprecatedEnvWarnings(os.Stderr, os.Environ())
+	cmdutil.PrintDeprecatedEnvWarnings(color.Error, os.Environ())
 	op := parseArgs(os.Args[1:])
-	if err := op.Run(os.Stdout, os.Stderr); err != nil {
-		printer.Error(os.Stderr, err.Error())
+	if err := op.Run(color.Output, color.Error); err != nil {
+		printer.Error(color.Error, err.Error())
 
 		if _, ok := os.LookupEnv(env.EnvDebug); ok {
 			// print stack trace in verbose mode
-			fmt.Fprintf(os.Stderr, "[DEBUG] error: %+v\n", err)
+			fmt.Fprintf(color.Error, "[DEBUG] error: %+v\n", err)
 		}
 		defer os.Exit(1)
 	}