feat: add interactive UI for commit message suggestion

{cocoide} · {cocoide} · commit afb3c6e184db · 2023-09-02T22:20:20.000+09:00
diff --git a/cmd/suggest.go b/cmd/suggest.go
@@ -4,37 +4,92 @@ import (
 	"context"
 	"fmt"
 	"log"
-	"os/exec"
+	"strings"
 
+	tea "github.com/charmbracelet/bubbletea"
 	"github.com/cocoide/commitify/internal/gateway"
+	"github.com/cocoide/commitify/internal/service"
 	"github.com/cocoide/commitify/util"
+	"github.com/fatih/color"
 	"github.com/spf13/cobra"
 )
 
-const (
-	CommitMessagePrompt = "Generate commit message for [%s]"
-	FormatNotice        = ", format your commit as:\n- feat: [feature description]\n- bugfix: [bugfix description]"
-)
+type model struct {
+	choices    []string
+	currentIdx int
+	errorMsg   string
+}
+
+func (m model) Init() tea.Cmd {
+	return nil
+}
+
+func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case tea.KeyMsg:
+		switch msg.Type {
+		case tea.KeyUp:
+			if m.currentIdx > 0 {
+				m.currentIdx--
+			}
+		case tea.KeyDown:
+			if m.currentIdx < len(m.choices)-1 {
+				m.currentIdx++
+			}
+		case tea.KeyEnter:
+			if err := util.ExecCommitMessage(m.choices[m.currentIdx]); err != nil {
+				m.errorMsg = err.Error()
+				return m, nil
+			}
+			return m, tea.Quit
+		case tea.KeyCtrlC, tea.KeyEsc:
+			return m, tea.Quit
+		}
+	}
+	return m, nil
+}
+
+func (m model) View() string {
+	var b strings.Builder
+	if m.errorMsg != "" {
+		// エラーメッセージを赤色で表示
+		red := color.New(color.FgRed).SprintFunc()
+		b.WriteString(red(m.errorMsg) + "\n\n")
+	}
+	b.WriteString("Please select an option:\n\n")
+
+	for i, choice := range m.choices {
+		blue := color.New(color.FgCyan).SprintFunc()
+		if i == m.currentIdx {
+			b.WriteString(fmt.Sprintf(blue("-> %s\n"), choice))
+		} else {
+			b.WriteString(fmt.Sprintf("   %s\n", choice))
+		}
+	}
+
+	b.WriteString("\nUse the arrow keys to navigate and press Enter to select.")
+	return b.String()
+}
 
 var suggestCmd = &cobra.Command{
-	Use:   "suggest",
-	Short: "Suggestion of commit message for staging repository",
+	Use:     "suggest",
+	Short:   "Suggestion of commit message for staging repository",
+	Aliases: []string{"s", "suggest"},
 	Run: func(cmd *cobra.Command, args []string) {
 		util.LoadEnv()
 		ctx := context.Background()
 		og := gateway.NewOpenAIGateway(ctx)
-		result, err := exec.Command("git", "diff", "--staged").Output()
-		if err != nil {
-			log.Fatal(err.Error())
-		}
-		// 設定に応じてPromptは動的に変化させる
-		prompt := fmt.Sprintf(CommitMessagePrompt, string(result))
-		answer, err := og.GetAnswerFromPrompt(prompt, 0.01)
-		// 生成中のUIを非同期で表示
+		ms := service.NewMessageService(og)
+		msgCh, err := ms.AsyncGenerateCommitMessage()
 		if err != nil {
 			log.Fatal(err.Error())
 		}
-		fmt.Println(answer)
+		suggestMsg := <-msgCh
+		fmt.Println(suggestMsg)
+		choices := []string{"32", "!", "!#!"}
+		m := model{choices: choices}
+		p := tea.NewProgram(m)
+		p.Run()
 	},
 }
 
