85 lines
2.7 KiB
TypeScript

import { END } from "@uncaged/nerve-core";
import type { Moderator } from "@uncaged/nerve-core";
import type { IntakeMeta } from "./roles/intake/index.js";
import type { IssueReaderMeta } from "./roles/issue-reader/index.js";
import type { PlannerMeta } from "./roles/planner/index.js";
import type { ImplementerMeta } from "./roles/implementer/index.js";
import type { ReviewerMeta } from "./roles/reviewer/index.js";
import type { TesterMeta } from "./roles/tester/index.js";
import type { PrPublisherMeta } from "./roles/pr-publisher/index.js";
export type WorkflowMeta = {
intake: IntakeMeta;
"issue-reader": IssueReaderMeta;
planner: PlannerMeta;
implementer: ImplementerMeta;
reviewer: ReviewerMeta;
tester: TesterMeta;
"pr-publisher": PrPublisherMeta;
};
const MAX_IMPLEMENTER_ROUNDS = 20;
const MAX_TOTAL_REJECTIONS = 10;
function implementerRounds(steps: { role: string }[]): number {
return steps.filter((s) => s.role === "implementer").length;
}
function totalRejections(steps: { role: string; meta: unknown }[]): number {
return steps.filter((s) => {
if (s.role === "reviewer") return !(s.meta as Record<string, boolean>).approved;
if (s.role === "tester") return !(s.meta as Record<string, boolean>).passed;
return false;
}).length;
}
function canRetryImplementer(steps: { role: string; meta: unknown }[]): boolean {
return implementerRounds(steps) < MAX_IMPLEMENTER_ROUNDS && totalRejections(steps) < MAX_TOTAL_REJECTIONS;
}
export const moderator: Moderator<WorkflowMeta> = (context) => {
if (context.steps.length === 0) {
return "intake";
}
const last = context.steps[context.steps.length - 1];
if (last.role === "intake") {
const meta = last.meta as WorkflowMeta["intake"];
return meta.valid ? "issue-reader" : END;
}
if (last.role === "issue-reader") {
const meta = last.meta as WorkflowMeta["issue-reader"];
return meta.fetchOk ? "planner" : END;
}
if (last.role === "planner") {
const meta = last.meta as WorkflowMeta["planner"];
return meta.planningOk ? "implementer" : END;
}
if (last.role === "implementer") {
const meta = last.meta as WorkflowMeta["implementer"];
if (meta.implementationOk) return "reviewer";
return canRetryImplementer(context.steps) ? "implementer" : END;
}
if (last.role === "reviewer") {
if (last.meta.approved) return "tester";
return canRetryImplementer(context.steps) ? "implementer" : END;
}
if (last.role === "tester") {
const meta = last.meta as WorkflowMeta["tester"];
if (meta.passed) return "pr-publisher";
return canRetryImplementer(context.steps) ? "implementer" : END;
}
if (last.role === "pr-publisher") {
return END;
}
return END;
};