import type { AgentFn, Role, ThreadContext } from "@uncaged/nerve-core";
import type { LlmExtractorConfig } from "@uncaged/nerve-workflow-utils";
import { createRole } from "@uncaged/nerve-workflow-utils";
import { z } from "zod";

import { reviewPrompt } from "./prompt.js";

export const reviewMetaSchema = z.object({
  approved: z.boolean().describe("true if diff is clean and ready for tests"),
});
export type ReviewMeta = z.infer<typeof reviewMetaSchema>;

export function createReviewRole(
  adapter: AgentFn,
  extract: LlmExtractorConfig,
  nerveRoot: string,
): Role<ReviewMeta> {
  return createRole(
    adapter,
    async (ctx: ThreadContext) =>
      reviewPrompt({ threadId: ctx.start.meta.threadId, nerveRoot }),
    reviewMetaSchema,
    extract,
  );
}