40 lines
1.3 KiB
TypeScript
40 lines
1.3 KiB
TypeScript
import { END } from "@uncaged/nerve-core";
|
|
import type { Moderator } from "@uncaged/nerve-core";
|
|
import type { PlannerMeta } from "./roles/planner/index.js";
|
|
import type { CoderMeta } from "./roles/coder/index.js";
|
|
import type { TesterMeta } from "./roles/tester/index.js";
|
|
import type { CommitterMeta } from "./roles/committer/index.js";
|
|
|
|
export type SenseMeta = {
|
|
planner: PlannerMeta;
|
|
coder: CoderMeta;
|
|
tester: TesterMeta;
|
|
committer: CommitterMeta;
|
|
};
|
|
|
|
const MAX_CODER_ITERATIONS = 5;
|
|
|
|
function countRole(steps: { role: string }[], name: string): number {
|
|
return steps.filter((s) => s.role === name).length;
|
|
}
|
|
|
|
export const moderator: Moderator<SenseMeta> = (context) => {
|
|
if (context.steps.length === 0) return "planner";
|
|
const last = context.steps[context.steps.length - 1];
|
|
const coderCount = context.steps.filter((s) => s.role === "coder").length;
|
|
|
|
if (last.role === "planner") return "coder";
|
|
if (last.role === "coder") return "tester";
|
|
if (last.role === "tester") {
|
|
if (last.meta.passed) return "committer";
|
|
const testerCount = countRole(context.steps, "tester");
|
|
if (testerCount < 3 && coderCount < MAX_CODER_ITERATIONS) return "coder";
|
|
return END;
|
|
}
|
|
if (last.role === "committer") {
|
|
if (last.meta.success) return END;
|
|
return coderCount < MAX_CODER_ITERATIONS ? "coder" : END;
|
|
}
|
|
return END;
|
|
};
|