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";

export type SenseMeta = {
  planner: PlannerMeta;
  coder: CoderMeta;
  tester: TesterMeta;
};

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];
  if (last.role === "planner") return "coder";
  if (last.role === "coder") return "tester";
  if (last.role === "tester") {
    if (last.meta.passed) return END;
    return countRole(context.steps, "tester") < 3 ? "coder" : END;
  }
  return END;
};