import { randomUUID } from "node:crypto";
import { requireRole } from "@/lib/auth/require-role";
import { requireSupportWrite } from "@/lib/impersonate/support";
import { getRequestPool } from "@/lib/agent-engine/db/request-pool";
import { ok, fail } from "@/lib/api/wrappers";
import { agentChatInputSchema } from "@/lib/prospecting/agent-chat-schema";
import { chatAboutAgent } from "@/lib/prospecting/agent-chat";
import { provedorOferecido } from "@/lib/ai/pontos/provedores-oferecidos";
import { createAdminClient } from "@/lib/supabase/admin";
import { AgentSetupError } from "@/lib/prospecting/agent-setup";
import { beginAgentChat, finishAgentChat } from "@/lib/prospecting/agent-session";

export const dynamic = "force-dynamic";
export const maxDuration = 120;

export async function POST(req: Request) {
  const requestId = randomUUID();
  const support = await requireSupportWrite();
  if (support) return support;
  const auth = await requireRole("admin", { requestId, resource: "prospecting" });
  if (!auth.ok) return auth.response;
  const parsed = agentChatInputSchema.safeParse(await req.json().catch(() => null));
  if (!parsed.success)
    return fail("validation_failed", "Confira a mensagem e o tamanho da conversa.", 422, {
      requestId,
    });
  try {
    const pool = getRequestPool();
    const actor = { orgId: auth.org.orgId, userId: auth.user.id, requestId };
    req.signal.throwIfAborted();
    const pending =
      parsed.data.revision === undefined ? null : await beginAgentChat(pool, actor, parsed.data);
    const data = await chatAboutAgent(
      pool,
      auth.org.orgId,
      parsed.data,
      await provedorOferecido(createAdminClient()),
      req.signal,
    );
    const saved = pending
      ? await finishAgentChat(
          pool,
          actor,
          parsed.data.campaign_id,
          pending.revision,
          data,
          req.signal,
        )
      : null;
    return ok({ ...data, ...saved }, { requestId, headers: { "Cache-Control": "no-store" } });
  } catch (error) {
    return fail(
      "prospecting_agent_chat_failed",
      error instanceof AgentSetupError
        ? error.message
        : "Não consegui conversar com a IA agora. Confira a credencial e o orçamento de IA; sua conversa foi mantida.",
      error instanceof AgentSetupError ? error.status : req.signal.aborted ? 499 : 503,
      { requestId, headers: { "Cache-Control": "no-store" } },
    );
  }
}