Initial implementation: AI code review bot for Gitea

- CLI binary with flag/env var configuration - Gitea API client (PR metadata, diff, CI status, post review) - OpenAI-compatible LLM client - Structured review prompt with conventions support - JSON response parser with validation - Markdown review formatter for Gitea - CI failure auto-detection (REQUEST_CHANGES) - Dry-run mode for testing
2026-05-01 09:42:45 -07:00
commit 700f186023
9 changed files with 673 additions and 0 deletions
@@ -0,0 +1,168 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"os"
+	"strconv"
+	"strings"
+
+	"gitea.weiker.me/rodin/review-bot/gitea"
+	"gitea.weiker.me/rodin/review-bot/llm"
+	"gitea.weiker.me/rodin/review-bot/review"
+)
+
+func main() {
+	// CLI flags
+	giteaURL := flag.String("gitea-url", envOrDefault("GITEA_URL", ""), "Gitea instance URL")
+	repo := flag.String("repo", envOrDefault("GITEA_REPO", ""), "Repository (owner/name)")
+	prNum := flag.String("pr", envOrDefault("PR_NUMBER", ""), "Pull request number")
+	reviewerName := flag.String("reviewer-name", envOrDefault("REVIEWER_NAME", ""), "Reviewer display name")
+	reviewerToken := flag.String("reviewer-token", envOrDefault("REVIEWER_TOKEN", ""), "Gitea token for posting review")
+	llmBaseURL := flag.String("llm-base-url", envOrDefault("LLM_BASE_URL", ""), "LLM API base URL")
+	llmAPIKey := flag.String("llm-api-key", envOrDefault("LLM_API_KEY", ""), "LLM API key")
+	llmModel := flag.String("llm-model", envOrDefault("LLM_MODEL", ""), "LLM model name")
+	conventionsFile := flag.String("conventions-file", envOrDefault("CONVENTIONS_FILE", ""), "Conventions file path in repo (e.g. CLAUDE.md)")
+	dryRun := flag.Bool("dry-run", false, "Print review to stdout instead of posting")
+
+	flag.Parse()
+
+	// Validate required fields
+	if *giteaURL == "" || *repo == "" || *prNum == "" || *reviewerToken == "" ||
+		*llmBaseURL == "" || *llmAPIKey == "" || *llmModel == "" {
+		fmt.Fprintf(os.Stderr, "Error: missing required flags or environment variables\n\n")
+		fmt.Fprintf(os.Stderr, "Required: --gitea-url, --repo, --pr, --reviewer-token, --llm-base-url, --llm-api-key, --llm-model\n")
+		os.Exit(1)
+	}
+
+	// Parse repo owner/name
+	parts := strings.SplitN(*repo, "/", 2)
+	if len(parts) != 2 {
+		log.Fatalf("Invalid repo format %q, expected owner/name", *repo)
+	}
+	owner, repoName := parts[0], parts[1]
+
+	// Parse PR number
+	prNumber, err := strconv.Atoi(*prNum)
+	if err != nil {
+		log.Fatalf("Invalid PR number %q: %v", *prNum, err)
+	}
+
+	// Initialize clients
+	giteaClient := gitea.NewClient(*giteaURL, *reviewerToken)
+	llmClient := llm.NewClient(*llmBaseURL, *llmAPIKey, *llmModel)
+
+	log.Printf("Reviewing PR #%d on %s/%s", prNumber, owner, repoName)
+
+	// Step 1: Fetch PR metadata
+	pr, err := giteaClient.GetPullRequest(owner, repoName, prNumber)
+	if err != nil {
+		log.Fatalf("Failed to fetch PR: %v", err)
+	}
+	log.Printf("PR: %s", pr.Title)
+
+	// Step 2: Fetch diff
+	diff, err := giteaClient.GetPullRequestDiff(owner, repoName, prNumber)
+	if err != nil {
+		log.Fatalf("Failed to fetch diff: %v", err)
+	}
+	log.Printf("Diff size: %d bytes", len(diff))
+
+	// Step 3: Check CI status
+	ciPassed := true
+	ciDetails := ""
+	if pr.Head.Sha != "" {
+		statuses, err := giteaClient.GetCommitStatuses(owner, repoName, pr.Head.Sha)
+		if err != nil {
+			log.Printf("Warning: could not fetch CI status: %v", err)
+		} else {
+			ciPassed, ciDetails = evaluateCIStatus(statuses)
+			log.Printf("CI status: passed=%v", ciPassed)
+		}
+	}
+
+	// Step 4: Load conventions file if specified
+	conventions := ""
+	if *conventionsFile != "" {
+		content, err := giteaClient.GetFileContent(owner, repoName, *conventionsFile)
+		if err != nil {
+			log.Printf("Warning: could not load conventions file %q: %v", *conventionsFile, err)
+		} else {
+			conventions = content
+			log.Printf("Loaded conventions file: %s (%d bytes)", *conventionsFile, len(conventions))
+		}
+	}
+
+	// Step 5: Build prompts
+	systemPrompt := review.BuildSystemPrompt(conventions)
+	userPrompt := review.BuildUserPrompt(pr.Title, pr.Body, diff, ciPassed, ciDetails)
+
+	// Step 6: Call LLM
+	log.Printf("Sending to LLM (%s)...", *llmModel)
+	messages := []llm.Message{
+		{Role: "system", Content: systemPrompt},
+		{Role: "user", Content: userPrompt},
+	}
+
+	response, err := llmClient.Complete(messages)
+	if err != nil {
+		log.Fatalf("LLM request failed: %v", err)
+	}
+	log.Printf("LLM response received (%d bytes)", len(response))
+
+	// Step 7: Parse response
+	result, err := review.ParseResponse(response)
+	if err != nil {
+		log.Fatalf("Failed to parse LLM response: %v", err)
+	}
+	log.Printf("Verdict: %s (%d findings)", result.Verdict, len(result.Findings))
+
+	// Step 8: Format and post review
+	reviewBody := review.FormatMarkdown(result, *reviewerName)
+	event := review.GiteaEvent(result.Verdict)
+
+	if *dryRun {
+		fmt.Println("--- DRY RUN ---")
+		fmt.Printf("Event: %s\n\n", event)
+		fmt.Println(reviewBody)
+		return
+	}
+
+	log.Printf("Posting review (event=%s)...", event)
+	if err := giteaClient.PostReview(owner, repoName, prNumber, event, reviewBody); err != nil {
+		log.Fatalf("Failed to post review: %v", err)
+	}
+	log.Printf("Review posted successfully!")
+}
+
+// evaluateCIStatus checks if all CI statuses indicate success.
+func evaluateCIStatus(statuses []gitea.CommitStatus) (passed bool, details string) {
+	if len(statuses) == 0 {
+		return true, "no CI statuses found"
+	}
+
+	var failed []string
+	for _, s := range statuses {
+		switch s.Status {
+		case "success":
+			// good
+		case "pending":
+			// treat pending as not-failed
+		case "failure", "error":
+			failed = append(failed, fmt.Sprintf("%s: %s", s.Context, s.Description))
+		}
+	}
+
+	if len(failed) > 0 {
+		return false, strings.Join(failed, "; ")
+	}
+	return true, "all checks passed"
+}
+
+func envOrDefault(key, defaultVal string) string {
+	if v := os.Getenv(key); v != "" {
+		return v
+	}
+	return defaultVal
+}