Add color to output

buildtree/buildtree.go

@@ -201,7 +201,7 @@ func (t *BuildTree) Build() error {
 	if err != nil {
 		return err
 	}
-	fmt.Println("Building new images")
+	utils.Info("Building new images")
 	for _, node := range t.rootNodes {
 		err = t.buildNodeAndChildren(node)
 		if err != nil {
@@ -217,7 +217,7 @@ func (t *BuildTree) TryBuild() error {
 	if err != nil {
 		return err
 	}
-	fmt.Println("Try building new images")
+	utils.Info("Try building new images")
 	for _, node := range t.rootNodes {
 		err = t.tryBuildNodeAndChildren(node)
 		if err != nil {
@@ -233,14 +233,14 @@ func (t *BuildTree) Push() error {
 	if err != nil {
 		return err
 	}
-	fmt.Println("Logging into registry")
+	utils.Info("Logging into registry")
 	for _, credential := range t.credentials {
 		err = utils.DockerLogin(credential.Registry, credential.Username, credential.Password)
 		if err != nil {
 			return err
 		}
 	}
-	fmt.Println("Pushing new images")
+	utils.Info("Pushing new images")
 	for _, node := range t.rootNodes {
 		err = t.pushNodeAndChildren(node)
 		if err != nil {
@@ -273,7 +273,7 @@ func (t *BuildTree) Clean() {
 		if node.buildRoot != "provided" {
 			err := utils.DockerRMI(node.name, node.tag)
 			if err != nil {
-				utils.PrintError(err)
+				utils.Error(err)
 			}
 		}
 	}
@@ -379,16 +379,16 @@ func (t *BuildTree) needBuild(node *buildNode) bool {
 
 func (t *BuildTree) buildNodeAndChildren(node *buildNode) error {
 	if !t.needBuild(node) {
-		fmt.Printf("====> Skipping %s\n", node.name)
+		utils.Info2("====> Skipping %s", node.name)
 	} else {
-		fmt.Printf("====> Building %s:%s\n", node.name, node.tag)
+		utils.Info2("====> Building %s:%s", node.name, node.tag)
 		err := t.buildNode(node, node.tag)
 		if err != nil {
 			return err
 		}
 		if node.pushLatest {
 			latestTag := "latest"
-			fmt.Printf("====> Building %s:%s\n", node.name, latestTag)
+			utils.Info2("====> Building %s:%s", node.name, latestTag)
 			err = t.buildNode(node, latestTag)
 			if err != nil {
 				return err
@@ -406,18 +406,18 @@ func (t *BuildTree) buildNodeAndChildren(node *buildNode) error {
 
 func (t *BuildTree) tryBuildNodeAndChildren(node *buildNode) error {
 	if !t.needBuild(node) {
-		fmt.Printf("====> Skipping %s\n", node.name)
+		utils.Info2("====> Skipping %s", node.name)
 	} else {
 		randomTag := fmt.Sprintf("%s-%d", node.tag, time.Now().UnixNano())
-		fmt.Printf("====> Building %s:%s\n", node.name, randomTag)
+		utils.Info2("====> Building %s:%s", node.name, randomTag)
 		err := t.buildNode(node, randomTag)
 		if err != nil {
 			return err
 		}
-		fmt.Printf("====> Removing %s:%s\n", node.name, randomTag)
+		utils.Info2("====> Removing %s:%s", node.name, randomTag)
 		err = utils.DockerRMI(node.name, randomTag)
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 		}
 	}
 	for _, child := range node.children {
@@ -455,16 +455,16 @@ func (t *BuildTree) resolveShellCommandPath(buildRoot, command string) string {
 
 func (t *BuildTree) pushNodeAndChildren(node *buildNode) error {
 	if !t.needBuild(node) {
-		fmt.Printf("====> Skipping %s\n", node.name)
+		utils.Info2("====> Skipping %s", node.name)
 	} else {
-		fmt.Printf("====> Pushing %s:%s\n", node.name, node.tag)
+		utils.Info2("====> Pushing %s:%s", node.name, node.tag)
 		err := utils.DockerPush(node.name, node.tag)
 		if err != nil {
 			return err
 		}
 		if node.pushLatest {
 			latestTag := "latest"
-			fmt.Printf("====> Pushing %s:%s\n", node.name, latestTag)
+			utils.Info2("====> Pushing %s:%s", node.name, latestTag)
 			err := utils.DockerPush(node.name, latestTag)
 			if err != nil {
 				return err

cmd/build.go

@@ -36,17 +36,17 @@ var buildCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		t, err := buildtree.ReadBuildTreeFromFile(cfgFile, variableMap, variableFiles)
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		err = t.Prepare()
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		err = t.Build()
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 	},

cmd/clean.go

@@ -36,7 +36,7 @@ var cleanCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		t, err := buildtree.ReadBuildTreeFromFile(cfgFile, variableMap, variableFiles)
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		t.Clean()

cmd/dryrun.go

@@ -38,12 +38,12 @@ var dryrunCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		t, err := buildtree.ReadBuildTreeFromFile(cfgFile, variableMap, variableFiles)
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		err = t.Prepare()
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		t.PrintTree(dryrunNoColor)

cmd/findlatest.go

@@ -38,12 +38,12 @@ Currently, only images hosted in gcr.io are supported by this command.
 	Run: func(cmd *cobra.Command, args []string) {
 		t, err := buildtree.ReadBuildTreeFromFile(cfgFile, variableMap, variableFiles)
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		tag, err := t.FindLatestTag(args[0])
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		fmt.Println(tag)

cmd/push.go

@@ -36,17 +36,17 @@ var pushCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		t, err := buildtree.ReadBuildTreeFromFile(cfgFile, variableMap, variableFiles)
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		err = t.Prepare()
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		err = t.Push()
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 	},

cmd/trybuild.go

@@ -36,17 +36,17 @@ var trybuildCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		t, err := buildtree.ReadBuildTreeFromFile(cfgFile, variableMap, variableFiles)
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		err = t.Prepare()
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 		err = t.TryBuild()
 		if err != nil {
-			utils.PrintError(err)
+			utils.Error(err)
 			os.Exit(1)
 		}
 	},

utils/common.go

@@ -1,18 +1,18 @@
 package utils
 
 import (
-	"fmt"
 	"os"
 	"os/exec"
 	"path/filepath"
 	"strings"
 
+	"github.com/fatih/color"
 	"github.com/palantir/stacktrace"
 )
 
 // Version returns doriath version
 func Version() string {
-	return "1.4.3"
+	return "1.4.4"
 }
 
 // ResolveDir appends a path to a rootDir
@@ -26,13 +26,47 @@ func ResolveDir(rootDir, path string) string {
 	return filepath.Join(rootDir, path)
 }
 
-// PrintError prints additional stack trace if DEBUG is set to true
-func PrintError(err error) {
-	if os.Getenv("DEBUG") == "true" {
-		fmt.Fprintln(os.Stderr, "ERROR:", err)
+// Info .
+func Info(msg string, args ...interface{}) {
+	color.New(color.FgBlue).Printf(appendNewLine(msg), args...)
+}
+
+// Info2 .
+func Info2(msg string, args ...interface{}) {
+	color.New(color.FgCyan).Printf(appendNewLine(msg), args...)
+}
+
+// Warn .
+func Warn(msg string, args ...interface{}) {
+	color.New(color.FgYellow).Printf(appendNewLine(msg), args...)
+}
+
+// Success .
+func Success(msg string, args ...interface{}) {
+	color.New(color.FgGreen).Printf(appendNewLine(msg), args...)
+}
+
+// Error .
+func Error(err error) {
+	debug := os.Getenv("DEBUG")
+	if debug == "true" || debug == "1" {
+		color.New(color.FgRed).Fprintln(os.Stderr, err)
 	} else {
-		fmt.Fprintln(os.Stderr, "ERROR:", stacktrace.RootCause(err))
+		color.New(color.FgRed).Fprintln(os.Stderr, stacktrace.RootCause(err))
+	}
+}
+
+// Fatal .
+func Fatal(err error) {
+	Error(err)
+	os.Exit(1)
+}
+
+func appendNewLine(msg string) string {
+	if strings.HasSuffix(msg, "\n") {
+		return msg
 	}
+	return msg + "\n"
 }
 
 // DetectRequirement detects requirement for doriath