blob: f305b4b0cf47901ba3dd7fd9a53b5bee342ccc51 [file] [log] [blame]
package claudetool
import (
"bytes"
"context"
"fmt"
"log/slog"
"os"
"os/exec"
"path/filepath"
"strings"
)
// A CodeReviewer manages quality checks.
type CodeReviewer struct {
repoRoot string
initialCommit string
initialStatus []fileStatus // git status of files at initial commit, absolute paths
reviewed []string // history of all commits which have been reviewed
initialWorktree string // git worktree at initial commit, absolute path
}
func NewCodeReviewer(ctx context.Context, repoRoot, initialCommit string) (*CodeReviewer, error) {
r := &CodeReviewer{
repoRoot: repoRoot,
initialCommit: initialCommit,
}
if r.repoRoot == "" {
return nil, fmt.Errorf("NewCodeReviewer: repoRoot must be non-empty")
}
if r.initialCommit == "" {
return nil, fmt.Errorf("NewCodeReviewer: initialCommit must be non-empty")
}
// Confirm that root is in fact the git repo root.
root, err := findRepoRoot(r.repoRoot)
if err != nil {
return nil, err
}
if root != r.repoRoot {
return nil, fmt.Errorf("NewCodeReviewer: repoRoot=%q but git repo root is %q", r.repoRoot, root)
}
// Get an initial list of dirty and untracked files.
// We'll filter them out later when deciding whether the worktree is clean.
status, err := r.repoStatus(ctx)
if err != nil {
return nil, err
}
r.initialStatus = status
return r, nil
}
// Autoformat formats all files changed in HEAD.
// It returns a list of all files that were formatted.
// It is best-effort only.
func (r *CodeReviewer) Autoformat(ctx context.Context) []string {
// Refuse to format if HEAD == r.InitialCommit
head, err := r.CurrentCommit(ctx)
if err != nil {
slog.WarnContext(ctx, "CodeReviewer.Autoformat unable to get current commit", "err", err)
return nil
}
parent, err := r.ResolveCommit(ctx, "HEAD^1")
if err != nil {
slog.WarnContext(ctx, "CodeReviewer.Autoformat unable to get parent commit", "err", err)
return nil
}
if head == r.initialCommit {
slog.WarnContext(ctx, "CodeReviewer.Autoformat refusing to format because HEAD == InitialCommit")
return nil
}
// Retrieve a list of all files changed
// TODO: instead of one git diff --name-only and then N --name-status, do one --name-status.
changedFiles, err := r.changedFiles(ctx, r.initialCommit, head)
if err != nil {
slog.WarnContext(ctx, "CodeReviewer.Autoformat unable to get changed files", "err", err)
return nil
}
// General strategy: For all changed files,
// run the strictest formatter that passes on the original version.
// TODO: add non-Go formatters?
// TODO: at a minimum, for common file types, ensure trailing newlines and maybe trim trailing whitespace per line?
var fmtFiles []string
for _, file := range changedFiles {
if !strings.HasSuffix(file, ".go") {
continue
}
fileStatus, err := r.gitFileStatus(ctx, file)
if err != nil {
slog.WarnContext(ctx, "CodeReviewer.Autoformat unable to get file status", "file", file, "err", err)
continue
}
if fileStatus == "D" { // deleted, nothing to format
continue
}
code, err := r.getFileContentAtCommit(ctx, file, head)
if err != nil {
slog.WarnContext(ctx, "CodeReviewer.Autoformat unable to get file content at head", "file", file, "err", err)
continue
}
if isAutogeneratedGoFile(code) { // leave autogenerated files alone
continue
}
onDisk, err := os.ReadFile(file)
if err != nil {
slog.WarnContext(ctx, "CodeReviewer.Autoformat unable to read file", "file", file, "err", err)
continue
}
if !bytes.Equal(code, onDisk) { // file has been modified since HEAD
slog.WarnContext(ctx, "CodeReviewer.Autoformat file modified since HEAD", "file", file, "err", err)
continue
}
var formatterToUse string
if fileStatus == "A" {
formatterToUse = "gofumpt" // newly added, so we can format how we please: use gofumpt
} else {
prev, err := r.getFileContentAtCommit(ctx, file, parent)
if err != nil {
slog.WarnContext(ctx, "CodeReviewer.Autoformat unable to get file content at parent", "file", file, "err", err)
continue
}
formatterToUse = r.pickFormatter(ctx, prev) // pick the strictest formatter that passes on the original version
}
// Apply the chosen formatter to the current file
newCode := r.runFormatter(ctx, formatterToUse, code)
if newCode == nil { // no changes made
continue
}
// write to disk
if err := os.WriteFile(file, newCode, 0o600); err != nil {
slog.WarnContext(ctx, "CodeReviewer.Autoformat unable to write formatted file", "file", file, "err", err)
continue
}
fmtFiles = append(fmtFiles, file)
}
return fmtFiles
}
// RequireNormalGitState checks that the git repo state is pretty normal.
func (r *CodeReviewer) RequireNormalGitState(_ context.Context) error {
rebaseDirs := []string{"rebase-merge", "rebase-apply"}
for _, dir := range rebaseDirs {
_, err := os.Stat(filepath.Join(r.repoRoot, dir))
if err == nil {
return fmt.Errorf("git repo is not clean: rebase in progress")
}
}
filesReason := map[string]string{
"MERGE_HEAD": "merge is in progress",
"CHERRY_PICK_HEAD": "cherry-pick is in progress",
"REVERT_HEAD": "revert is in progress",
"BISECT_LOG": "bisect is in progress",
}
for file, reason := range filesReason {
_, err := os.Stat(filepath.Join(r.repoRoot, file))
if err == nil {
return fmt.Errorf("git repo is not clean: %s", reason)
}
}
return nil
}
func (r *CodeReviewer) RequireNoUncommittedChanges(ctx context.Context) error {
// Check that there are no uncommitted changes, whether staged or not.
// (Changes in r.initialStatus are OK, no other changes are.)
statuses, err := r.repoStatus(ctx)
if err != nil {
return fmt.Errorf("unable to get repo status: %w", err)
}
uncommitted := new(strings.Builder)
for _, status := range statuses {
if !r.initialStatusesContainFile(status.Path) {
fmt.Fprintf(uncommitted, "%s %s\n", status.Path, status.RawStatus)
}
}
if uncommitted.Len() > 0 {
return fmt.Errorf("uncommitted changes in repo, please commit or revert:\n%s", uncommitted.String())
}
return nil
}
func (r *CodeReviewer) initialStatusesContainFile(file string) bool {
for _, s := range r.initialStatus {
if s.Path == file {
return true
}
}
return false
}
type fileStatus struct {
Path string
RawStatus string // always 2 characters
}
func (r *CodeReviewer) repoStatus(ctx context.Context) ([]fileStatus, error) {
// Run git status --porcelain, split into lines
cmd := exec.CommandContext(ctx, "git", "status", "--porcelain")
cmd.Dir = r.repoRoot
out, err := cmd.CombinedOutput()
if err != nil {
return nil, fmt.Errorf("failed to run git status: %w\n%s", err, out)
}
var statuses []fileStatus
for line := range strings.Lines(string(out)) {
if len(line) == 0 {
continue
}
if len(line) < 3 {
return nil, fmt.Errorf("invalid status line: %s", line)
}
path := line[3:]
status := line[:2]
absPath := r.absPath(path)
statuses = append(statuses, fileStatus{Path: absPath, RawStatus: status})
}
return statuses, nil
}
// CurrentCommit retrieves the current git commit hash
func (r *CodeReviewer) CurrentCommit(ctx context.Context) (string, error) {
return r.ResolveCommit(ctx, "HEAD")
}
func (r *CodeReviewer) ResolveCommit(ctx context.Context, ref string) (string, error) {
cmd := exec.CommandContext(ctx, "git", "rev-parse", ref)
cmd.Dir = r.repoRoot
out, err := cmd.CombinedOutput()
if err != nil {
return "", fmt.Errorf("failed to get current commit hash: %w\n%s", err, out)
}
return strings.TrimSpace(string(out)), nil
}
func (r *CodeReviewer) absPath(relPath string) string {
return filepath.Clean(filepath.Join(r.repoRoot, relPath))
}
// gitFileStatus returns the status of a file (A for added, M for modified, D for deleted, etc.)
func (r *CodeReviewer) gitFileStatus(ctx context.Context, file string) (string, error) {
cmd := exec.CommandContext(ctx, "git", "diff", "--name-status", r.initialCommit, "HEAD", "--", file)
cmd.Dir = r.repoRoot
out, err := cmd.CombinedOutput()
if err != nil {
return "", fmt.Errorf("failed to get file status: %w\n%s", err, out)
}
status := strings.TrimSpace(string(out))
if status == "" {
return "", fmt.Errorf("no status found for file: %s", file)
}
return string(status[0]), nil
}
// getFileContentAtCommit retrieves file content at a specific commit
func (r *CodeReviewer) getFileContentAtCommit(ctx context.Context, file, commit string) ([]byte, error) {
relFile, err := filepath.Rel(r.repoRoot, file)
if err != nil {
slog.WarnContext(ctx, "CodeReviewer.getFileContentAtCommit: failed to get relative path", "repo_root", r.repoRoot, "file", file, "err", err)
file = relFile
}
cmd := exec.CommandContext(ctx, "git", "show", fmt.Sprintf("%s:%s", commit, relFile))
cmd.Dir = r.repoRoot
out, err := cmd.CombinedOutput()
if err != nil {
return nil, fmt.Errorf("failed to get file content at commit %s: %w\n%s", commit, err, out)
}
return out, nil
}
// runFormatter runs the specified formatter on a file and returns the results.
// A nil result indicates that the file is unchanged, or that an error occurred.
func (r *CodeReviewer) runFormatter(ctx context.Context, formatter string, content []byte) []byte {
if formatter == "" {
return nil // no formatter
}
// Run the formatter and capture the output
cmd := exec.CommandContext(ctx, formatter)
cmd.Dir = r.repoRoot
cmd.Stdin = bytes.NewReader(content)
out, err := cmd.CombinedOutput()
if err != nil {
// probably a parse error, err on the side of safety
return nil
}
if bytes.Equal(content, out) {
return nil // no changes
}
return out
}
// formatterWouldChange reports whether a formatter would make changes to the content.
// If the contents are invalid, it returns false.
// It works by piping the content to the formatter with the -l flag.
func (r *CodeReviewer) formatterWouldChange(ctx context.Context, formatter string, content []byte) bool {
cmd := exec.CommandContext(ctx, formatter, "-l")
cmd.Dir = r.repoRoot
cmd.Stdin = bytes.NewReader(content)
out, err := cmd.CombinedOutput()
if err != nil {
// probably a parse error, err on the side of safety
return false
}
// If the output is empty, the file passes the formatter
// If the output contains "<standard input>", the file would be changed
return len(bytes.TrimSpace(out)) > 0
}
// pickFormatter picks a formatter to use for code.
// If something goes wrong, it recommends no formatter (empty string).
func (r *CodeReviewer) pickFormatter(ctx context.Context, code []byte) string {
// Test each formatter from strictest to least strict.
// Keep the first one that doesn't make changes.
formatters := []string{"gofumpt", "goimports", "gofmt"}
for _, formatter := range formatters {
if r.formatterWouldChange(ctx, formatter, code) {
continue
}
return formatter
}
return "" // no safe formatter found
}
// changedFiles retrieves a list of all files changed between two commits
func (r *CodeReviewer) changedFiles(ctx context.Context, fromCommit, toCommit string) ([]string, error) {
cmd := exec.CommandContext(ctx, "git", "diff", "--name-only", fromCommit, toCommit)
cmd.Dir = r.repoRoot
out, err := cmd.CombinedOutput()
if err != nil {
return nil, fmt.Errorf("failed to get changed files: %w\n%s", err, out)
}
var files []string
for line := range strings.Lines(string(out)) {
line = strings.TrimSpace(line)
if len(line) == 0 {
continue
}
path := r.absPath(line)
if r.initialStatusesContainFile(path) {
continue
}
files = append(files, path)
}
return files, nil
}