101 lines
3.0 KiB
TypeScript
101 lines
3.0 KiB
TypeScript
import { NextRequest, NextResponse } from "next/server";
|
|
|
|
import { auth } from "@/app/(auth)/auth";
|
|
import {
|
|
BoxCutoutServiceError,
|
|
executeBoxCutoutJob,
|
|
} from "@/lib/ai/box-cutout/service";
|
|
import { RmbgServiceError } from "@/lib/ai/rmbg/service";
|
|
import { getPointsSystemConfig } from "@/lib/db/queries";
|
|
import { prepareFeaturePointsCharge } from "@/lib/points/feature-charge";
|
|
import { getImageToolCreditCost } from "@/lib/points/config";
|
|
|
|
export const runtime = "nodejs";
|
|
export const maxDuration = 60;
|
|
|
|
type BoxCutoutRequest = {
|
|
imageUrl?: string;
|
|
cropRect?: {
|
|
x?: number;
|
|
y?: number;
|
|
width?: number;
|
|
height?: number;
|
|
};
|
|
paddingRatio?: number;
|
|
prompt?: string;
|
|
chatId?: string | null;
|
|
messageId?: string | null;
|
|
skipGallerySave?: boolean;
|
|
sourceNodeId?: string;
|
|
};
|
|
|
|
export async function POST(request: NextRequest) {
|
|
try {
|
|
const session = await auth();
|
|
if (!session?.user?.id) {
|
|
return NextResponse.json({ success: false, error: "未授权，请先登录" }, { status: 401 });
|
|
}
|
|
|
|
const body = (await request.json()) as BoxCutoutRequest;
|
|
const rawImageUrl = String(body.imageUrl || "").trim();
|
|
if (!rawImageUrl) {
|
|
return NextResponse.json({ success: false, error: "缺少 imageUrl 参数" }, { status: 400 });
|
|
}
|
|
|
|
const resolvedImageUrl = new URL(rawImageUrl, request.url).toString();
|
|
const pointsConfig = await getPointsSystemConfig();
|
|
const pointsCharge = await prepareFeaturePointsCharge({
|
|
userId: session.user.id,
|
|
unitCost: getImageToolCreditCost(pointsConfig, "box-cutout"),
|
|
reason: "image_box_cutout",
|
|
modelId: "box-cutout",
|
|
note: "框选抠图消耗积分",
|
|
pointsConfig,
|
|
});
|
|
if (!pointsCharge.ok) {
|
|
return NextResponse.json(pointsCharge.payload, { status: pointsCharge.status });
|
|
}
|
|
|
|
const result = await executeBoxCutoutJob({
|
|
userId: session.user.id,
|
|
imageUrl: resolvedImageUrl,
|
|
cropRect: body.cropRect || {},
|
|
paddingRatio: body.paddingRatio,
|
|
prompt: body.prompt || "框选抠图",
|
|
chatId: body.chatId || null,
|
|
messageId: body.messageId || null,
|
|
skipGallerySave: body.skipGallerySave === true,
|
|
sourceNodeId: body.sourceNodeId,
|
|
});
|
|
|
|
if (pointsCharge.chargeable) {
|
|
try {
|
|
await pointsCharge.consume();
|
|
} catch (pointsError) {
|
|
console.error("[Points] Failed to consume box cutout points:", pointsError);
|
|
}
|
|
}
|
|
|
|
return NextResponse.json(result);
|
|
} catch (error) {
|
|
if (error instanceof BoxCutoutServiceError || error instanceof RmbgServiceError) {
|
|
return NextResponse.json(
|
|
{
|
|
success: false,
|
|
error: error.detail || error.message,
|
|
},
|
|
{ status: error.status },
|
|
);
|
|
}
|
|
|
|
console.error("[BoxCutout] 处理失败:", error);
|
|
return NextResponse.json(
|
|
{
|
|
success: false,
|
|
error: error instanceof Error ? error.message : "框选抠图失败",
|
|
},
|
|
{ status: 500 },
|
|
);
|
|
}
|
|
}
|