feat: implement commit message editing with bubbletea UI (#187)

- Add `tea` import from `github.com/charmbracelet/bubbletea`
- Trim commit message before displaying it
- Change confirmation prompt default to `Yes`
- Add prompt to change commit message with default `No`
- Add logic to handle commit message change using `bubbletea` program
- Create new file `textarea.go` for handling commit message input
- Implement `model` struct and functions for `textarea` interaction
- Add `View` function to display commit message confirmation prompt

Signed-off-by: Bo-Yi Wu <[email protected]>

Author: appleboy

cmd/commit.go

@@ -13,6 +13,7 @@ import (
 	"github.com/appleboy/CodeGPT/prompt"
 	"github.com/appleboy/CodeGPT/util"
 
+	tea "github.com/charmbracelet/bubbletea"
 	"github.com/erikgeiser/promptkit/confirmation"
 	"github.com/fatih/color"
 	"github.com/joho/godotenv"
@@ -264,10 +265,11 @@ var commitCmd = &cobra.Command{
 
 		// unescape html entities in commit message
 		commitMessage = html.UnescapeString(commitMessage)
+		commitMessage = strings.TrimSpace(commitMessage)
 
 		// Output commit summary data from AI
 		color.Yellow("================Commit Summary====================")
-		color.Yellow("\n" + strings.TrimSpace(commitMessage) + "\n\n")
+		color.Yellow("\n" + commitMessage + "\n\n")
 		color.Yellow("==================================================")
 
 		outputFile := viper.GetString("output.file")
@@ -286,7 +288,7 @@ var commitCmd = &cobra.Command{
 		}
 
 		if preview {
-			input := confirmation.New("Commit preview summary?", confirmation.Undecided)
+			input := confirmation.New("Commit preview summary?", confirmation.Yes)
 			ready, err := input.RunPrompt()
 			if err != nil {
 				return err
@@ -296,6 +298,20 @@ var commitCmd = &cobra.Command{
 			}
 		}
 
+		input := confirmation.New("Do you want to change the commit message?", confirmation.No)
+		change, err := input.RunPrompt()
+		if err != nil {
+			return err
+		}
+
+		if change {
+			p := tea.NewProgram(initialPrompt(commitMessage))
+			if _, err := p.Run(); err != nil {
+				return err
+			}
+			p.Wait()
+		}
+
 		// git commit automatically
 		color.Cyan("Git record changes to the repository")
 		output, err := g.Commit(commitMessage)

cmd/textarea.go

@@ -0,0 +1,72 @@
+package cmd
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/charmbracelet/bubbles/textarea"
+	tea "github.com/charmbracelet/bubbletea"
+)
+
+type errMsg error
+
+type model struct {
+	textarea textarea.Model
+	err      error
+}
+
+func initialPrompt(value string) model {
+	ti := textarea.New()
+	ti.InsertString(value)
+	ti.SetWidth(80)
+	ti.SetHeight(len(strings.Split(value, "\n")))
+	ti.Focus()
+
+	return model{
+		textarea: ti,
+		err:      nil,
+	}
+}
+
+func (m model) Init() tea.Cmd {
+	return textarea.Blink
+}
+
+func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	var cmds []tea.Cmd
+	var cmd tea.Cmd
+
+	switch msg := msg.(type) {
+	case tea.KeyMsg:
+		switch msg.Type { //nolint:exhaustive
+		case tea.KeyEsc:
+			if m.textarea.Focused() {
+				m.textarea.Blur()
+			}
+		case tea.KeyCtrlC:
+			return m, tea.Quit
+		default:
+			if !m.textarea.Focused() {
+				cmd = m.textarea.Focus()
+				cmds = append(cmds, cmd)
+			}
+		}
+
+	// We handle errors just like any other message
+	case errMsg:
+		m.err = msg
+		return m, nil
+	}
+
+	m.textarea, cmd = m.textarea.Update(msg)
+	cmds = append(cmds, cmd)
+	return m, tea.Batch(cmds...)
+}
+
+func (m model) View() string {
+	return fmt.Sprintf(
+		"Please confirm the following commit message.\n\n%s\n\n%s",
+		m.textarea.View(),
+		"(ctrl+c to continue.)",
+	) + "\n\n"
+}