137 lines
3.9 KiB
TypeScript
137 lines
3.9 KiB
TypeScript
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 }
|
|
);
|
|
}
|
|
}
|