48 lines
1.7 KiB
TypeScript
48 lines
1.7 KiB
TypeScript
import { getAsset } from "@/lib/server/data-store";
|
|
import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
|
|
import { requireAppUser } from "@/lib/server/auth/current-user";
|
|
import { requestOrigin } from "@/lib/server/runtime";
|
|
import { saveMaskDataUrl } from "@/lib/server/storage";
|
|
import { submitImageJob } from "@/lib/server/generation-service";
|
|
|
|
export const runtime = "nodejs";
|
|
|
|
export async function POST(request: Request, context: { params: Promise<{ id: string }> }) {
|
|
try {
|
|
const user = await requireAppUser();
|
|
const { id } = await context.params;
|
|
const asset = await getAsset(id);
|
|
if (!asset || asset.ownerId !== user.id) return jsonError(new Error("Asset not found."), 404);
|
|
const body = await readJsonBody<{
|
|
prompt?: string;
|
|
maskDataUrl?: string;
|
|
maskUrl?: string;
|
|
seed?: number;
|
|
}>(request);
|
|
let maskUrl = body.maskUrl;
|
|
let maskAssetId: string | undefined;
|
|
if (body.maskDataUrl) {
|
|
const mask = await saveMaskDataUrl({
|
|
ownerId: asset.ownerId,
|
|
dataUrl: body.maskDataUrl,
|
|
origin: requestOrigin(request),
|
|
jobHint: asset.id
|
|
});
|
|
maskUrl = mask.url;
|
|
maskAssetId = mask.id;
|
|
}
|
|
if (!maskUrl) throw new Error("maskDataUrl or maskUrl is required for inpainting.");
|
|
const job = await submitImageJob({
|
|
ownerId: user.id,
|
|
capability: "image.inpaint",
|
|
prompt: body.prompt || "删除",
|
|
imageUrls: [asset.url, maskUrl],
|
|
inputAssetIds: [asset.id, ...(maskAssetId ? [maskAssetId] : [])],
|
|
seed: typeof body.seed === "number" ? body.seed : undefined
|
|
}, requestOrigin(request));
|
|
return jsonOk({ job }, { status: 202 });
|
|
} catch (error) {
|
|
return jsonError(error);
|
|
}
|
|
}
|