import type { WorkflowDefinition } from "@uncaged/nerve-core";
import { END } from "@uncaged/nerve-core";
import { buildPlannerRole } from "./roles/planner/index.js";
import { buildCoderRole } from "./roles/coder/index.js";
import { tester } from "./roles/tester/index.js";

import type { SenseMeta } from "./roles/types.js";

const workflow: WorkflowDefinition<SenseMeta> = {
  name: "sense-generator",
  roles: {
    planner: await buildPlannerRole(),
    coder: await buildCoderRole(),
    tester,
  },
  moderator(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 last.meta.attempt < 3 ? "coder" : END;
    }
    return END;
  },
};

export default workflow;