53 lines
1.4 KiB
Go
53 lines
1.4 KiB
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()
|
|
},
|
|
}
|