From eea5c2a32a460ab1b914e4d7c4c37f04be5cfa2a Mon Sep 17 00:00:00 2001
From: Christian Rocha <christian@rocha.is>
Date: Fri, 27 Mar 2020 14:10:09 -0400
Subject: [PATCH] Add color settings to spinner

---
 spinner/spinner.go | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/spinner/spinner.go b/spinner/spinner.go
index efbeaea..0ffe42f 100644
--- a/spinner/spinner.go
+++ b/spinner/spinner.go
@@ -5,6 +5,7 @@ import (
 	"time"
 
 	"github.com/charmbracelet/tea"
+	"github.com/muesli/termenv"
 )
 
 // Spinner denotes a type of spinner
@@ -24,13 +25,18 @@ var (
 	}
 
 	assertionErr = errors.New("could not perform assertion on model to what the spinner expects. are you sure you passed the right value?")
+
+	color = termenv.ColorProfile().Color
 )
 
 // Model contains the state for the spinner. Use NewModel to create new models
 // rather than using Model as a struct literal.
 type Model struct {
-	Type  Spinner
-	FPS   int
+	Type            Spinner
+	FPS             int
+	ForegroundColor string
+	BackgroundColor string
+
 	frame int
 }
 
@@ -66,7 +72,18 @@ func View(model Model) string {
 	if model.frame >= len(s) {
 		return "[error]"
 	}
-	return s[model.frame]
+
+	str := s[model.frame]
+
+	if model.ForegroundColor != "" || model.BackgroundColor != "" {
+		return termenv.
+			String(str).
+			Foreground(color(model.ForegroundColor)).
+			Background(color(model.BackgroundColor)).
+			String()
+	}
+
+	return str
 }
 
 // Sub is the subscription that allows the spinner to spin