Greg Gauthier
0aa806be70
- Implemented `grokkit docs` command for generating language-specific documentation comments (godoc, PEP 257, Doxygen, etc.) with previews, backups, and auto-apply option - Extracted message builder functions for commit, history, pr-describe, and review commands - Added comprehensive unit tests for all command message builders (commit_test.go, docs_test.go, history_test.go, lint_test.go, prdescribe_test.go, review_test.go) - Enforced 70% test coverage threshold in CI workflow - Added .golangci.yml configuration with linters like govet, errcheck, staticcheck - Updated Makefile to include -race in tests and add help target - Updated README.md with new docs command details, workflows, and quality features - Added .claude/ to .gitignore - Configured default model for docs command in config.go
45 lines
1.3 KiB
Go
45 lines
1.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/spf13/cobra"
|
|
"gmgauthier.com/grokkit/config"
|
|
"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) {
|
|
diff, err := git.Run([]string{"diff", "main..HEAD", "--no-color"})
|
|
if err != nil || diff == "" {
|
|
diff, err = git.Run([]string{"diff", "origin/main..HEAD", "--no-color"})
|
|
if err != nil {
|
|
color.Red("Failed to get branch diff: %v", err)
|
|
return
|
|
}
|
|
}
|
|
if diff == "" {
|
|
color.Yellow("No changes on this branch compared to main/origin/main.")
|
|
return
|
|
}
|
|
modelFlag, _ := cmd.Flags().GetString("model")
|
|
model := config.GetModel("prdescribe", modelFlag)
|
|
|
|
client := grok.NewClient()
|
|
messages := buildPRDescribeMessages(diff)
|
|
color.Yellow("Writing PR description...")
|
|
client.Stream(messages, model)
|
|
},
|
|
}
|
|
|
|
func buildPRDescribeMessages(diff string) []map[string]string {
|
|
return []map[string]string{
|
|
{"role": "system", "content": "Write a professional GitHub PR title + detailed body (changes, motivation, testing notes)."},
|
|
{"role": "user", "content": fmt.Sprintf("Diff:\n%s", diff)},
|
|
}
|
|
}
|