Merge remote-tracking branch 'origin/main' into ao/reverbcode-2/issue-2272-prompts

This commit is contained in:
whoisasx 2026-06-29 22:42:31 +05:30
commit 71411052f3
58 changed files with 2988 additions and 459 deletions

View File

@ -150,6 +150,7 @@ func (p *Plugin) GetLaunchCommand(ctx context.Context, cfg ports.LaunchConfig) (
permissions = cfg.Config.Permissions
}
appendPermissionFlags(&cmd, permissions)
appendToolFlags(&cmd, cfg.AllowedTools, cfg.DisallowedTools)
if model := strings.TrimSpace(cfg.Config.Model); model != "" {
cmd = append(cmd, "--model", model)
@ -328,6 +329,20 @@ func appendPermissionFlags(cmd *[]string, permissions ports.PermissionMode) {
}
}
// appendToolFlags emits --allowedTools / --disallowedTools for a tool-scoped
// launch. Each list is joined with commas into one value so rules that contain
// spaces (e.g. "Bash(git diff:*)") are not split into separate tool names.
// Empty lists emit nothing, so an unrestricted launch is unchanged. These rules
// only bite when the launch is off bypassPermissions, which ignores them.
func appendToolFlags(cmd *[]string, allowed, disallowed []string) {
if len(allowed) > 0 {
*cmd = append(*cmd, "--allowedTools", strings.Join(allowed, ","))
}
if len(disallowed) > 0 {
*cmd = append(*cmd, "--disallowedTools", strings.Join(disallowed, ","))
}
}
func normalizePermissionMode(mode ports.PermissionMode) ports.PermissionMode {
switch mode {
case ports.PermissionModeDefault,

View File

@ -604,6 +604,38 @@ func readJSON(t *testing.T, path string) map[string]any {
return m
}
func TestGetLaunchCommandEmitsToolAllowlist(t *testing.T) {
p := &Plugin{resolvedBinary: "claude"}
cmd, err := p.GetLaunchCommand(context.Background(), ports.LaunchConfig{
AllowedTools: []string{"Read", "Grep", "Bash(git diff:*)"},
DisallowedTools: []string{"Edit", "Write", "Bash(git push:*)"},
})
if err != nil {
t.Fatal(err)
}
// Each list is one comma-joined value so a rule with spaces stays intact.
if !containsSubsequence(cmd, []string{"--allowedTools", "Read,Grep,Bash(git diff:*)"}) {
t.Fatalf("missing joined --allowedTools value; got %#v", cmd)
}
if !containsSubsequence(cmd, []string{"--disallowedTools", "Edit,Write,Bash(git push:*)"}) {
t.Fatalf("missing joined --disallowedTools value; got %#v", cmd)
}
}
func TestGetLaunchCommandOmitsToolFlagsWhenUnset(t *testing.T) {
p := &Plugin{resolvedBinary: "claude"}
cmd, err := p.GetLaunchCommand(context.Background(), ports.LaunchConfig{Prompt: "do it"})
if err != nil {
t.Fatal(err)
}
if contains(cmd, "--allowedTools") || contains(cmd, "--disallowedTools") {
t.Fatalf("unrestricted launch should emit no tool flags; got %#v", cmd)
}
}
func contains(values []string, needle string) bool {
for _, v := range values {
if v == needle {

View File

@ -31,6 +31,37 @@ func (r *Reviewer) Harness() domain.ReviewerHarness {
var _ ports.Reviewer = (*Reviewer)(nil)
// reviewerAllowedTools is the read-only tool allowlist the reviewer launches
// with. The reviewer runs headless (no human to approve prompts) but must stay
// read-only, so instead of bypassPermissions — which skips the permission
// system entirely and ignores allow/deny rules — it launches in the default
// mode where these rules are honored: allow rules auto-approve without
// prompting, so the reviewer can read the checkout and run the few commands it
// needs (git diff/log/show to inspect the PR, gh to post the review, and
// `ao review submit` to record the verdict) without stalling.
var reviewerAllowedTools = []string{
"Read",
"Grep",
"Glob",
"Bash(gh:*)",
"Bash(git diff:*)",
"Bash(git log:*)",
"Bash(git show:*)",
"Bash(git status:*)",
"Bash(ao review submit:*)",
}
// reviewerDisallowedTools hard-denies the write paths as defense in depth, so a
// misbehaving model cannot edit files or move the branch even if a future
// allowlist entry would otherwise admit it.
var reviewerDisallowedTools = []string{
"Edit",
"Write",
"NotebookEdit",
"Bash(git push:*)",
"Bash(git commit:*)",
}
// ReviewCommand builds a claude-code invocation that reviews the worker's
// checkout for the PR, with the review prompt baked in.
func (r *Reviewer) ReviewCommand(ctx context.Context, inv ports.ReviewInvocation) (ports.ReviewCommandSpec, error) {
@ -39,10 +70,12 @@ func (r *Reviewer) ReviewCommand(ctx context.Context, inv ports.ReviewInvocation
WorkspacePath: inv.WorkspacePath,
Prompt: inv.Prompt,
SystemPrompt: inv.SystemPrompt,
// The reviewer runs headless with no human to approve tool prompts; it
// is read-only by prompt and must run gh/ao on its own, so bypass the
// permission gate rather than stall on the first prompt.
Permissions: ports.PermissionModeBypassPermissions,
// Launch off bypassPermissions so the allow/deny lists are enforced.
// Set an explicit non-bypass mode instead of deferring to the user's
// Claude defaultMode, which may itself be bypassPermissions.
Permissions: ports.PermissionModeAuto,
AllowedTools: reviewerAllowedTools,
DisallowedTools: reviewerDisallowedTools,
})
if err != nil {
return ports.ReviewCommandSpec{}, err

View File

@ -0,0 +1,71 @@
package claudecode
import (
"context"
"testing"
"github.com/aoagents/agent-orchestrator/backend/internal/ports"
)
// captureAgent is a stub ports.Agent that records the LaunchConfig the reviewer
// builds, so the test asserts the reviewer's tool policy without needing the
// real claude binary on PATH.
type captureAgent struct {
got ports.LaunchConfig
}
func (a *captureAgent) GetConfigSpec(context.Context) (ports.ConfigSpec, error) {
return ports.ConfigSpec{}, nil
}
func (a *captureAgent) GetLaunchCommand(_ context.Context, cfg ports.LaunchConfig) ([]string, error) {
a.got = cfg
return []string{"claude"}, nil
}
func (a *captureAgent) GetPromptDeliveryStrategy(context.Context, ports.LaunchConfig) (ports.PromptDeliveryStrategy, error) {
return ports.PromptDeliveryInCommand, nil
}
func (a *captureAgent) GetAgentHooks(context.Context, ports.WorkspaceHookConfig) error { return nil }
func (a *captureAgent) GetRestoreCommand(context.Context, ports.RestoreConfig) ([]string, bool, error) {
return nil, false, nil
}
func (a *captureAgent) SessionInfo(context.Context, ports.SessionRef) (ports.SessionInfo, bool, error) {
return ports.SessionInfo{}, false, nil
}
func TestReviewCommandLaunchesReadOnlyOffBypass(t *testing.T) {
agent := &captureAgent{}
r := &Reviewer{agent: agent}
if _, err := r.ReviewCommand(context.Background(), ports.ReviewInvocation{
ReviewerID: "review-w1",
WorkspacePath: "/ws/w1",
Prompt: "review it",
SystemPrompt: "you are a reviewer",
}); err != nil {
t.Fatalf("ReviewCommand: %v", err)
}
// The allowlist is what enforces read-only, so it must launch in an
// explicit non-bypass mode: bypassPermissions ignores allow/deny rules
// entirely, and an empty mode would defer to a user's defaultMode.
if agent.got.Permissions != ports.PermissionModeAuto {
t.Fatalf("reviewer must launch in auto permission mode; got %q", agent.got.Permissions)
}
if !contains(agent.got.AllowedTools, "Read") || !contains(agent.got.AllowedTools, "Bash(ao review submit:*)") {
t.Fatalf("allowlist missing read-only review tools: %#v", agent.got.AllowedTools)
}
for _, denied := range []string{"Edit", "Write", "Bash(git push:*)", "Bash(git commit:*)"} {
if !contains(agent.got.DisallowedTools, denied) {
t.Fatalf("disallow list missing %q: %#v", denied, agent.got.DisallowedTools)
}
}
}
func contains(values []string, needle string) bool {
for _, v := range values {
if v == needle {
return true
}
}
return false
}

View File

@ -33,6 +33,8 @@ const (
githubReviewCommentLimitPerThread = 5
// githubReviewThreadMaxPages bounds the explicit older-thread fallback.
githubReviewThreadMaxPages = 2
// githubReviewSummaryLimit bounds submitted decisive reviews used for summary links.
githubReviewSummaryLimit = 20
)
// ParseRepository normalizes a GitHub remote/origin URL into a provider-neutral
@ -156,12 +158,12 @@ func (p *Provider) FetchFailedCheckLogTail(ctx context.Context, repo ports.SCMRe
// FetchReviewThreads fetches review threads separately from the fast PR/CI path.
func (p *Provider) FetchReviewThreads(ctx context.Context, ref ports.SCMPRRef) (ports.SCMReviewObservation, error) {
latest, decision, pi, err := p.fetchReviewThreadPage(ctx, ref, "")
latest, reviews, decision, pi, err := p.fetchReviewThreadPage(ctx, ref, "", true)
if err != nil {
return ports.SCMReviewObservation{}, err
}
if !boolv(pi["hasPreviousPage"]) {
return ports.SCMReviewObservation{Decision: decision, Threads: latest}, nil
return ports.SCMReviewObservation{Decision: decision, Reviews: reviews, Threads: latest}, nil
}
out := latest
startCursor := str(pi["startCursor"])
@ -175,7 +177,7 @@ func (p *Provider) FetchReviewThreads(ctx context.Context, ref ports.SCMPRRef) (
p.logger.Warn("github scm: review thread page is partial but missing start cursor",
"repo", repoFullName(ref.Repo), "pr", ref.Number)
} else {
older, _, olderPI, err := p.fetchReviewThreadPage(ctx, ref, startCursor)
older, _, _, olderPI, err := p.fetchReviewThreadPage(ctx, ref, startCursor, false)
if err != nil {
return ports.SCMReviewObservation{}, err
}
@ -190,7 +192,7 @@ func (p *Provider) FetchReviewThreads(ctx context.Context, ref ports.SCMPRRef) (
}
}
}
return ports.SCMReviewObservation{Decision: decision, Threads: out, Partial: true}, nil
return ports.SCMReviewObservation{Decision: decision, Reviews: reviews, Threads: out, Partial: true}, nil
}
type restListPull struct {
@ -518,38 +520,76 @@ func mergeabilityObservation(providerMergeable, providerMergeState, ci, review s
return out
}
func (p *Provider) fetchReviewThreadPage(ctx context.Context, ref ports.SCMPRRef, beforeCursor string) ([]ports.SCMReviewThreadObservation, string, map[string]any, error) {
query := buildReviewThreadsQuery(ref, beforeCursor)
func (p *Provider) fetchReviewThreadPage(ctx context.Context, ref ports.SCMPRRef, beforeCursor string, includeReviews bool) ([]ports.SCMReviewThreadObservation, []ports.SCMReviewSummaryObservation, string, map[string]any, error) {
query := buildReviewThreadsQuery(ref, beforeCursor, includeReviews)
data, err := p.client.doGraphQL(ctx, query, nil)
if err != nil {
return nil, "", nil, err
return nil, nil, "", nil, err
}
repoData, _ := data["repo"].(map[string]any)
pr, _ := repoData["pullRequest"].(map[string]any)
if pr == nil {
return nil, "", nil, fmt.Errorf("%w: pull request not found in review response", ErrNotFound)
return nil, nil, "", nil, fmt.Errorf("%w: pull request not found in review response", ErrNotFound)
}
decision := string(reviewDecisionFromGraphQL(pr))
reviewSummaries, _ := pr["reviewSummaries"].(map[string]any)
reviews := make([]ports.SCMReviewSummaryObservation, 0, len(nodes(reviewSummaries["nodes"])))
for _, review := range nodes(reviewSummaries["nodes"]) {
summary := scmReviewSummaryFromGraphQL(review)
if summary.ID == "" && summary.URL == "" {
continue
}
reviews = append(reviews, summary)
}
threads, _ := pr["reviewThreads"].(map[string]any)
out := make([]ports.SCMReviewThreadObservation, 0, len(nodes(threads["nodes"])))
for _, th := range nodes(threads["nodes"]) {
out = append(out, scmThreadFromGraphQL(th))
}
pi, _ := threads["pageInfo"].(map[string]any)
return out, decision, pi, nil
return out, reviews, decision, pi, nil
}
func buildReviewThreadsQuery(ref ports.SCMPRRef, beforeCursor string) string {
func buildReviewThreadsQuery(ref ports.SCMPRRef, beforeCursor string, includeReviews bool) string {
before := "null"
if beforeCursor != "" {
before = graphQLString(beforeCursor)
}
reviewSelection := ""
if includeReviews {
reviewSelection = fmt.Sprintf(" reviewSummaries: reviews(last:%d, states:[APPROVED,CHANGES_REQUESTED]){ nodes{ id state url submittedAt author{ login __typename } } }", githubReviewSummaryLimit)
}
return fmt.Sprintf(`query{
repo: repository(owner:%s,name:%s){ pullRequest(number:%d){ reviewDecision reviewThreads(last:%d, before:%s){ nodes{
repo: repository(owner:%s,name:%s){ pullRequest(number:%d){ reviewDecision%s reviewThreads(last:%d, before:%s){ nodes{
id isResolved path line
comments(first:%d){ nodes{ id body url author{ login __typename } } }
} pageInfo{ hasPreviousPage startCursor } } } }
}`, graphQLString(ref.Repo.Owner), graphQLString(ref.Repo.Name), ref.Number, githubReviewThreadPageSize, before, githubReviewCommentLimitPerThread)
}`, graphQLString(ref.Repo.Owner), graphQLString(ref.Repo.Name), ref.Number, reviewSelection, githubReviewThreadPageSize, before, githubReviewCommentLimitPerThread)
}
func scmReviewSummaryFromGraphQL(review map[string]any) ports.SCMReviewSummaryObservation {
author, _ := review["author"].(map[string]any)
return ports.SCMReviewSummaryObservation{
ID: str(review["id"]),
Author: str(author["login"]),
State: string(reviewStateFromGraphQL(review["state"])),
URL: str(review["url"]),
IsBot: isBotAuthor(author),
SubmittedAt: parseGitHubTime(str(review["submittedAt"])),
}
}
func reviewStateFromGraphQL(state any) domain.ReviewDecision {
switch strings.ToUpper(strings.TrimSpace(str(state))) {
case "APPROVED":
return domain.ReviewApproved
case "CHANGES_REQUESTED":
return domain.ReviewChangesRequest
case "REVIEW_REQUIRED":
return domain.ReviewRequired
default:
return domain.ReviewNone
}
}
func scmThreadFromGraphQL(th map[string]any) ports.SCMReviewThreadObservation {

View File

@ -1369,6 +1369,9 @@ func TestFetchReviewThreadsUsesLatestWindowWithoutFallbackWhenOldestResolved(t *
if !strings.Contains(string(body), "reviewThreads(last:50, before:null)") {
t.Fatalf("review query should fetch latest 50, body=%s", body)
}
if !strings.Contains(string(body), "reviews(last:20, states:[APPROVED,CHANGES_REQUESTED])") {
t.Fatalf("review query should fetch decisive review summaries, body=%s", body)
}
if !strings.Contains(string(body), "comments(first:5)") {
t.Fatalf("review query should cap comments per thread, body=%s", body)
}
@ -1376,8 +1379,17 @@ func TestFetchReviewThreadsUsesLatestWindowWithoutFallbackWhenOldestResolved(t *
_ = json.NewEncoder(w).Encode(map[string]any{
"data": map[string]any{"repo": map[string]any{"pullRequest": map[string]any{
"reviewDecision": "CHANGES_REQUESTED",
"reviewSummaries": map[string]any{"nodes": []any{map[string]any{
"id": "review-1",
"state": "CHANGES_REQUESTED",
"url": "https://github.com/o/r/pull/1#pullrequestreview-1",
"submittedAt": "2026-06-15T00:00:00Z",
"author": map[string]any{"login": "alice", "__typename": "User"},
}}},
"reviewThreads": map[string]any{
"nodes": []any{map[string]any{"id": "latest-resolved", "path": "main.go", "line": 1, "isResolved": true, "comments": map[string]any{"nodes": []any{}}}},
"nodes": []any{map[string]any{"id": "latest-resolved", "path": "main.go", "line": 1, "isResolved": true, "comments": map[string]any{"nodes": []any{map[string]any{
"id": "comment-1", "body": "fix", "url": "https://github.com/o/r/pull/1#discussion_r1", "author": map[string]any{"login": "alice", "__typename": "User"},
}}}}},
"pageInfo": map[string]any{"hasPreviousPage": true, "startCursor": "latest-start"},
},
}}},
@ -1397,6 +1409,12 @@ func TestFetchReviewThreadsUsesLatestWindowWithoutFallbackWhenOldestResolved(t *
if len(review.Threads) != 1 || review.Threads[0].ID != "latest-resolved" {
t.Fatalf("threads = %#v", review.Threads)
}
if len(review.Reviews) != 1 || review.Reviews[0].Author != "alice" || review.Reviews[0].URL != "https://github.com/o/r/pull/1#pullrequestreview-1" {
t.Fatalf("reviews = %#v", review.Reviews)
}
if len(review.Threads[0].Comments) != 1 || review.Threads[0].Comments[0].URL != "https://github.com/o/r/pull/1#discussion_r1" {
t.Fatalf("thread comments = %#v", review.Threads[0].Comments)
}
}
func TestFetchReviewThreadsFetchesOneOlderPageWhenOldestUnresolved(t *testing.T) {

View File

@ -1,6 +1,7 @@
package cli
import (
"encoding/json"
"errors"
"fmt"
"io"
@ -15,29 +16,42 @@ import (
// reviewRun mirrors the daemon's domain.ReviewRun for the CLI client.
type reviewRun struct {
ID string `json:"id"`
SessionID string `json:"sessionId"`
Harness string `json:"harness"`
PRURL string `json:"prUrl"`
TargetSHA string `json:"targetSha"`
Status string `json:"status"`
Verdict string `json:"verdict"`
Body string `json:"body"`
CreatedAt time.Time `json:"createdAt"`
ID string `json:"id"`
SessionID string `json:"sessionId"`
BatchID string `json:"batchId"`
Harness string `json:"harness"`
PRURL string `json:"prUrl"`
TargetSHA string `json:"targetSha"`
Status string `json:"status"`
Verdict string `json:"verdict"`
Body string `json:"body"`
GithubReviewID string `json:"githubReviewId"`
CreatedAt time.Time `json:"createdAt"`
DeliveredAt *time.Time `json:"deliveredAt,omitempty"`
}
// reviewRunResponse mirrors controllers.ReviewRunResponse.
type reviewRunResponse struct {
Review reviewRun `json:"review"`
ReviewerHandleID string `json:"reviewerHandleId"`
Review reviewRun `json:"review"`
Reviews []reviewRun `json:"reviews"`
ReviewerHandleID string `json:"reviewerHandleId"`
}
// submitReviewItem mirrors controllers.SubmitReviewItem.
type submitReviewItem struct {
RunID string `json:"runId"`
Verdict string `json:"verdict"`
Body string `json:"body,omitempty"`
GithubReviewID string `json:"githubReviewId,omitempty"`
}
// submitReviewRequest mirrors controllers.SubmitReviewInput.
type submitReviewRequest struct {
RunID string `json:"runId"`
Verdict string `json:"verdict"`
Body string `json:"body"`
GithubReviewID string `json:"githubReviewId"`
RunID string `json:"runId,omitempty"`
Verdict string `json:"verdict,omitempty"`
Body string `json:"body,omitempty"`
GithubReviewID string `json:"githubReviewId,omitempty"`
Reviews []submitReviewItem `json:"reviews,omitempty"`
}
type reviewSubmitOptions struct {
@ -46,6 +60,7 @@ type reviewSubmitOptions struct {
verdict string
body string
reviewID string
reviews string
}
func newReviewCommand(ctx *commandContext) *cobra.Command {
@ -77,6 +92,7 @@ func newReviewSubmitCommand(ctx *commandContext) *cobra.Command {
cmd.Flags().StringVar(&opts.verdict, "verdict", "", "Review verdict: approved or changes_requested (required)")
cmd.Flags().StringVar(&opts.body, "body", "", "Review body: a path to a Markdown file, or - to read from stdin (so nothing is written into the worktree)")
cmd.Flags().StringVar(&opts.reviewID, "review-id", "", "Id of the GitHub PR review just posted (the .id from the gh api POST that created the review)")
cmd.Flags().StringVar(&opts.reviews, "reviews", "", "JSON review results array or object: a path, or - to read from stdin")
return cmd
}
@ -88,6 +104,9 @@ func (c *commandContext) submitReview(cmd *cobra.Command, args []string, opts re
if session == "" {
return usageError{errors.New("usage: worker session id is required (positional or --session)")}
}
if strings.TrimSpace(opts.reviews) != "" {
return c.submitReviewBatch(cmd, session, opts)
}
runID := strings.TrimSpace(opts.runID)
if runID == "" {
return usageError{errors.New("usage: --run is required")}
@ -121,3 +140,49 @@ func (c *commandContext) submitReview(cmd *cobra.Command, args []string, opts re
_, err := fmt.Fprintf(cmd.OutOrStdout(), "recorded %s review for %s\n", res.Review.Verdict, session)
return err
}
func (c *commandContext) submitReviewBatch(cmd *cobra.Command, session string, opts reviewSubmitOptions) error {
if strings.TrimSpace(opts.runID) != "" || strings.TrimSpace(opts.verdict) != "" || strings.TrimSpace(opts.body) != "" || strings.TrimSpace(opts.reviewID) != "" {
return usageError{errors.New("usage: --reviews cannot be combined with --run, --verdict, --body, or --review-id")}
}
reviews, err := readReviewItems(cmd, strings.TrimSpace(opts.reviews))
if err != nil {
return err
}
path := "sessions/" + url.PathEscape(session) + "/reviews/submit"
var res reviewRunResponse
if err := c.postJSON(cmd.Context(), path, submitReviewRequest{Reviews: reviews}, &res); err != nil {
return err
}
count := len(res.Reviews)
if count == 0 {
count = len(reviews)
}
_, err = fmt.Fprintf(cmd.OutOrStdout(), "recorded %d review(s) for %s\n", count, session)
return err
}
func readReviewItems(cmd *cobra.Command, path string) ([]submitReviewItem, error) {
var raw []byte
var err error
if path == "-" {
raw, err = io.ReadAll(cmd.InOrStdin())
} else {
raw, err = os.ReadFile(path)
}
if err != nil {
return nil, usageError{fmt.Errorf("read review results: %w", err)}
}
var req submitReviewRequest
if err := json.Unmarshal(raw, &req); err == nil && len(req.Reviews) > 0 {
return req.Reviews, nil
}
var reviews []submitReviewItem
if err := json.Unmarshal(raw, &reviews); err != nil {
return nil, usageError{fmt.Errorf("decode review results JSON: %w", err)}
}
if len(reviews) == 0 {
return nil, usageError{errors.New("usage: --reviews requires at least one review result")}
}
return reviews, nil
}

View File

@ -104,6 +104,32 @@ func TestReviewSubmitAcceptsUnderscoreFlags(t *testing.T) {
}
}
func TestReviewSubmitBatchReadsReviewsFromStdin(t *testing.T) {
cfg := setConfigEnv(t)
srv, capture := reviewServer(t, http.StatusOK, `{"reviews":[{"id":"run-1","verdict":"changes_requested"},{"id":"run-2","verdict":"approved"}]}`)
writeRunFileFor(t, cfg, srv)
deps := aliveDeps()
deps.In = strings.NewReader(`{"reviews":[{"runId":"run-1","verdict":"changes_requested","body":"fix auth","githubReviewId":"101"},{"runId":"run-2","verdict":"approved","body":"looks good"}]}`)
out, errOut, err := executeCLI(t, deps, "review", "submit", "mer-1", "--reviews", "-")
if err != nil {
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
}
if !strings.Contains(out, "recorded 2 review(s) for mer-1") {
t.Fatalf("stdout = %q", out)
}
var req submitReviewRequest
if err := json.Unmarshal([]byte(capture.body), &req); err != nil {
t.Fatalf("decode body: %v", err)
}
if len(req.Reviews) != 2 || req.Reviews[0].RunID != "run-1" || req.Reviews[0].GithubReviewID != "101" || req.Reviews[1].Verdict != "approved" {
t.Fatalf("request = %+v", req)
}
if req.RunID != "" || req.Verdict != "" {
t.Fatalf("batch request should not also set legacy fields: %+v", req)
}
}
func TestReviewSubmitUsesSessionFlag(t *testing.T) {
cfg := setConfigEnv(t)
srv, capture := reviewServer(t, http.StatusOK, `{"review":{"verdict":"approved"}}`)

View File

@ -107,6 +107,16 @@ type PullRequestReviewThread struct {
UpdatedAt time.Time
}
// PullRequestReview is one submitted provider review for a pull request.
type PullRequestReview struct {
ID string
Author string
State ReviewDecision
URL string
IsBot bool
SubmittedAt time.Time
}
// CIState is the aggregate CI status of a PR.
type CIState string

View File

@ -29,11 +29,15 @@ type Review struct {
// ReviewRun is one review pass against a worker's PR.
type ReviewRun struct {
ID string `json:"id"`
ReviewID string `json:"reviewId"`
SessionID SessionID `json:"sessionId"`
Harness ReviewerHarness `json:"harness"`
PRURL string `json:"prUrl"`
ID string `json:"id"`
ReviewID string `json:"reviewId"`
SessionID SessionID `json:"sessionId"`
// BatchID groups review runs created by one trigger so worker feedback can
// be delivered once after the whole trigger batch is terminal. Empty marks
// legacy/single-run delivery.
BatchID string `json:"batchId"`
Harness ReviewerHarness `json:"harness"`
PRURL string `json:"prUrl"`
// TargetSHA is the PR head commit this pass reviewed.
TargetSHA string `json:"targetSha"`
Status ReviewRunStatus `json:"status"`

View File

@ -1292,13 +1292,13 @@ paths:
content:
application/json:
schema:
$ref: '#/components/schemas/ReviewRunResponse'
$ref: '#/components/schemas/TriggerReviewResponse'
description: OK
"201":
content:
application/json:
schema:
$ref: '#/components/schemas/ReviewRunResponse'
$ref: '#/components/schemas/TriggerReviewResponse'
description: Created
"404":
content:
@ -1719,7 +1719,7 @@ components:
type: string
reviews:
items:
$ref: '#/components/schemas/ReviewRun'
$ref: '#/components/schemas/PRReviewState'
type: array
required:
- reviewerHandleId
@ -1855,6 +1855,33 @@ components:
- id
- projectId
type: object
PRReviewState:
properties:
latestRun:
$ref: '#/components/schemas/ReviewRun'
prNumber:
type: integer
prUrl:
type: string
status:
enum:
- needs_review
- running
- up_to_date
- changes_requested
- ineligible
type: string
targetSha:
type: string
title:
type: string
required:
- prUrl
- prNumber
- title
- targetSha
- status
type: object
Project:
properties:
agent:
@ -2022,6 +2049,8 @@ components:
type: object
ReviewRun:
properties:
batchId:
type: string
body:
type: string
createdAt:
@ -2054,6 +2083,7 @@ components:
- id
- reviewId
- sessionId
- batchId
- harness
- prUrl
- targetSha
@ -2069,8 +2099,13 @@ components:
$ref: '#/components/schemas/ReviewRun'
reviewerHandleId:
type: string
reviews:
items:
$ref: '#/components/schemas/ReviewRun'
type: array
required:
- review
- reviews
- reviewerHandleId
type: object
RoleOverride:
@ -2342,10 +2377,14 @@ components:
properties:
count:
type: integer
isBot:
type: boolean
links:
items:
$ref: '#/components/schemas/SessionPRReviewCommentLink'
type: array
reviewUrl:
type: string
reviewerId:
type: string
required:
@ -2473,6 +2512,26 @@ components:
- projectId
type: object
SubmitReviewInput:
properties:
body:
description: Review body recorded by AO. Required for changes_requested.
type: string
githubReviewId:
description: Id of the GitHub PR review the reviewer posted, if any.
type: string
reviews:
description: Batched review results recorded by one reviewer CLI command.
items:
$ref: '#/components/schemas/SubmitReviewItem'
type: array
runId:
description: Review run id being completed.
type: string
verdict:
description: 'Review verdict: approved or changes_requested.'
type: string
type: object
SubmitReviewItem:
properties:
body:
description: Review body recorded by AO. Required for changes_requested.
@ -2489,8 +2548,18 @@ components:
required:
- runId
- verdict
- body
- githubReviewId
type: object
TriggerReviewResponse:
properties:
reviewerHandleId:
type: string
reviews:
items:
$ref: '#/components/schemas/PRReviewState'
type: array
required:
- reviewerHandleId
- reviews
type: object
WorkspaceRepo:
properties:

View File

@ -183,11 +183,14 @@ var schemaNames = map[string]string{
"ControllersResolveCommentsRequest": "ResolveCommentsRequest",
"ControllersResolveCommentsResponse": "ResolveCommentsResponse",
// httpd/controllers — review wire envelopes
"ControllersListReviewsResponse": "ListReviewsResponse",
"ControllersReviewRunResponse": "ReviewRunResponse",
"ControllersSubmitReviewInput": "SubmitReviewInput",
"ControllersListReviewsResponse": "ListReviewsResponse",
"ControllersReviewRunResponse": "ReviewRunResponse",
"ControllersTriggerReviewResponse": "TriggerReviewResponse",
"ControllersSubmitReviewItem": "SubmitReviewItem",
"ControllersSubmitReviewInput": "SubmitReviewInput",
// domain review entities
"DomainReviewRun": "ReviewRun",
"DomainReviewRun": "ReviewRun",
"ReviewPRReviewState": "PRReviewState",
// httpd/controllers: import wire envelopes
"ControllersImportStatusResponse": "ImportStatusResponse",
"ControllersImportRunResponse": "ImportRunResponse",
@ -378,8 +381,8 @@ func reviewOperations() []operation {
summary: "Trigger a code review of a worker's PR",
pathParams: []any{controllers.SessionIDParam{}},
resps: []respUnit{
{http.StatusOK, controllers.ReviewRunResponse{}},
{http.StatusCreated, controllers.ReviewRunResponse{}},
{http.StatusOK, controllers.TriggerReviewResponse{}},
{http.StatusCreated, controllers.TriggerReviewResponse{}},
{http.StatusUnprocessableEntity, envelope.APIError{}},
{http.StatusNotFound, envelope.APIError{}},
{http.StatusNotImplemented, envelope.APIError{}},

View File

@ -295,6 +295,8 @@ type SessionPRUnresolvedReviewer struct {
ReviewerID string `json:"reviewerId"`
Count int `json:"count"`
Links []SessionPRReviewCommentLink `json:"links"`
ReviewURL string `json:"reviewUrl,omitempty"`
IsBot bool `json:"isBot,omitempty"`
}
// SessionPRReviewCommentLink points to one unresolved review comment.
@ -366,7 +368,7 @@ func newSessionPRReviewSummary(in sessionsvc.PRReviewSummary) SessionPRReviewSum
for _, link := range reviewer.Links {
links = append(links, SessionPRReviewCommentLink{URL: link.URL, File: link.File, Line: link.Line})
}
reviewers = append(reviewers, SessionPRUnresolvedReviewer{ReviewerID: reviewer.ReviewerID, Count: reviewer.Count, Links: links})
reviewers = append(reviewers, SessionPRUnresolvedReviewer{ReviewerID: reviewer.ReviewerID, Count: reviewer.Count, Links: links, ReviewURL: reviewer.ReviewURL, IsBot: reviewer.IsBot})
}
return SessionPRReviewSummary{Decision: in.Decision, HasUnresolvedHumanComments: in.HasUnresolvedHumanComments, UnresolvedBy: reviewers}
}

View File

@ -10,6 +10,7 @@ import (
"github.com/aoagents/agent-orchestrator/backend/internal/domain"
"github.com/aoagents/agent-orchestrator/backend/internal/httpd/apispec"
"github.com/aoagents/agent-orchestrator/backend/internal/httpd/envelope"
reviewcore "github.com/aoagents/agent-orchestrator/backend/internal/review"
reviewsvc "github.com/aoagents/agent-orchestrator/backend/internal/service/review"
)
@ -17,23 +18,40 @@ import (
// reviewerHandleId is the live reviewer pane's runtime handle, for the UI to
// attach its terminal over /mux (empty when no reviewer has run).
type ListReviewsResponse struct {
ReviewerHandleID string `json:"reviewerHandleId"`
Reviews []domain.ReviewRun `json:"reviews"`
ReviewerHandleID string `json:"reviewerHandleId"`
Reviews []reviewcore.PRReviewState `json:"reviews"`
}
// ReviewRunResponse is the body of trigger (200/201) and submit (200). It
// carries the run plus the reviewer pane handle so the UI can attach a terminal.
// ReviewRunResponse is the body of submit (200). It carries the run plus the
// reviewer pane handle so the UI can attach a terminal.
type ReviewRunResponse struct {
Review domain.ReviewRun `json:"review"`
ReviewerHandleID string `json:"reviewerHandleId"`
Review domain.ReviewRun `json:"review"`
Reviews []domain.ReviewRun `json:"reviews"`
ReviewerHandleID string `json:"reviewerHandleId"`
}
// TriggerReviewResponse is the body of trigger (200/201). reviews carries the
// PR-scoped review state after the trigger.
type TriggerReviewResponse struct {
ReviewerHandleID string `json:"reviewerHandleId"`
Reviews []reviewcore.PRReviewState `json:"reviews"`
}
// SubmitReviewItem is one review result in a batched submit request.
type SubmitReviewItem struct {
RunID string `json:"runId" description:"Review run id being completed."`
Verdict string `json:"verdict" description:"Review verdict: approved or changes_requested."`
Body string `json:"body,omitempty" description:"Review body recorded by AO. Required for changes_requested."`
GithubReviewID string `json:"githubReviewId,omitempty" description:"Id of the GitHub PR review the reviewer posted, if any."`
}
// SubmitReviewInput is the body of POST /api/v1/sessions/{sessionId}/reviews/submit.
type SubmitReviewInput struct {
RunID string `json:"runId" description:"Review run id being completed."`
Verdict string `json:"verdict" description:"Review verdict: approved or changes_requested."`
Body string `json:"body" description:"Review body recorded by AO. Required for changes_requested."`
GithubReviewID string `json:"githubReviewId" description:"Id of the GitHub PR review the reviewer posted, if any."`
RunID string `json:"runId,omitempty" description:"Review run id being completed."`
Verdict string `json:"verdict,omitempty" description:"Review verdict: approved or changes_requested."`
Body string `json:"body,omitempty" description:"Review body recorded by AO. Required for changes_requested."`
GithubReviewID string `json:"githubReviewId,omitempty" description:"Id of the GitHub PR review the reviewer posted, if any."`
Reviews []SubmitReviewItem `json:"reviews,omitempty" description:"Batched review results recorded by one reviewer CLI command."`
}
// ReviewsController owns the session-scoped /reviews routes. A nil Svc returns 501.
@ -58,11 +76,11 @@ func (c *ReviewsController) list(w http.ResponseWriter, r *http.Request) {
writeReviewError(w, r, err)
return
}
runs := res.Runs
if runs == nil {
runs = []domain.ReviewRun{}
reviews := res.Reviews
if reviews == nil {
reviews = []reviewcore.PRReviewState{}
}
envelope.WriteJSON(w, http.StatusOK, ListReviewsResponse{ReviewerHandleID: res.ReviewerHandleID, Reviews: runs})
envelope.WriteJSON(w, http.StatusOK, ListReviewsResponse{ReviewerHandleID: res.ReviewerHandleID, Reviews: reviews})
}
func (c *ReviewsController) trigger(w http.ResponseWriter, r *http.Request) {
@ -81,7 +99,14 @@ func (c *ReviewsController) trigger(w http.ResponseWriter, r *http.Request) {
if res.Created {
status = http.StatusCreated
}
envelope.WriteJSON(w, status, ReviewRunResponse{Review: res.Run, ReviewerHandleID: res.ReviewerHandleID})
reviews := res.Reviews
if reviews == nil {
reviews = []reviewcore.PRReviewState{}
}
envelope.WriteJSON(w, status, TriggerReviewResponse{
ReviewerHandleID: res.ReviewerHandleID,
Reviews: reviews,
})
}
func (c *ReviewsController) submit(w http.ResponseWriter, r *http.Request) {
@ -94,12 +119,34 @@ func (c *ReviewsController) submit(w http.ResponseWriter, r *http.Request) {
envelope.WriteAPIError(w, r, http.StatusBadRequest, "bad_request", "INVALID_BODY", "Invalid request body", nil)
return
}
run, err := c.Svc.Submit(r.Context(), sessionID(r), in.RunID, domain.ReviewVerdict(in.Verdict), in.Body, in.GithubReviewID)
reviews := make([]reviewsvc.SubmittedReview, 0, len(in.Reviews))
if len(in.Reviews) > 0 {
for _, item := range in.Reviews {
reviews = append(reviews, reviewsvc.SubmittedReview{
RunID: item.RunID,
Verdict: domain.ReviewVerdict(item.Verdict),
Body: item.Body,
GithubReviewID: item.GithubReviewID,
})
}
} else {
reviews = append(reviews, reviewsvc.SubmittedReview{
RunID: in.RunID,
Verdict: domain.ReviewVerdict(in.Verdict),
Body: in.Body,
GithubReviewID: in.GithubReviewID,
})
}
runs, err := c.Svc.SubmitMany(r.Context(), sessionID(r), reviews)
if err != nil {
writeReviewError(w, r, err)
return
}
envelope.WriteJSON(w, http.StatusOK, ReviewRunResponse{Review: run})
first := domain.ReviewRun{}
if len(runs) > 0 {
first = runs[0]
}
envelope.WriteJSON(w, http.StatusOK, ReviewRunResponse{Review: first, Reviews: runs})
}
func writeReviewError(w http.ResponseWriter, r *http.Request, err error) {

View File

@ -20,12 +20,18 @@ import (
type fakeReviewService struct {
triggerErr error
trigger reviewcore.TriggerResult
list reviewcore.SessionReviews
submitted []reviewsvc.SubmittedReview
}
func (f *fakeReviewService) Trigger(context.Context, domain.SessionID) (reviewcore.TriggerResult, error) {
if f.triggerErr != nil {
return reviewcore.TriggerResult{}, f.triggerErr
}
if f.trigger.ReviewerHandleID != "" || f.trigger.Run.ID != "" || f.trigger.Reviews != nil || f.trigger.CreatedRuns != nil {
return f.trigger, nil
}
return reviewcore.TriggerResult{Run: domain.ReviewRun{ID: "run-1"}, Created: true}, nil
}
@ -33,8 +39,17 @@ func (f *fakeReviewService) Submit(context.Context, domain.SessionID, string, do
return domain.ReviewRun{}, nil
}
func (f *fakeReviewService) SubmitMany(_ context.Context, _ domain.SessionID, reviews []reviewsvc.SubmittedReview) ([]domain.ReviewRun, error) {
f.submitted = append([]reviewsvc.SubmittedReview(nil), reviews...)
runs := make([]domain.ReviewRun, 0, len(reviews))
for _, review := range reviews {
runs = append(runs, domain.ReviewRun{ID: review.RunID, Verdict: review.Verdict, Body: review.Body, GithubReviewID: review.GithubReviewID})
}
return runs, nil
}
func (f *fakeReviewService) List(context.Context, domain.SessionID) (reviewcore.SessionReviews, error) {
return reviewcore.SessionReviews{}, nil
return f.list, nil
}
func newReviewTestServer(t *testing.T, svc reviewsvc.Manager) *httptest.Server {
@ -59,3 +74,73 @@ func TestReviewsTrigger_MissingReviewerBinaryReturns422WithCause(t *testing.T) {
t.Fatalf("message = %q, want reviewer binary cause", got.Message)
}
}
func TestReviewsListIncludesReviewStates(t *testing.T) {
srv := newReviewTestServer(t, &fakeReviewService{list: reviewcore.SessionReviews{
ReviewerHandleID: "review-mer-1",
Runs: []domain.ReviewRun{{ID: "run-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1"}},
Reviews: []reviewcore.PRReviewState{{PRURL: "https://github.com/o/r/pull/1", PRNumber: 1, TargetSHA: "sha1", Status: reviewcore.ReviewStateUpToDate}},
}})
body, status, headers := doRequest(t, srv, "GET", "/api/v1/sessions/mer-1/reviews", "")
assertJSON(t, headers)
if status != http.StatusOK {
t.Fatalf("status = %d body=%s", status, body)
}
if !strings.Contains(string(body), `"reviews"`) || !strings.Contains(string(body), `"up_to_date"`) || !strings.Contains(string(body), `"reviewerHandleId":"review-mer-1"`) {
t.Fatalf("body missing review states/handle: %s", body)
}
if strings.Contains(string(body), `"items"`) || strings.Contains(string(body), `"reviewItems"`) || strings.Contains(string(body), `"reviewRuns"`) {
t.Fatalf("body contains deprecated review item aliases: %s", body)
}
}
func TestReviewsTriggerIncludesBatchFields(t *testing.T) {
run1 := domain.ReviewRun{ID: "run-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1"}
run2 := domain.ReviewRun{ID: "run-2", PRURL: "https://github.com/o/r/pull/2", TargetSHA: "sha2"}
srv := newReviewTestServer(t, &fakeReviewService{trigger: reviewcore.TriggerResult{
Run: run1,
ReviewerHandleID: "review-mer-1",
Created: true,
CreatedRuns: []domain.ReviewRun{run1, run2},
Reviews: []reviewcore.PRReviewState{
{PRURL: run1.PRURL, PRNumber: 1, TargetSHA: run1.TargetSHA, Status: reviewcore.ReviewStateRunning, LatestRun: &run1},
{PRURL: run2.PRURL, PRNumber: 2, TargetSHA: run2.TargetSHA, Status: reviewcore.ReviewStateRunning, LatestRun: &run2},
},
}})
body, status, headers := doRequest(t, srv, "POST", "/api/v1/sessions/mer-1/reviews/trigger", "")
assertJSON(t, headers)
if status != http.StatusCreated {
t.Fatalf("status = %d body=%s", status, body)
}
for _, want := range []string{`"reviews"`, `"running"`, `"run-1"`, `"run-2"`, `"reviewerHandleId":"review-mer-1"`} {
if !strings.Contains(string(body), want) {
t.Fatalf("body missing %s: %s", want, body)
}
}
for _, unwanted := range []string{`"reviewItems"`, `"items"`, `"createdReviews"`, `"createdRuns"`, `"reviewRuns"`, `"review":`} {
if strings.Contains(string(body), unwanted) {
t.Fatalf("body contains deprecated field %s: %s", unwanted, body)
}
}
}
func TestReviewsSubmitAcceptsBatchedReviews(t *testing.T) {
svc := &fakeReviewService{}
srv := newReviewTestServer(t, svc)
body, status, headers := doRequest(t, srv, "POST", "/api/v1/sessions/mer-1/reviews/submit", `{"reviews":[{"runId":"run-1","verdict":"changes_requested","body":"fix auth","githubReviewId":"101"},{"runId":"run-2","verdict":"approved"}]}`)
assertJSON(t, headers)
if status != http.StatusOK {
t.Fatalf("status = %d body=%s", status, body)
}
if len(svc.submitted) != 2 || svc.submitted[0].RunID != "run-1" || svc.submitted[1].Verdict != domain.VerdictApproved {
t.Fatalf("submitted = %+v", svc.submitted)
}
for _, want := range []string{`"reviews"`, `"run-1"`, `"run-2"`} {
if !strings.Contains(string(body), want) {
t.Fatalf("body missing %s: %s", want, body)
}
}
}

View File

@ -192,6 +192,7 @@ func (f *fakeSessionService) ListPRSummaries(_ context.Context, id domain.Sessio
UnresolvedBy: []sessionsvc.PRUnresolvedReviewer{{
ReviewerID: "reviewer-a",
Count: 1,
ReviewURL: "https://github.com/aoagents/agent-orchestrator/pull/142#pullrequestreview-1",
Links: []sessionsvc.PRReviewCommentLink{{URL: "https://github.com/aoagents/agent-orchestrator/pull/142#discussion_r1", File: "main.go", Line: 12}},
}},
},
@ -844,6 +845,7 @@ func TestSessionsAPI_PRRoutes(t *testing.T) {
UnresolvedBy []struct {
ReviewerID string `json:"reviewerId"`
Count int `json:"count"`
ReviewURL string `json:"reviewUrl"`
Links []struct {
URL string `json:"url"`
File string `json:"file"`
@ -869,7 +871,7 @@ func TestSessionsAPI_PRRoutes(t *testing.T) {
if checks := listed.PRs[0].CI.FailingChecks; len(checks) != 1 || checks[0].Name != "unit" || checks[0].LogTail != "" {
t.Fatalf("failing checks = %#v", checks)
}
if reviewers := listed.PRs[0].Review.UnresolvedBy; len(reviewers) != 1 || reviewers[0].ReviewerID != "reviewer-a" || reviewers[0].Links[0].Body != "" {
if reviewers := listed.PRs[0].Review.UnresolvedBy; len(reviewers) != 1 || reviewers[0].ReviewerID != "reviewer-a" || reviewers[0].ReviewURL == "" || reviewers[0].Links[0].Body != "" {
t.Fatalf("reviewers = %#v", reviewers)
}
if merge := listed.PRs[0].Mergeability; merge.State != "conflicting" || len(merge.ConflictFiles) != 0 || merge.PRURL == "" {

View File

@ -543,8 +543,10 @@ func TestApplyReviewResultSendsAndDedupsThroughPRSignature(t *testing.T) {
t.Fatalf("outcome/messages = %q/%v, want sent once", outcome, msg.msgs)
}
got := msg.msgs[0]
if !strings.Contains(got, "[AO reviewer]") || !strings.Contains(got, "fix the bug") || !strings.Contains(got, "98[2J765") {
t.Fatalf("AO review nudge missing label/body/review id: %q", got)
for _, want := range []string{"[AO reviewer]", "PR: " + result.PRURL, "Verdict: changes_requested", "Review body:\nfix the bug", "GitHub review: 98[2J765"} {
if !strings.Contains(got, want) {
t.Fatalf("AO review nudge missing %q: %q", want, got)
}
}
if strings.Contains(got, "\x1b") {
t.Fatalf("AO review nudge should sanitize control bytes: %q", got)
@ -572,6 +574,65 @@ func TestApplyReviewResultSendsAndDedupsThroughPRSignature(t *testing.T) {
}
}
func TestApplyReviewBatchSendsCombinedAndDedups(t *testing.T) {
st := newFakeStore()
st.sessions["mer-1"] = working("mer-1")
msg := &fakeMessenger{}
m := New(st, msg)
results := []ReviewResult{
{RunID: "run-2", BatchID: "batch-1", WorkerID: "mer-1", PRURL: "https://github.com/o/r/pull/2", TargetSHA: "sha2", Verdict: domain.VerdictChangesRequested, Body: "fix tests", GithubReviewID: "102"},
{RunID: "run-1", BatchID: "batch-1", WorkerID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1", Verdict: domain.VerdictChangesRequested, Body: "fix auth", GithubReviewID: "101"},
}
outcome, err := m.ApplyReviewBatch(ctx, "mer-1", "batch-1", results)
if err != nil {
t.Fatalf("ApplyReviewBatch: %v", err)
}
if outcome != ReviewDeliverySent || len(msg.msgs) != 1 {
t.Fatalf("outcome/messages = %q/%v, want sent once", outcome, msg.msgs)
}
got := msg.msgs[0]
for _, want := range []string{
"submitted 2 review(s) requesting changes",
"PR: https://github.com/o/r/pull/1",
"GitHub review: 101",
"Review body:\nfix auth",
"PR: https://github.com/o/r/pull/2",
"GitHub review: 102",
"Review body:\nfix tests",
} {
if !strings.Contains(got, want) {
t.Fatalf("batch nudge missing %q: %q", want, got)
}
}
if st.signatures["https://github.com/o/r/pull/1"] == "" {
t.Fatal("batch nudge did not persist signature on anchor PR")
}
outcome, err = m.ApplyReviewBatch(ctx, "mer-1", "batch-1", results)
if err != nil {
t.Fatalf("repeat ApplyReviewBatch: %v", err)
}
if outcome != ReviewDeliverySent || len(msg.msgs) != 1 {
t.Fatalf("repeat should suppress duplicate send, outcome=%q msgs=%v", outcome, msg.msgs)
}
}
func TestApplyReviewBatchNoopsWithoutDeliverableResults(t *testing.T) {
st := newFakeStore()
st.sessions["mer-1"] = working("mer-1")
msg := &fakeMessenger{}
m := New(st, msg)
outcome, err := m.ApplyReviewBatch(ctx, "mer-1", "batch-1", nil)
if err != nil {
t.Fatalf("ApplyReviewBatch: %v", err)
}
if outcome != ReviewDeliveryNoop || len(msg.msgs) != 0 || st.signatureWrites != 0 {
t.Fatalf("empty batch should no-op, outcome=%q msgs=%v signatureWrites=%d", outcome, msg.msgs, st.signatureWrites)
}
}
func TestApplyReviewResultNoopsWhenIrrelevant(t *testing.T) {
deliveredAt := time.Unix(100, 0).UTC()
tests := []struct {

View File

@ -5,6 +5,7 @@ import (
"encoding/json"
"fmt"
"log/slog"
"sort"
"strings"
"sync"
"time"
@ -33,6 +34,7 @@ const (
// review_run row.
type ReviewResult struct {
RunID string
BatchID string
WorkerID domain.SessionID
PRURL string
TargetSHA string
@ -42,6 +44,56 @@ type ReviewResult struct {
DeliveredAt *time.Time
}
// ApplyReviewBatch reacts to one reviewer CLI submission after the review
// service has decided which current-head changes-requested results are
// deliverable.
func (m *Manager) ApplyReviewBatch(ctx context.Context, workerID domain.SessionID, batchID string, results []ReviewResult) (ReviewDeliveryOutcome, error) {
if batchID == "" || len(results) == 0 {
return ReviewDeliveryNoop, nil
}
rec, ok, err := m.store.GetSession(ctx, workerID)
if err != nil || !ok {
return ReviewDeliveryNoop, err
}
if rec.IsTerminated || rec.Activity.State == domain.ActivityWaitingInput {
return ReviewDeliveryNoop, nil
}
if m.messenger == nil {
return ReviewDeliveryNoop, nil
}
sort.Slice(results, func(i, j int) bool {
if results[i].PRURL != results[j].PRURL {
return results[i].PRURL < results[j].PRURL
}
return results[i].RunID < results[j].RunID
})
var msg strings.Builder
fmt.Fprintf(&msg, "[AO reviewer] AO's internal code reviewer submitted %d review(s) requesting changes.\n", len(results))
var sigParts []string
for i, r := range results {
fmt.Fprintf(&msg, "\nReview %d\nPR: %s\nVerdict: %s", i+1, domain.SanitizeControlChars(r.PRURL), domain.SanitizeControlChars(string(r.Verdict)))
if r.TargetSHA != "" {
fmt.Fprintf(&msg, "\nHead commit: %s", domain.SanitizeControlChars(r.TargetSHA))
}
if r.GithubReviewID != "" {
safeReviewID := domain.SanitizeControlChars(r.GithubReviewID)
fmt.Fprintf(&msg, "\nGitHub review: %s", safeReviewID)
fmt.Fprintf(&msg, "\nOnce you have addressed it, reply on GitHub review %s with how you addressed it, then resolve the review comment threads you addressed.", safeReviewID)
}
if r.Body != "" {
fmt.Fprintf(&msg, "\n\nReview body:\n%s\n", domain.SanitizeControlChars(r.Body))
}
sigParts = append(sigParts, strings.Join([]string{r.RunID, r.PRURL, r.TargetSHA, r.GithubReviewID, r.Body}, "\x00"))
}
anchorPR := results[0].PRURL
key := "review-batch:" + anchorPR + ":" + batchID
sig := strings.Join(sigParts, "\x01")
if err := m.sendOnce(ctx, workerID, anchorPR, key, sig, msg.String(), reviewMaxNudge); err != nil {
return ReviewDeliveryNoop, err
}
return ReviewDeliverySent, nil
}
type reactionState struct {
mu sync.Mutex
seen map[string]string
@ -154,13 +206,14 @@ func (m *Manager) ApplyReviewResult(ctx context.Context, workerID domain.Session
if m.messenger == nil {
return ReviewDeliveryNoop, nil
}
msg := "[AO reviewer] AO's internal code reviewer requested changes on your PR. Review the feedback below and address it."
msg := fmt.Sprintf("[AO reviewer] AO's internal code reviewer submitted a review.\n\nPR: %s\nVerdict: %s", domain.SanitizeControlChars(r.PRURL), domain.SanitizeControlChars(string(r.Verdict)))
if r.GithubReviewID != "" {
safeReviewID := domain.SanitizeControlChars(r.GithubReviewID)
msg += fmt.Sprintf(" This feedback is GitHub review %s. Once you have addressed it, reply on that review referencing id %s with how you addressed it, then resolve the review comment threads you addressed.", safeReviewID, safeReviewID)
msg += fmt.Sprintf("\nGitHub review: %s", safeReviewID)
msg += fmt.Sprintf("\n\nOnce you have addressed it, reply on GitHub review %s with how you addressed it, then resolve the review comment threads you addressed.", safeReviewID)
}
if r.Body != "" {
msg += "\n\n" + domain.SanitizeControlChars(r.Body)
msg += "\n\nReview body:\n" + domain.SanitizeControlChars(r.Body)
}
key := "review:" + r.PRURL + ":ao:" + r.RunID
sig := strings.Join([]string{r.TargetSHA, r.RunID, r.GithubReviewID, r.Body}, "\x00")

View File

@ -53,7 +53,7 @@ type Store interface {
UpsertProject(ctx context.Context, row domain.ProjectRecord) error
ListPRsBySession(ctx context.Context, sessionID domain.SessionID) ([]domain.PullRequest, error)
ListChecks(ctx context.Context, prURL string) ([]domain.PullRequestCheck, error)
WriteSCMObservation(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode) error
WriteSCMObservation(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, reviews []domain.PullRequestReview, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode) error
}
// Lifecycle is the provider-neutral lifecycle notification sink.
@ -344,8 +344,8 @@ func (o *Observer) Poll(ctx context.Context) error {
prRefreshOK[key] = true
continue
}
finalPR, finalChecks, finalThreads, finalComments := domainFromObservation(subj.session.ID, prepared, local, opts, now)
pr, checks, threads, comments := finalPR, finalChecks, finalThreads, finalComments
finalPR, finalChecks, finalReviews, finalThreads, finalComments := domainFromObservation(subj.session.ID, prepared, local, opts, now)
pr, checks, reviews, threads, comments := finalPR, finalChecks, finalReviews, finalThreads, finalComments
// Lifecycle is allowed to run only after the observed facts are durable,
// but semantic hashes are the observer's acknowledgement cursor. Keep
// changed hashes at their local values until lifecycle succeeds; if the
@ -362,9 +362,9 @@ func (o *Observer) Poll(ctx context.Context) error {
if prepared.Changed.Review {
pendingOpts.preserveLocalReviewHash = true
}
pr, checks, threads, comments = domainFromObservation(subj.session.ID, prepared, local, pendingOpts, now)
pr, checks, reviews, threads, comments = domainFromObservation(subj.session.ID, prepared, local, pendingOpts, now)
}
if err := o.store.WriteSCMObservation(ctx, pr, checks, threads, comments, reviewMode); err != nil {
if err := o.store.WriteSCMObservation(ctx, pr, checks, reviews, threads, comments, reviewMode); err != nil {
o.logger.Error("scm observer: DB write failed", "session", subj.session.ID, "pr", pr.URL, "err", err)
markRepoRefreshFailed(subj.repo)
continue
@ -375,7 +375,7 @@ func (o *Observer) Poll(ctx context.Context) error {
markRepoRefreshFailed(subj.repo)
continue
}
if err := o.store.WriteSCMObservation(ctx, finalPR, finalChecks, nil, nil, ports.ReviewWritePreserve); err != nil {
if err := o.store.WriteSCMObservation(ctx, finalPR, finalChecks, nil, nil, nil, ports.ReviewWritePreserve); err != nil {
o.logger.Error("scm observer: DB lifecycle acknowledgement failed", "session", subj.session.ID, "pr", finalPR.URL, "err", err)
markRepoRefreshFailed(subj.repo)
continue
@ -586,7 +586,7 @@ func (o *Observer) discoverNewPRs(ctx context.Context, sessionRepos []sessionRep
// reads every PR of the session from the store. Without this write, an
// open sibling/child discovered in the same poll would not yet be
// durable, and the session could terminate while that PR is still open.
if err := o.store.WriteSCMObservation(ctx, known, nil, nil, nil, ports.ReviewWritePreserve); err != nil {
if err := o.store.WriteSCMObservation(ctx, known, nil, nil, nil, nil, ports.ReviewWritePreserve); err != nil {
o.logger.Error("scm observer: persist discovered PR failed", "session", sr.session.ID, "pr", known.URL, "err", err)
if markRepoFailed != nil {
markRepoFailed(repo)
@ -793,6 +793,7 @@ func (o *Observer) refreshReviews(ctx context.Context, subjects map[string]*subj
if review.Decision != "" {
obs.Review.Decision = review.Decision
}
obs.Review.Reviews = review.Reviews
obs.Review.Threads = review.Threads
obs.Review.Partial = review.Partial
obs.ObservedAt = now
@ -850,7 +851,7 @@ func (o *Observer) prepareForPersistence(obs ports.SCMObservation, local domain.
return obs
}
func domainFromObservation(sessionID domain.SessionID, obs ports.SCMObservation, local domain.PullRequest, opts persistenceOptions, now time.Time) (domain.PullRequest, []domain.PullRequestCheck, []domain.PullRequestReviewThread, []domain.PullRequestComment) {
func domainFromObservation(sessionID domain.SessionID, obs ports.SCMObservation, local domain.PullRequest, opts persistenceOptions, now time.Time) (domain.PullRequest, []domain.PullRequestCheck, []domain.PullRequestReview, []domain.PullRequestReviewThread, []domain.PullRequestComment) {
metadataHash := metadataSemanticHash(obs)
if opts.preserveLocalMetadataHash {
metadataHash = local.MetadataHash
@ -924,6 +925,17 @@ func domainFromObservation(sessionID domain.SessionID, obs ports.SCMObservation,
for _, ch := range obs.CI.Checks {
checks = append(checks, domain.PullRequestCheck{Name: ch.Name, CommitHash: obs.CI.HeadSHA, Status: domain.PRCheckStatus(ch.Status), Conclusion: ch.Conclusion, URL: ch.URL, Details: ch.ProviderID, LogTail: ch.LogTail, CreatedAt: now})
}
reviews := make([]domain.PullRequestReview, 0, len(obs.Review.Reviews))
for _, review := range obs.Review.Reviews {
reviews = append(reviews, domain.PullRequestReview{
ID: review.ID,
Author: review.Author,
State: domain.ReviewDecision(firstNonEmpty(review.State, string(domain.ReviewNone))),
URL: review.URL,
IsBot: review.IsBot,
SubmittedAt: firstTime(review.SubmittedAt, now),
})
}
threads := make([]domain.PullRequestReviewThread, 0, len(obs.Review.Threads))
commentCount := 0
for _, th := range obs.Review.Threads {
@ -936,7 +948,7 @@ func domainFromObservation(sessionID domain.SessionID, obs ports.SCMObservation,
comments = append(comments, domain.PullRequestComment{ThreadID: th.ID, ID: c.ID, Author: c.Author, File: th.Path, Line: th.Line, Body: c.Body, URL: c.URL, Resolved: th.Resolved, IsBot: c.IsBot || th.IsBot, CreatedAt: now})
}
}
return pr, checks, threads, comments
return pr, checks, reviews, threads, comments
}
func observationFromLocal(repo ports.SCMRepo, pr domain.PullRequest, checks []domain.PullRequestCheck) ports.SCMObservation {
@ -1119,10 +1131,11 @@ func ciSemanticHash(ci ports.SCMCIObservation) string {
func reviewSemanticHash(review ports.SCMReviewObservation) string {
type reviewHashPayload struct {
Decision string
Reviews []ports.SCMReviewSummaryObservation
Threads []ports.SCMReviewThreadObservation
Partial bool `json:",omitempty"`
}
return stableHash(reviewHashPayload{Decision: review.Decision, Threads: review.Threads, Partial: review.Partial})
return stableHash(reviewHashPayload{Decision: review.Decision, Reviews: review.Reviews, Threads: review.Threads, Partial: review.Partial})
}
func threadSemanticHash(th ports.SCMReviewThreadObservation) string {

View File

@ -41,6 +41,7 @@ type fakeStore struct {
type fakeWrite struct {
pr domain.PullRequest
checks []domain.PullRequestCheck
reviews []domain.PullRequestReview
comments []domain.PullRequestComment
reviewMode ports.ReviewWriteMode
}
@ -94,13 +95,13 @@ func (s *fakeStore) ListChecks(_ context.Context, prURL string) ([]domain.PullRe
return append([]domain.PullRequestCheck(nil), s.checks[prURL]...), nil
}
func (s *fakeStore) WriteSCMObservation(_ context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode) error {
func (s *fakeStore) WriteSCMObservation(_ context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, reviews []domain.PullRequestReview, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode) error {
s.mu.Lock()
defer s.mu.Unlock()
if s.writeErr != nil {
return s.writeErr
}
s.writes = append(s.writes, fakeWrite{pr: pr, checks: append([]domain.PullRequestCheck(nil), checks...), comments: append([]domain.PullRequestComment(nil), comments...), reviewMode: reviewMode})
s.writes = append(s.writes, fakeWrite{pr: pr, checks: append([]domain.PullRequestCheck(nil), checks...), reviews: append([]domain.PullRequestReview(nil), reviews...), comments: append([]domain.PullRequestComment(nil), comments...), reviewMode: reviewMode})
return nil
}
@ -229,7 +230,7 @@ func testObs(num int) ports.SCMObservation {
func knownPR(num int) domain.PullRequest {
obs := testObs(num)
pr, _, _, _ := domainFromObservation("p-1", obs, domain.PullRequest{}, persistenceOptions{}, time.Unix(1, 0).UTC())
pr, _, _, _, _ := domainFromObservation("p-1", obs, domain.PullRequest{}, persistenceOptions{}, time.Unix(1, 0).UTC())
return pr
}
@ -731,7 +732,11 @@ func TestPoll_ReviewHashDrivesPersistenceAndLifecycle(t *testing.T) {
local.ReviewHash = "old"
local.Review = domain.ReviewChangesRequest
store.prs["p-1"] = []domain.PullRequest{local}
review := ports.SCMReviewObservation{Decision: string(domain.ReviewChangesRequest), Threads: []ports.SCMReviewThreadObservation{{ID: "t1", Path: "f.go", Line: 2, Comments: []ports.SCMReviewCommentObservation{{ID: "c1", Author: "ann", Body: "fix this"}}}}}
review := ports.SCMReviewObservation{
Decision: string(domain.ReviewChangesRequest),
Reviews: []ports.SCMReviewSummaryObservation{{ID: "review-1", Author: "ann", State: string(domain.ReviewChangesRequest), URL: "https://github.com/o/r/pull/1#pullrequestreview-1", SubmittedAt: time.Unix(199, 0).UTC()}},
Threads: []ports.SCMReviewThreadObservation{{ID: "t1", Path: "f.go", Line: 2, Comments: []ports.SCMReviewCommentObservation{{ID: "c1", Author: "ann", Body: "fix this"}}}},
}
provider := &fakeProvider{repoGuards: map[string]ports.SCMGuardResult{prKey(testRepo, 0): {ETag: "repo", NotModified: true}}, observations: map[string]ports.SCMObservation{}, reviews: map[string]ports.SCMReviewObservation{prKey(testRepo, 1): review}}
lc := &fakeLifecycle{}
obs := newTestObserver(store, provider, lc, time.Unix(200, 0).UTC())
@ -742,6 +747,9 @@ func TestPoll_ReviewHashDrivesPersistenceAndLifecycle(t *testing.T) {
if len(store.writes) == 0 || len(store.writes[0].comments) != 1 {
t.Fatalf("review change not persisted: %#v", store.writes)
}
if len(store.writes[0].reviews) != 1 || store.writes[0].reviews[0].URL != "https://github.com/o/r/pull/1#pullrequestreview-1" {
t.Fatalf("review summaries not persisted: %#v", store.writes[0].reviews)
}
if len(store.writes) != 2 {
t.Fatalf("review change with lifecycle should write held-back facts then acknowledgement, got %d writes", len(store.writes))
}

View File

@ -100,11 +100,19 @@ const (
// LaunchConfig carries inputs needed to build a new agent launch command.
type LaunchConfig struct {
Config AgentConfig
IssueID string
Permissions PermissionMode
Prompt string
SessionID string
Config AgentConfig
IssueID string
Permissions PermissionMode
Prompt string
SessionID string
// AllowedTools and DisallowedTools scope the agent to a tool allowlist when
// it runs in a non-bypass permission mode (allow rules auto-approve, deny
// rules auto-reject). They are the enforced read-only guarantee the reviewer
// relies on: bypassPermissions ignores both lists, so a restricted launch
// must leave Permissions off bypass. Empty means no restriction, so worker
// sessions are unaffected.
AllowedTools []string
DisallowedTools []string
SystemPrompt string
SystemPromptFile string
WorkspacePath string

View File

@ -38,13 +38,13 @@ const (
// whether review facts are preserved, replaced with a complete snapshot, or
// merged as a bounded partial window.
type SCMWriter interface {
WriteSCMObservation(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ReviewWriteMode) error
WriteSCMObservation(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, reviews []domain.PullRequestReview, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ReviewWriteMode) error
}
// PRClaimer atomically moves (or creates) a PR row for a target session and
// persists the live SCM facts observed for that PR in the same transaction.
type PRClaimer interface {
ClaimPR(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ReviewWriteMode, allowActiveTakeover bool) (ClaimOutcome, error)
ClaimPR(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, reviews []domain.PullRequestReview, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ReviewWriteMode, allowActiveTakeover bool) (ClaimOutcome, error)
}
// ErrPRClaimedByActiveSession is returned by PRClaimer.ClaimPR when takeover is

View File

@ -37,6 +37,11 @@ type ReviewInvocation struct {
PRURL string
// TargetSHA is the PR head commit under review.
TargetSHA string
// ReviewQueue lists all review tasks created by the same trigger so a shared
// reviewer pane can review multiple PRs and submit the results together.
ReviewQueue []ReviewTask
// ReviewIndex is this invocation's zero-based position in ReviewQueue.
ReviewIndex int
// WorkspacePath is the worker's checkout the reviewer reads.
WorkspacePath string
// Prompt and SystemPrompt are the review instructions AO authored centrally,
@ -48,6 +53,13 @@ type ReviewInvocation struct {
SystemPrompt string
}
// ReviewTask is one PR/run in a multi-PR review trigger queue.
type ReviewTask struct {
RunID string
PRURL string
TargetSHA string
}
// ReviewCommandSpec is how to launch a reviewer: the argv and any extra env the
// adapter needs. AO supplies the workspace and review-tracking env around it.
type ReviewCommandSpec struct {

View File

@ -184,6 +184,8 @@ type SCMCheckObservation struct {
type SCMReviewObservation struct {
// Decision is AO's normalized review decision.
Decision string
// Reviews contains submitted review summaries fetched on the slower review cadence.
Reviews []SCMReviewSummaryObservation
// Threads contains normalized review threads fetched on the slower review cadence.
Threads []SCMReviewThreadObservation
// Partial is true when the provider intentionally fetched and persisted a
@ -192,6 +194,22 @@ type SCMReviewObservation struct {
Partial bool
}
// SCMReviewSummaryObservation is one submitted review with its provider summary URL.
type SCMReviewSummaryObservation struct {
// ID is the provider's stable submitted-review identifier.
ID string
// Author is the provider login/name of the reviewer.
Author string
// State is AO's normalized review decision for this review.
State string
// URL is a provider link to the submitted review summary.
URL string
// IsBot is true when the provider identifies the reviewer as a bot.
IsBot bool
// SubmittedAt is the provider's review submission timestamp.
SubmittedAt time.Time
}
// SCMReviewThreadObservation is a normalized review thread with comments.
type SCMReviewThreadObservation struct {
// ID is the provider's stable review thread identifier.

View File

@ -31,6 +31,8 @@ type LaunchSpec struct {
WorkspacePath string
PRURL string
TargetSHA string
ReviewQueue []ports.ReviewTask
ReviewIndex int
}
// reviewerRuntime is the runtime surface the launcher needs: create a pane,
@ -73,6 +75,8 @@ func (l *agentLauncher) invocation(spec LaunchSpec) ports.ReviewInvocation {
WorkerSessionID: spec.WorkerID,
PRURL: spec.PRURL,
TargetSHA: spec.TargetSHA,
ReviewQueue: spec.ReviewQueue,
ReviewIndex: spec.ReviewIndex,
WorkspacePath: spec.WorkspacePath,
Prompt: prompt,
SystemPrompt: systemPrompt,

View File

@ -0,0 +1,95 @@
package review
import (
"sort"
"github.com/aoagents/agent-orchestrator/backend/internal/domain"
)
// StateStatus is the per-PR review planning state.
type StateStatus string
const (
// ReviewStateNeedsReview means an eligible PR has no current AO approval or running pass.
ReviewStateNeedsReview StateStatus = "needs_review"
// ReviewStateRunning means a review run is already active for the PR's current head.
ReviewStateRunning StateStatus = "running"
// ReviewStateUpToDate means AO approved the PR's current head.
ReviewStateUpToDate StateStatus = "up_to_date"
// ReviewStateChangesRequested means AO requested changes on the PR's current head.
ReviewStateChangesRequested StateStatus = "changes_requested"
// ReviewStateIneligible means the PR is draft, closed, merged, or missing required facts.
ReviewStateIneligible StateStatus = "ineligible"
)
// PRReviewState is one PR-scoped review decision for a worker session.
type PRReviewState struct {
PRURL string `json:"prUrl"`
PRNumber int `json:"prNumber"`
Title string `json:"title"`
TargetSHA string `json:"targetSha"`
Status StateStatus `json:"status" enum:"needs_review,running,up_to_date,changes_requested,ineligible"`
LatestRun *domain.ReviewRun `json:"latestRun,omitempty"`
}
// Plan computes per-PR review work from the currently observed PRs and existing
// review runs. It is pure so the trigger path and API list path share exactly
// the same eligibility/status rules.
func Plan(prs []domain.PullRequest, runs []domain.ReviewRun) []PRReviewState {
latest := latestRunsByPRAndSHA(runs)
reviews := make([]PRReviewState, 0, len(prs))
for _, pr := range prs {
review := PRReviewState{
PRURL: pr.URL,
PRNumber: pr.Number,
Title: pr.Title,
TargetSHA: pr.HeadSHA,
Status: ReviewStateNeedsReview,
}
if pr.URL == "" || pr.HeadSHA == "" || pr.Draft || pr.Merged || pr.Closed {
review.Status = ReviewStateIneligible
if run, ok := latest[review.PRURL+"\x00"+review.TargetSHA]; ok {
review.LatestRun = &run
}
reviews = append(reviews, review)
continue
}
if run, ok := latest[review.PRURL+"\x00"+review.TargetSHA]; ok {
review.LatestRun = &run
switch {
case run.Status == domain.ReviewRunRunning:
review.Status = ReviewStateRunning
case run.Verdict == domain.VerdictApproved:
review.Status = ReviewStateUpToDate
case run.Verdict == domain.VerdictChangesRequested:
review.Status = ReviewStateChangesRequested
case run.Status == domain.ReviewRunFailed:
review.Status = ReviewStateNeedsReview
default:
review.Status = ReviewStateNeedsReview
}
}
reviews = append(reviews, review)
}
sort.SliceStable(reviews, func(i, j int) bool {
if reviews[i].PRNumber != reviews[j].PRNumber {
return reviews[i].PRNumber < reviews[j].PRNumber
}
return reviews[i].PRURL < reviews[j].PRURL
})
return reviews
}
func latestRunsByPRAndSHA(runs []domain.ReviewRun) map[string]domain.ReviewRun {
latest := make(map[string]domain.ReviewRun)
for _, run := range runs {
if run.PRURL == "" || run.TargetSHA == "" {
continue
}
key := run.PRURL + "\x00" + run.TargetSHA
if existing, ok := latest[key]; !ok || run.CreatedAt.After(existing.CreatedAt) {
latest[key] = run
}
}
return latest
}

View File

@ -0,0 +1,68 @@
package review
import (
"testing"
"time"
"github.com/aoagents/agent-orchestrator/backend/internal/domain"
)
func TestPlanStatuses(t *testing.T) {
now := time.Unix(100, 0).UTC()
tests := []struct {
name string
pr domain.PullRequest
runs []domain.ReviewRun
want StateStatus
}{
{name: "open needs review", pr: planPR("pr1", 1, "sha1"), want: ReviewStateNeedsReview},
{name: "draft ineligible", pr: withDraft(planPR("pr1", 1, "sha1")), want: ReviewStateIneligible},
{name: "merged ineligible", pr: withMerged(planPR("pr1", 1, "sha1")), want: ReviewStateIneligible},
{name: "closed ineligible", pr: withClosed(planPR("pr1", 1, "sha1")), want: ReviewStateIneligible},
{name: "approved current sha up to date", pr: planPR("pr1", 1, "sha1"), runs: []domain.ReviewRun{
{ID: "run-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunComplete, Verdict: domain.VerdictApproved, CreatedAt: now},
}, want: ReviewStateUpToDate},
{name: "changes requested current sha", pr: planPR("pr1", 1, "sha1"), runs: []domain.ReviewRun{
{ID: "run-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunComplete, Verdict: domain.VerdictChangesRequested, CreatedAt: now},
}, want: ReviewStateChangesRequested},
{name: "running current sha", pr: planPR("pr1", 1, "sha1"), runs: []domain.ReviewRun{
{ID: "run-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunRunning, CreatedAt: now},
}, want: ReviewStateRunning},
{name: "different sha needs review", pr: planPR("pr1", 1, "sha2"), runs: []domain.ReviewRun{
{ID: "run-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunComplete, Verdict: domain.VerdictApproved, CreatedAt: now},
}, want: ReviewStateNeedsReview},
{name: "failed current sha retryable", pr: planPR("pr1", 1, "sha1"), runs: []domain.ReviewRun{
{ID: "run-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunFailed, CreatedAt: now},
}, want: ReviewStateNeedsReview},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got := Plan([]domain.PullRequest{tt.pr}, tt.runs)
if len(got) != 1 {
t.Fatalf("review states = %d, want 1", len(got))
}
if got[0].Status != tt.want {
t.Fatalf("status = %s, want %s; item=%+v", got[0].Status, tt.want, got[0])
}
})
}
}
func planPR(url string, n int, sha string) domain.PullRequest {
return domain.PullRequest{URL: url, Number: n, HeadSHA: sha}
}
func withDraft(pr domain.PullRequest) domain.PullRequest {
pr.Draft = true
return pr
}
func withMerged(pr domain.PullRequest) domain.PullRequest {
pr.Merged = true
return pr
}
func withClosed(pr domain.PullRequest) domain.PullRequest {
pr.Closed = true
return pr
}

View File

@ -1,6 +1,9 @@
package review
import "fmt"
import (
"fmt"
"strings"
)
// reviewTexts returns the user-facing prompt and the system prompt to deliver to
// a reviewer, authored in one place — the reviewer analogue of
@ -14,14 +17,18 @@ import "fmt"
func reviewTexts(spec LaunchSpec) (prompt, systemPrompt string) {
systemPrompt = `## Code reviewer role
You are an AO code reviewer. You review a single pull request's changes in the current checkout do not start unrelated work. Inspect what the PR changed by diffing the checkout against the PR's base branch, and review for correctness bugs, missing error handling, security issues, test coverage, and clear deviations from the surrounding code's conventions. Prefer a few high-confidence findings over nitpicks.
You are an AO code reviewer. You review the requested pull request changes in the current checkout do not start unrelated work. Inspect what each PR changed by diffing the checkout against the PR's base branch, and review for correctness bugs, missing error handling, security issues, test coverage, and clear deviations from the surrounding code's conventions. Prefer a few high-confidence findings over nitpicks.
Post your review as a comment on the pull request, stating clearly whether it needs changes or is ready, with inline comments for specific findings. Do not push commits, edit files, or modify the branch review only.`
prompt = fmt.Sprintf(`Review pull request %s (head commit %s).
queueText := reviewQueueText(spec)
prompt = fmt.Sprintf(`Review the requested pull request(s) for worker session %s.
%s
Complete every review task in the queue autonomously. Do not ask the user whether to continue to the next PR, and do not stop after the first PR unless the provider or checkout is genuinely unusable for every queued task.
Do these steps in order:
1. Post your review on the pull request and capture its id in one call. Post with `+"`gh api`"+` rather than `+"`gh pr review`"+`: it is the only way to attach inline comments, and its response carries the created review's id, so AO can tell the worker exactly which review to address. Send the review as a JSON body so the inline comments form a proper array of objects:
1. For each PR below, post a separate review on that pull request and capture its id in one call. Post with `+"`gh api`"+` rather than `+"`gh pr review`"+`: it is the only way to attach inline comments, and its response carries the created review's id, so AO can tell the worker exactly which review to address. Send the review as a JSON body so the inline comments form a proper array of objects:
gh api --method POST repos/{owner}/{repo}/pulls/{number}/reviews --input - --jq '.id' <<'JSON'
{ "event": "COMMENT", "body": "<summary>",
@ -31,13 +38,29 @@ Do these steps in order:
- Substitute the PR's owner/repo/number. Add one object to "comments" per inline finding; omit the field for a review with no inline comments.
- Always use "event": "COMMENT": reviews are posted from the PR author's own account, and GitHub rejects both APPROVE and REQUEST_CHANGES on your own PR. State in the body whether you are requesting changes or approving; the machine-readable verdict goes to AO in step 2.
- The printed number is the review id. If the call fails on the provider, leave the id empty.
2. Record the result with AO, passing your full review on stdin with --body - so nothing is ever written into the worktree (a file there could be committed onto the worker's branch):
2. After every PR has its own GitHub review from step 1, record AO's bookkeeping for those already-posted reviews using one command. Pass JSON on stdin so nothing is ever written into the worktree (a file there could be committed onto the worker's branch). Include one object per PR/run from the queue:
ao review submit --session %s --run %s --verdict <approved|changes_requested> --review-id <id-from-step-1> --body - <<'MD'
<your full review markdown>
MD
ao review submit --session %s --reviews - <<'JSON'
{
"reviews": [
{ "runId": "<run-id>", "verdict": "<approved|changes_requested>", "githubReviewId": "<id-from-step-1-or-empty>", "body": "<your full review markdown>" }
]
}
JSON
Only if step 1 genuinely fails on the provider, still run step 2 (without --review-id) so the result is recorded.`,
spec.PRURL, spec.TargetSHA, spec.WorkerID, spec.RunID)
Only if step 1 genuinely fails on the provider for a PR, still include that run in step 2 with an empty githubReviewId so the result is recorded.`,
spec.WorkerID, queueText, spec.WorkerID)
return prompt, systemPrompt
}
func reviewQueueText(spec LaunchSpec) string {
if len(spec.ReviewQueue) <= 1 {
return fmt.Sprintf("\nReview task queue:\n* 1. %s (head commit %s, run %s)\n", spec.PRURL, spec.TargetSHA, spec.RunID)
}
var b strings.Builder
fmt.Fprintf(&b, "\nAO created %d review tasks for this worker session. Review every queued PR, then submit all results together.\n\nReview task queue:\n", len(spec.ReviewQueue))
for i, task := range spec.ReviewQueue {
fmt.Fprintf(&b, "* %d. %s (head commit %s, run %s)\n", i+1, task.PRURL, task.TargetSHA, task.RunID)
}
return b.String()
}

View File

@ -0,0 +1,37 @@
package review
import (
"strings"
"testing"
"github.com/aoagents/agent-orchestrator/backend/internal/ports"
)
func TestReviewTextsIncludesMultiPRQueue(t *testing.T) {
spec := launchSpec()
spec.RunID = "run-2"
spec.PRURL = "https://github.com/o/r/pull/2"
spec.TargetSHA = "sha2"
spec.ReviewIndex = 1
spec.ReviewQueue = []ports.ReviewTask{
{RunID: "run-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1"},
{RunID: "run-2", PRURL: "https://github.com/o/r/pull/2", TargetSHA: "sha2"},
}
prompt, _ := reviewTexts(spec)
for _, want := range []string{
"AO created 2 review tasks",
"Review every queued PR, then submit all results together",
"Complete every review task in the queue autonomously",
"Do not ask the user whether to continue to the next PR",
"* 1. https://github.com/o/r/pull/1 (head commit sha1, run run-1)",
"* 2. https://github.com/o/r/pull/2 (head commit sha2, run run-2)",
"After every PR has its own GitHub review from step 1",
"ao review submit --session mer-1 --reviews -",
`"reviews": [`,
} {
if !strings.Contains(prompt, want) {
t.Fatalf("prompt missing %q:\n%s", want, prompt)
}
}
}

View File

@ -18,6 +18,7 @@ import (
"github.com/google/uuid"
"github.com/aoagents/agent-orchestrator/backend/internal/domain"
"github.com/aoagents/agent-orchestrator/backend/internal/ports"
)
// ErrInvalid and ErrNotFound let the transport layer map failures to 422/404.
@ -34,9 +35,9 @@ type Store interface {
InsertReviewRun(ctx stdctx.Context, r domain.ReviewRun) error
UpdateReviewRunResult(ctx stdctx.Context, id string, status domain.ReviewRunStatus, verdict domain.ReviewVerdict, body, githubReviewID string) (bool, error)
SupersedeReviewRun(ctx stdctx.Context, id, body string) (bool, error)
SupersedeStaleRunningReviewRuns(ctx stdctx.Context, sessionID domain.SessionID, targetSHA, body string) (int64, error)
SupersedeStaleRunningReviewRuns(ctx stdctx.Context, sessionID domain.SessionID, prURL, targetSHA, body string) (int64, error)
GetReviewRun(ctx stdctx.Context, id string) (domain.ReviewRun, bool, error)
GetReviewRunBySessionAndSHA(ctx stdctx.Context, id domain.SessionID, targetSHA string) (domain.ReviewRun, bool, error)
GetReviewRunBySessionPRAndSHA(ctx stdctx.Context, id domain.SessionID, prURL, targetSHA string) (domain.ReviewRun, bool, error)
ListReviewRunsBySession(ctx stdctx.Context, id domain.SessionID) ([]domain.ReviewRun, error)
}
@ -134,6 +135,8 @@ type TriggerResult struct {
Run domain.ReviewRun
ReviewerHandleID string
Created bool
Reviews []PRReviewState
CreatedRuns []domain.ReviewRun
}
// SessionReviews is a worker's review state: the live reviewer handle plus its
@ -141,14 +144,12 @@ type TriggerResult struct {
type SessionReviews struct {
ReviewerHandleID string
Runs []domain.ReviewRun
Reviews []PRReviewState
}
// Trigger starts (or reuses) a review of a worker's PR at its current head:
// - if a non-failed run already exists for this commit, it is returned unchanged;
// - otherwise, if a live reviewer pane exists, it is messaged to review the
// new commit; if not, a fresh reviewer is spawned;
// - the run is recorded before launch so startup failures leave a visible
// failed pass instead of an empty gap.
// Trigger starts reviews for every PR on the worker session that needs review.
// It reuses running/up-to-date runs, retries failed/current changes-requested
// heads, and uses one reviewer pane for every new run in the batch.
func (e *Engine) Trigger(ctx stdctx.Context, workerID domain.SessionID) (TriggerResult, error) {
if workerID == "" {
return TriggerResult{}, fmt.Errorf("%w: worker session id is required", ErrInvalid)
@ -175,39 +176,21 @@ func (e *Engine) Trigger(ctx stdctx.Context, workerID domain.SessionID) (Trigger
return TriggerResult{}, fmt.Errorf("%w: worker session %q has no workspace to review", ErrInvalid, workerID)
}
pr, err := e.workerPR(ctx, workerID)
prs, err := e.prs.ListPRsBySession(ctx, workerID)
if err != nil {
return TriggerResult{}, err
}
targetSHA := pr.HeadSHA
review, hasReview, err := e.store.GetReviewBySession(ctx, workerID)
if len(prs) == 0 {
return TriggerResult{}, fmt.Errorf("%w: worker %q has no PR to review", ErrInvalid, workerID)
}
runs, err := e.store.ListReviewRunsBySession(ctx, workerID)
if err != nil {
return TriggerResult{}, err
}
reviews := Plan(prs, runs)
// Idempotency: a pass for this commit is reusable while it is still running
// or once it carries a verdict. The fallback branch below is defensive for
// any non-running, non-failed row that somehow lacks a verdict; normal
// Submit paths complete a run only with a valid verdict (#342).
if existing, ok, err := e.store.GetReviewRunBySessionAndSHA(ctx, workerID, targetSHA); err != nil {
return TriggerResult{}, err
} else if ok && (existing.Status == domain.ReviewRunRunning || existing.Verdict != domain.VerdictNone) {
return TriggerResult{Run: existing, ReviewerHandleID: review.ReviewerHandleID, Created: false}, nil
} else if ok && existing.Status != domain.ReviewRunFailed {
superseded, err := e.store.SupersedeReviewRun(ctx, existing.ID, "superseded by a new review trigger")
if err != nil {
return TriggerResult{}, err
}
if !superseded {
if latest, ok, err := e.store.GetReviewRun(ctx, existing.ID); err != nil {
return TriggerResult{}, err
} else if ok {
return TriggerResult{Run: latest, ReviewerHandleID: review.ReviewerHandleID, Created: false}, nil
}
}
}
if _, err := e.store.SupersedeStaleRunningReviewRuns(ctx, workerID, targetSHA, "superseded by a review trigger for a newer commit"); err != nil {
reviewRow, hasReview, err := e.store.GetReviewBySession(ctx, workerID)
if err != nil {
return TriggerResult{}, err
}
@ -217,78 +200,156 @@ func (e *Engine) Trigger(ctx stdctx.Context, workerID domain.SessionID) (Trigger
}
now := e.clock()
runID := e.newID()
spec := LaunchSpec{
RunID: runID,
WorkerID: workerID,
Harness: harness,
WorkspacePath: worker.Metadata.WorkspacePath,
PRURL: pr.URL,
TargetSHA: targetSHA,
}
review, err = e.upsertReview(ctx, worker, harness, pr.URL, review.ReviewerHandleID, now)
reviewRow, err = e.upsertReview(ctx, worker, harness, reviewRow.ReviewerHandleID, now)
if err != nil {
return TriggerResult{}, err
}
run := domain.ReviewRun{
ID: runID,
ReviewID: review.ID,
SessionID: workerID,
Harness: harness,
PRURL: pr.URL,
TargetSHA: targetSHA,
Status: domain.ReviewRunRunning,
Verdict: domain.VerdictNone,
CreatedAt: now,
}
if err := e.store.InsertReviewRun(ctx, run); err != nil {
if errors.Is(err, domain.ErrDuplicateReviewRun) {
if existing, ok, getErr := e.store.GetReviewRunBySessionAndSHA(ctx, workerID, targetSHA); getErr != nil {
return TriggerResult{}, getErr
} else if ok {
return TriggerResult{Run: existing, ReviewerHandleID: review.ReviewerHandleID, Created: false}, nil
var created []domain.ReviewRun
batchID := ""
for _, reviewState := range reviews {
if reviewState.Status != ReviewStateNeedsReview && reviewState.Status != ReviewStateChangesRequested {
continue
}
if reviewState.LatestRun != nil && reviewState.LatestRun.Status != domain.ReviewRunFailed && reviewState.LatestRun.Status != domain.ReviewRunRunning && reviewState.LatestRun.Verdict == domain.VerdictNone {
superseded, err := e.store.SupersedeReviewRun(ctx, reviewState.LatestRun.ID, "superseded by a new review trigger")
if err != nil {
return TriggerResult{}, err
}
if !superseded {
if latest, ok, err := e.store.GetReviewRun(ctx, reviewState.LatestRun.ID); err != nil {
return TriggerResult{}, err
} else if ok {
reviews = replaceReviewLatestRun(reviews, reviewState.PRURL, reviewState.TargetSHA, latest)
continue
}
}
}
return TriggerResult{}, err
if _, err := e.store.SupersedeStaleRunningReviewRuns(ctx, workerID, reviewState.PRURL, reviewState.TargetSHA, "superseded by a review trigger for a newer commit"); err != nil {
return TriggerResult{}, err
}
if batchID == "" {
batchID = e.newID()
}
run := domain.ReviewRun{
ID: e.newID(),
ReviewID: reviewRow.ID,
SessionID: workerID,
BatchID: batchID,
Harness: harness,
PRURL: reviewState.PRURL,
TargetSHA: reviewState.TargetSHA,
Status: domain.ReviewRunRunning,
Verdict: domain.VerdictNone,
CreatedAt: now,
}
if err := e.store.InsertReviewRun(ctx, run); err != nil {
if errors.Is(err, domain.ErrDuplicateReviewRun) {
if existing, ok, getErr := e.store.GetReviewRunBySessionPRAndSHA(ctx, workerID, reviewState.PRURL, reviewState.TargetSHA); getErr != nil {
return TriggerResult{}, getErr
} else if ok {
reviews = replaceReviewLatestRun(reviews, reviewState.PRURL, reviewState.TargetSHA, existing)
continue
}
}
return TriggerResult{}, err
}
created = append(created, run)
reviews = replaceReviewLatestRun(reviews, reviewState.PRURL, reviewState.TargetSHA, run)
}
if len(created) == 0 {
return TriggerResult{Run: firstReusableRun(reviews), ReviewerHandleID: reviewRow.ReviewerHandleID, Created: false, Reviews: reviews}, nil
}
failRun := func(err error) error {
if _, updateErr := e.store.UpdateReviewRunResult(ctx, run.ID, domain.ReviewRunFailed, domain.VerdictNone, err.Error(), ""); updateErr != nil {
return updateErr
failRuns := func(start int, err error) error {
for _, run := range created[start:] {
if _, updateErr := e.store.UpdateReviewRunResult(ctx, run.ID, domain.ReviewRunFailed, domain.VerdictNone, err.Error(), ""); updateErr != nil {
return updateErr
}
}
return err
}
// Reuse a live reviewer pane if there is one; otherwise spawn a fresh one.
handleID := ""
if hasReview && review.ReviewerHandleID != "" {
alive, err := e.launcher.Alive(ctx, review.ReviewerHandleID)
queue := reviewQueue(created)
if hasReview && reviewRow.ReviewerHandleID != "" {
alive, err := e.launcher.Alive(ctx, reviewRow.ReviewerHandleID)
if err != nil {
return TriggerResult{}, failRun(err)
return TriggerResult{}, failRuns(0, err)
}
if alive {
if err := e.launcher.Notify(ctx, review.ReviewerHandleID, spec); err != nil {
return TriggerResult{}, failRun(fmt.Errorf("notify reviewer: %w", err))
}
handleID = review.ReviewerHandleID
handleID = reviewRow.ReviewerHandleID
}
}
if handleID == "" {
h, err := e.launcher.Spawn(ctx, spec)
h, err := e.launcher.Spawn(ctx, reviewLaunchSpec(worker, harness, created[0], queue, 0))
if err != nil {
return TriggerResult{}, failRun(fmt.Errorf("launch reviewer: %w", err))
return TriggerResult{}, failRuns(0, fmt.Errorf("launch reviewer: %w", err))
}
handleID = h
} else {
if err := e.launcher.Notify(ctx, handleID, reviewLaunchSpec(worker, harness, created[0], queue, 0)); err != nil {
return TriggerResult{}, failRuns(0, fmt.Errorf("notify reviewer: %w", err))
}
}
// The reviewer is running; now record the pass.
review, err = e.upsertReview(ctx, worker, harness, pr.URL, handleID, now)
reviewRow, err = e.upsertReview(ctx, worker, harness, handleID, now)
if err != nil {
return TriggerResult{}, err
}
run.ReviewID = review.ID
return TriggerResult{Run: run, ReviewerHandleID: handleID, Created: true}, nil
for i := range created {
created[i].ReviewID = reviewRow.ID
}
return TriggerResult{Run: created[0], ReviewerHandleID: handleID, Created: true, Reviews: reviews, CreatedRuns: created}, nil
}
func reviewLaunchSpec(worker domain.SessionRecord, harness domain.ReviewerHarness, run domain.ReviewRun, queue []ports.ReviewTask, index int) LaunchSpec {
return LaunchSpec{
RunID: run.ID,
WorkerID: worker.ID,
Harness: harness,
WorkspacePath: worker.Metadata.WorkspacePath,
PRURL: run.PRURL,
TargetSHA: run.TargetSHA,
ReviewQueue: queue,
ReviewIndex: index,
}
}
func reviewQueue(runs []domain.ReviewRun) []ports.ReviewTask {
queue := make([]ports.ReviewTask, 0, len(runs))
for _, run := range runs {
queue = append(queue, ports.ReviewTask{
RunID: run.ID,
PRURL: run.PRURL,
TargetSHA: run.TargetSHA,
})
}
return queue
}
func replaceReviewLatestRun(reviews []PRReviewState, prURL, targetSHA string, run domain.ReviewRun) []PRReviewState {
for i := range reviews {
if reviews[i].PRURL == prURL && reviews[i].TargetSHA == targetSHA {
reviews[i].LatestRun = &run
if run.Status == domain.ReviewRunRunning {
reviews[i].Status = ReviewStateRunning
}
break
}
}
return reviews
}
func firstReusableRun(reviews []PRReviewState) domain.ReviewRun {
// Legacy compatibility only: in the multi-PR model the authoritative state
// is Reviews. When no run is created, this field is just a best-effort
// non-empty run for older clients.
for _, review := range reviews {
if review.LatestRun != nil {
return *review.LatestRun
}
}
return domain.ReviewRun{}
}
// List returns a worker's review state: the live reviewer handle and its passes.
@ -306,18 +367,11 @@ func (e *Engine) List(ctx stdctx.Context, workerID domain.SessionID) (SessionRev
} else if ok {
handle = review.ReviewerHandleID
}
return SessionReviews{ReviewerHandleID: handle, Runs: runs}, nil
}
func (e *Engine) workerPR(ctx stdctx.Context, workerID domain.SessionID) (domain.PullRequest, error) {
prs, err := e.prs.ListPRsBySession(ctx, workerID)
if err != nil {
return domain.PullRequest{}, err
return SessionReviews{}, err
}
if len(prs) == 0 {
return domain.PullRequest{}, fmt.Errorf("%w: worker %q has no PR to review", ErrInvalid, workerID)
}
return prs[0], nil
return SessionReviews{ReviewerHandleID: handle, Runs: runs, Reviews: Plan(prs, runs)}, nil
}
// reviewerHarness resolves which harness reviews the worker's PR: a configured
@ -335,7 +389,7 @@ func (e *Engine) reviewerHarness(ctx stdctx.Context, worker domain.SessionRecord
return cfg.ResolveReviewerHarness(worker.Harness), nil
}
func (e *Engine) upsertReview(ctx stdctx.Context, worker domain.SessionRecord, harness domain.ReviewerHarness, prURL, handleID string, now time.Time) (domain.Review, error) {
func (e *Engine) upsertReview(ctx stdctx.Context, worker domain.SessionRecord, harness domain.ReviewerHarness, handleID string, now time.Time) (domain.Review, error) {
existing, ok, err := e.store.GetReviewBySession(ctx, worker.ID)
if err != nil {
return domain.Review{}, err
@ -345,7 +399,7 @@ func (e *Engine) upsertReview(ctx stdctx.Context, worker domain.SessionRecord, h
SessionID: worker.ID,
ProjectID: worker.ProjectID,
Harness: harness,
PRURL: prURL,
PRURL: "",
ReviewerHandleID: handleID,
CreatedAt: now,
UpdatedAt: now,

View File

@ -43,6 +43,16 @@ func (f *fakeStore) InsertReviewRun(_ context.Context, r domain.ReviewRun) error
f.runs = append(f.runs, winner)
return f.insertErr
}
for _, existing := range f.runs {
if existing.SessionID == r.SessionID &&
existing.PRURL == r.PRURL &&
existing.TargetSHA == r.TargetSHA &&
existing.TargetSHA != "" &&
existing.Status != domain.ReviewRunFailed &&
existing.Verdict != domain.VerdictChangesRequested {
return domain.ErrDuplicateReviewRun
}
}
f.runs = append(f.runs, r)
return nil
}
@ -74,10 +84,10 @@ func (f *fakeStore) SupersedeReviewRun(_ context.Context, id, body string) (bool
}
return false, nil
}
func (f *fakeStore) SupersedeStaleRunningReviewRuns(_ context.Context, sessionID domain.SessionID, targetSHA, body string) (int64, error) {
func (f *fakeStore) SupersedeStaleRunningReviewRuns(_ context.Context, sessionID domain.SessionID, prURL, targetSHA, body string) (int64, error) {
var n int64
for i := range f.runs {
if f.runs[i].SessionID == sessionID && f.runs[i].TargetSHA != targetSHA && f.runs[i].Status == domain.ReviewRunRunning && f.runs[i].Verdict == domain.VerdictNone {
if f.runs[i].SessionID == sessionID && f.runs[i].PRURL == prURL && f.runs[i].TargetSHA != targetSHA && f.runs[i].Status == domain.ReviewRunRunning && f.runs[i].Verdict == domain.VerdictNone {
f.runs[i].Status = domain.ReviewRunFailed
f.runs[i].Body = body
n++
@ -93,9 +103,9 @@ func (f *fakeStore) GetReviewRun(_ context.Context, id string) (domain.ReviewRun
}
return domain.ReviewRun{}, false, nil
}
func (f *fakeStore) GetReviewRunBySessionAndSHA(_ context.Context, _ domain.SessionID, sha string) (domain.ReviewRun, bool, error) {
func (f *fakeStore) GetReviewRunBySessionPRAndSHA(_ context.Context, sessionID domain.SessionID, prURL, sha string) (domain.ReviewRun, bool, error) {
for i := len(f.runs) - 1; i >= 0; i-- {
if f.runs[i].TargetSHA == sha {
if f.runs[i].SessionID == sessionID && f.runs[i].PRURL == prURL && f.runs[i].TargetSHA == sha {
return f.runs[i], true, nil
}
}
@ -136,12 +146,15 @@ type fakeLauncher struct {
notified bool
gotSpec LaunchSpec
gotHandle string
specs []LaunchSpec
handles []string
}
func (f *fakeLauncher) Spawn(_ context.Context, spec LaunchSpec) (string, error) {
f.spawned = true
f.spawnCount++
f.gotSpec = spec
f.specs = append(f.specs, spec)
if f.spawnErr != nil {
return "", f.spawnErr
}
@ -151,6 +164,8 @@ func (f *fakeLauncher) Notify(_ context.Context, handleID string, spec LaunchSpe
f.notified = true
f.gotHandle = handleID
f.gotSpec = spec
f.handles = append(f.handles, handleID)
f.specs = append(f.specs, spec)
return f.notifyErr
}
func (f *fakeLauncher) Alive(_ context.Context, _ string) (bool, error) {
@ -176,7 +191,7 @@ func newEngineForTest(store Store, sessions Sessions, prs PRs, projects Projects
}
func prAt(sha string) fakePRs {
return fakePRs{prs: []domain.PullRequest{{URL: "https://github.com/o/r/pull/1", HeadSHA: sha}}}
return fakePRs{prs: []domain.PullRequest{{URL: "https://github.com/o/r/pull/1", Number: 1, HeadSHA: sha}}}
}
// --- tests ---
@ -259,7 +274,7 @@ func TestTriggerFallsBackToExistingRunOnUniqueConflict(t *testing.T) {
if res.Created {
t.Fatalf("expected Created=false on unique conflict: %+v", res)
}
if res.Run.TargetSHA != "sha1" || res.Run.ID != "winner-id-1" {
if res.Run.TargetSHA != "sha1" || !strings.HasPrefix(res.Run.ID, "winner-") {
t.Fatalf("expected the recorded winner run, got %+v", res.Run)
}
if launcher.spawnCount != 0 {
@ -271,7 +286,7 @@ func TestTriggerIsIdempotentForSameCommit(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{{
ID: "run-1", SessionID: "mer-1", TargetSHA: "sha1",
ID: "run-1", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1",
Status: domain.ReviewRunComplete, Verdict: domain.VerdictApproved,
}},
}
@ -296,7 +311,7 @@ func TestTriggerIsIdempotentForSameCommit(t *testing.T) {
func TestTriggerReusesRunningRowWithNoVerdict(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{{ID: "run-1", SessionID: "mer-1", TargetSHA: "sha1", Status: domain.ReviewRunRunning}},
runs: []domain.ReviewRun{{ID: "run-1", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1", Status: domain.ReviewRunRunning}},
}
launcher := &fakeLauncher{alive: false, handle: "review-mer-2"}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prAt("sha1"), fakeProjects{}, launcher)
@ -319,7 +334,7 @@ func TestTriggerReusesRunningRowWithNoVerdict(t *testing.T) {
func TestTriggerSupersedesNonRunningRowWithNoVerdict(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{{ID: "run-1", SessionID: "mer-1", TargetSHA: "sha1", Status: domain.ReviewRunComplete}},
runs: []domain.ReviewRun{{ID: "run-1", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1", Status: domain.ReviewRunComplete}},
}
launcher := &fakeLauncher{alive: true, handle: "review-mer-1"}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prAt("sha1"), fakeProjects{}, launcher)
@ -342,7 +357,7 @@ func TestTriggerSupersedesNonRunningRowWithNoVerdict(t *testing.T) {
func TestTriggerNotifiesLiveReviewerOnNewCommit(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{{ID: "run-0", SessionID: "mer-1", TargetSHA: "sha0", Status: domain.ReviewRunComplete}},
runs: []domain.ReviewRun{{ID: "run-0", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha0", Status: domain.ReviewRunComplete}},
}
launcher := &fakeLauncher{alive: true}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prAt("sha1"), fakeProjects{}, launcher)
@ -365,7 +380,7 @@ func TestTriggerNotifiesLiveReviewerOnNewCommit(t *testing.T) {
func TestTriggerSupersedesOlderRunningRunOnNewCommit(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{{ID: "run-old", SessionID: "mer-1", TargetSHA: "sha0", Status: domain.ReviewRunRunning}},
runs: []domain.ReviewRun{{ID: "run-old", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha0", Status: domain.ReviewRunRunning}},
}
launcher := &fakeLauncher{alive: true, handle: "review-mer-1"}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prAt("sha1"), fakeProjects{}, launcher)
@ -388,7 +403,7 @@ func TestTriggerSupersedesOlderRunningRunOnNewCommit(t *testing.T) {
func TestTriggerSpawnsWhenReviewerDead(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{{ID: "run-0", SessionID: "mer-1", TargetSHA: "sha0", Status: domain.ReviewRunComplete}},
runs: []domain.ReviewRun{{ID: "run-0", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha0", Status: domain.ReviewRunComplete}},
}
launcher := &fakeLauncher{alive: false, handle: "review-mer-1"}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prAt("sha1"), fakeProjects{}, launcher)
@ -424,7 +439,7 @@ func TestTriggerLaunchFailureRecordsFailedRun(t *testing.T) {
func TestTriggerRetriesAfterFailedRunForSameCommit(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{{ID: "run-failed", ReviewID: "rev-1", SessionID: "mer-1", TargetSHA: "sha1", Status: domain.ReviewRunFailed}},
runs: []domain.ReviewRun{{ID: "run-failed", ReviewID: "rev-1", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1", Status: domain.ReviewRunFailed}},
}
launcher := &fakeLauncher{handle: "review-mer-1"}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prAt("sha1"), fakeProjects{}, launcher)
@ -441,6 +456,111 @@ func TestTriggerRetriesAfterFailedRunForSameCommit(t *testing.T) {
}
}
func TestTriggerCreatesRunsForMultipleEligiblePRsWithOneReviewer(t *testing.T) {
store := &fakeStore{}
launcher := &fakeLauncher{handle: "review-mer-1"}
prs := fakePRs{prs: []domain.PullRequest{
{URL: "https://github.com/o/r/pull/1", Number: 1, HeadSHA: "sha1"},
{URL: "https://github.com/o/r/pull/2", Number: 2, HeadSHA: "sha2"},
}}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prs, fakeProjects{}, launcher)
res, err := eng.Trigger(context.Background(), "mer-1")
if err != nil {
t.Fatalf("Trigger: %v", err)
}
if !res.Created || len(res.CreatedRuns) != 2 || len(store.runs) != 2 {
t.Fatalf("created batch = %+v runs=%+v", res, store.runs)
}
if res.CreatedRuns[0].BatchID == "" || res.CreatedRuns[0].BatchID != res.CreatedRuns[1].BatchID {
t.Fatalf("created runs should share one batch id: %+v", res.CreatedRuns)
}
if launcher.spawnCount != 1 || len(launcher.handles) != 0 {
t.Fatalf("expected one spawn and no extra notify, launcher=%+v", launcher)
}
if len(launcher.specs) != 1 {
t.Fatalf("launch specs = %d, want 1: %+v", len(launcher.specs), launcher.specs)
}
spec := launcher.specs[0]
if spec.ReviewIndex != 0 || len(spec.ReviewQueue) != 2 {
t.Fatalf("spec queue context = index %d queue %+v", spec.ReviewIndex, spec.ReviewQueue)
}
if spec.ReviewQueue[0].PRURL != "https://github.com/o/r/pull/1" || spec.ReviewQueue[1].PRURL != "https://github.com/o/r/pull/2" {
t.Fatalf("spec queue URLs = %+v", spec.ReviewQueue)
}
if store.review == nil || store.review.ReviewerHandleID != "review-mer-1" || store.review.PRURL != "" {
t.Fatalf("review row = %+v, want shared handle and no behavioral pr_url", store.review)
}
}
func TestTriggerAllowsTwoPRsWithSameHeadSHA(t *testing.T) {
store := &fakeStore{}
launcher := &fakeLauncher{handle: "review-mer-1"}
prs := fakePRs{prs: []domain.PullRequest{
{URL: "https://github.com/o/r/pull/1", Number: 1, HeadSHA: "same"},
{URL: "https://github.com/o/r/pull/2", Number: 2, HeadSHA: "same"},
}}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prs, fakeProjects{}, launcher)
res, err := eng.Trigger(context.Background(), "mer-1")
if err != nil {
t.Fatalf("Trigger: %v", err)
}
if len(res.CreatedRuns) != 2 {
t.Fatalf("created runs = %d, want 2: %+v", len(res.CreatedRuns), res.CreatedRuns)
}
if store.runs[0].PRURL == store.runs[1].PRURL || store.runs[0].TargetSHA != store.runs[1].TargetSHA {
t.Fatalf("runs should differ by PR only: %+v", store.runs)
}
}
func TestTriggerSkipsApprovedAndRunningCurrentHead(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{
{ID: "approved", ReviewID: "rev-1", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1", Status: domain.ReviewRunComplete, Verdict: domain.VerdictApproved, CreatedAt: time.Unix(1, 0)},
{ID: "running", ReviewID: "rev-1", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/2", TargetSHA: "sha2", Status: domain.ReviewRunRunning, CreatedAt: time.Unix(2, 0)},
},
}
launcher := &fakeLauncher{alive: true}
prs := fakePRs{prs: []domain.PullRequest{
{URL: "https://github.com/o/r/pull/1", Number: 1, HeadSHA: "sha1"},
{URL: "https://github.com/o/r/pull/2", Number: 2, HeadSHA: "sha2"},
}}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prs, fakeProjects{}, launcher)
res, err := eng.Trigger(context.Background(), "mer-1")
if err != nil {
t.Fatalf("Trigger: %v", err)
}
if res.Created || len(res.CreatedRuns) != 0 || launcher.spawned || launcher.notified {
t.Fatalf("expected no new work: res=%+v launcher=%+v", res, launcher)
}
if len(res.Reviews) != 2 || res.Reviews[0].Status != ReviewStateUpToDate || res.Reviews[1].Status != ReviewStateRunning {
t.Fatalf("review states = %+v", res.Reviews)
}
}
func TestTriggerCreatesRunForChangesRequestedCurrentHead(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{{
ID: "changes", ReviewID: "rev-1", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1",
Status: domain.ReviewRunComplete, Verdict: domain.VerdictChangesRequested, CreatedAt: time.Unix(1, 0),
}},
}
launcher := &fakeLauncher{alive: true, handle: "review-mer-1"}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prAt("sha1"), fakeProjects{}, launcher)
res, err := eng.Trigger(context.Background(), "mer-1")
if err != nil {
t.Fatalf("Trigger: %v", err)
}
if !res.Created || len(res.CreatedRuns) != 1 || !launcher.notified || launcher.spawned {
t.Fatalf("expected rerun on changes_requested current head: res=%+v launcher=%+v", res, launcher)
}
}
func TestTriggerUsesConfiguredReviewerHarness(t *testing.T) {
store := &fakeStore{}
projects := fakeProjects{cfg: domain.ProjectConfig{Reviewers: []domain.ReviewerConfig{{Harness: domain.ReviewerHarness("greptile")}}}}
@ -474,7 +594,7 @@ func TestTriggerRejectsBadWorkerState(t *testing.T) {
func TestListReturnsHandleAndRuns(t *testing.T) {
store := &fakeStore{
review: &domain.Review{ID: "rev-1", SessionID: "mer-1", ReviewerHandleID: "review-mer-1"},
runs: []domain.ReviewRun{{ID: "run-1", SessionID: "mer-1", TargetSHA: "sha1"}},
runs: []domain.ReviewRun{{ID: "run-1", SessionID: "mer-1", PRURL: "https://github.com/o/r/pull/1", TargetSHA: "sha1"}},
}
eng := newEngineForTest(store, fakeSessions{rec: liveWorker(), ok: true}, prAt("sha1"), fakeProjects{}, &fakeLauncher{})
got, err := eng.List(context.Background(), "mer-1")

View File

@ -27,6 +27,7 @@ var (
type Manager interface {
Trigger(ctx context.Context, workerID domain.SessionID) (reviewcore.TriggerResult, error)
Submit(ctx context.Context, workerID domain.SessionID, runID string, verdict domain.ReviewVerdict, body, githubReviewID string) (domain.ReviewRun, error)
SubmitMany(ctx context.Context, workerID domain.SessionID, reviews []SubmittedReview) ([]domain.ReviewRun, error)
List(ctx context.Context, workerID domain.SessionID) (reviewcore.SessionReviews, error)
}
@ -45,12 +46,14 @@ type Store interface {
GetReviewRun(ctx context.Context, id string) (domain.ReviewRun, bool, error)
UpdateReviewRunResult(ctx context.Context, id string, status domain.ReviewRunStatus, verdict domain.ReviewVerdict, body, githubReviewID string) (bool, error)
MarkReviewRunDelivered(ctx context.Context, id string, deliveredAt time.Time) (bool, error)
ListPRsBySession(ctx context.Context, id domain.SessionID) ([]domain.PullRequest, error)
}
// Reducer is the lifecycle reaction boundary used after a review result has
// been persisted.
type Reducer interface {
ApplyReviewResult(ctx context.Context, workerID domain.SessionID, result lifecycle.ReviewResult) (lifecycle.ReviewDeliveryOutcome, error)
ApplyReviewBatch(ctx context.Context, workerID domain.SessionID, batchID string, results []lifecycle.ReviewResult) (lifecycle.ReviewDeliveryOutcome, error)
}
// Option customizes the review service.
@ -68,7 +71,11 @@ func WithClock(clock func() time.Time) Option {
// New wraps a core review engine as the API-facing service.
func New(engine *reviewcore.Engine, store Store, opts ...Option) *Service {
s := &Service{engine: engine, store: store, clock: func() time.Time { return time.Now().UTC() }}
s := &Service{
engine: engine,
store: store,
clock: func() time.Time { return time.Now().UTC() },
}
for _, opt := range opts {
opt(s)
}
@ -80,11 +87,76 @@ func (s *Service) Trigger(ctx context.Context, workerID domain.SessionID) (revie
return s.engine.Trigger(ctx, workerID)
}
// SubmittedReview is one review result supplied by the reviewer CLI.
type SubmittedReview struct {
RunID string
Verdict domain.ReviewVerdict
Body string
GithubReviewID string
}
// Submit records a reviewer's result for a specific worker review pass.
func (s *Service) Submit(ctx context.Context, workerID domain.SessionID, runID string, verdict domain.ReviewVerdict, body, githubReviewID string) (domain.ReviewRun, error) {
if workerID == "" {
return domain.ReviewRun{}, fmt.Errorf("%w: worker session id is required", ErrInvalid)
runs, err := s.SubmitMany(ctx, workerID, []SubmittedReview{{
RunID: runID,
Verdict: verdict,
Body: body,
GithubReviewID: githubReviewID,
}})
if err != nil {
return domain.ReviewRun{}, err
}
if len(runs) == 0 {
return domain.ReviewRun{}, fmt.Errorf("%w: no review result submitted", ErrInvalid)
}
return runs[0], nil
}
// SubmitMany records one reviewer CLI submission containing results for one or
// more PR-scoped runs. Delivery is scoped to the runs in this submission, so a
// missing/stuck result for another PR in the same trigger cannot block feedback.
func (s *Service) SubmitMany(ctx context.Context, workerID domain.SessionID, reviews []SubmittedReview) ([]domain.ReviewRun, error) {
if workerID == "" {
return nil, fmt.Errorf("%w: worker session id is required", ErrInvalid)
}
if len(reviews) == 0 {
return nil, fmt.Errorf("%w: at least one review result is required", ErrInvalid)
}
if s.store == nil {
return nil, fmt.Errorf("review service store is not configured")
}
runs := make([]domain.ReviewRun, 0, len(reviews))
for _, review := range reviews {
run, err := s.submitOne(ctx, workerID, review)
if err != nil {
return nil, err
}
runs = append(runs, run)
}
if s.lifecycle == nil {
return runs, nil
}
delivered, err := s.deliverSubmitted(ctx, workerID, runs)
if err != nil {
return nil, err
}
byID := make(map[string]domain.ReviewRun, len(delivered))
for _, run := range delivered {
byID[run.ID] = run
}
for i, run := range runs {
if deliveredRun, ok := byID[run.ID]; ok {
runs[i] = deliveredRun
}
}
return runs, nil
}
func (s *Service) submitOne(ctx context.Context, workerID domain.SessionID, review SubmittedReview) (domain.ReviewRun, error) {
runID := review.RunID
verdict := review.Verdict
body := review.Body
githubReviewID := review.GithubReviewID
if runID == "" {
return domain.ReviewRun{}, fmt.Errorf("%w: review run id is required", ErrInvalid)
}
@ -94,9 +166,6 @@ func (s *Service) Submit(ctx context.Context, workerID domain.SessionID, runID s
if verdict == domain.VerdictChangesRequested && body == "" {
return domain.ReviewRun{}, fmt.Errorf("%w: a changes_requested review requires a body", ErrInvalid)
}
if s.store == nil {
return domain.ReviewRun{}, fmt.Errorf("review service store is not configured")
}
run, ok, err := s.store.GetReviewRun(ctx, runID)
if err != nil {
return domain.ReviewRun{}, err
@ -136,35 +205,92 @@ func (s *Service) Submit(ctx context.Context, workerID domain.SessionID, runID s
default:
return domain.ReviewRun{}, fmt.Errorf("%w: review run %q is not running", ErrInvalid, runID)
}
return run, nil
}
if s.lifecycle == nil {
return run, nil
}
outcome, err := s.lifecycle.ApplyReviewResult(ctx, workerID, lifecycle.ReviewResult{
RunID: run.ID,
WorkerID: workerID,
PRURL: run.PRURL,
TargetSHA: run.TargetSHA,
Verdict: run.Verdict,
Body: run.Body,
GithubReviewID: run.GithubReviewID,
DeliveredAt: run.DeliveredAt,
})
func (s *Service) deliverSubmitted(ctx context.Context, workerID domain.SessionID, runs []domain.ReviewRun) ([]domain.ReviewRun, error) {
deliverable, err := s.deliverableRuns(ctx, workerID, runs)
if err != nil {
return domain.ReviewRun{}, err
return nil, err
}
if outcome == lifecycle.ReviewDeliverySent {
deliveredAt := s.clock()
if len(deliverable) == 0 {
return nil, nil
}
results := reviewResults(workerID, deliverable)
var outcome lifecycle.ReviewDeliveryOutcome
if len(results) == 1 && results[0].BatchID == "" {
outcome, err = s.lifecycle.ApplyReviewResult(ctx, workerID, results[0])
} else {
outcome, err = s.lifecycle.ApplyReviewBatch(ctx, workerID, results[0].BatchID, results)
}
if err != nil {
return nil, err
}
if outcome != lifecycle.ReviewDeliverySent {
return nil, nil
}
deliveredAt := s.clock()
delivered := make([]domain.ReviewRun, 0, len(deliverable))
for _, run := range deliverable {
updated, err := s.store.MarkReviewRunDelivered(ctx, run.ID, deliveredAt)
if err != nil {
return domain.ReviewRun{}, err
return nil, err
}
if updated {
run.Status = domain.ReviewRunDelivered
run.DeliveredAt = &deliveredAt
delivered = append(delivered, run)
}
}
return run, nil
return delivered, nil
}
func (s *Service) deliverableRuns(ctx context.Context, workerID domain.SessionID, runs []domain.ReviewRun) ([]domain.ReviewRun, error) {
currentHeads, err := s.currentHeadsByPR(ctx, workerID)
if err != nil {
return nil, err
}
deliverable := make([]domain.ReviewRun, 0, len(runs))
for _, run := range runs {
if run.Status != domain.ReviewRunComplete || run.Verdict != domain.VerdictChangesRequested || run.DeliveredAt != nil {
continue
}
if run.BatchID != "" && currentHeads[run.PRURL] != run.TargetSHA {
continue
}
deliverable = append(deliverable, run)
}
return deliverable, nil
}
func reviewResults(workerID domain.SessionID, runs []domain.ReviewRun) []lifecycle.ReviewResult {
results := make([]lifecycle.ReviewResult, 0, len(runs))
for _, run := range runs {
results = append(results, lifecycle.ReviewResult{
RunID: run.ID,
BatchID: run.BatchID,
WorkerID: workerID,
PRURL: run.PRURL,
TargetSHA: run.TargetSHA,
Verdict: run.Verdict,
Body: run.Body,
GithubReviewID: run.GithubReviewID,
DeliveredAt: run.DeliveredAt,
})
}
return results
}
func (s *Service) currentHeadsByPR(ctx context.Context, workerID domain.SessionID) (map[string]string, error) {
prs, err := s.store.ListPRsBySession(ctx, workerID)
if err != nil {
return nil, err
}
current := make(map[string]string, len(prs))
for _, pr := range prs {
current[pr.URL] = pr.HeadSHA
}
return current, nil
}
// List returns a worker's review state.

View File

@ -11,18 +11,45 @@ import (
)
type fakeStore struct {
run domain.ReviewRun
ok bool
run domain.ReviewRun
ok bool
batchRuns []domain.ReviewRun
prs []domain.PullRequest
updateCalls int
markCalls int
markedIDs []string
}
func (f *fakeStore) GetReviewRun(context.Context, string) (domain.ReviewRun, bool, error) {
return f.run, f.ok, nil
func (f *fakeStore) GetReviewRun(_ context.Context, id string) (domain.ReviewRun, bool, error) {
for _, run := range f.batchRuns {
if run.ID == id {
return run, true, nil
}
}
if f.ok && f.run.ID == id {
return f.run, true, nil
}
return domain.ReviewRun{}, false, nil
}
func (f *fakeStore) UpdateReviewRunResult(_ context.Context, _ string, status domain.ReviewRunStatus, verdict domain.ReviewVerdict, body, githubReviewID string) (bool, error) {
func (f *fakeStore) UpdateReviewRunResult(_ context.Context, id string, status domain.ReviewRunStatus, verdict domain.ReviewVerdict, body, githubReviewID string) (bool, error) {
for i := range f.batchRuns {
if f.batchRuns[i].ID == id {
if f.batchRuns[i].Status != domain.ReviewRunRunning {
return false, nil
}
f.updateCalls++
f.batchRuns[i].Status = status
f.batchRuns[i].Verdict = verdict
f.batchRuns[i].Body = body
f.batchRuns[i].GithubReviewID = githubReviewID
if f.run.ID == id {
f.run = f.batchRuns[i]
}
return true, nil
}
}
if f.run.Status != domain.ReviewRunRunning {
return false, nil
}
@ -34,21 +61,44 @@ func (f *fakeStore) UpdateReviewRunResult(_ context.Context, _ string, status do
return true, nil
}
func (f *fakeStore) MarkReviewRunDelivered(_ context.Context, _ string, deliveredAt time.Time) (bool, error) {
func (f *fakeStore) MarkReviewRunDelivered(_ context.Context, id string, deliveredAt time.Time) (bool, error) {
f.markCalls++
if f.run.Status != domain.ReviewRunComplete || f.run.DeliveredAt != nil {
f.markedIDs = append(f.markedIDs, id)
if f.run.ID == id && f.run.Status == domain.ReviewRunComplete && f.run.DeliveredAt == nil {
f.run.Status = domain.ReviewRunDelivered
f.run.DeliveredAt = &deliveredAt
}
for i := range f.batchRuns {
if f.batchRuns[i].ID == id && f.batchRuns[i].Status == domain.ReviewRunComplete && f.batchRuns[i].DeliveredAt == nil {
f.batchRuns[i].Status = domain.ReviewRunDelivered
f.batchRuns[i].DeliveredAt = &deliveredAt
return true, nil
}
}
if f.run.ID != id || f.run.Status != domain.ReviewRunDelivered {
return false, nil
}
f.run.Status = domain.ReviewRunDelivered
f.run.DeliveredAt = &deliveredAt
return true, nil
}
func (f *fakeStore) ListReviewRunsByBatch(context.Context, domain.SessionID, string) ([]domain.ReviewRun, error) {
out := append([]domain.ReviewRun(nil), f.batchRuns...)
return out, nil
}
func (f *fakeStore) ListPRsBySession(context.Context, domain.SessionID) ([]domain.PullRequest, error) {
out := append([]domain.PullRequest(nil), f.prs...)
return out, nil
}
type fakeReducer struct {
outcome lifecycle.ReviewDeliveryOutcome
err error
calls int
got lifecycle.ReviewResult
outcome lifecycle.ReviewDeliveryOutcome
err error
calls int
batchCalls int
got lifecycle.ReviewResult
gotBatchID string
gotBatch []lifecycle.ReviewResult
}
func (f *fakeReducer) ApplyReviewResult(_ context.Context, _ domain.SessionID, result lifecycle.ReviewResult) (lifecycle.ReviewDeliveryOutcome, error) {
@ -57,6 +107,13 @@ func (f *fakeReducer) ApplyReviewResult(_ context.Context, _ domain.SessionID, r
return f.outcome, f.err
}
func (f *fakeReducer) ApplyReviewBatch(_ context.Context, _ domain.SessionID, batchID string, results []lifecycle.ReviewResult) (lifecycle.ReviewDeliveryOutcome, error) {
f.batchCalls++
f.gotBatchID = batchID
f.gotBatch = append([]lifecycle.ReviewResult(nil), results...)
return f.outcome, f.err
}
func TestSubmitPersistsThenAppliesThenStampsDelivered(t *testing.T) {
now := time.Unix(100, 0).UTC()
st := &fakeStore{ok: true, run: domain.ReviewRun{ID: "run-1", SessionID: "mer-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunRunning}}
@ -78,6 +135,98 @@ func TestSubmitPersistsThenAppliesThenStampsDelivered(t *testing.T) {
}
}
func TestSubmitBatchRunDoesNotWaitForOtherRunningRuns(t *testing.T) {
now := time.Unix(100, 0).UTC()
st := &fakeStore{
ok: true,
run: domain.ReviewRun{ID: "run-1", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunRunning},
batchRuns: []domain.ReviewRun{
{ID: "run-1", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunRunning},
{ID: "run-2", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr2", TargetSHA: "sha2", Status: domain.ReviewRunRunning},
},
prs: []domain.PullRequest{{URL: "pr1", HeadSHA: "sha1"}, {URL: "pr2", HeadSHA: "sha2"}},
}
reducer := &fakeReducer{outcome: lifecycle.ReviewDeliverySent}
svc := New(nil, st, WithLifecycleReducer(reducer), WithClock(func() time.Time { return now }))
run, err := svc.Submit(context.Background(), "mer-1", "run-1", domain.VerdictChangesRequested, "fix pr1", "101")
if err != nil {
t.Fatalf("Submit: %v", err)
}
if run.Status != domain.ReviewRunDelivered || run.DeliveredAt == nil || !run.DeliveredAt.Equal(now) {
t.Fatalf("first submit status = %+v, want delivered", run)
}
if reducer.batchCalls != 1 || len(reducer.gotBatch) != 1 || reducer.gotBatch[0].RunID != "run-1" || st.markCalls != 1 {
t.Fatalf("submitted run should deliver independently: batchCalls=%d got=%+v markCalls=%d", reducer.batchCalls, reducer.gotBatch, st.markCalls)
}
}
func TestSubmitManySendsCombinedChangesRequested(t *testing.T) {
now := time.Unix(100, 0).UTC()
st := &fakeStore{
ok: true,
batchRuns: []domain.ReviewRun{
{ID: "run-1", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunRunning},
{ID: "run-2", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr2", TargetSHA: "sha2", Status: domain.ReviewRunRunning},
{ID: "run-3", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr3", TargetSHA: "sha3", Status: domain.ReviewRunComplete, Verdict: domain.VerdictApproved},
{ID: "run-4", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr4", TargetSHA: "old", Status: domain.ReviewRunComplete, Verdict: domain.VerdictChangesRequested, Body: "stale"},
{ID: "run-5", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr5", TargetSHA: "sha5", Status: domain.ReviewRunFailed},
},
prs: []domain.PullRequest{
{URL: "pr1", HeadSHA: "sha1"},
{URL: "pr2", HeadSHA: "sha2"},
{URL: "pr3", HeadSHA: "sha3"},
{URL: "pr4", HeadSHA: "new"},
{URL: "pr5", HeadSHA: "sha5"},
},
}
reducer := &fakeReducer{outcome: lifecycle.ReviewDeliverySent}
svc := New(nil, st, WithLifecycleReducer(reducer), WithClock(func() time.Time { return now }))
runs, err := svc.SubmitMany(context.Background(), "mer-1", []SubmittedReview{
{RunID: "run-1", Verdict: domain.VerdictChangesRequested, Body: "fix pr1", GithubReviewID: "101"},
{RunID: "run-2", Verdict: domain.VerdictChangesRequested, Body: "fix pr2", GithubReviewID: "102"},
{RunID: "run-3", Verdict: domain.VerdictApproved},
})
if err != nil {
t.Fatalf("SubmitMany: %v", err)
}
if reducer.batchCalls != 1 || reducer.gotBatchID != "batch-1" {
t.Fatalf("batch delivery calls/id = %d/%q", reducer.batchCalls, reducer.gotBatchID)
}
if len(reducer.gotBatch) != 2 || reducer.gotBatch[0].RunID != "run-1" || reducer.gotBatch[1].RunID != "run-2" {
t.Fatalf("delivered batch = %+v, want run-1 and run-2 only", reducer.gotBatch)
}
if st.markCalls != 2 {
t.Fatalf("markCalls = %d, want 2", st.markCalls)
}
if runs[0].Status != domain.ReviewRunDelivered || runs[0].DeliveredAt == nil || !runs[0].DeliveredAt.Equal(now) ||
runs[1].Status != domain.ReviewRunDelivered || runs[1].DeliveredAt == nil || !runs[1].DeliveredAt.Equal(now) {
t.Fatalf("submitted runs not stamped delivered: %+v", runs)
}
}
func TestSubmitBatchApprovedOnlySendsNothing(t *testing.T) {
st := &fakeStore{
ok: true,
run: domain.ReviewRun{ID: "run-2", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr2", TargetSHA: "sha2", Status: domain.ReviewRunRunning},
batchRuns: []domain.ReviewRun{
{ID: "run-1", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunComplete, Verdict: domain.VerdictApproved},
{ID: "run-2", SessionID: "mer-1", BatchID: "batch-1", PRURL: "pr2", TargetSHA: "sha2", Status: domain.ReviewRunRunning},
},
prs: []domain.PullRequest{{URL: "pr1", HeadSHA: "sha1"}, {URL: "pr2", HeadSHA: "sha2"}},
}
reducer := &fakeReducer{outcome: lifecycle.ReviewDeliverySent}
svc := New(nil, st, WithLifecycleReducer(reducer))
if _, err := svc.Submit(context.Background(), "mer-1", "run-2", domain.VerdictApproved, "", "102"); err != nil {
t.Fatalf("Submit: %v", err)
}
if reducer.batchCalls != 0 || st.markCalls != 0 {
t.Fatalf("approved-only batch should not deliver: batchCalls=%d markCalls=%d", reducer.batchCalls, st.markCalls)
}
}
func TestSubmitDeliveryFailureLeavesCompletedUndeliveredForRetry(t *testing.T) {
sendErr := errors.New("dead pane")
st := &fakeStore{ok: true, run: domain.ReviewRun{ID: "run-1", SessionID: "mer-1", PRURL: "pr1", TargetSHA: "sha1", Status: domain.ReviewRunRunning}}

View File

@ -113,8 +113,8 @@ func (s *Service) ClaimPR(ctx context.Context, id domain.SessionID, ref string,
return ClaimPRResult{}, err
}
now := s.clock().UTC()
pr, checks, threads, comments := claimRowsFromSCM(id, obs, now)
outcome, err := s.prClaimer.ClaimPR(ctx, pr, checks, threads, comments, reviewMode, opts.AllowTakeover)
pr, checks, reviews, threads, comments := claimRowsFromSCM(id, obs, now)
outcome, err := s.prClaimer.ClaimPR(ctx, pr, checks, reviews, threads, comments, reviewMode, opts.AllowTakeover)
if err != nil {
return ClaimPRResult{}, err
}
@ -163,6 +163,7 @@ func (s *Service) enrichClaimReviews(ctx context.Context, ref ports.SCMPRRef, ob
obs.Review.Decision = review.Decision
}
obs.Review.Threads = review.Threads
obs.Review.Reviews = review.Reviews
obs.Review.Partial = review.Partial
if review.Partial {
return ports.ReviewWriteMerge, nil
@ -181,7 +182,7 @@ func scmRepoForClaim(provider scmProvider, projectOrigin, prURL string) (ports.S
return ports.SCMRepo{Provider: "github", Host: "github.com", Owner: owner, Name: name, Repo: owner + "/" + name}, nil
}
func claimRowsFromSCM(sessionID domain.SessionID, obs ports.SCMObservation, now time.Time) (domain.PullRequest, []domain.PullRequestCheck, []domain.PullRequestReviewThread, []domain.PullRequestComment) {
func claimRowsFromSCM(sessionID domain.SessionID, obs ports.SCMObservation, now time.Time) (domain.PullRequest, []domain.PullRequestCheck, []domain.PullRequestReview, []domain.PullRequestReviewThread, []domain.PullRequestComment) {
observedAt := obs.ObservedAt
if observedAt.IsZero() {
observedAt = now
@ -226,6 +227,21 @@ func claimRowsFromSCM(sessionID domain.SessionID, obs ports.SCMObservation, now
for _, ch := range obs.CI.Checks {
checks = append(checks, domain.PullRequestCheck{Name: ch.Name, CommitHash: obs.CI.HeadSHA, Status: domain.PRCheckStatus(ch.Status), Conclusion: ch.Conclusion, URL: ch.URL, Details: ch.ProviderID, LogTail: ch.LogTail, CreatedAt: now})
}
reviews := make([]domain.PullRequestReview, 0, len(obs.Review.Reviews))
for _, review := range obs.Review.Reviews {
submittedAt := review.SubmittedAt
if submittedAt.IsZero() {
submittedAt = now
}
reviews = append(reviews, domain.PullRequestReview{
ID: review.ID,
Author: review.Author,
State: domain.ReviewDecision(firstNonEmpty(review.State, string(domain.ReviewNone))),
URL: review.URL,
IsBot: review.IsBot,
SubmittedAt: submittedAt,
})
}
threads := make([]domain.PullRequestReviewThread, 0, len(obs.Review.Threads))
commentCount := 0
for _, th := range obs.Review.Threads {
@ -238,7 +254,7 @@ func claimRowsFromSCM(sessionID domain.SessionID, obs ports.SCMObservation, now
comments = append(comments, domain.PullRequestComment{ThreadID: th.ID, ID: c.ID, Author: c.Author, File: th.Path, Line: th.Line, Body: c.Body, URL: c.URL, Resolved: th.Resolved, IsBot: c.IsBot || th.IsBot, CreatedAt: now})
}
}
return pr, checks, threads, comments
return pr, checks, reviews, threads, comments
}
func sessionmanagerAPIError(code, message string) error {

View File

@ -62,6 +62,8 @@ type PRUnresolvedReviewer struct {
ReviewerID string
Count int
Links []PRReviewCommentLink
ReviewURL string
IsBot bool
}
// PRReviewCommentLink points to one unresolved review comment.
@ -107,17 +109,21 @@ func (s *Service) ListPRSummaries(ctx context.Context, id domain.SessionID) ([]P
if err != nil {
return nil, err
}
reviews, err := s.store.ListPRReviews(ctx, pr.URL)
if err != nil {
return nil, err
}
comments, err := s.store.ListPRComments(ctx, pr.URL)
if err != nil {
return nil, err
}
out = append(out, summarizePR(pr, checks, threads, comments))
out = append(out, summarizePR(pr, checks, reviews, threads, comments))
}
sortPRSummaries(out)
return out, nil
}
func summarizePR(pr domain.PullRequest, checks []domain.PullRequestCheck, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment) PRSummary {
func summarizePR(pr domain.PullRequest, checks []domain.PullRequestCheck, reviews []domain.PullRequestReview, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment) PRSummary {
return PRSummary{
URL: pr.URL,
HTMLURL: firstNonEmpty(pr.HTMLURL, pr.URL),
@ -134,7 +140,7 @@ func summarizePR(pr domain.PullRequest, checks []domain.PullRequestCheck, thread
Deletions: pr.Deletions,
ChangedFiles: pr.ChangedFiles,
CI: summarizeCI(pr, checks),
Review: summarizeReview(pr, comments),
Review: summarizeReview(pr, comments, reviews),
Mergeability: summarizeMergeability(pr, threads),
UpdatedAt: pr.UpdatedAt,
ObservedAt: pr.ObservedAt,
@ -166,7 +172,7 @@ func summarizeCI(pr domain.PullRequest, checks []domain.PullRequestCheck) PRCISu
return out
}
func summarizeReview(pr domain.PullRequest, comments []domain.PullRequestComment) PRReviewSummary {
func summarizeReview(pr domain.PullRequest, comments []domain.PullRequestComment, reviews []domain.PullRequestReview) PRReviewSummary {
out := PRReviewSummary{Decision: reviewOrNone(pr.Review)}
if pr.Merged || pr.Closed {
return out
@ -174,6 +180,7 @@ func summarizeReview(pr domain.PullRequest, comments []domain.PullRequestComment
byReviewer := map[string]int{}
order := []string{}
links := map[string][]PRReviewCommentLink{}
isBot := map[string]bool{}
for _, c := range comments {
if c.Resolved || c.IsBot {
continue
@ -186,24 +193,74 @@ func summarizeReview(pr domain.PullRequest, comments []domain.PullRequestComment
order = append(order, reviewer)
}
byReviewer[reviewer]++
isBot[reviewer] = c.IsBot
links[reviewer] = append(links[reviewer], PRReviewCommentLink{
URL: c.URL,
File: c.File,
Line: c.Line,
})
}
reviewURLByAuthor := map[string]string{}
for reviewer, review := range latestChangesRequestedReviews(reviews) {
if _, ok := byReviewer[reviewer]; !ok {
order = append(order, reviewer)
}
reviewURLByAuthor[reviewer] = review.URL
isBot[reviewer] = review.IsBot
}
sort.Strings(order)
for _, reviewer := range order {
out.UnresolvedBy = append(out.UnresolvedBy, PRUnresolvedReviewer{
ReviewerID: reviewer,
Count: byReviewer[reviewer],
Links: links[reviewer],
ReviewURL: reviewURLByAuthor[reviewer],
IsBot: isBot[reviewer],
})
}
out.HasUnresolvedHumanComments = len(out.UnresolvedBy) > 0
for _, reviewer := range out.UnresolvedBy {
if reviewer.Count > 0 && !reviewer.IsBot {
out.HasUnresolvedHumanComments = true
break
}
}
return out
}
func latestChangesRequestedReviews(reviews []domain.PullRequestReview) map[string]domain.PullRequestReview {
latestByReviewer := map[string]domain.PullRequestReview{}
for _, review := range reviews {
if review.State != domain.ReviewChangesRequest && review.State != domain.ReviewApproved {
continue
}
reviewer := strings.TrimSpace(review.Author)
if reviewer == "" {
reviewer = "unknown"
}
current, ok := latestByReviewer[reviewer]
if !ok || reviewAfter(review, current) {
latestByReviewer[reviewer] = review
}
}
out := map[string]domain.PullRequestReview{}
for reviewer, review := range latestByReviewer {
if review.State == domain.ReviewChangesRequest {
out[reviewer] = review
}
}
return out
}
func reviewAfter(a, b domain.PullRequestReview) bool {
if a.SubmittedAt.IsZero() || b.SubmittedAt.IsZero() {
return a.SubmittedAt.IsZero() == b.SubmittedAt.IsZero() && a.ID > b.ID
}
if a.SubmittedAt.Equal(b.SubmittedAt) {
return a.ID > b.ID
}
return a.SubmittedAt.After(b.SubmittedAt)
}
func summarizeMergeability(pr domain.PullRequest, _ []domain.PullRequestReviewThread) PRMergeabilitySummary {
return PRMergeabilitySummary{
State: mergeabilityOrUnknown(pr.Mergeability),

View File

@ -25,6 +25,7 @@ type Store interface {
ListPRFactsForSession(ctx context.Context, id domain.SessionID) ([]domain.PRFacts, error)
ListPRsBySession(ctx context.Context, sessionID domain.SessionID) ([]domain.PullRequest, error)
ListChecks(ctx context.Context, prURL string) ([]domain.PullRequestCheck, error)
ListPRReviews(ctx context.Context, prURL string) ([]domain.PullRequestReview, error)
ListPRReviewThreads(ctx context.Context, prURL string) ([]domain.PullRequestReviewThread, error)
ListPRComments(ctx context.Context, prURL string) ([]domain.PullRequestComment, error)
GetProject(ctx context.Context, id string) (domain.ProjectRecord, bool, error)

View File

@ -25,6 +25,7 @@ type fakeStore struct {
pr map[domain.SessionID]domain.PRFacts
projects map[string]domain.ProjectRecord
checks map[string][]domain.PullRequestCheck
reviews map[string][]domain.PullRequestReview
threads map[string][]domain.PullRequestReviewThread
comments map[string][]domain.PullRequestComment
num int
@ -36,6 +37,7 @@ func newFakeStore() *fakeStore {
pr: map[domain.SessionID]domain.PRFacts{},
projects: map[string]domain.ProjectRecord{},
checks: map[string][]domain.PullRequestCheck{},
reviews: map[string][]domain.PullRequestReview{},
threads: map[string][]domain.PullRequestReviewThread{},
comments: map[string][]domain.PullRequestComment{},
}
@ -118,6 +120,10 @@ func (f *fakeStore) ListChecks(_ context.Context, prURL string) ([]domain.PullRe
return append([]domain.PullRequestCheck(nil), f.checks[prURL]...), nil
}
func (f *fakeStore) ListPRReviews(_ context.Context, prURL string) ([]domain.PullRequestReview, error) {
return append([]domain.PullRequestReview(nil), f.reviews[prURL]...), nil
}
func (f *fakeStore) ListPRReviewThreads(_ context.Context, prURL string) ([]domain.PullRequestReviewThread, error) {
return append([]domain.PullRequestReviewThread(nil), f.threads[prURL]...), nil
}
@ -622,7 +628,7 @@ type errorFreeClaimOutcome struct {
ports.ClaimOutcome
}
func (f fakePRClaimer) ClaimPR(context.Context, domain.PullRequest, []domain.PullRequestCheck, []domain.PullRequestReviewThread, []domain.PullRequestComment, ports.ReviewWriteMode, bool) (ports.ClaimOutcome, error) {
func (f fakePRClaimer) ClaimPR(context.Context, domain.PullRequest, []domain.PullRequestCheck, []domain.PullRequestReview, []domain.PullRequestReviewThread, []domain.PullRequestComment, ports.ReviewWriteMode, bool) (ports.ClaimOutcome, error) {
return f.out.ClaimOutcome, f.err
}
@ -740,6 +746,9 @@ func TestListPRSummariesOmitsRawLogsAndReviewBodies(t *testing.T) {
{Name: "unit", Status: domain.PRCheckFailed, Conclusion: "failure", URL: "https://github.com/acme/repo/actions/runs/1", LogTail: "panic: secret"},
{Name: "lint", Status: domain.PRCheckPassed, Conclusion: "success", URL: "https://github.com/acme/repo/actions/runs/2"},
}
stList.reviews[prURL] = []domain.PullRequestReview{
{ID: "review-1", Author: "reviewer-a", State: domain.ReviewChangesRequest, URL: "https://github.com/acme/repo/pull/7#pullrequestreview-1", SubmittedAt: now.Add(-30 * time.Second)},
}
stList.comments[prURL] = []domain.PullRequestComment{
{Author: "reviewer-a", File: "main.go", Line: 12, Body: "raw body must stay private", URL: "https://github.com/acme/repo/pull/7#discussion_r1"},
{Author: "ci-bot", File: "main.go", Line: 13, Body: "bot body", URL: "https://github.com/acme/repo/pull/7#discussion_r2", IsBot: true},
@ -765,6 +774,8 @@ func TestListPRSummariesOmitsRawLogsAndReviewBodies(t *testing.T) {
}
if reviewer := pr.Review.UnresolvedBy[0]; reviewer.ReviewerID != "reviewer-a" || reviewer.Count != 2 || len(reviewer.Links) != 2 {
t.Fatalf("reviewer = %+v", reviewer)
} else if reviewer.ReviewURL != "https://github.com/acme/repo/pull/7#pullrequestreview-1" {
t.Fatalf("review url = %q", reviewer.ReviewURL)
}
if pr.Mergeability.State != domain.MergeConflicting || len(pr.Mergeability.ConflictFiles) != 0 || !containsString(pr.Mergeability.Reasons, "conflicts") {
t.Fatalf("mergeability = %+v", pr.Mergeability)

View File

@ -101,6 +101,16 @@ type PRComment struct {
IsBot int64
}
type PRReview struct {
PRURL string
ReviewID string
Author string
State string
URL string
IsBot int64
SubmittedAt time.Time
}
type PRReviewThread struct {
PRURL string
ThreadID string
@ -147,6 +157,7 @@ type ReviewRun struct {
CreatedAt time.Time
GithubReviewID string
DeliveredAt sql.NullTime
BatchID string
}
type Session struct {

View File

@ -0,0 +1,90 @@
// Code generated by sqlc. DO NOT EDIT.
// versions:
// sqlc v1.31.1
// source: pr_reviews.sql
package gen
import (
"context"
"time"
)
const deletePRReviews = `-- name: DeletePRReviews :exec
DELETE FROM pr_reviews WHERE pr_url = ?
`
func (q *Queries) DeletePRReviews(ctx context.Context, prUrl string) error {
_, err := q.db.ExecContext(ctx, deletePRReviews, prUrl)
return err
}
const listPRReviews = `-- name: ListPRReviews :many
SELECT pr_url, review_id, author, state, url, is_bot, submitted_at
FROM pr_reviews WHERE pr_url = ? ORDER BY submitted_at, review_id
`
func (q *Queries) ListPRReviews(ctx context.Context, prUrl string) ([]PRReview, error) {
rows, err := q.db.QueryContext(ctx, listPRReviews, prUrl)
if err != nil {
return nil, err
}
defer rows.Close()
items := []PRReview{}
for rows.Next() {
var i PRReview
if err := rows.Scan(
&i.PRURL,
&i.ReviewID,
&i.Author,
&i.State,
&i.URL,
&i.IsBot,
&i.SubmittedAt,
); err != nil {
return nil, err
}
items = append(items, i)
}
if err := rows.Close(); err != nil {
return nil, err
}
if err := rows.Err(); err != nil {
return nil, err
}
return items, nil
}
const upsertPRReview = `-- name: UpsertPRReview :exec
INSERT INTO pr_reviews (pr_url, review_id, author, state, url, is_bot, submitted_at)
VALUES (?, ?, ?, ?, ?, ?, ?)
ON CONFLICT (pr_url, review_id) DO UPDATE SET
author = excluded.author,
state = excluded.state,
url = excluded.url,
is_bot = excluded.is_bot,
submitted_at = excluded.submitted_at
`
type UpsertPRReviewParams struct {
PRURL string
ReviewID string
Author string
State string
URL string
IsBot int64
SubmittedAt time.Time
}
func (q *Queries) UpsertPRReview(ctx context.Context, arg UpsertPRReviewParams) error {
_, err := q.db.ExecContext(ctx, upsertPRReview,
arg.PRURL,
arg.ReviewID,
arg.Author,
arg.State,
arg.URL,
arg.IsBot,
arg.SubmittedAt,
)
return err
}

View File

@ -35,7 +35,7 @@ func (q *Queries) GetReviewBySession(ctx context.Context, sessionID domain.Sessi
}
const getReviewRun = `-- name: GetReviewRun :one
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at, batch_id
FROM review_run WHERE id = ?
`
@ -55,22 +55,24 @@ func (q *Queries) GetReviewRun(ctx context.Context, id string) (ReviewRun, error
&i.CreatedAt,
&i.GithubReviewID,
&i.DeliveredAt,
&i.BatchID,
)
return i, err
}
const getReviewRunBySessionAndSHA = `-- name: GetReviewRunBySessionAndSHA :one
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at
FROM review_run WHERE session_id = ? AND target_sha = ? ORDER BY created_at DESC LIMIT 1
const getReviewRunBySessionPRAndSHA = `-- name: GetReviewRunBySessionPRAndSHA :one
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at, batch_id
FROM review_run WHERE session_id = ? AND pr_url = ? AND target_sha = ? ORDER BY created_at DESC LIMIT 1
`
type GetReviewRunBySessionAndSHAParams struct {
type GetReviewRunBySessionPRAndSHAParams struct {
SessionID domain.SessionID
PRURL string
TargetSha string
}
func (q *Queries) GetReviewRunBySessionAndSHA(ctx context.Context, arg GetReviewRunBySessionAndSHAParams) (ReviewRun, error) {
row := q.db.QueryRowContext(ctx, getReviewRunBySessionAndSHA, arg.SessionID, arg.TargetSha)
func (q *Queries) GetReviewRunBySessionPRAndSHA(ctx context.Context, arg GetReviewRunBySessionPRAndSHAParams) (ReviewRun, error) {
row := q.db.QueryRowContext(ctx, getReviewRunBySessionPRAndSHA, arg.SessionID, arg.PRURL, arg.TargetSha)
var i ReviewRun
err := row.Scan(
&i.ID,
@ -85,19 +87,21 @@ func (q *Queries) GetReviewRunBySessionAndSHA(ctx context.Context, arg GetReview
&i.CreatedAt,
&i.GithubReviewID,
&i.DeliveredAt,
&i.BatchID,
)
return i, err
}
const insertReviewRun = `-- name: InsertReviewRun :exec
INSERT INTO review_run (id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, github_review_id, created_at)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
INSERT INTO review_run (id, review_id, session_id, batch_id, harness, pr_url, target_sha, status, verdict, body, github_review_id, created_at)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
`
type InsertReviewRunParams struct {
ID string
ReviewID string
SessionID domain.SessionID
BatchID string
Harness domain.ReviewerHarness
PRURL string
TargetSha string
@ -113,6 +117,7 @@ func (q *Queries) InsertReviewRun(ctx context.Context, arg InsertReviewRunParams
arg.ID,
arg.ReviewID,
arg.SessionID,
arg.BatchID,
arg.Harness,
arg.PRURL,
arg.TargetSha,
@ -125,8 +130,55 @@ func (q *Queries) InsertReviewRun(ctx context.Context, arg InsertReviewRunParams
return err
}
const listReviewRunsByBatch = `-- name: ListReviewRunsByBatch :many
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at, batch_id
FROM review_run WHERE session_id = ? AND batch_id = ? ORDER BY created_at ASC, id ASC
`
type ListReviewRunsByBatchParams struct {
SessionID domain.SessionID
BatchID string
}
func (q *Queries) ListReviewRunsByBatch(ctx context.Context, arg ListReviewRunsByBatchParams) ([]ReviewRun, error) {
rows, err := q.db.QueryContext(ctx, listReviewRunsByBatch, arg.SessionID, arg.BatchID)
if err != nil {
return nil, err
}
defer rows.Close()
items := []ReviewRun{}
for rows.Next() {
var i ReviewRun
if err := rows.Scan(
&i.ID,
&i.ReviewID,
&i.SessionID,
&i.Harness,
&i.PRURL,
&i.TargetSha,
&i.Status,
&i.Verdict,
&i.Body,
&i.CreatedAt,
&i.GithubReviewID,
&i.DeliveredAt,
&i.BatchID,
); err != nil {
return nil, err
}
items = append(items, i)
}
if err := rows.Close(); err != nil {
return nil, err
}
if err := rows.Err(); err != nil {
return nil, err
}
return items, nil
}
const listReviewRunsBySession = `-- name: ListReviewRunsBySession :many
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at, batch_id
FROM review_run WHERE session_id = ? ORDER BY created_at DESC
`
@ -152,6 +204,7 @@ func (q *Queries) ListReviewRunsBySession(ctx context.Context, sessionID domain.
&i.CreatedAt,
&i.GithubReviewID,
&i.DeliveredAt,
&i.BatchID,
); err != nil {
return nil, err
}
@ -201,17 +254,23 @@ func (q *Queries) SupersedeReviewRun(ctx context.Context, arg SupersedeReviewRun
}
const supersedeStaleRunningReviewRuns = `-- name: SupersedeStaleRunningReviewRuns :execrows
UPDATE review_run SET status = 'failed', body = ? WHERE session_id = ? AND target_sha != ? AND status = 'running' AND verdict = ''
UPDATE review_run SET status = 'failed', body = ? WHERE session_id = ? AND pr_url = ? AND target_sha != ? AND status = 'running' AND verdict = ''
`
type SupersedeStaleRunningReviewRunsParams struct {
Body string
SessionID domain.SessionID
PRURL string
TargetSha string
}
func (q *Queries) SupersedeStaleRunningReviewRuns(ctx context.Context, arg SupersedeStaleRunningReviewRunsParams) (int64, error) {
result, err := q.db.ExecContext(ctx, supersedeStaleRunningReviewRuns, arg.Body, arg.SessionID, arg.TargetSha)
result, err := q.db.ExecContext(ctx, supersedeStaleRunningReviewRuns,
arg.Body,
arg.SessionID,
arg.PRURL,
arg.TargetSha,
)
if err != nil {
return 0, err
}

View File

@ -0,0 +1,22 @@
-- Summary: persist provider review submissions so the UI can link to review summaries.
-- +goose Up
-- +goose StatementBegin
CREATE TABLE pr_reviews (
pr_url TEXT NOT NULL REFERENCES pr (url) ON DELETE CASCADE,
review_id TEXT NOT NULL,
author TEXT NOT NULL DEFAULT '',
state TEXT NOT NULL DEFAULT 'none'
CHECK (state IN ('none', 'approved', 'changes_requested', 'review_required')),
url TEXT NOT NULL DEFAULT '',
is_bot INTEGER NOT NULL DEFAULT 0,
submitted_at TIMESTAMP NOT NULL,
PRIMARY KEY (pr_url, review_id)
);
CREATE INDEX idx_pr_reviews_lookup ON pr_reviews (pr_url, submitted_at);
-- +goose StatementEnd
-- +goose Down
-- +goose StatementBegin
DROP INDEX IF EXISTS idx_pr_reviews_lookup;
DROP TABLE IF EXISTS pr_reviews;
-- +goose StatementEnd

View File

@ -0,0 +1,88 @@
-- Review runs are PR-scoped. Two PRs in one worker session can legitimately
-- share a head SHA, so the idempotency index must include pr_url. Completed
-- changes_requested runs are intentionally excluded so the same head can be
-- reviewed again after the worker applies feedback. Runs created by one trigger
-- also share batch_id so one reviewer CLI submission can notify the worker about
-- multiple PRs at once.
-- +goose Up
-- +goose StatementBegin
ALTER TABLE review_run ADD COLUMN batch_id TEXT NOT NULL DEFAULT '';
-- +goose StatementEnd
-- +goose StatementBegin
DROP INDEX idx_review_run_session_sha;
-- +goose StatementEnd
-- +goose StatementBegin
DELETE FROM review_run
WHERE target_sha != ''
AND pr_url != ''
AND status != 'failed'
AND verdict != 'changes_requested'
AND rowid NOT IN (
SELECT rowid FROM (
SELECT rowid,
ROW_NUMBER() OVER (
PARTITION BY session_id, pr_url, target_sha
ORDER BY CASE status WHEN 'complete' THEN 0 WHEN 'delivered' THEN 0 WHEN 'running' THEN 1 ELSE 2 END,
created_at DESC,
rowid DESC
) AS rn
FROM review_run
WHERE target_sha != '' AND pr_url != '' AND status != 'failed' AND verdict != 'changes_requested'
)
WHERE rn = 1
);
-- +goose StatementEnd
-- +goose StatementBegin
CREATE UNIQUE INDEX idx_review_run_session_pr_sha
ON review_run (session_id, pr_url, target_sha)
WHERE target_sha != '' AND status != 'failed' AND verdict != 'changes_requested';
-- +goose StatementEnd
-- +goose StatementBegin
CREATE INDEX idx_review_run_session_batch
ON review_run (session_id, batch_id, created_at)
WHERE batch_id != '';
-- +goose StatementEnd
-- +goose Down
-- +goose StatementBegin
DROP INDEX idx_review_run_session_batch;
-- +goose StatementEnd
-- +goose StatementBegin
DROP INDEX idx_review_run_session_pr_sha;
-- +goose StatementEnd
-- +goose StatementBegin
DELETE FROM review_run
WHERE target_sha != ''
AND status != 'failed'
AND rowid NOT IN (
SELECT rowid FROM (
SELECT rowid,
ROW_NUMBER() OVER (
PARTITION BY session_id, target_sha
ORDER BY CASE status WHEN 'complete' THEN 0 WHEN 'delivered' THEN 0 WHEN 'running' THEN 1 ELSE 2 END,
created_at DESC,
rowid DESC
) AS rn
FROM review_run
WHERE target_sha != '' AND status != 'failed'
)
WHERE rn = 1
);
-- +goose StatementEnd
-- +goose StatementBegin
CREATE UNIQUE INDEX idx_review_run_session_sha
ON review_run (session_id, target_sha)
WHERE target_sha != '' AND status != 'failed';
-- +goose StatementEnd
-- +goose StatementBegin
ALTER TABLE review_run DROP COLUMN batch_id;
-- +goose StatementEnd

View File

@ -0,0 +1,16 @@
-- name: UpsertPRReview :exec
INSERT INTO pr_reviews (pr_url, review_id, author, state, url, is_bot, submitted_at)
VALUES (?, ?, ?, ?, ?, ?, ?)
ON CONFLICT (pr_url, review_id) DO UPDATE SET
author = excluded.author,
state = excluded.state,
url = excluded.url,
is_bot = excluded.is_bot,
submitted_at = excluded.submitted_at;
-- name: DeletePRReviews :exec
DELETE FROM pr_reviews WHERE pr_url = ?;
-- name: ListPRReviews :many
SELECT pr_url, review_id, author, state, url, is_bot, submitted_at
FROM pr_reviews WHERE pr_url = ? ORDER BY submitted_at, review_id;

View File

@ -12,8 +12,8 @@ SELECT id, session_id, project_id, harness, pr_url, reviewer_handle_id, created_
FROM review WHERE session_id = ?;
-- name: InsertReviewRun :exec
INSERT INTO review_run (id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, github_review_id, created_at)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?);
INSERT INTO review_run (id, review_id, session_id, batch_id, harness, pr_url, target_sha, status, verdict, body, github_review_id, created_at)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?);
-- name: UpdateReviewRunResult :execrows
UPDATE review_run SET status = ?, verdict = ?, body = ?, github_review_id = ? WHERE id = ? AND status = 'running';
@ -22,19 +22,23 @@ UPDATE review_run SET status = ?, verdict = ?, body = ?, github_review_id = ? WH
UPDATE review_run SET status = 'failed', body = ? WHERE id = ? AND verdict = '' AND status != 'failed';
-- name: SupersedeStaleRunningReviewRuns :execrows
UPDATE review_run SET status = 'failed', body = ? WHERE session_id = ? AND target_sha != ? AND status = 'running' AND verdict = '';
UPDATE review_run SET status = 'failed', body = ? WHERE session_id = ? AND pr_url = ? AND target_sha != ? AND status = 'running' AND verdict = '';
-- name: MarkReviewRunDelivered :execrows
UPDATE review_run SET status = 'delivered', delivered_at = ? WHERE id = ? AND status = 'complete' AND delivered_at IS NULL;
-- name: GetReviewRun :one
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at, batch_id
FROM review_run WHERE id = ?;
-- name: GetReviewRunBySessionAndSHA :one
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at
FROM review_run WHERE session_id = ? AND target_sha = ? ORDER BY created_at DESC LIMIT 1;
-- name: GetReviewRunBySessionPRAndSHA :one
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at, batch_id
FROM review_run WHERE session_id = ? AND pr_url = ? AND target_sha = ? ORDER BY created_at DESC LIMIT 1;
-- name: ListReviewRunsBySession :many
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at, batch_id
FROM review_run WHERE session_id = ? ORDER BY created_at DESC;
-- name: ListReviewRunsByBatch :many
SELECT id, review_id, session_id, harness, pr_url, target_sha, status, verdict, body, created_at, github_review_id, delivered_at, batch_id
FROM review_run WHERE session_id = ? AND batch_id = ? ORDER BY created_at ASC, id ASC;

View File

@ -64,17 +64,17 @@ func TestPRReviewThreadsCDC_EmitsOnInsertAndResolvedTransition(t *testing.T) {
now := time.Now().UTC().Truncate(time.Second)
pr := domain.PullRequest{URL: "https://example/pr/9", SessionID: rec.ID, Number: 9, UpdatedAt: now}
if err := s.WriteSCMObservation(ctx, pr, nil, []domain.PullRequestReviewThread{{
if err := s.WriteSCMObservation(ctx, pr, nil, nil, []domain.PullRequestReviewThread{{
ThreadID: "t1", Path: "main.go", Line: 7, IsBot: true, SemanticHash: "v1", UpdatedAt: now,
}}, nil, ports.ReviewWriteReplace); err != nil {
t.Fatal(err)
}
if err := s.WriteSCMObservation(ctx, pr, nil, []domain.PullRequestReviewThread{{
if err := s.WriteSCMObservation(ctx, pr, nil, nil, []domain.PullRequestReviewThread{{
ThreadID: "t1", Path: "main.go", Line: 8, Resolved: true, IsBot: true, SemanticHash: "v2", UpdatedAt: now.Add(time.Second),
}}, nil, ports.ReviewWriteMerge); err != nil {
t.Fatal(err)
}
if err := s.WriteSCMObservation(ctx, pr, nil, []domain.PullRequestReviewThread{{
if err := s.WriteSCMObservation(ctx, pr, nil, nil, []domain.PullRequestReviewThread{{
ThreadID: "t1", Path: "main.go", Line: 9, Resolved: true, IsBot: true, SemanticHash: "v3", UpdatedAt: now.Add(2 * time.Second),
}}, nil, ports.ReviewWriteMerge); err != nil {
t.Fatal(err)
@ -138,7 +138,7 @@ func TestPRReviewThreadsCDC_EmitsResolvedOnReplacePoll(t *testing.T) {
// First poll: seed via Replace (no Partial pagination). Same shape as the
// real GitHub provider path when the review-thread listing fits in one page.
if err := s.WriteSCMObservation(ctx, pr, nil, []domain.PullRequestReviewThread{{
if err := s.WriteSCMObservation(ctx, pr, nil, nil, []domain.PullRequestReviewThread{{
ThreadID: "t1", Path: "main.go", Line: 7, IsBot: true, SemanticHash: "v1", UpdatedAt: now,
}}, nil, ports.ReviewWriteReplace); err != nil {
t.Fatal(err)
@ -148,7 +148,7 @@ func TestPRReviewThreadsCDC_EmitsResolvedOnReplacePoll(t *testing.T) {
// On the buggy code this fired the INSERT trigger again (because the
// DELETE-all removed the row first) and the UPDATE trigger never saw the
// resolved transition.
if err := s.WriteSCMObservation(ctx, pr, nil, []domain.PullRequestReviewThread{{
if err := s.WriteSCMObservation(ctx, pr, nil, nil, []domain.PullRequestReviewThread{{
ThreadID: "t1", Path: "main.go", Line: 7, Resolved: true, IsBot: true, SemanticHash: "v2", UpdatedAt: now.Add(time.Second),
}}, nil, ports.ReviewWriteReplace); err != nil {
t.Fatal(err)
@ -197,13 +197,13 @@ func TestPRReviewThreadsReplace_PrunesOrphansWithoutReinserting(t *testing.T) {
now := time.Now().UTC().Truncate(time.Second)
pr := domain.PullRequest{URL: "https://example/pr/56", SessionID: rec.ID, Number: 56, UpdatedAt: now}
if err := s.WriteSCMObservation(ctx, pr, nil, []domain.PullRequestReviewThread{
if err := s.WriteSCMObservation(ctx, pr, nil, nil, []domain.PullRequestReviewThread{
{ThreadID: "keep", Path: "a.go", Line: 1, IsBot: true, SemanticHash: "k1", UpdatedAt: now},
{ThreadID: "drop", Path: "b.go", Line: 2, IsBot: true, SemanticHash: "d1", UpdatedAt: now},
}, nil, ports.ReviewWriteReplace); err != nil {
t.Fatal(err)
}
if err := s.WriteSCMObservation(ctx, pr, nil, []domain.PullRequestReviewThread{
if err := s.WriteSCMObservation(ctx, pr, nil, nil, []domain.PullRequestReviewThread{
{ThreadID: "keep", Path: "a.go", Line: 1, Resolved: true, IsBot: true, SemanticHash: "k2", UpdatedAt: now.Add(time.Second)},
}, nil, ports.ReviewWriteReplace); err != nil {
t.Fatal(err)
@ -283,7 +283,7 @@ func TestClaimPR_CreatesMovesAndGuardsActiveOwner(t *testing.T) {
url := "https://github.com/acme/repo/pull/42"
pr := domain.PullRequest{URL: url, SessionID: first.ID, Number: 42, CI: domain.CIPassing, Mergeability: domain.MergeMergeable, UpdatedAt: time.Now().UTC()}
out, err := s.ClaimPR(ctx, pr, nil, nil, nil, ports.ReviewWritePreserve, true)
out, err := s.ClaimPR(ctx, pr, nil, nil, nil, nil, ports.ReviewWritePreserve, true)
if err != nil {
t.Fatalf("initial claim: %v", err)
}
@ -296,7 +296,7 @@ func TestClaimPR_CreatesMovesAndGuardsActiveOwner(t *testing.T) {
}
pr.SessionID = second.ID
if _, err := s.ClaimPR(ctx, pr, nil, nil, nil, ports.ReviewWritePreserve, false); !errors.Is(err, ports.ErrPRClaimedByActiveSession) {
if _, err := s.ClaimPR(ctx, pr, nil, nil, nil, nil, ports.ReviewWritePreserve, false); !errors.Is(err, ports.ErrPRClaimedByActiveSession) {
t.Fatalf("no-takeover err = %v, want ErrPRClaimedByActiveSession", err)
}
got, _, _ = s.GetPR(ctx, url)
@ -304,7 +304,7 @@ func TestClaimPR_CreatesMovesAndGuardsActiveOwner(t *testing.T) {
t.Fatalf("active-owner refusal moved row to %s", got.SessionID)
}
out, err = s.ClaimPR(ctx, pr, nil, nil, nil, ports.ReviewWritePreserve, true)
out, err = s.ClaimPR(ctx, pr, nil, nil, nil, nil, ports.ReviewWritePreserve, true)
if err != nil {
t.Fatalf("takeover: %v", err)
}
@ -333,7 +333,7 @@ func TestClaimPRCreatedCDCUsesClaimReviewDecision(t *testing.T) {
Review: domain.ReviewChangesRequest,
UpdatedAt: time.Now().UTC(),
}
if _, err := s.ClaimPR(ctx, pr, nil, nil, nil, ports.ReviewWritePreserve, true); err != nil {
if _, err := s.ClaimPR(ctx, pr, nil, nil, nil, nil, ports.ReviewWritePreserve, true); err != nil {
t.Fatal(err)
}
@ -361,7 +361,7 @@ func TestClaimPR_TakesOverTerminatedOwnerAndEmitsSessionChangedCDC(t *testing.T)
second, _ := s.CreateSession(ctx, sampleRecord("mer"))
url := "https://github.com/acme/repo/pull/99"
pr := domain.PullRequest{URL: url, SessionID: first.ID, Number: 99, CI: domain.CIPassing, UpdatedAt: time.Now().UTC()}
if _, err := s.ClaimPR(ctx, pr, nil, nil, nil, ports.ReviewWritePreserve, true); err != nil {
if _, err := s.ClaimPR(ctx, pr, nil, nil, nil, nil, ports.ReviewWritePreserve, true); err != nil {
t.Fatal(err)
}
first.IsTerminated = true
@ -371,7 +371,7 @@ func TestClaimPR_TakesOverTerminatedOwnerAndEmitsSessionChangedCDC(t *testing.T)
}
pr.SessionID = second.ID
out, err := s.ClaimPR(ctx, pr, nil, nil, nil, ports.ReviewWritePreserve, false)
out, err := s.ClaimPR(ctx, pr, nil, nil, nil, nil, ports.ReviewWritePreserve, false)
if err != nil {
t.Fatalf("terminated takeover: %v", err)
}

View File

@ -28,7 +28,7 @@ func TestListPRFactsForSessionProjectsAllPRsNewestFirst(t *testing.T) {
// historical PR. Distinct updated_at so newest-first ordering is observable.
write := func(pr domain.PullRequest) {
t.Helper()
if err := s.WriteSCMObservation(ctx, pr, nil, nil, nil, ports.ReviewWritePreserve); err != nil {
if err := s.WriteSCMObservation(ctx, pr, nil, nil, nil, nil, ports.ReviewWritePreserve); err != nil {
t.Fatalf("write %s: %v", pr.URL, err)
}
}

View File

@ -36,7 +36,7 @@ var (
// not touch pr_review_threads: those rows are owned by WriteSCMObservation's
// slower review-thread refresh path.
func (s *Store) WritePR(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, comments []domain.PullRequestComment) error {
return s.writePR(ctx, pr, checks, nil, comments, ports.ReviewWritePreserve, true)
return s.writePR(ctx, pr, checks, nil, nil, comments, ports.ReviewWritePreserve, true)
}
// WriteSCMObservation persists a provider-neutral SCM observation in one write
@ -44,14 +44,14 @@ func (s *Store) WritePR(ctx context.Context, pr domain.PullRequest, checks []dom
// and comments are preserved, replaced, or merged according to reviewMode
// because review polling runs at a slower and sometimes intentionally bounded
// cadence than metadata/CI polling.
func (s *Store) WriteSCMObservation(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode) error {
return s.writePR(ctx, pr, checks, threads, comments, reviewMode, false)
func (s *Store) WriteSCMObservation(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, reviews []domain.PullRequestReview, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode) error {
return s.writePR(ctx, pr, checks, reviews, threads, comments, reviewMode, false)
}
// ClaimPR moves (or creates) a PR row to pr.SessionID and applies the live SCM
// observation in the same transaction. The session_id update is what fires the
// pr_session_changed CDC trigger added in migration 0005.
func (s *Store) ClaimPR(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode, allowActiveTakeover bool) (ports.ClaimOutcome, error) {
func (s *Store) ClaimPR(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, reviews []domain.PullRequestReview, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode, allowActiveTakeover bool) (ports.ClaimOutcome, error) {
s.writeMu.Lock()
defer s.writeMu.Unlock()
var outcome ports.ClaimOutcome
@ -73,20 +73,20 @@ func (s *Store) ClaimPR(ctx context.Context, pr domain.PullRequest, checks []dom
}); err != nil {
return err
}
return writePRRows(ctx, q, pr, checks, threads, comments, reviewMode, false, false)
return writePRRows(ctx, q, pr, checks, reviews, threads, comments, reviewMode, false, false)
})
return outcome, err
}
func (s *Store) writePR(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode, replaceLegacyComments bool) error {
func (s *Store) writePR(ctx context.Context, pr domain.PullRequest, checks []domain.PullRequestCheck, reviews []domain.PullRequestReview, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode, replaceLegacyComments bool) error {
s.writeMu.Lock()
defer s.writeMu.Unlock()
return s.inTx(ctx, "write pr observation", func(q *gen.Queries) error {
return writePRRows(ctx, q, pr, checks, threads, comments, reviewMode, replaceLegacyComments, true)
return writePRRows(ctx, q, pr, checks, reviews, threads, comments, reviewMode, replaceLegacyComments, true)
})
}
func writePRRows(ctx context.Context, q *gen.Queries, pr domain.PullRequest, checks []domain.PullRequestCheck, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode, replaceLegacyComments, rejectReassignment bool) error {
func writePRRows(ctx context.Context, q *gen.Queries, pr domain.PullRequest, checks []domain.PullRequestCheck, reviews []domain.PullRequestReview, threads []domain.PullRequestReviewThread, comments []domain.PullRequestComment, reviewMode ports.ReviewWriteMode, replaceLegacyComments, rejectReassignment bool) error {
if rejectReassignment {
existing, err := q.GetPR(ctx, pr.URL)
if err != nil && !errors.Is(err, sql.ErrNoRows) {
@ -114,11 +114,21 @@ func writePRRows(ctx context.Context, q *gen.Queries, pr domain.PullRequest, che
if err := q.DeletePRComments(ctx, pr.URL); err != nil {
return err
}
if err := q.DeletePRReviews(ctx, pr.URL); err != nil {
return err
}
} else if replaceLegacyComments {
if err := q.DeleteLegacyPRComments(ctx, pr.URL); err != nil {
return err
}
}
if reviewMode == ports.ReviewWriteReplace || reviewMode == ports.ReviewWriteMerge {
for _, review := range reviews {
if err := q.UpsertPRReview(ctx, genReviewParams(pr.URL, review)); err != nil {
return fmt.Errorf("review %q: %w", review.ID, err)
}
}
}
if reviewMode == ports.ReviewWriteReplace || reviewMode == ports.ReviewWriteMerge {
for _, th := range threads {
if err := q.UpsertPRReviewThread(ctx, genReviewThreadParams(pr.URL, th)); err != nil {
@ -282,6 +292,19 @@ func (s *Store) ListPRReviewThreads(ctx context.Context, prURL string) ([]domain
return out, nil
}
// ListPRReviews returns a PR's submitted reviews, oldest first.
func (s *Store) ListPRReviews(ctx context.Context, prURL string) ([]domain.PullRequestReview, error) {
rows, err := s.qr.ListPRReviews(ctx, prURL)
if err != nil {
return nil, fmt.Errorf("list pr reviews %s: %w", prURL, err)
}
out := make([]domain.PullRequestReview, 0, len(rows))
for _, review := range rows {
out = append(out, reviewFromGen(review))
}
return out, nil
}
// ---- domain <-> gen mapping ----
// prState collapses the PR's bools into the single pr.state column value.
@ -481,6 +504,33 @@ func reviewThreadFromGen(th gen.PRReviewThread) domain.PullRequestReviewThread {
}
}
func genReviewParams(prURL string, review domain.PullRequestReview) gen.UpsertPRReviewParams {
id := review.ID
if id == "" {
id = review.URL
}
return gen.UpsertPRReviewParams{
PRURL: prURL,
ReviewID: id,
Author: review.Author,
State: string(reviewOrDefault(review.State)),
URL: review.URL,
IsBot: boolInt(review.IsBot),
SubmittedAt: review.SubmittedAt,
}
}
func reviewFromGen(review gen.PRReview) domain.PullRequestReview {
return domain.PullRequestReview{
ID: review.ReviewID,
Author: review.Author,
State: domain.ReviewDecision(review.State),
URL: review.URL,
IsBot: review.IsBot != 0,
SubmittedAt: review.SubmittedAt,
}
}
func boolInt(v bool) int64 {
if v {
return 1

View File

@ -41,7 +41,7 @@ func (s *Store) GetReviewBySession(ctx context.Context, id domain.SessionID) (do
}
// InsertReviewRun records a new review pass. A unique-constraint hit on the
// (session_id, target_sha) index (migration 0013) is surfaced as the sentinel
// (session_id, pr_url, target_sha) index (migration 0020) is surfaced as the sentinel
// domain.ErrDuplicateReviewRun so the engine can fall back to the existing run.
func (s *Store) InsertReviewRun(ctx context.Context, r domain.ReviewRun) error {
s.writeMu.Lock()
@ -50,6 +50,7 @@ func (s *Store) InsertReviewRun(ctx context.Context, r domain.ReviewRun) error {
ID: r.ID,
ReviewID: r.ReviewID,
SessionID: r.SessionID,
BatchID: r.BatchID,
Harness: r.Harness,
PRURL: r.PRURL,
TargetSha: r.TargetSHA,
@ -60,7 +61,7 @@ func (s *Store) InsertReviewRun(ctx context.Context, r domain.ReviewRun) error {
CreatedAt: r.CreatedAt,
})
if isSQLiteUnique(err) {
return fmt.Errorf("insert review run for session %s sha %s: %w", r.SessionID, r.TargetSHA, domain.ErrDuplicateReviewRun)
return fmt.Errorf("insert review run for session %s pr %s sha %s: %w", r.SessionID, r.PRURL, r.TargetSHA, domain.ErrDuplicateReviewRun)
}
return err
}
@ -100,12 +101,13 @@ func (s *Store) SupersedeReviewRun(ctx context.Context, id, body string) (bool,
// SupersedeStaleRunningReviewRuns marks older running unverdicted passes for a
// worker failed before starting a review for a newer commit.
func (s *Store) SupersedeStaleRunningReviewRuns(ctx context.Context, sessionID domain.SessionID, targetSHA, body string) (int64, error) {
func (s *Store) SupersedeStaleRunningReviewRuns(ctx context.Context, sessionID domain.SessionID, prURL, targetSHA, body string) (int64, error) {
s.writeMu.Lock()
defer s.writeMu.Unlock()
return s.qw.SupersedeStaleRunningReviewRuns(ctx, gen.SupersedeStaleRunningReviewRunsParams{
Body: body,
SessionID: sessionID,
PRURL: prURL,
TargetSha: targetSHA,
})
}
@ -137,16 +139,17 @@ func (s *Store) GetReviewRun(ctx context.Context, id string) (domain.ReviewRun,
return reviewRunFromRow(row), true, nil
}
// GetReviewRunBySessionAndSHA returns the most recent review pass for a worker
// session at a specific commit, ok=false if none. It lets a repeat trigger for
// the same PR head short-circuit to the existing run.
func (s *Store) GetReviewRunBySessionAndSHA(ctx context.Context, id domain.SessionID, targetSHA string) (domain.ReviewRun, bool, error) {
row, err := s.qr.GetReviewRunBySessionAndSHA(ctx, gen.GetReviewRunBySessionAndSHAParams{SessionID: id, TargetSha: targetSHA})
// GetReviewRunBySessionPRAndSHA returns the most recent review pass for a
// worker session, PR, and commit, ok=false if none. It lets a repeat trigger for
// the same PR head short-circuit to the existing run without colliding with
// another PR that happens to share the same head commit.
func (s *Store) GetReviewRunBySessionPRAndSHA(ctx context.Context, id domain.SessionID, prURL, targetSHA string) (domain.ReviewRun, bool, error) {
row, err := s.qr.GetReviewRunBySessionPRAndSHA(ctx, gen.GetReviewRunBySessionPRAndSHAParams{SessionID: id, PRURL: prURL, TargetSha: targetSHA})
if errors.Is(err, sql.ErrNoRows) {
return domain.ReviewRun{}, false, nil
}
if err != nil {
return domain.ReviewRun{}, false, fmt.Errorf("get review run for session %s sha %s: %w", id, targetSHA, err)
return domain.ReviewRun{}, false, fmt.Errorf("get review run for session %s pr %s sha %s: %w", id, prURL, targetSHA, err)
}
return reviewRunFromRow(row), true, nil
}
@ -164,6 +167,19 @@ func (s *Store) ListReviewRunsBySession(ctx context.Context, id domain.SessionID
return out, nil
}
// ListReviewRunsByBatch returns all passes in one trigger-created batch, oldest first.
func (s *Store) ListReviewRunsByBatch(ctx context.Context, id domain.SessionID, batchID string) ([]domain.ReviewRun, error) {
rows, err := s.qr.ListReviewRunsByBatch(ctx, gen.ListReviewRunsByBatchParams{SessionID: id, BatchID: batchID})
if err != nil {
return nil, fmt.Errorf("list review runs for session %s batch %s: %w", id, batchID, err)
}
out := make([]domain.ReviewRun, 0, len(rows))
for _, row := range rows {
out = append(out, reviewRunFromRow(row))
}
return out, nil
}
func reviewFromRow(r gen.Review) domain.Review {
return domain.Review{
ID: r.ID,
@ -187,6 +203,7 @@ func reviewRunFromRow(r gen.ReviewRun) domain.ReviewRun {
ID: r.ID,
ReviewID: r.ReviewID,
SessionID: r.SessionID,
BatchID: r.BatchID,
Harness: r.Harness,
PRURL: r.PRURL,
TargetSHA: r.TargetSha,

View File

@ -9,7 +9,7 @@ import (
"github.com/aoagents/agent-orchestrator/backend/internal/domain"
)
func TestInsertReviewRunDuplicateSHAMapsToSentinel(t *testing.T) {
func TestInsertReviewRunDuplicatePRSHAMapsToSentinel(t *testing.T) {
s := newTestStore(t)
ctx := context.Background()
seedProject(t, s, "mer")
@ -26,21 +26,28 @@ func TestInsertReviewRunDuplicateSHAMapsToSentinel(t *testing.T) {
}
run := domain.ReviewRun{
ID: "run-1", ReviewID: "rev-1", SessionID: rec.ID, Harness: domain.ReviewerClaudeCode,
TargetSHA: "sha1", Status: domain.ReviewRunRunning, Verdict: domain.VerdictNone, CreatedAt: now,
PRURL: "https://example/pr/1", TargetSHA: "sha1", Status: domain.ReviewRunRunning, Verdict: domain.VerdictNone, CreatedAt: now,
}
if err := s.InsertReviewRun(ctx, run); err != nil {
t.Fatalf("first insert: %v", err)
}
// A second run for the same (session_id, target_sha) hits the partial unique
// index (migration 0013) and must surface as the sentinel so the engine can
// fall back to the existing run.
// A second run for the same (session_id, pr_url, target_sha) hits the
// partial unique index (migration 0020) and must surface as the sentinel so
// the engine can fall back to the existing run.
dup := run
dup.ID = "run-2"
if err := s.InsertReviewRun(ctx, dup); !errors.Is(err, domain.ErrDuplicateReviewRun) {
t.Fatalf("duplicate insert err = %v, want ErrDuplicateReviewRun", err)
}
otherPR := run
otherPR.ID = "run-other-pr"
otherPR.PRURL = "https://example/pr/2"
if err := s.InsertReviewRun(ctx, otherPR); err != nil {
t.Fatalf("same sha on different PR should insert: %v", err)
}
if ok, err := s.UpdateReviewRunResult(ctx, "run-1", domain.ReviewRunFailed, domain.VerdictNone, "claude: not found", ""); err != nil {
t.Fatalf("mark failed: %v", err)
} else if !ok {
@ -60,6 +67,42 @@ func TestInsertReviewRunDuplicateSHAMapsToSentinel(t *testing.T) {
}
}
func TestInsertReviewRunAllowsRerunAfterChangesRequested(t *testing.T) {
s := newTestStore(t)
ctx := context.Background()
seedProject(t, s, "mer")
rec, err := s.CreateSession(ctx, sampleRecord("mer"))
if err != nil {
t.Fatalf("create session: %v", err)
}
now := time.Now().UTC().Truncate(time.Second)
if err := s.UpsertReview(ctx, domain.Review{
ID: "rev-1", SessionID: rec.ID, ProjectID: rec.ProjectID,
Harness: domain.ReviewerClaudeCode, CreatedAt: now, UpdatedAt: now,
}); err != nil {
t.Fatalf("upsert review: %v", err)
}
run := domain.ReviewRun{
ID: "run-1", ReviewID: "rev-1", SessionID: rec.ID, Harness: domain.ReviewerClaudeCode,
PRURL: "https://example/pr/1", TargetSHA: "sha1", Status: domain.ReviewRunRunning, Verdict: domain.VerdictNone, CreatedAt: now,
}
if err := s.InsertReviewRun(ctx, run); err != nil {
t.Fatalf("first insert: %v", err)
}
if ok, err := s.UpdateReviewRunResult(ctx, "run-1", domain.ReviewRunComplete, domain.VerdictChangesRequested, "please fix", "rev-1"); err != nil {
t.Fatalf("mark changes requested: %v", err)
} else if !ok {
t.Fatal("mark changes requested: got ok=false")
}
rerun := run
rerun.ID = "run-2"
rerun.CreatedAt = now.Add(time.Second)
if err := s.InsertReviewRun(ctx, rerun); err != nil {
t.Fatalf("rerun after changes_requested insert: %v", err)
}
}
func TestReviewUpsertReusesRowAndRunRoundTrip(t *testing.T) {
s := newTestStore(t)
ctx := context.Background()
@ -102,7 +145,7 @@ func TestReviewUpsertReusesRowAndRunRoundTrip(t *testing.T) {
// A run inserts running and updates to complete/changes_requested.
if err := s.InsertReviewRun(ctx, domain.ReviewRun{
ID: "run-1", ReviewID: got.ID, SessionID: rec.ID, Harness: domain.ReviewerHarness("greptile"),
ID: "run-1", ReviewID: got.ID, SessionID: rec.ID, BatchID: "batch-1", Harness: domain.ReviewerHarness("greptile"),
PRURL: got.PRURL, TargetSHA: "sha1", Status: domain.ReviewRunRunning, Verdict: domain.VerdictNone,
CreatedAt: now,
}); err != nil {
@ -118,18 +161,18 @@ func TestReviewUpsertReusesRowAndRunRoundTrip(t *testing.T) {
if err != nil || !ok {
t.Fatalf("get run: ok=%v err=%v", ok, err)
}
if gotRun.ID != "run-1" || gotRun.SessionID != rec.ID || gotRun.TargetSHA != "sha1" {
if gotRun.ID != "run-1" || gotRun.SessionID != rec.ID || gotRun.BatchID != "batch-1" || gotRun.TargetSHA != "sha1" {
t.Fatalf("get run = %+v", gotRun)
}
bySHA, ok, err := s.GetReviewRunBySessionAndSHA(ctx, rec.ID, "sha1")
bySHA, ok, err := s.GetReviewRunBySessionPRAndSHA(ctx, rec.ID, got.PRURL, "sha1")
if err != nil || !ok {
t.Fatalf("by sha: ok=%v err=%v", ok, err)
}
if bySHA.Status != domain.ReviewRunComplete || bySHA.Verdict != domain.VerdictChangesRequested || bySHA.Body != "please fix" || bySHA.GithubReviewID != "rev-987" {
t.Fatalf("run result not persisted: %+v", bySHA)
}
if _, ok, _ := s.GetReviewRunBySessionAndSHA(ctx, rec.ID, "other"); ok {
if _, ok, _ := s.GetReviewRunBySessionPRAndSHA(ctx, rec.ID, got.PRURL, "other"); ok {
t.Fatal("unexpected run for a different sha")
}
@ -140,6 +183,13 @@ func TestReviewUpsertReusesRowAndRunRoundTrip(t *testing.T) {
if len(runs) != 1 || runs[0].ID != "run-1" {
t.Fatalf("list runs = %+v", runs)
}
batchRuns, err := s.ListReviewRunsByBatch(ctx, rec.ID, "batch-1")
if err != nil {
t.Fatalf("list batch runs: %v", err)
}
if len(batchRuns) != 1 || batchRuns[0].ID != "run-1" || batchRuns[0].BatchID != "batch-1" {
t.Fatalf("batch runs = %+v", batchRuns)
}
if ok, err := s.UpdateReviewRunResult(ctx, "run-1", domain.ReviewRunComplete, domain.VerdictApproved, "again", ""); err != nil {
t.Fatalf("second update: %v", err)
@ -154,7 +204,7 @@ func TestReviewGettersMissing(t *testing.T) {
if _, ok, err := s.GetReviewBySession(ctx, "mer-1"); err != nil || ok {
t.Fatalf("missing review: ok=%v err=%v", ok, err)
}
if _, ok, err := s.GetReviewRunBySessionAndSHA(ctx, "mer-1", "sha1"); err != nil || ok {
if _, ok, err := s.GetReviewRunBySessionPRAndSHA(ctx, "mer-1", "pr1", "sha1"); err != nil || ok {
t.Fatalf("missing run: ok=%v err=%v", ok, err)
}
if _, ok, err := s.GetReviewRun(ctx, "run-missing"); err != nil || ok {

View File

@ -415,10 +415,11 @@ func TestWriteSCMObservationPersistsMetadataChecksReviewsAndComments(t *testing.
UpdatedAt: now, ObservedAt: now, CIObservedAt: now, ReviewObservedAt: now,
}
checks := []domain.PullRequestCheck{{Name: "build", CommitHash: "h1", Status: domain.PRCheckFailed, Conclusion: "failure", URL: "ci", Details: "99", LogTail: "boom", CreatedAt: now}}
reviews := []domain.PullRequestReview{{ID: "review-1", Author: "reviewer", State: domain.ReviewChangesRequest, URL: "https://github.com/o/r/pull/1#pullrequestreview-1", SubmittedAt: now}}
threads := []domain.PullRequestReviewThread{{ThreadID: "t1", Path: "main.go", Line: 7, SemanticHash: "th", UpdatedAt: now}}
comments := []domain.PullRequestComment{{ThreadID: "t1", ID: "c1", Author: "reviewer", File: "main.go", Line: 7, Body: "fix", URL: "comment", CreatedAt: now}}
if err := s.WriteSCMObservation(ctx, pr, checks, threads, comments, ports.ReviewWriteReplace); err != nil {
if err := s.WriteSCMObservation(ctx, pr, checks, reviews, threads, comments, ports.ReviewWriteReplace); err != nil {
t.Fatal(err)
}
got, ok, err := s.GetPR(ctx, pr.URL)
@ -436,6 +437,10 @@ func TestWriteSCMObservationPersistsMetadataChecksReviewsAndComments(t *testing.
if len(gotThreads) != 1 || gotThreads[0].ThreadID != "t1" || gotThreads[0].SemanticHash != "th" {
t.Fatalf("threads not persisted: %+v", gotThreads)
}
gotReviews, _ := s.ListPRReviews(ctx, pr.URL)
if len(gotReviews) != 1 || gotReviews[0].ID != "review-1" || gotReviews[0].URL != "https://github.com/o/r/pull/1#pullrequestreview-1" {
t.Fatalf("reviews not persisted: %+v", gotReviews)
}
gotComments, _ := s.ListPRComments(ctx, pr.URL)
if len(gotComments) != 1 || gotComments[0].ThreadID != "t1" || gotComments[0].URL != "comment" {
t.Fatalf("comments not persisted: %+v", gotComments)
@ -458,7 +463,7 @@ func TestWriteSCMObservationMergeUpdatesFetchedReviewWindow(t *testing.T) {
{ThreadID: "older", ID: "older-c1", Author: "ann", Body: "old", CreatedAt: now},
{ThreadID: "latest", ID: "latest-c1", Author: "bob", Body: "before", CreatedAt: now},
}
if err := s.WriteSCMObservation(ctx, pr, nil, initialThreads, initialComments, ports.ReviewWriteReplace); err != nil {
if err := s.WriteSCMObservation(ctx, pr, nil, nil, initialThreads, initialComments, ports.ReviewWriteReplace); err != nil {
t.Fatal(err)
}
@ -470,7 +475,7 @@ func TestWriteSCMObservationMergeUpdatesFetchedReviewWindow(t *testing.T) {
{ThreadID: "latest", ID: "latest-c2", Author: "bob", Body: "after", CreatedAt: now.Add(time.Second)},
{ThreadID: "new", ID: "new-c1", Author: "cat", Body: "new", CreatedAt: now.Add(time.Second)},
}
if err := s.WriteSCMObservation(ctx, pr, nil, mergedThreads, mergedComments, ports.ReviewWriteMerge); err != nil {
if err := s.WriteSCMObservation(ctx, pr, nil, nil, mergedThreads, mergedComments, ports.ReviewWriteMerge); err != nil {
t.Fatal(err)
}
@ -527,7 +532,7 @@ func TestWritePRPreservesSCMReviewThreads(t *testing.T) {
threads := []domain.PullRequestReviewThread{{ThreadID: "t1", Path: "main.go", Line: 7, SemanticHash: "thread-v1", UpdatedAt: now}}
comments := []domain.PullRequestComment{{ThreadID: "t1", ID: "c1", Author: "reviewer", Body: "scm", URL: "https://example/comment/c1", CreatedAt: now}}
if err := s.WriteSCMObservation(ctx, pr, nil, threads, comments, ports.ReviewWriteReplace); err != nil {
if err := s.WriteSCMObservation(ctx, pr, nil, nil, threads, comments, ports.ReviewWriteReplace); err != nil {
t.Fatal(err)
}
legacyComments := []domain.PullRequestComment{
@ -578,7 +583,7 @@ func TestWritePRPreservesSCMReviewThreads(t *testing.T) {
mergedThreads := []domain.PullRequestReviewThread{{ThreadID: "t1", Path: "main.go", Line: 8, Resolved: true, SemanticHash: "thread-v2", UpdatedAt: now.Add(2 * time.Second)}}
mergedComments := []domain.PullRequestComment{{ThreadID: "t1", ID: "c2", Author: "reviewer", Body: "updated", URL: "https://example/comment/c2", CreatedAt: now.Add(2 * time.Second)}}
if err := s.WriteSCMObservation(ctx, pr, nil, mergedThreads, mergedComments, ports.ReviewWriteMerge); err != nil {
if err := s.WriteSCMObservation(ctx, pr, nil, nil, mergedThreads, mergedComments, ports.ReviewWriteMerge); err != nil {
t.Fatal(err)
}
gotComments, err = s.ListPRComments(ctx, pr.URL)

View File

@ -595,7 +595,7 @@ export interface components {
};
ListReviewsResponse: {
reviewerHandleId: string;
reviews: components["schemas"]["ReviewRun"][];
reviews: components["schemas"]["PRReviewState"][];
};
ListSessionPRsResponse: {
prs: components["schemas"]["SessionPRSummary"][];
@ -648,6 +648,15 @@ export interface components {
projectId: string;
projectName?: string;
};
PRReviewState: {
latestRun?: components["schemas"]["ReviewRun"];
prNumber: number;
prUrl: string;
/** @enum {string} */
status: "needs_review" | "running" | "up_to_date" | "changes_requested" | "ineligible";
targetSha: string;
title: string;
};
Project: {
agent?: string;
config?: components["schemas"]["ProjectConfig"];
@ -714,6 +723,7 @@ export interface components {
sessionId: string;
};
ReviewRun: {
batchId: string;
body: string;
/** Format: date-time */
createdAt: string;
@ -732,6 +742,7 @@ export interface components {
ReviewRunResponse: {
review: components["schemas"]["ReviewRun"];
reviewerHandleId: string;
reviews: components["schemas"]["ReviewRun"][];
};
RoleOverride: {
agent?: string;
@ -831,7 +842,9 @@ export interface components {
};
SessionPRUnresolvedReviewer: {
count: number;
isBot?: boolean;
links: components["schemas"]["SessionPRReviewCommentLink"][];
reviewUrl?: string;
reviewerId: string;
};
SessionPreviewResponse: {
@ -880,14 +893,30 @@ export interface components {
};
SubmitReviewInput: {
/** @description Review body recorded by AO. Required for changes_requested. */
body: string;
body?: string;
/** @description Id of the GitHub PR review the reviewer posted, if any. */
githubReviewId: string;
githubReviewId?: string;
/** @description Batched review results recorded by one reviewer CLI command. */
reviews?: components["schemas"]["SubmitReviewItem"][];
/** @description Review run id being completed. */
runId?: string;
/** @description Review verdict: approved or changes_requested. */
verdict?: string;
};
SubmitReviewItem: {
/** @description Review body recorded by AO. Required for changes_requested. */
body?: string;
/** @description Id of the GitHub PR review the reviewer posted, if any. */
githubReviewId?: string;
/** @description Review run id being completed. */
runId: string;
/** @description Review verdict: approved or changes_requested. */
verdict: string;
};
TriggerReviewResponse: {
reviewerHandleId: string;
reviews: components["schemas"]["PRReviewState"][];
};
WorkspaceRepo: {
name: string;
relativePath: string;
@ -2535,7 +2564,7 @@ export interface operations {
[name: string]: unknown;
};
content: {
"application/json": components["schemas"]["ReviewRunResponse"];
"application/json": components["schemas"]["TriggerReviewResponse"];
};
};
/** @description Created */
@ -2544,7 +2573,7 @@ export interface operations {
[name: string]: unknown;
};
content: {
"application/json": components["schemas"]["ReviewRunResponse"];
"application/json": components["schemas"]["TriggerReviewResponse"];
};
};
/** @description Not Found */

View File

@ -73,6 +73,100 @@ function respondWithProjectAndPRs() {
});
}
function respondWithAttentionPR() {
getMock.mockImplementation(async (url: string) => {
if (url === "/api/v1/projects") {
return { data: { projects: [{ id: "proj-1", name: "my-app", path: "/repo/my-app" }] }, error: undefined };
}
if (url === "/api/v1/sessions/sess-1/pr") {
return {
data: {
sessionId: "sess-1",
prs: [
{
url: "https://github.com/aoagents/ReverbCode/pull/278",
htmlUrl: "https://github.com/aoagents/ReverbCode/pull/278",
number: 278,
title: "fix the bug",
state: "open",
provider: "github",
repo: "aoagents/ReverbCode",
author: "worker",
sourceBranch: "fix/bug",
targetBranch: "main",
headSha: "abc123",
additions: 1,
deletions: 1,
changedFiles: 1,
ci: { state: "passing", failingChecks: [] },
review: {
decision: "changes_requested",
hasUnresolvedHumanComments: true,
unresolvedBy: [
{
reviewerId: "reviewer-a",
count: 1,
reviewUrl: "https://github.com/aoagents/ReverbCode/pull/278#pullrequestreview-1",
links: [
{
url: "https://github.com/aoagents/ReverbCode/pull/278#discussion_r1",
file: "main.go",
line: 12,
},
],
},
],
},
mergeability: {
state: "conflicting",
reasons: ["conflicts"],
prUrl: "https://github.com/aoagents/ReverbCode/pull/278",
conflictFiles: [],
},
updatedAt: "2026-06-10T16:15:04Z",
observedAt: "2026-06-10T16:15:04Z",
ciObservedAt: "2026-06-10T16:15:04Z",
reviewObservedAt: "2026-06-10T16:15:04Z",
},
],
},
error: undefined,
};
}
if (url === "/api/v1/sessions") {
return {
data: {
sessions: [
{
id: "sess-1",
projectId: "proj-1",
displayName: "fix the bug",
harness: "claude-code",
status: "changes_requested",
isTerminated: false,
updatedAt: "2026-06-10T16:15:04Z",
prs: [
{
number: 278,
state: "open",
url: "https://github.com/aoagents/ReverbCode/pull/278",
ci: "passing",
review: "changes_requested",
mergeability: "conflicting",
reviewComments: true,
updatedAt: "2026-06-10T16:15:04Z",
},
],
},
],
},
error: undefined,
};
}
throw new Error(`unexpected GET ${url}`);
});
}
function renderWithProviders(node: ReactNode) {
const queryClient = new QueryClient({ defaultOptions: { queries: { retry: false } } });
render(<QueryClientProvider client={queryClient}>{node}</QueryClientProvider>);
@ -93,6 +187,20 @@ describe("PR hydration for a normal project (#251)", () => {
expect(screen.queryByText("no PR yet")).not.toBeInTheDocument();
});
it("links Board attention states to PR fallback and merge conflict targets", async () => {
respondWithAttentionPR();
renderWithProviders(<SessionsBoard />);
expect(await screen.findByRole("link", { name: "PR" })).toHaveAttribute(
"href",
"https://github.com/aoagents/ReverbCode/pull/278",
);
expect(screen.getByRole("link", { name: "conflicts" })).toHaveAttribute(
"href",
"https://github.com/aoagents/ReverbCode/pull/278/conflicts",
);
});
it("lists every session PR on the PR page instead of being empty", async () => {
renderWithProviders(<PullRequestsPage />);

View File

@ -56,7 +56,7 @@ function renderWithQuery(children: ReactNode) {
return render(<QueryClientProvider client={client}>{children}</QueryClientProvider>);
}
function mockCommonGets(reviews: unknown[] = [], reviewerHandleId = "") {
function mockCommonGets(_unusedRuns: unknown[] = [], reviewerHandleId = "", reviews: unknown[] = []) {
getMock.mockImplementation(async (path: string) => {
if (path === "/api/v1/sessions/{sessionId}/reviews") {
return { data: { reviewerHandleId, reviews } };
@ -94,6 +94,24 @@ const approvedReview = {
createdAt: "2026-06-16T10:06:00Z",
};
const failedReview = {
...approvedReview,
id: "run-failed",
status: "failed",
verdict: "",
body: "reviewer crashed",
};
const reviewState = (n: number, status: string, targetSha = `sha-${n}`) => ({
prUrl: `https://example.com/pr/${n}`,
prNumber: n,
title: `Reviewable change ${n}`,
targetSha,
status,
latestRun:
status === "up_to_date" ? { ...approvedReview, prUrl: `https://example.com/pr/${n}`, targetSha } : undefined,
});
beforeEach(() => {
getMock.mockReset();
postMock.mockReset();
@ -155,12 +173,13 @@ describe("SessionInspector reviews tab", () => {
const openReviewsTab = async () => userEvent.click(screen.getByRole("tab", { name: /Reviews/ }));
it("triggers a review and opens the returned reviewer terminal", async () => {
mockCommonGets();
mockCommonGets([], "", [reviewState(3, "needs_review")]);
const runningReview = { ...approvedReview, status: "running", verdict: "", body: "" };
postMock.mockResolvedValue({
response: { status: 201 },
data: {
reviewerHandleId: "reviewer-pane",
review: { ...approvedReview, status: "running", verdict: "", body: "" },
reviews: [{ ...reviewState(3, "running"), latestRun: runningReview }],
},
});
const onOpenReviewerTerminal = vi.fn();
@ -180,11 +199,34 @@ describe("SessionInspector reviews tab", () => {
expect(onOpenReviewerTerminal).toHaveBeenCalledWith({ handleId: "reviewer-pane", harness: "codex" });
});
it("shows an up-to-date notice instead of opening the terminal when the backend reuses a run", async () => {
mockCommonGets([approvedReview], "reviewer-pane");
it("shows eligible and up-to-date PR review rows", async () => {
mockCommonGets([approvedReview], "reviewer-pane", [
reviewState(3, "needs_review", "abc123"),
reviewState(4, "up_to_date", "def456"),
]);
renderWithQuery(<SessionInspector session={session([pr(3, "open"), pr(4, "open")])} />);
await openReviewsTab();
expect(await screen.findByText("Reviewable change 3")).toBeInTheDocument();
expect(screen.getByText("#3")).toBeInTheDocument();
expect(screen.getByText("Reviewable change 4")).toBeInTheDocument();
expect(screen.getByText("#4")).toBeInTheDocument();
expect(screen.getAllByText("Not run")).not.toHaveLength(0);
expect(screen.getAllByText("Approved")).not.toHaveLength(0);
expect(screen.getByRole("button", { name: "Re-run review" })).toBeInTheDocument();
expect(screen.queryByRole("button", { name: "Run" })).not.toBeInTheDocument();
expect(screen.queryByRole("button", { name: "Re-run" })).not.toBeInTheDocument();
});
it("shows a no-needed-reviews notice instead of opening the terminal when the backend reuses runs", async () => {
mockCommonGets([approvedReview], "reviewer-pane", [reviewState(3, "up_to_date")]);
postMock.mockResolvedValue({
response: { status: 200 },
data: { reviewerHandleId: "reviewer-pane", review: approvedReview },
data: {
reviewerHandleId: "reviewer-pane",
reviews: [],
},
});
const onOpenReviewerTerminal = vi.fn();
@ -195,12 +237,15 @@ describe("SessionInspector reviews tab", () => {
await userEvent.click(await screen.findByRole("button", { name: /re-run review/i }));
expect(await screen.findByText("Review is already up to date for this commit.")).toBeInTheDocument();
expect(await screen.findByText("No needed reviews were started.")).toBeInTheDocument();
expect(onOpenReviewerTerminal).not.toHaveBeenCalled();
});
it("shows an approved review and opens its terminal", async () => {
mockCommonGets([approvedReview], "reviewer-pane");
it("shows one shared terminal action", async () => {
mockCommonGets([approvedReview], "reviewer-pane", [
reviewState(3, "running", "abc123"),
reviewState(4, "up_to_date", "def456"),
]);
const onOpenReviewerTerminal = vi.fn();
renderWithQuery(
@ -208,12 +253,38 @@ describe("SessionInspector reviews tab", () => {
);
await openReviewsTab();
await waitFor(() => expect(screen.getAllByText("Approved").length).toBeGreaterThan(0));
await waitFor(() => expect(screen.getAllByText("Open terminal")).toHaveLength(1));
expect(screen.getAllByRole("button", { name: /review/i })).toHaveLength(1);
await userEvent.click(screen.getByRole("button", { name: /open terminal/i }));
expect(onOpenReviewerTerminal).toHaveBeenCalledWith({ handleId: "reviewer-pane", harness: "codex" });
});
it("shows the reviewer identity and aggregate verdict", async () => {
mockCommonGets([approvedReview], "reviewer-pane", [reviewState(3, "changes_requested", "abc123")]);
renderWithQuery(<SessionInspector session={session([pr(3, "open")])} />);
await openReviewsTab();
expect(await screen.findByText("codex")).toBeInTheDocument();
expect(screen.getByText("reviewer")).toBeInTheDocument();
expect(screen.queryByText("sess-1")).not.toBeInTheDocument();
expect(screen.queryByText("review session")).not.toBeInTheDocument();
expect(screen.getAllByText("Changes requested")).not.toHaveLength(0);
});
it("shows failed latest runs as failed and still allows rerun", async () => {
mockCommonGets([failedReview], "reviewer-pane", [
{ ...reviewState(3, "needs_review", "abc123"), latestRun: failedReview },
]);
renderWithQuery(<SessionInspector session={session([pr(3, "open")])} />);
await openReviewsTab();
expect(await screen.findAllByText("Failed")).not.toHaveLength(0);
expect(screen.getByRole("button", { name: "Re-run review" })).toBeEnabled();
});
it("shows the no-PR empty state when the session has no PRs", async () => {
mockCommonGets();
renderWithQuery(<SessionInspector session={session([])} />);

View File

@ -1,21 +1,12 @@
import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
import { useState, type ReactNode } from "react";
import {
AlertCircle,
ArrowUpRight,
CheckCircle2,
CircleMinus,
GitPullRequest,
Play,
Shield,
Terminal,
} from "lucide-react";
import { ArrowUpRight, GitPullRequest, Play, Shield, Terminal } from "lucide-react";
import type { components } from "../../api/schema";
import { apiClient, apiErrorMessage } from "../lib/api-client";
import { workspaceQueryKey } from "../hooks/useWorkspaceQuery";
import { formatTimeCompact } from "../lib/format-time";
import { useSessionScmSummary, type SessionPRSummary } from "../hooks/useSessionScmSummary";
import { prStatusRows, sessionPRDisplaySummaries, type PRDisplayTone } from "../lib/pr-display";
import { prBrowserUrl, prStatusRows, sessionPRDisplaySummaries, type PRDisplayTone } from "../lib/pr-display";
import type { SessionStatus, WorkspaceSession } from "../types/workspace";
import { sortedPRs, workerDisplayStatus } from "../types/workspace";
import { BrowserPanelView } from "./BrowserPanel";
@ -26,7 +17,7 @@ import { cn } from "../lib/utils";
import { PRAttentionPanel, PRSummaryMeta } from "./PRSummaryDisplay";
type ProjectConfig = components["schemas"]["ProjectConfig"];
type ReviewRun = components["schemas"]["ReviewRun"];
type PRReviewState = components["schemas"]["PRReviewState"];
type ReviewsResponse = components["schemas"]["ListReviewsResponse"];
type OpenReviewerTerminal = (target: { handleId: string; harness: string }) => void;
@ -219,7 +210,7 @@ function PRSummaryCard({ pr }: { pr: SessionPRSummary }) {
{pr.state}
</Badge>
<a
href={pr.htmlUrl || pr.url}
href={prBrowserUrl(pr)}
target="_blank"
rel="noopener noreferrer"
className="ml-auto inline-flex items-center gap-0.5 text-[11px] font-medium text-accent hover:underline"
@ -387,7 +378,8 @@ function ReviewsView({
enabled: hasPr,
refetchInterval: (query) => {
const data = query.state.data as ReviewsResponse | undefined;
return data?.reviews.some((review) => review.status === "running") ? 2500 : false;
const reviews = data?.reviews ?? [];
return reviews.some((review) => review.status === "running") ? 2500 : false;
},
queryFn: async () => {
const { data, error } = await apiClient.GET("/api/v1/sessions/{sessionId}/reviews", {
@ -422,16 +414,18 @@ function ReviewsView({
onSuccess: ({ data, reused }) => {
void queryClient.invalidateQueries({ queryKey: ["session-reviews", session.id] });
void queryClient.invalidateQueries({ queryKey: workspaceQueryKey });
if (reused) {
setReviewNotice("Review is already up to date for this commit.");
const started = data?.reviews?.find((review) => review.status === "running" && review.latestRun);
if (reused || !started?.latestRun) {
setReviewNotice("No needed reviews were started.");
return;
}
if (data?.reviewerHandleId) {
onOpenReviewerTerminal?.({ handleId: data.reviewerHandleId, harness: data.review.harness || "reviewer" });
const harness = started.latestRun.harness || "reviewer";
onOpenReviewerTerminal?.({ handleId: data.reviewerHandleId, harness });
}
},
});
const reviews = reviewsQuery.data?.reviews ?? [];
const reviewStates = reviewsQuery.data?.reviews ?? [];
return (
<div role="tabpanel">
@ -444,7 +438,7 @@ function ReviewsView({
onOpenTerminal={onOpenReviewerTerminal}
onTrigger={() => triggerReview.mutate()}
reviewerHandleId={reviewsQuery.data?.reviewerHandleId ?? ""}
reviews={reviews}
reviewStates={reviewStates}
notice={reviewNotice}
session={session}
/>
@ -461,7 +455,7 @@ function projectConfig(project: components["schemas"]["ProjectOrDegraded"] | und
function ReviewPanel({
session,
config,
reviews,
reviewStates,
reviewerHandleId,
isLoading,
isTriggering,
@ -472,7 +466,7 @@ function ReviewPanel({
}: {
session: WorkspaceSession;
config?: ProjectConfig;
reviews: ReviewRun[];
reviewStates: PRReviewState[];
reviewerHandleId: string;
isLoading: boolean;
isTriggering: boolean;
@ -488,108 +482,142 @@ function ReviewPanel({
return <p className="inspector-empty">Loading reviews...</p>;
}
const latest = latestReview(reviews);
const latest = reviewStates.find((review) => review.latestRun)?.latestRun;
const harness = latest?.harness || config?.reviewers?.[0]?.harness || session.provider || "reviewer";
const terminalEnabled = Boolean(reviewerHandleId && onOpenTerminal);
const aggregateVerdict = sessionReviewVerdict(reviewStates);
const runAction = reviewSessionRunAction(reviewStates, isTriggering);
const runDisabled =
isTriggering ||
reviewStates.length === 0 ||
reviewStates.some((reviewState) => reviewState.status === "running") ||
reviewStates.every((reviewState) => reviewState.status === "ineligible");
return (
<div className="reviewer-list">
{error ? <p className="reviewer-error">{apiErrorMessage(error, "Review request failed")}</p> : null}
{notice ? <p className="reviewer-notice">{notice}</p> : null}
<ReviewerCard
handleId={reviewerHandleId}
harness={harness}
isTriggering={isTriggering}
onOpenTerminal={onOpenTerminal}
onTrigger={onTrigger}
review={latest}
/>
</div>
);
}
function latestReview(reviews: ReviewRun[]): ReviewRun | undefined {
return [...reviews].sort((a, b) => Date.parse(b.createdAt) - Date.parse(a.createdAt))[0];
}
function ReviewerCard({
harness,
review,
handleId,
isTriggering,
onTrigger,
onOpenTerminal,
}: {
harness: string;
review?: ReviewRun;
handleId: string;
isTriggering: boolean;
onTrigger: () => void;
onOpenTerminal?: OpenReviewerTerminal;
}) {
const status = reviewStatus(review);
const terminalEnabled = Boolean(handleId && onOpenTerminal);
const runLabel = review ? "Re-run review" : "Run review";
return (
<div className={cn("reviewer-card", status.tone && `reviewer-card--${status.tone}`)}>
<div className="reviewer-card__top">
<div className="reviewer-card__name">
<Shield aria-hidden="true" />
<span>{harness}</span>
</div>
<span className={cn("reviewer-status", `reviewer-status--${status.tone}`)}>
{status.icon}
{status.label}
</span>
<div className="reviewer-kicker">
<Shield aria-hidden="true" />
<span>{harness}</span>
<span>reviewer</span>
</div>
<div className="reviewer-card__actions">
<button
className="reviewer-card__action reviewer-card__action--primary"
disabled={isTriggering}
onClick={onTrigger}
type="button"
>
<Play aria-hidden="true" />
{isTriggering ? "Starting..." : runLabel}
</button>
{review ? (
<div className="reviewer-card">
<div className="reviewer-card__top">
<span className="reviewer-card__label">Pull requests</span>
<span className={cn("reviewer-status", `reviewer-status--${aggregateVerdict.tone}`)}>
{aggregateVerdict.label}
</span>
</div>
<div className="reviewer-summary-list">
{reviewStates.length === 0 ? <p className="inspector-empty">No review state loaded yet.</p> : null}
{reviewStates.map((reviewState) => (
<ReviewStateRow key={`${reviewState.prUrl}:${reviewState.targetSha}`} reviewState={reviewState} />
))}
</div>
<div className="reviewer-card__actions">
<button
className="reviewer-card__action reviewer-card__action--primary"
disabled={runDisabled}
onClick={onTrigger}
type="button"
>
<Play aria-hidden="true" />
{runAction}
</button>
<button
className="reviewer-card__action"
disabled={!terminalEnabled}
onClick={() => {
if (!terminalEnabled) return;
onOpenTerminal?.({ handleId, harness });
onOpenTerminal?.({ handleId: reviewerHandleId, harness });
}}
type="button"
>
<Terminal aria-hidden="true" />
Open terminal
</button>
) : null}
</div>
</div>
</div>
);
}
function reviewStatus(review?: ReviewRun): {
function ReviewStateRow({ reviewState }: { reviewState: PRReviewState }) {
const verdict = reviewVerdict(reviewState);
const title = reviewState.title?.trim() || `PR #${reviewState.prNumber}`;
return (
<div
className={cn(
"reviewer-row",
`reviewer-row--${verdict.tone}`,
reviewState.status === "ineligible" && "opacity-70",
)}
>
<div className="reviewer-row__main">
<span className={cn("reviewer-row__dot", `reviewer-row__dot--${verdict.tone}`)} />
<div className="reviewer-row__copy">
<GitPullRequest aria-hidden="true" />
<a href={reviewState.prUrl} target="_blank" rel="noopener noreferrer">
{title}
</a>
<span className="reviewer-row__number">#{reviewState.prNumber}</span>
</div>
</div>
<span className={cn("reviewer-row__verdict", `reviewer-row__verdict--${verdict.tone}`)}>{verdict.label}</span>
</div>
);
}
function sessionReviewVerdict(reviewStates: PRReviewState[]): {
label: string;
tone: "neutral" | "running" | "success" | "danger";
icon: ReactNode;
} {
if (!review) return { label: "Not run", tone: "neutral", icon: null };
if (review.status === "running") {
return { label: "Running", tone: "running", icon: <Play aria-hidden="true" /> };
if (reviewStates.some((reviewState) => reviewState.status === "running")) {
return { label: "Reviewing...", tone: "running" };
}
if (review.status === "failed") {
return { label: "Failed", tone: "danger", icon: <AlertCircle aria-hidden="true" /> };
if (reviewStates.some((reviewState) => reviewState.latestRun?.status === "failed")) {
return { label: "Failed", tone: "danger" };
}
if (review.verdict === "approved") {
return { label: "Approved", tone: "success", icon: <CheckCircle2 aria-hidden="true" /> };
if (reviewStates.some((reviewState) => reviewState.status === "changes_requested")) {
return { label: "Changes requested", tone: "danger" };
}
if (review.verdict === "changes_requested") {
return { label: "Changes requested", tone: "danger", icon: <CircleMinus aria-hidden="true" /> };
const eligibleReviews = reviewStates.filter((reviewState) => reviewState.status !== "ineligible");
if (eligibleReviews.length > 0 && eligibleReviews.every((reviewState) => reviewState.status === "up_to_date")) {
return { label: "Approved", tone: "success" };
}
return { label: "Complete", tone: "success", icon: <CheckCircle2 aria-hidden="true" /> };
return { label: "Not run", tone: "neutral" };
}
function reviewVerdict(reviewState: PRReviewState): {
label: string;
tone: "neutral" | "running" | "success" | "danger";
} {
if (reviewState.latestRun?.status === "failed") {
return { label: "Failed", tone: "danger" };
}
switch (reviewState.status) {
case "running":
return { label: "Reviewing...", tone: "running" };
case "up_to_date":
return { label: "Approved", tone: "success" };
case "changes_requested":
return { label: "Changes requested", tone: "danger" };
case "needs_review":
case "ineligible":
return { label: "Not run", tone: "neutral" };
}
return { label: "Not run", tone: "neutral" };
}
function reviewSessionRunAction(reviewStates: PRReviewState[], isTriggering: boolean): string {
if (isTriggering || reviewStates.some((reviewState) => reviewState.status === "running")) {
return "Reviewing...";
}
if (reviewStates.some((reviewState) => reviewState.status === "changes_requested" || reviewState.latestRun)) {
return "Re-run review";
}
return "Run review";
}
function BrowserView({

View File

@ -1,4 +1,4 @@
import { useState } from "react";
import { useState, type KeyboardEvent } from "react";
import { useQueryClient } from "@tanstack/react-query";
import { useNavigate } from "@tanstack/react-router";
import { Plus } from "lucide-react";
@ -263,11 +263,22 @@ function SessionCard({ session, onOpen }: { session: WorkspaceSession; onOpen: (
const branch = session.branch || "";
const showBranch = branch !== "" && !sameLabel(branch, session.title) && !sameLabel(branch, session.id);
const prSummaries = sessionPRDisplaySummaries(session, useSessionScmSummary(session.id).data);
const handleKeyDown = (event: KeyboardEvent<HTMLDivElement>) => {
if (event.target !== event.currentTarget) {
return;
}
if (event.key === "Enter" || event.key === " ") {
event.preventDefault();
onOpen();
}
};
return (
<button
<div
className="w-full rounded-[7px] border border-border bg-surface text-left transition-colors hover:border-border-strong"
onClick={onOpen}
type="button"
onKeyDown={handleKeyDown}
role="button"
tabIndex={0}
>
<div className="flex items-center gap-2 px-[13px] pb-[9px] pt-3">
<span className={cn("inline-flex items-center gap-1.5 text-[11px] font-medium", badge.className)}>
@ -303,7 +314,7 @@ function SessionCard({ session, onOpen }: { session: WorkspaceSession; onOpen: (
"no PR yet"
)}
</div>
</button>
</div>
);
}
@ -316,7 +327,7 @@ function BoardPRSummary({ className, pr }: { className?: string; pr: SessionPRSu
</span>
{diffSummary ? <span className="truncate">{diffSummary}</span> : null}
<PRStatusStrip pr={pr} />
<PRAttentionPanel className="mt-1.5 pt-1.5" interactiveLinks={false} maxItems={2} pr={pr} />
<PRAttentionPanel className="mt-1.5 pt-1.5" maxItems={2} pr={pr} />
</div>
);
}

View File

@ -1,6 +1,6 @@
import { describe, expect, it } from "vitest";
import type { SessionPRSummary } from "../hooks/useSessionScmSummary";
import { prAttentionItems, prDiffSummary, prStatusRows } from "./pr-display";
import { prAttentionItems, prBrowserUrl, prDiffSummary, prStatusRows } from "./pr-display";
const summary = (overrides: Partial<SessionPRSummary> = {}): SessionPRSummary => ({
url: "https://github.com/acme/repo/pull/7",
@ -56,6 +56,19 @@ describe("prDiffSummary", () => {
});
});
describe("prBrowserUrl", () => {
it("normalizes issue-shaped GitHub PR URLs to the pull request page", () => {
expect(
prBrowserUrl(
summary({
url: "https://github.com/acme/repo/issues/7",
htmlUrl: "https://github.com/acme/repo/issues/7",
}),
),
).toBe("https://github.com/acme/repo/pull/7");
});
});
describe("prAttentionItems", () => {
it("returns no attention for clean open PRs", () => {
expect(prAttentionItems(summary())).toEqual([]);
@ -100,6 +113,150 @@ describe("prAttentionItems", () => {
});
});
it("links failing CI checks to their provider URLs", () => {
const items = prAttentionItems(
summary({
ci: {
state: "failing",
failingChecks: [
{ name: "unit", status: "failed", conclusion: "failure", url: "https://checks.example/unit" },
{ name: "lint", status: "failed", conclusion: "failure", url: "https://checks.example/lint" },
{ name: "build", status: "failed", conclusion: "failure", url: "https://checks.example/build" },
{ name: "types", status: "failed", conclusion: "failure", url: "https://checks.example/types" },
],
},
}),
);
const ciItem = items.find((item) => item.kind === "ci_failing");
expect(ciItem?.links).toEqual([
{ label: "unit", href: "https://checks.example/unit", title: "failure" },
{ label: "lint", href: "https://checks.example/lint", title: "failure" },
{ label: "build", href: "https://checks.example/build", title: "failure" },
]);
expect(ciItem?.overflowLabel).toBe("+1 check");
});
it("prefers the submitted review summary over inline comments", () => {
const items = prAttentionItems(
summary({
review: {
decision: "changes_requested",
hasUnresolvedHumanComments: true,
unresolvedBy: [
{
reviewerId: "alice",
count: 2,
reviewUrl: "https://github.com/acme/repo/pull/7#pullrequestreview-1",
links: [
{ url: "https://github.com/acme/repo/pull/7#discussion_r1", file: "main.go", line: 12 },
{ url: "https://github.com/acme/repo/pull/7#discussion_r2", file: "test.go", line: 20 },
],
},
],
},
}),
);
expect(items.find((item) => item.kind === "review_changes_requested")?.links[0]).toMatchObject({
label: "alice +1",
href: "https://github.com/acme/repo/pull/7#pullrequestreview-1",
title: "Open requested-changes review from alice",
});
});
it("falls back to the first inline comment when no review summary exists", () => {
const items = prAttentionItems(
summary({
review: {
decision: "changes_requested",
hasUnresolvedHumanComments: true,
unresolvedBy: [
{
reviewerId: "alice",
count: 2,
links: [
{ url: "https://github.com/acme/repo/pull/7#discussion_r1", file: "main.go", line: 12 },
{ url: "https://github.com/acme/repo/pull/7#discussion_r2", file: "test.go", line: 20 },
],
},
],
},
}),
);
expect(items.find((item) => item.kind === "review_changes_requested")?.links[0]).toMatchObject({
label: "alice +1",
href: "https://github.com/acme/repo/pull/7#discussion_r1",
title: "2 unresolved comments from alice",
});
});
it("falls back to the PR page when review summary and inline comment URLs are missing", () => {
const items = prAttentionItems(
summary({
url: "https://github.com/acme/repo/issues/7",
htmlUrl: "https://github.com/acme/repo/issues/7",
review: {
decision: "changes_requested",
hasUnresolvedHumanComments: true,
unresolvedBy: [{ reviewerId: "alice", count: 1, links: [] }],
},
}),
);
expect(items.find((item) => item.kind === "review_changes_requested")?.links[0]).toMatchObject({
label: "alice",
href: "https://github.com/acme/repo/pull/7",
title: "Open pull request for alice",
});
});
it("shows bot reviewers with a bot label", () => {
const items = prAttentionItems(
summary({
review: {
decision: "changes_requested",
hasUnresolvedHumanComments: false,
unresolvedBy: [
{
reviewerId: "copilot",
count: 0,
reviewUrl: "https://github.com/acme/repo/pull/7#pullrequestreview-2",
isBot: true,
links: [],
},
],
},
}),
);
expect(items.find((item) => item.kind === "review_changes_requested")?.links[0]).toMatchObject({
label: "copilot bot",
href: "https://github.com/acme/repo/pull/7#pullrequestreview-2",
title: "Open requested-changes review from copilot bot",
});
});
it("links merge conflicts to GitHub's conflict resolution page", () => {
const items = prAttentionItems(
summary({
url: "https://github.com/acme/repo/issues/7",
htmlUrl: "https://github.com/acme/repo/issues/7",
mergeability: {
state: "conflicting",
reasons: [],
prUrl: "https://github.com/acme/repo/issues/7",
},
}),
);
expect(items.find((item) => item.kind === "merge_conflict")?.links[0]).toMatchObject({
label: "conflicts",
href: "https://github.com/acme/repo/pull/7/conflicts",
});
});
it("suppresses attention once the PR is closed or merged", () => {
expect(
prAttentionItems(

View File

@ -46,6 +46,10 @@ export function comparePRDisplaySummaries(a: SessionPRSummary, b: SessionPRSumma
return prStateRank[a.state] - prStateRank[b.state] || a.number - b.number;
}
export function prBrowserUrl(pr: SessionPRSummary): string {
return prBaseUrl(pr) ?? pr.htmlUrl ?? pr.url;
}
export function sessionPRDisplaySummaries(
session: WorkspaceSession,
summaries: SessionPRSummary[] = [],
@ -175,11 +179,10 @@ export function prAttentionItems(pr: SessionPRSummary): PRAttentionItem[] {
}
if (pr.review.decision === "changes_requested" || pr.review.hasUnresolvedHumanComments) {
const reviewers = pr.review.unresolvedBy.slice(0, 3);
const links = reviewers.map((reviewer) => ({
label: reviewerLabel(reviewer),
href: reviewer.links.find((link) => link.url)?.url || undefined,
title: `${reviewer.count} unresolved ${pluralize("comment", reviewer.count)}`,
}));
const links = reviewers.map((reviewer) => reviewAttentionLink(pr, reviewer));
if (links.length === 0 && pr.review.decision === "changes_requested") {
links.push({ label: "PR", href: prBrowserUrl(pr), title: "Open pull request" });
}
items.push({
kind: "review_changes_requested",
title: "Address requested changes",
@ -331,18 +334,23 @@ function mergeAttention(
fallback: string,
tone: PRDisplayTone,
): PRAttentionItem {
const href =
kind === "merge_conflict" ? mergeConflictUrl(pr) : pr.mergeability.prUrl || pr.htmlUrl || pr.url || undefined;
const fileLinks = (pr.mergeability.conflictFiles ?? []).slice(0, 3).map((file) => ({
label: file.path,
href: file.url || pr.mergeability.prUrl || undefined,
href: file.url || href,
title: kind === "merge_conflict" ? "Open merge conflicts" : undefined,
}));
const reasonLinks =
fileLinks.length > 0
fileLinks.length > 0 || kind === "merge_conflict"
? []
: pr.mergeability.reasons.slice(0, 3).map((reason) => ({
label: mergeReasonLabel(reason),
href: pr.mergeability.prUrl || undefined,
href,
}));
const links = fileLinks.length > 0 ? fileLinks : reasonLinks;
const fallbackLink =
kind === "merge_conflict" && href ? [{ label: "conflicts", href, title: "Open merge conflicts" }] : [];
const links = fileLinks.length > 0 ? fileLinks : reasonLinks.length > 0 ? reasonLinks : fallbackLink;
return {
kind,
title,
@ -356,11 +364,78 @@ function mergeAttention(
};
}
function reviewerLabel(reviewer: SessionPRSummary["review"]["unresolvedBy"][number]): string {
if (reviewer.count <= 1) {
return reviewer.reviewerId;
function mergeConflictUrl(pr: SessionPRSummary): string | undefined {
return prSubpageUrl(pr, "conflicts") ?? pr.mergeability.prUrl ?? prBrowserUrl(pr);
}
function prBaseUrl(pr: SessionPRSummary): string | undefined {
return prURL(pr);
}
function prSubpageUrl(pr: SessionPRSummary, subpage: "conflicts"): string | undefined {
const base = prURL(pr);
return base ? `${base}/${subpage}` : undefined;
}
function prURL(pr: SessionPRSummary): string | undefined {
const raw = pr.htmlUrl || pr.mergeability.prUrl || pr.url;
if (!raw) {
return undefined;
}
return `${reviewer.reviewerId} +${reviewer.count - 1}`;
try {
const url = new URL(raw);
const match = url.pathname.match(/^(\/[^/]+\/[^/]+)\/(?:pull|issues)\/(\d+)(?:\/.*)?$/);
if (!match) {
return undefined;
}
url.pathname = `${match[1]}/pull/${match[2]}`;
url.search = "";
url.hash = "";
return url.toString();
} catch {
return undefined;
}
}
function reviewerLabel(reviewer: SessionPRSummary["review"]["unresolvedBy"][number]): string {
const name = reviewerDisplayName(reviewer);
if (reviewer.count <= 1) {
return name;
}
return `${name} +${reviewer.count - 1}`;
}
function reviewerDisplayName(reviewer: SessionPRSummary["review"]["unresolvedBy"][number]): string {
return reviewer.isBot ? `${reviewer.reviewerId} bot` : reviewer.reviewerId;
}
function reviewAttentionLink(
pr: SessionPRSummary,
reviewer: SessionPRSummary["review"]["unresolvedBy"][number],
): PRAttentionLink {
const inlineURL = reviewer.links.find((link) => link.url)?.url;
if (reviewer.reviewUrl) {
return {
label: reviewerLabel(reviewer),
href: reviewer.reviewUrl,
title: `Open requested-changes review from ${reviewerDisplayName(reviewer)}`,
};
}
if (inlineURL) {
return {
label: reviewerLabel(reviewer),
href: inlineURL,
title:
reviewer.count > 0
? `${reviewer.count} unresolved ${pluralize("comment", reviewer.count)} from ${reviewerDisplayName(reviewer)}`
: `Open review comments from ${reviewerDisplayName(reviewer)}`,
};
}
return {
label: reviewerLabel(reviewer),
href: prBrowserUrl(pr),
title: `Open pull request for ${reviewerDisplayName(reviewer)}`,
};
}
function mergeReasonLabel(reason: string): string {

View File

@ -945,7 +945,39 @@ body.is-resizing-x [data-slot="sidebar-container"] {
.reviewer-list {
display: flex;
flex-direction: column;
gap: 12px;
gap: 14px;
}
.reviewer-kicker {
display: inline-flex;
min-width: 0;
align-items: center;
gap: 8px;
font-family: var(--font-mono);
font-size: 13px;
font-weight: 700;
color: var(--fg);
}
.reviewer-kicker svg {
width: 15px;
height: 15px;
flex-shrink: 0;
color: var(--fg-passive);
}
.reviewer-kicker span:first-of-type {
min-width: 0;
overflow: hidden;
text-overflow: ellipsis;
white-space: nowrap;
}
.reviewer-kicker span:last-child {
font-family: var(--font-sans);
font-size: 11.5px;
font-weight: 500;
color: var(--fg-passive);
}
.reviewer-error {
@ -973,11 +1005,11 @@ body.is-resizing-x [data-slot="sidebar-container"] {
.reviewer-card {
display: flex;
flex-direction: column;
gap: 14px;
gap: 12px;
border-radius: 8px;
border: 1px solid var(--border);
background: var(--bg-1);
padding: 14px;
padding: 12px;
}
.reviewer-card--success {
@ -1000,9 +1032,9 @@ body.is-resizing-x [data-slot="sidebar-container"] {
display: inline-flex;
min-width: 0;
align-items: center;
gap: 9px;
gap: 8px;
font-family: var(--font-mono);
font-size: 13px;
font-size: 12.5px;
font-weight: 600;
color: var(--fg);
}
@ -1021,17 +1053,34 @@ body.is-resizing-x [data-slot="sidebar-container"] {
white-space: nowrap;
}
.reviewer-card__identity {
display: flex;
min-width: 0;
flex-direction: column;
gap: 4px;
}
.reviewer-card__label {
min-width: 0;
overflow: hidden;
text-overflow: ellipsis;
white-space: nowrap;
font-size: 12px;
font-weight: 650;
color: var(--fg-muted);
}
.reviewer-status {
display: inline-flex;
height: 24px;
height: 20px;
max-width: 58%;
flex-shrink: 0;
align-items: center;
gap: 5px;
overflow: hidden;
border-radius: 7px;
padding: 0 9px;
font-size: 11.5px;
border-radius: 6px;
padding: 0 8px;
font-size: 10.5px;
font-weight: 650;
line-height: 1;
white-space: nowrap;
@ -1068,6 +1117,7 @@ body.is-resizing-x [data-slot="sidebar-container"] {
display: grid;
grid-template-columns: repeat(2, minmax(0, 1fr));
gap: 10px;
padding-top: 4px;
}
.reviewer-card__actions:has(.reviewer-card__action:only-child) {
@ -1119,6 +1169,138 @@ body.is-resizing-x [data-slot="sidebar-container"] {
color: #8bdc75;
}
.reviewer-card__action--wide {
width: 100%;
}
.reviewer-summary-list {
display: flex;
flex-direction: column;
gap: 0;
overflow: hidden;
border-radius: 7px;
border: 1px solid var(--border);
background: color-mix(in srgb, var(--bg-0) 35%, transparent);
}
.reviewer-row {
display: grid;
grid-template-columns: minmax(0, 1fr) auto;
align-items: center;
gap: 10px;
border-radius: 0;
border: 0;
border-bottom: 1px solid var(--border);
background: transparent;
padding: 11px 12px;
min-height: 52px;
}
.reviewer-row:last-child {
border-bottom: 0;
}
.reviewer-row--success {
border-color: var(--border);
}
.reviewer-row--danger {
border-color: var(--border);
}
.reviewer-row__main {
display: inline-flex;
min-width: 0;
align-items: center;
gap: 9px;
}
.reviewer-row__dot {
width: 7px;
height: 7px;
flex-shrink: 0;
border-radius: 999px;
background: var(--fg-passive);
}
.reviewer-row__dot--running {
background: var(--orange);
}
.reviewer-row__dot--success {
background: var(--green);
}
.reviewer-row__dot--danger {
background: var(--red);
}
.reviewer-row__copy {
display: grid;
grid-template-columns: auto auto;
min-width: 0;
align-items: baseline;
column-gap: 6px;
row-gap: 2px;
font-size: 12px;
font-weight: 650;
color: var(--fg);
}
.reviewer-row__copy svg {
display: none;
width: 14px;
height: 14px;
flex-shrink: 0;
color: var(--fg-passive);
}
.reviewer-row__copy a {
grid-column: 1 / -1;
min-width: 0;
overflow: hidden;
text-overflow: ellipsis;
white-space: nowrap;
color: inherit;
text-decoration: none;
}
.reviewer-row__copy a:hover {
text-decoration: underline;
}
.reviewer-row__number {
grid-column: 1;
font-family: var(--font-mono);
font-size: 11px;
color: var(--fg-passive);
}
.reviewer-row__verdict {
white-space: nowrap;
font-size: 11px;
font-weight: 650;
color: var(--fg-muted);
}
.reviewer-row__verdict--running {
color: var(--orange);
}
.reviewer-row__verdict--success {
color: var(--green);
}
.reviewer-row__verdict--danger {
color: var(--red);
}
@media (max-width: 680px) {
.reviewer-row {
grid-template-columns: minmax(0, 1fr) auto;
}
}
:root[data-theme="light"] .reviewer-card__action--primary {
color: var(--green);
}