grokkit/cmd/prdescribe.go

package cmd

import (
	"fmt"

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

var prDescribeCmd = &cobra.Command{
	Use:   "pr-describe",
	Short: "Generate full PR description from current branch",
	Run: func(cmd *cobra.Command, args []string) {
		model := viper.GetString("model")

		client := grok.NewClient()

		if !git.IsRepo() {
			color.Yellow("Not a git repo.")
		}

		logMain := git.Run([]string{"log", "--oneline", "main..HEAD"})
		if logMain == "" {
			logMain = git.Run([]string{"log", "--oneline", "origin/main..HEAD"})
		}
		diffMain := git.Run([]string{"diff", "main..HEAD"})
		if diffMain == "" {
			diffMain = git.Run([]string{"diff", "origin/main..HEAD"})
		}
		if logMain == "" && diffMain == "" {
			color.Yellow("No changes vs main/origin/main. Nothing to describe.")
			return
		}

		systemPrompt := `Write a professional GitHub PR title + detailed body (changes, motivation, testing notes).`

		messages := []map[string]string{
			{"role": "system", "content": systemPrompt},
			{"role": "user", "content": "Branch changes:\n" + logMain + "\n\nDiff:\n" + diffMain},
		}

		prDesc := client.Stream(messages, model)

		fmt.Println(color.YellowString("📝 Suggested PR Description:"))
		fmt.Println()
		fmt.Print(color.CyanString(prDesc))
		fmt.Println()
	},
}