Add some color option

main.go

@@ -50,8 +50,10 @@ func main() {
 		completer,
 		prompt.MaxCompletionsOption(8),
 		prompt.PrefixOption(">>> "),
-		prompt.PrefixColorOption(prompt.Blue),
 		prompt.TitleOption("SQLITE CLI"),
+		prompt.TextColorOption(prompt.Red),
+		prompt.ResultTextColorOption(prompt.DarkGray),
+		prompt.CompletedTextColorOption(prompt.Green),
 		prompt.CompletionTextColor(prompt.Black),
 		prompt.SelectedCompletionTextColor(prompt.White),
 	)

prompt/option.go

@@ -39,6 +39,27 @@ func PrefixColorOption(x Color) option {
 	}
 }
 
+func TextColorOption(x Color) option {
+	return func(p *Prompt) error {
+		p.renderer.textColor = x
+		return nil
+	}
+}
+
+func CompletedTextColorOption(x Color) option {
+	return func(p *Prompt) error {
+		p.renderer.completedTextColor = x
+		return nil
+	}
+}
+
+func ResultTextColorOption(x Color) option {
+	return func(p *Prompt) error {
+		p.renderer.outputTextColor = x
+		return nil
+	}
+}
+
 func CompletionTextColor(x Color) option {
 	return func(p *Prompt) error {
 		p.renderer.completionTextColor = x
@@ -80,7 +101,10 @@ func NewPrompt(executor Executor, completer Completer, opts ...option) *Prompt {
 		renderer: &Render{
 			prefix:      "> ",
 			out:         &VT100Writer{fd: syscall.Stdout},
-			prefixColor: Green,
+			prefixColor: Blue,
+			textColor: DefaultColor,
+			outputTextColor: DefaultColor,
+			completedTextColor: Green,
 			completionTextColor: White,
 			completionBGColor: Cyan,
 			selectedCompletionTextColor: Black,

prompt/render.go

@@ -1,15 +1,17 @@
 package prompt
 
 type Render struct {
+	out            ConsoleWriter
 	prefix         string
 	title          string
-	out            ConsoleWriter
 	row            uint16
 	col            uint16
 	maxCompletions uint16
 	// colors
 	prefixColor                 Color
 	textColor                   Color
+	outputTextColor             Color
+	completedTextColor          Color
 	completionTextColor         Color
 	completionBGColor           Color
 	selectedCompletionTextColor Color
@@ -104,22 +106,27 @@ func (r *Render) Erase(buffer *Buffer) {
 
 func (r *Render) Render(buffer *Buffer, completions []string, chosen int) {
 	line := buffer.Document().CurrentLine()
+	r.out.SetColor(r.textColor, DefaultColor)
 	r.out.WriteStr(line)
+	r.out.SetColor(DefaultColor, DefaultColor)
 	r.out.CursorBackward(len(line) - buffer.CursorPosition)
 	r.renderCompletion(buffer, completions, chosen)
 	if chosen != -1 {
 		c := completions[chosen]
 		r.out.CursorBackward(len([]rune(buffer.Document().GetWordBeforeCursor())))
+		r.out.SetColor(r.completedTextColor, DefaultColor)
 		r.out.WriteStr(c)
+		r.out.SetColor(DefaultColor, DefaultColor)
 	}
 	r.out.Flush()
 }
 
 func (r *Render) BreakLine(buffer *Buffer, result string) {
-	r.out.WriteStr(buffer.Document().Text)
-	r.out.WriteStr("\n")
-	r.out.WriteStr(result)
-	r.out.WriteStr("\n")
+	r.out.SetColor(r.textColor, DefaultColor)
+	r.out.WriteStr(buffer.Document().Text + "\n")
+	r.out.SetColor(r.outputTextColor, DefaultColor)
+	r.out.WriteStr(result + "\n")
+	r.out.SetColor(DefaultColor, DefaultColor)
 	r.renderPrefix()
 }