From 68774581b78574d76520e7bf23e469779988eb1f Mon Sep 17 00:00:00 2001
From: Christian Rocha <christian@rocha.is>
Date: Sat, 1 Feb 2020 22:47:38 -0500
Subject: [PATCH] Cursor color option on input field

---
 input/input.go | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/input/input.go b/input/input.go
index 0142f39..8a9e370 100644
--- a/input/input.go
+++ b/input/input.go
@@ -15,6 +15,7 @@ type Model struct {
 	BlinkSpeed       time.Duration
 	Placeholder      string
 	PlaceholderColor string
+	CursorColor      string
 
 	blink        bool
 	pos          int
@@ -30,6 +31,7 @@ func DefaultModel() Model {
 		BlinkSpeed:       time.Millisecond * 600,
 		Placeholder:      "",
 		PlaceholderColor: "240",
+		CursorColor:      "",
 
 		blink:        false,
 		pos:          0,
@@ -111,10 +113,10 @@ func View(model tea.Model) string {
 	v := m.Value[:m.pos]
 
 	if m.pos < len(m.Value) {
-		v += cursor(string(m.Value[m.pos]), m.blink)
+		v += cursorView(string(m.Value[m.pos]), m)
 		v += m.Value[m.pos+1:]
 	} else {
-		v += cursor(" ", m.blink)
+		v += cursorView(" ", m)
 	}
 	return m.Prompt + v
 }
@@ -129,14 +131,14 @@ func placeholderView(m Model) string {
 
 	// Cursor
 	if m.blink && m.PlaceholderColor != "" {
-		v += cursor(
+		v += cursorView(
 			termenv.String(p[:1]).
 				Foreground(color(c)).
 				String(),
-			m.blink,
+			m,
 		)
 	} else {
-		v += cursor(p[:1], m.blink)
+		v += cursorView(p[:1], m)
 	}
 
 	// The rest of the palceholder text
@@ -148,9 +150,14 @@ func placeholderView(m Model) string {
 }
 
 // Style the cursor
-func cursor(s string, blink bool) string {
-	if blink {
+func cursorView(s string, m Model) string {
+	if m.blink {
 		return s
+	} else if m.CursorColor != "" {
+		return termenv.String(s).
+			Foreground(m.colorProfile.Color(m.CursorColor)).
+			Reverse().
+			String()
 	}
 	return termenv.String(s).Reverse().String()
 }