import type { LlmProvider } from "@uncaged/nerve-workflow-utils";
import { createCursorRole } from "@uncaged/nerve-workflow-utils";
import { plannerPrompt } from "./prompt.js";
import { z } from "zod";

export const plannerMetaSchema = z.object({
  ready: z.boolean().describe("true if requirements are clear and a workflow can be implemented"),
});
export type PlannerMeta = z.infer<typeof plannerMetaSchema>;

export type BuildPlannerDeps = {
  provider: LlmProvider;
  cwd: string;
};

export function buildPlannerRole({ provider, cwd }: BuildPlannerDeps) {
  return createCursorRole<PlannerMeta>({
    cwd,
    mode: "ask",
    prompt: async (threadId) => plannerPrompt({ threadId }),
    extract: { provider, schema: plannerMetaSchema },
  });
}