import { NextResponse } from "next/server";
import { auth } from "@/app/(auth)/auth";
import {
  createCanvasImageTask,
  enqueueCanvasImageTask,
  getCanvasImageTaskForUser,
} from "@/lib/canvas/image-task-queue";
import {
  generateCanvasImage,
  type CanvasGenerateImageRequest,
} from "@/lib/canvas/generate-image";

export const maxDuration = 120;

interface GenerateImageRequest extends CanvasGenerateImageRequest {
  async?: boolean;
}

function toGenerateImagePayload(body: GenerateImageRequest): CanvasGenerateImageRequest {
  return {
    prompt: String(body.prompt || "").trim(),
    negativePrompt:
      typeof body.negativePrompt === "string" ? body.negativePrompt : undefined,
    model: typeof body.model === "string" ? body.model : undefined,
    aspectRatio:
      typeof body.aspectRatio === "string" && body.aspectRatio.trim()
        ? body.aspectRatio.trim()
        : "auto",
    imageSize:
      typeof body.imageSize === "string" && body.imageSize.trim()
        ? body.imageSize.trim()
        : undefined,
    imageCount:
      typeof body.imageCount === "number" && Number.isFinite(body.imageCount)
        ? body.imageCount
        : 1,
    referenceImage:
      typeof body.referenceImage === "string" ? body.referenceImage : undefined,
    referenceImages: Array.isArray(body.referenceImages)
      ? body.referenceImages.map((item) => String(item || "").trim()).filter(Boolean)
      : [],
  };
}

export async function GET(request: Request) {
  try {
    const session = await auth();
    if (!session?.user?.id) {
      return NextResponse.json({ success: false, error: "未登录" }, { status: 401 });
    }

    const { searchParams } = new URL(request.url);
    const taskId = String(searchParams.get("taskId") || "").trim();
    if (!taskId) {
      return NextResponse.json(
        { success: false, error: "缺少 taskId 参数" },
        { status: 400 }
      );
    }

    const task = await getCanvasImageTaskForUser({ taskId, userId: session.user.id });
    if (!task) {
      return NextResponse.json({ success: false, error: "任务不存在" }, { status: 404 });
    }

    return NextResponse.json({
      success: true,
      task: {
        id: task.id,
        status: task.status,
        attempts: task.attempts,
        error: task.error || null,
        result: task.result || null,
        createdAt: task.createdAt,
        startedAt: task.startedAt,
        completedAt: task.completedAt,
      },
    });
  } catch (error) {
    console.error("[Canvas Generate Image][TaskStatus] Error:", error);
    return NextResponse.json(
      {
        success: false,
        error: error instanceof Error ? error.message : "查询任务失败",
      },
      { status: 500 }
    );
  }
}

export async function POST(request: Request) {
  try {
    const session = await auth();
    if (!session?.user?.id) {
      return NextResponse.json(
        { success: false, error: "未登录" },
        { status: 401 }
      );
    }

    const body: GenerateImageRequest = await request.json();
    const payload = toGenerateImagePayload(body);
    if (!payload.prompt) {
      return NextResponse.json(
        { success: false, error: "请输入提示词" },
        { status: 400 }
      );
    }

    const asyncMode = body.async === true;
    if (asyncMode) {
      const task = await createCanvasImageTask({
        userId: session.user.id,
        request: payload,
      });
      await enqueueCanvasImageTask(task.id);
      return NextResponse.json({
        success: true,
        taskId: task.id,
        status: task.status,
      });
    }

    const result = await generateCanvasImage(payload);
    return NextResponse.json({
      success: true,
      data: result,
    });
  } catch (error) {
    console.error("[Canvas Generate Image] Error:", error);
    return NextResponse.json(
      { success: false, error: error instanceof Error ? error.message : "生成失败" },
      { status: 500 }
    );
  }
}