Add colored text for user and Otto messages in chat and config options for colors

Author: chand1012

cmd/chat.go

@@ -47,16 +47,17 @@ No code context is passed in this mode. Emulates web chat.`,
 
 		client := openai.NewClient(conf.APIKey)
 
-		fmt.Println("Otto: Hello! I am Otto. Use Ctrl+C to exit at any time.")
+		utils.PrintColoredText("Otto: ", "#008080")
+		fmt.Println("Hello! I am Otto. Use Ctrl+C to exit at any time.")
 
 		for {
-			question, err := utils.Input("You: ")
+			question, err := utils.InputWithColor("You: ", "#007bff")
 			if err != nil {
 				log.Error(err)
 				os.Exit(1)
 			}
 
-			fmt.Print("Otto: ")
+			utils.PrintColoredText("Otto: ", "#008080")
 
 			log.Debugf("Adding question '%s' to context...", question)
 

cmd/config.go

@@ -39,7 +39,7 @@ GitHub Token Generation: https://github.com/settings/tokens
 		}
 
 		// if none of the config options are provided, print a warning
-		if apiKey == "" && model == "" && ghToken == "" {
+		if apiKey == "" && model == "" && ghToken == "" && userColor == "" && ottoColor == "" {
 			log.Warn("No configuration options provided")
 			os.Exit(0)
 		}
@@ -67,6 +67,18 @@ GitHub Token Generation: https://github.com/settings/tokens
 			c.GHToken = ghToken
 		}
 
+		// if the userColor is provided, set it
+		if userColor != "" {
+			fmt.Println("Setting user color...")
+			c.UserColor = userColor
+		}
+
+		// if the ottoColor is provided, set it
+		if ottoColor != "" {
+			fmt.Println("Setting Otto color...")
+			c.OttoColor = ottoColor
+		}
+
 		// save the config
 		err = c.Save()
 		if err != nil {
@@ -87,4 +99,8 @@ func init() {
 	configCmd.Flags().StringVarP(&model, "model", "m", "", "Model to use for documentation")
 	// set gh token
 	configCmd.Flags().StringVarP(&ghToken, "ghtoken", "t", "", "GitHub token to use for documentation")
-}
+	// set user color
+	configCmd.Flags().StringVarP(&userColor, "userColor", "u", "", "User color for configuration")
+	// set otto color
+	configCmd.Flags().StringVarP(&ottoColor, "ottoColor", "o", "", "Otto color for configuration")
+}

cmd/issue.go

@@ -1,5 +1,5 @@
 /*
-Copyright © 2023 NAME HERE <EMAIL ADDRESS>
+Copyright © 2023 Chandler <chandler@chand1012.dev>
 */
 package cmd
 

cmd/vars.go

@@ -34,6 +34,8 @@ var model string
 var apiKey string
 var ghToken string
 var remote string
+var userColor string
+var ottoColor string
 
 var issuePRNumber int
 var useComments bool

pkg/config/config.go

@@ -13,6 +13,8 @@ type Config struct {
 	Model     string `json:"model"`
 	GHToken   string `json:"gh_token"`
 	Signature string `json:"signature"`
+	UserColor string `json:"user_color"`
+	OttoColor string `json:"otto_color"`
 }
 
 // also returns the path to the config file
@@ -47,6 +49,8 @@ func createIfNotExists() (string, error) {
 			Model:     "gpt-3.5-turbo",
 			GHToken:   "",
 			Signature: "Created by [OttoDocs 🦦](https://ottodocs.chand1012.dev/)",
+			UserColor: "#87CEEB",
+			OttoColor: "#1BFFE4",
 		}
 		err = json.NewEncoder(file).Encode(blankConfig)
 		if err != nil {
@@ -113,4 +117,4 @@ func Save(c *Config) error {
 	}
 
 	return json.NewEncoder(file).Encode(c)
-}
+}

pkg/utils/color_text.go

@@ -0,0 +1,31 @@
+package utils
+
+import (
+	"fmt"
+
+	"github.com/charmbracelet/lipgloss"
+)
+
+// PrintColoredText prints the given text in the specified CSS color code using fmt.Print.
+func PrintColoredText(text, cssColorCode string) {
+	// Create a new Lipgloss style with a foreground color
+	style := lipgloss.NewStyle().Foreground(lipgloss.Color(cssColorCode))
+
+	// Style the input text
+	styledText := style.Render(text)
+
+	// Print the styled text to the terminal without a newline
+	fmt.Print(styledText)
+}
+
+// PrintColoredTextLn prints the given text in the specified CSS color code using fmt.Println.
+func PrintColoredTextLn(text, cssColorCode string) {
+	// Create a new Lipgloss style with a foreground color
+	style := lipgloss.NewStyle().Foreground(lipgloss.Color(cssColorCode))
+
+	// Style the input text
+	styledText := style.Render(text)
+
+	// Print the styled text to the terminal with a newline
+	fmt.Println(styledText)
+}

pkg/utils/input.go

@@ -18,3 +18,15 @@ func Input(prompt string) (string, error) {
 
 	return strings.TrimSpace(input), nil
 }
+
+func InputWithColor(prompt, cssColorCode string) (string, error) {
+	PrintColoredText(prompt, cssColorCode)
+
+	reader := bufio.NewReader(os.Stdin)
+	input, err := reader.ReadString('\n')
+	if err != nil {
+		return "", err
+	}
+
+	return strings.TrimSpace(input), nil
+}