import { NextResponse } from "next/server";
import { z } from "zod";
import { auth } from "@/app/(auth)/auth";
import { getChatById } from "@/lib/db/queries";
import {
  buildClawdCancelKey,
  cancelClawdRequest,
} from "@/lib/openclaw-chat/cancel-registry";
import {
  OPENCLAW_AUTH_REQUIRED_CODE,
  OPENCLAW_AUTH_REQUIRED_MESSAGE,
} from "@/lib/openclaw-chat/errors";
import { deletePersistedClawdTask } from "@/lib/openclaw-chat/task-store";

const cancelSchema = z.object({
  id: z.string().uuid(),
  userMessageId: z.string().uuid(),
});

export async function POST(request: Request) {
  const session = await auth().catch(() => null);
  if (!session?.user?.id) {
    return NextResponse.json(
      {
        error: OPENCLAW_AUTH_REQUIRED_MESSAGE,
        code: OPENCLAW_AUTH_REQUIRED_CODE,
      },
      { status: 401 }
    );
  }

  const parsed = cancelSchema.safeParse(await request.json().catch(() => null));
  if (!parsed.success) {
    return NextResponse.json({ error: "请求参数不正确" }, { status: 400 });
  }

  const existingChat = await getChatById({ id: parsed.data.id }).catch(() => null);
  if (existingChat && existingChat.userId !== session.user.id) {
    return NextResponse.json({ error: "Forbidden" }, { status: 403 });
  }

  const active = cancelClawdRequest(
    buildClawdCancelKey({
      userId: session.user.id,
      chatId: parsed.data.id,
      userMessageId: parsed.data.userMessageId,
    })
  );

  const deleted = await deletePersistedClawdTask({
    chatId: parsed.data.id,
    userMessageId: parsed.data.userMessageId,
  }).catch(() => []);

  return NextResponse.json({
    canceled: true,
    active,
    deleted: deleted.length > 0,
  });
}