grokkit/cmd/commitmsg.go

package cmd

import (
	"fmt"
	"os"

	"github.com/fatih/color"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
	"gmgauthier.com/grokkit/internal/git"
	"gmgauthier.com/grokkit/internal/grok"
)

var commitMsgCmd = &cobra.Command{
	Use:   "commit-msg",
	Short: "Generate conventional commit message from staged changes",
	Run: func(cmd *cobra.Command, args []string) {
		model := viper.GetString("model")

		client := grok.NewClient()

		if !git.IsRepo() {
			color.Red("Error: Not inside a git repository")
			os.Exit(1)
		}

		diffOutput := git.Run([]string{"diff", "--cached"})

		if diffOutput == "" {
			color.Red("Error: No staged changes. Stage files with `git add` first.")
			os.Exit(1)
		}

		systemPrompt := `You are an expert at writing conventional commit messages.

Analyze the code diff below and generate a concise commit message:

- Prefix: feat|fix|docs|style|refactor|perf|test|ci|chore
- Imperative mood (e.g., 'Add', 'Fix'), capitalize first letter
- Max 72 chars for subject line
- Optional body lines wrapped at 72 chars
- No trailing period on subject`

		messages := []map[string]string{
			{"role": "system", "content": systemPrompt},
			{"role": "user", "content": diffOutput},
		}

		commitMsg := client.Stream(messages, model)

		fmt.Println(color.YellowString("Suggested commit message:"))
		fmt.Println()
		fmt.Print(color.CyanString(commitMsg))
		fmt.Println()
	},
}