1 files changed, 56 insertions, 0 deletions

diff --git a/cli/internal/colorcache/colorcache.go b/cli/internal/colorcache/colorcache.go
new file mode 100644
index 0000000..08a15e8
--- /dev/null
+++ b/cli/internal/colorcache/colorcache.go
@@ -0,0 +1,56 @@
+package colorcache
+
+import (
+	"sync"
+
+	"github.com/vercel/turbo/cli/internal/util"
+
+	"github.com/fatih/color"
+)
+
+type colorFn = func(format string, a ...interface{}) string
+
+func getTerminalPackageColors() []colorFn {
+	return []colorFn{color.CyanString, color.MagentaString, color.GreenString, color.YellowString, color.BlueString}
+}
+
+type ColorCache struct {
+	mu         sync.Mutex
+	index      int
+	TermColors []colorFn
+	Cache      map[interface{}]colorFn
+}
+
+// New creates an instance of ColorCache with helpers for adding colors to task outputs
+func New() *ColorCache {
+	return &ColorCache{
+		TermColors: getTerminalPackageColors(),
+		index:      0,
+		Cache:      make(map[interface{}]colorFn),
+	}
+}
+
+// colorForKey returns a color function for a given package name
+func (c *ColorCache) colorForKey(key string) colorFn {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	colorFn, ok := c.Cache[key]
+	if ok {
+		return colorFn
+	}
+	c.index++
+	colorFn = c.TermColors[util.PositiveMod(c.index, len(c.TermColors))] // 5 possible colors
+	c.Cache[key] = colorFn
+	return colorFn
+}
+
+// PrefixWithColor returns a string consisting of the provided prefix in a consistent
+// color based on the cacheKey
+func (c *ColorCache) PrefixWithColor(cacheKey string, prefix string) string {
+	colorFn := c.colorForKey(cacheKey)
+	if prefix != "" {
+		return colorFn("%s: ", prefix)
+	}
+
+	return ""
+}