Add color choosing support

Signed-off-by: Ulysses Souza <ulyssessouza@gmail.com>

util/progress/progressui/colors.go

@@ -0,0 +1,40 @@
+package progressui
+
+import (
+	"os"
+	"strings"
+
+	"github.com/morikuni/aec"
+)
+
+var termColorMap = map[string]aec.ANSI{
+	"":        colorRun,
+	"default": aec.DefaultF,
+
+	"black":   aec.BlackF,
+	"red":     aec.RedF,
+	"green":   aec.GreenF,
+	"yellow":  aec.YellowF,
+	"blue":    aec.BlueF,
+	"magenta": aec.MagentaF,
+	"cyan":    aec.CyanF,
+	"white":   aec.WhiteF,
+
+	"light-black":   aec.LightBlackF,
+	"light-red":     aec.LightRedF,
+	"light-green":   aec.LightGreenF,
+	"light-yellow":  aec.LightYellowF,
+	"light-blue":    aec.LightBlueF,
+	"light-magenta": aec.LightMagentaF,
+	"light-cyan":    aec.LightCyanF,
+	"light-white":   aec.LightWhiteF,
+}
+
+func getTermColor() aec.ANSI {
+	envColorString := os.Getenv("BUILDKIT_TERM_COLOR")
+	c, ok := termColorMap[strings.ToLower(envColorString)]
+	if !ok {
+		return colorRun
+	}
+	return c
+}

util/progress/progressui/display.go

@@ -521,7 +521,7 @@ func (disp *display) print(d displayInfo, width, height int, all bool) {
 
 		out = align(out, timer, width)
 		if j.completedTime != nil {
-			color := colorRun
+			color := getTermColor()
 			if j.isCanceled {
 				color = colorCancel
 			} else if j.hasError {