111 lines
3.3 KiB
TypeScript
111 lines
3.3 KiB
TypeScript
import { NextRequest, NextResponse } from "next/server";
|
||
import { auth } from "@/app/(auth)/auth";
|
||
import {
|
||
executeExtendJob,
|
||
ExtendServiceError,
|
||
} from "@/lib/ai/extend/service";
|
||
|
||
export const maxDuration = 180;
|
||
|
||
export async function POST(request: NextRequest) {
|
||
try {
|
||
const session = await auth();
|
||
if (!session?.user?.id) {
|
||
return NextResponse.json({ error: "未授权，请先登录" }, { status: 401 });
|
||
}
|
||
|
||
const contentType = request.headers.get("content-type") || "";
|
||
|
||
let imageUrl: string | null = null;
|
||
let imageBase64: string | null = null;
|
||
let prompt = "图像拓展处理";
|
||
let chatId: string | null = null;
|
||
let messageId: string | null = null;
|
||
|
||
if (contentType.includes("application/json")) {
|
||
const body = await request.json();
|
||
imageUrl = body.image_url ? String(body.image_url).trim() : null;
|
||
imageBase64 = body.image_base64 ? String(body.image_base64).trim() : null;
|
||
prompt = body.prompt || prompt;
|
||
chatId = body.chatId || null;
|
||
messageId = body.messageId || null;
|
||
|
||
if (!imageUrl && !imageBase64) {
|
||
return NextResponse.json(
|
||
{ error: "缺少image_base64或image_url参数" },
|
||
{ status: 400 },
|
||
);
|
||
}
|
||
|
||
const result = await executeExtendJob({
|
||
userId: session.user.id,
|
||
imageUrl,
|
||
imageBase64,
|
||
prompt,
|
||
chatId,
|
||
messageId,
|
||
xScale: body.xScale ? Number(body.xScale) : undefined,
|
||
yScale: body.yScale ? Number(body.yScale) : undefined,
|
||
});
|
||
|
||
return NextResponse.json({
|
||
success: true,
|
||
original_url: result.original_url,
|
||
thumbnail_url: result.thumbnail_url,
|
||
base64: result.image_base64,
|
||
processing_time: result.processing_time,
|
||
});
|
||
}
|
||
|
||
if (contentType.includes("multipart/form-data")) {
|
||
const formData = await request.formData();
|
||
const file = formData.get("file") as File | null;
|
||
if (!file) {
|
||
return NextResponse.json({ error: "缺少file参数" }, { status: 400 });
|
||
}
|
||
|
||
const arrayBuffer = await file.arrayBuffer();
|
||
const buffer = Buffer.from(arrayBuffer);
|
||
const result = await executeExtendJob({
|
||
userId: session.user.id,
|
||
imageBase64: buffer.toString("base64"),
|
||
prompt: String(formData.get("prompt") || prompt),
|
||
chatId: String(formData.get("chatId") || "") || null,
|
||
messageId: String(formData.get("messageId") || "") || null,
|
||
});
|
||
|
||
return NextResponse.json({
|
||
success: true,
|
||
original_url: result.original_url,
|
||
thumbnail_url: result.thumbnail_url,
|
||
base64: result.image_base64,
|
||
processing_time: result.processing_time,
|
||
});
|
||
}
|
||
|
||
return NextResponse.json(
|
||
{ error: "不支持的Content-Type，请使用application/json或multipart/form-data" },
|
||
{ status: 400 },
|
||
);
|
||
} catch (error) {
|
||
if (error instanceof ExtendServiceError) {
|
||
return NextResponse.json(
|
||
{
|
||
error: error.message,
|
||
detail: error.detail || error.message,
|
||
},
|
||
{ status: error.status },
|
||
);
|
||
}
|
||
|
||
console.error("[Extend API] 错误:", error);
|
||
return NextResponse.json(
|
||
{
|
||
error: "图像拓展失败",
|
||
detail: error instanceof Error ? error.message : String(error),
|
||
},
|
||
{ status: 500 },
|
||
);
|
||
}
|
||
}
|