ai-chatbot/app/api/canvas/generate/image/route.ts

import { NextResponse } from "next/server";
import { auth } from "@/app/(auth)/auth";
import {
  createCanvasImageTask,
  enqueueCanvasImageTask,
  getCanvasImageTaskForUser,
  getCanvasDetailReplacementTasksForUserContext,
} from "@/lib/canvas/image-task-queue";
import {
  generateCanvasImage,
  type CanvasGenerateImageRequest,
} from "@/lib/canvas/generate-image";
import { getPointsSystemConfig } from "@/lib/db/queries";
import { prepareFeaturePointsCharge } from "@/lib/points/feature-charge";
import {
  getImageToolCreditCost,
  type ImageToolCreditFeature,
} from "@/lib/points/config";

export const maxDuration = 1200;

interface GenerateImageRequest extends CanvasGenerateImageRequest {
  async?: boolean;
}

function normalizeBillingFeature(
  value: unknown
): Extract<ImageToolCreditFeature, "multi-view" | "repaint"> | null {
  if (value === "multi-view" || value === "repaint") return value;
  return null;
}

function toGenerateImagePayload(body: GenerateImageRequest): CanvasGenerateImageRequest {
  const requestedQuality =
    body.providerImageParams &&
    typeof body.providerImageParams === "object" &&
    !Array.isArray(body.providerImageParams)
      ? String(body.providerImageParams.quality || "").trim().toLowerCase()
      : "";
  return {
    prompt: String(body.prompt || "").trim(),
    displayPrompt:
      typeof body.displayPrompt === "string" ? body.displayPrompt : undefined,
    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,
    promptExtend:
      typeof body.promptExtend === "boolean" ? body.promptExtend : undefined,
    providerImageParams: new Set(["low", "medium", "high"]).has(
      requestedQuality,
    )
      ? { quality: requestedQuality }
      : undefined,
    referenceImage:
      typeof body.referenceImage === "string" ? body.referenceImage : undefined,
    referenceImages: Array.isArray(body.referenceImages)
      ? body.referenceImages.map((item) => String(item || "").trim()).filter(Boolean)
      : [],
    debugLabel:
      typeof body.debugLabel === "string" ? body.debugLabel : undefined,
    debugMeta:
      body.debugMeta &&
      typeof body.debugMeta === "object" &&
      !Array.isArray(body.debugMeta)
        ? body.debugMeta
        : undefined,
  };
}

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();
    const chatId = String(searchParams.get("chatId") || "").trim();
    const projectId = String(searchParams.get("projectId") || "").trim();
    if (!taskId && chatId) {
      const tasks = await getCanvasDetailReplacementTasksForUserContext({
        userId: session.user.id,
        chatId,
        projectId: projectId || null,
      });
      return NextResponse.json({
        success: true,
        tasks,
      });
    }
    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,
        queueStatus: task.queueStatus,
      },
    });
  } 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: CanvasGenerateImageRequest = {
      ...toGenerateImagePayload(body),
      ownerKey: session.user.id,
    };
    if (!payload.prompt) {
      return NextResponse.json(
        { success: false, error: "请输入提示词" },
        { status: 400 }
      );
    }

    const billingFeature = normalizeBillingFeature(body.billingFeature);
    const pointsCharge = billingFeature
      ? await (async () => {
          const pointsConfig = await getPointsSystemConfig();
          return prepareFeaturePointsCharge({
            userId: session.user.id,
            unitCost: getImageToolCreditCost(pointsConfig, billingFeature),
            reason:
              billingFeature === "multi-view"
                ? "image_multi_view"
                : "image_repaint",
            modelId: "canvas-generate-image",
            note:
              billingFeature === "multi-view"
                ? "图片多视角生成消耗积分"
                : "图片重绘消耗积分",
            pointsConfig,
          });
        })()
      : null;
    if (pointsCharge && !pointsCharge.ok) {
      return NextResponse.json(pointsCharge.payload, { status: pointsCharge.status });
    }

    const asyncMode = body.async === true;
    if (asyncMode) {
      const task = await createCanvasImageTask({
        userId: session.user.id,
        request: payload as unknown as Record<string, unknown>,
      });
      await enqueueCanvasImageTask(task.id);
      const currentTask = await getCanvasImageTaskForUser({
        taskId: task.id,
        userId: session.user.id,
      });
      if (pointsCharge?.chargeable) {
        try {
          await pointsCharge.consume({
            note:
              billingFeature === "multi-view"
                ? `图片多视角生成消耗积分（${task.id}）`
                : `图片重绘消耗积分（${task.id}）`,
          });
        } catch (pointsError) {
          console.error("[Points] Failed to consume canvas image task points:", pointsError);
        }
      }
      return NextResponse.json({
        success: true,
        taskId: task.id,
        status: currentTask?.status || task.status,
        queueStatus: currentTask?.queueStatus,
      });
    }

    const result = await generateCanvasImage(payload);
    if (pointsCharge?.chargeable) {
      try {
        await pointsCharge.consume();
      } catch (pointsError) {
        console.error("[Points] Failed to consume canvas image points:", pointsError);
      }
    }
    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 }
    );
  }
}