Initial 智念AIGC platform

app/api/assets/[id]/inpaint/route.ts

@@ -0,0 +1,44 @@
+import { getAsset } from "@/lib/server/data-store";
+import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
+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 { id } = await context.params;
+    const asset = await getAsset(id);
+    if (!asset) 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({
+      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);
+  }
+}

app/api/assets/[id]/route.ts

@@ -0,0 +1,18 @@
+import { deleteAsset, getAsset } from "@/lib/server/data-store";
+import { jsonError, jsonOk } from "@/lib/server/api";
+import { deleteStoredAsset } from "@/lib/server/storage";
+
+export const runtime = "nodejs";
+
+export async function DELETE(_request: Request, context: { params: Promise<{ id: string }> }) {
+  try {
+    const { id } = await context.params;
+    const asset = await getAsset(id);
+    if (!asset) return jsonError("资产不存在", 404);
+    await deleteStoredAsset(asset);
+    await deleteAsset(id);
+    return jsonOk({ ok: true, deletedAssetId: id });
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}

app/api/assets/[id]/upscale/route.ts

@@ -0,0 +1,28 @@
+import { getAsset } from "@/lib/server/data-store";
+import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
+import { requestOrigin } from "@/lib/server/runtime";
+import { submitImageJob } from "@/lib/server/generation-service";
+
+export const runtime = "nodejs";
+
+export async function POST(request: Request, context: { params: Promise<{ id: string }> }) {
+  try {
+    const { id } = await context.params;
+    const asset = await getAsset(id);
+    if (!asset) return jsonError(new Error("Asset not found."), 404);
+    const body = await readJsonBody<{
+      resolution?: "4k" | "8k";
+      scale?: number;
+    }>(request);
+    const job = await submitImageJob({
+      capability: "image.upscale",
+      imageUrls: [asset.url],
+      inputAssetIds: [asset.id],
+      resolution: body.resolution === "8k" ? "8k" : "4k",
+      scale: typeof body.scale === "number" ? body.scale : undefined
+    }, requestOrigin(request));
+    return jsonOk({ job }, { status: 202 });
+  } catch (error) {
+    return jsonError(error);
+  }
+}

app/api/assets/route.ts

@@ -0,0 +1,41 @@
+import { createAsset, listAssets } from "@/lib/server/data-store";
+import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
+import { DEFAULT_OWNER_ID } from "@/lib/server/runtime";
+import type { AssetKind } from "@/lib/types";
+
+export const runtime = "nodejs";
+
+export async function GET() {
+  try {
+    return jsonOk({ assets: await listAssets(DEFAULT_OWNER_ID) });
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}
+
+export async function POST(request: Request) {
+  try {
+    const body = await readJsonBody<{
+      url?: string;
+      name?: string;
+      kind?: AssetKind;
+      tags?: string[];
+      source?: "upload" | "generated" | "edited" | "upscaled" | "external" | "seed";
+    }>(request);
+    if (!body.url) throw new Error("url is required");
+    const asset = await createAsset({
+      ownerId: DEFAULT_OWNER_ID,
+      kind: body.kind || "image",
+      name: body.name || "外部图片",
+      url: body.url,
+      source: body.source || "external",
+      tags: body.tags || ["external"],
+      metadata: {
+        registeredFrom: "api"
+      }
+    });
+    return jsonOk({ asset }, { status: 201 });
+  } catch (error) {
+    return jsonError(error);
+  }
+}

app/api/assets/upload/route.ts

@@ -0,0 +1,26 @@
+import { jsonError, jsonOk } from "@/lib/server/api";
+import { DEFAULT_OWNER_ID, requestOrigin } from "@/lib/server/runtime";
+import { saveUploadAsset } from "@/lib/server/storage";
+
+export const runtime = "nodejs";
+
+export async function POST(request: Request) {
+  try {
+    const form = await request.formData();
+    const files = form.getAll("files").filter((item): item is File => item instanceof File);
+    if (!files.length) throw new Error("No files uploaded.");
+    const assets = await Promise.all(files.map(async (file) => {
+      return saveUploadAsset({
+        ownerId: DEFAULT_OWNER_ID,
+        bytes: Buffer.from(await file.arrayBuffer()),
+        fileName: file.name,
+        contentType: file.type || "application/octet-stream",
+        origin: requestOrigin(request),
+        tags: ["upload"]
+      });
+    }));
+    return jsonOk({ assets }, { status: 201 });
+  } catch (error) {
+    return jsonError(error);
+  }
+}

app/api/generations/image/[id]/retry/route.ts

@@ -0,0 +1,15 @@
+import { jsonError, jsonOk } from "@/lib/server/api";
+import { requestOrigin } from "@/lib/server/runtime";
+import { retryImageJob } from "@/lib/server/generation-service";
+
+export const runtime = "nodejs";
+
+export async function POST(request: Request, context: { params: Promise<{ id: string }> }) {
+  try {
+    const { id } = await context.params;
+    const job = await retryImageJob(id, requestOrigin(request));
+    return jsonOk({ job }, { status: 202 });
+  } catch (error) {
+    return jsonError(error);
+  }
+}

app/api/generations/image/[id]/route.ts

@@ -0,0 +1,39 @@
+import { deleteAsset, deleteGenerationJob, getAsset, getGenerationJob } from "@/lib/server/data-store";
+import { jsonError, jsonOk } from "@/lib/server/api";
+import { requestOrigin } from "@/lib/server/runtime";
+import { syncImageJob } from "@/lib/server/generation-service";
+import { deleteStoredAsset } from "@/lib/server/storage";
+
+export const runtime = "nodejs";
+
+export async function GET(request: Request, context: { params: Promise<{ id: string }> }) {
+  try {
+    const { id } = await context.params;
+    const existing = await getGenerationJob(id);
+    if (!existing) return jsonError(new Error("Generation job not found."), 404);
+    const job = await syncImageJob(id, requestOrigin(request));
+    return jsonOk({ job });
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}
+
+export async function DELETE(_request: Request, context: { params: Promise<{ id: string }> }) {
+  try {
+    const { id } = await context.params;
+    const job = await getGenerationJob(id);
+    if (!job || job.capability === "video.generate") return jsonError("任务不存在", 404);
+    const deletedAssetIds: string[] = [];
+    for (const assetId of job.outputAssetIds) {
+      const asset = await getAsset(assetId);
+      if (!asset) continue;
+      await deleteStoredAsset(asset);
+      await deleteAsset(asset.id);
+      deletedAssetIds.push(asset.id);
+    }
+    await deleteGenerationJob(id);
+    return jsonOk({ ok: true, deletedJobId: id, deletedAssetIds });
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}

app/api/generations/image/route.ts

@@ -0,0 +1,64 @@
+import { getGenerationJob, listGenerationJobs } from "@/lib/server/data-store";
+import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
+import { requestOrigin } from "@/lib/server/runtime";
+import { submitImageJob } from "@/lib/server/generation-service";
+import { assemblePrompt, type PromptAssemblyInput, type PromptMaterial } from "@/lib/prompt/assembler";
+import type { EnabledImageCapability } from "@/lib/types";
+
+export const runtime = "nodejs";
+
+export async function GET() {
+  try {
+    const jobs = (await listGenerationJobs()).filter((job) => job.capability !== "video.generate");
+    return jsonOk({ jobs });
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}
+
+export async function POST(request: Request) {
+  try {
+    const body = await readJsonBody<{
+      capability?: EnabledImageCapability;
+      prompt?: string;
+      imageUrls?: string[];
+      materials?: PromptMaterial[];
+      promptAssembly?: PromptAssemblyInput;
+      inputAssetIds?: string[];
+      scale?: number;
+      width?: number;
+      height?: number;
+      min_ratio?: number;
+      max_ratio?: number;
+      force_single?: boolean;
+    }>(request);
+    const capability = body.capability || "image.generate";
+    const assembled = body.promptAssembly
+      ? assemblePrompt({ ...body.promptAssembly, mode: "image", materials: body.materials || body.promptAssembly.materials || [] })
+      : undefined;
+    const materialImages = (body.materials || assembled?.materials || [])
+      .filter((material) => material.type === "image")
+      .map((material) => material.url);
+    const job = await submitImageJob({
+      capability,
+      prompt: body.prompt || assembled?.prompt,
+      imageUrls: body.imageUrls || materialImages,
+      inputAssetIds: body.inputAssetIds || (body.materials || []).map((material) => material.id).filter(Boolean) as string[],
+      scale: asNumber(body.scale),
+      width: asNumber(body.width),
+      height: asNumber(body.height),
+      min_ratio: asNumber(body.min_ratio),
+      max_ratio: asNumber(body.max_ratio),
+      force_single: Boolean(body.force_single)
+    }, requestOrigin(request));
+    return jsonOk({ job: await getGenerationJob(job.id) }, { status: 202 });
+  } catch (error) {
+    return jsonError(error);
+  }
+}
+
+function asNumber(value: unknown): number | undefined {
+  if (value === undefined || value === null || value === "") return undefined;
+  const parsed = Number(value);
+  return Number.isFinite(parsed) ? parsed : undefined;
+}

app/api/generations/video/[id]/route.ts

@@ -0,0 +1,37 @@
+import { deleteAsset, deleteGenerationJob, getAsset, getGenerationJob } from "@/lib/server/data-store";
+import { jsonError, jsonOk } from "@/lib/server/api";
+import { requestOrigin } from "@/lib/server/runtime";
+import { syncVideoJob } from "@/lib/server/video-generation-service";
+import { deleteStoredAsset } from "@/lib/server/storage";
+
+export const runtime = "nodejs";
+
+export async function GET(request: Request, context: { params: Promise<{ id: string }> }) {
+  try {
+    const { id } = await context.params;
+    const job = await syncVideoJob(id, requestOrigin(request));
+    return jsonOk({ job });
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}
+
+export async function DELETE(_request: Request, context: { params: Promise<{ id: string }> }) {
+  try {
+    const { id } = await context.params;
+    const job = await getGenerationJob(id);
+    if (!job || job.capability !== "video.generate") return jsonError("任务不存在", 404);
+    const deletedAssetIds: string[] = [];
+    for (const assetId of job.outputAssetIds) {
+      const asset = await getAsset(assetId);
+      if (!asset) continue;
+      await deleteStoredAsset(asset);
+      await deleteAsset(asset.id);
+      deletedAssetIds.push(asset.id);
+    }
+    await deleteGenerationJob(id);
+    return jsonOk({ ok: true, deletedJobId: id, deletedAssetIds });
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}

app/api/generations/video/route.ts

@@ -0,0 +1,25 @@
+import { listGenerationJobs } from "@/lib/server/data-store";
+import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
+import { requestOrigin } from "@/lib/server/runtime";
+import { submitVideoJob, type SubmitVideoJobInput } from "@/lib/server/video-generation-service";
+
+export const runtime = "nodejs";
+
+export async function GET() {
+  try {
+    const jobs = (await listGenerationJobs()).filter((job) => job.capability === "video.generate");
+    return jsonOk({ jobs });
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}
+
+export async function POST(request: Request) {
+  try {
+    const body = await readJsonBody<SubmitVideoJobInput & Record<string, unknown>>(request);
+    const job = await submitVideoJob(body, requestOrigin(request));
+    return jsonOk({ job }, { status: 202 });
+  } catch (error) {
+    return jsonError(error);
+  }
+}

app/api/health/route.ts

@@ -0,0 +1,38 @@
+import { jsonOk } from "@/lib/server/api";
+import { getEffectiveImageEngine, getEvolinkImageSettings, shouldMockEvolinkApi } from "@/lib/evolink/image-client";
+import { getVisibleImageCapabilities } from "@/lib/jimeng/capabilities";
+import { shouldMockVisualApi } from "@/lib/volcengine/visual-client";
+import { getSeedanceConfig, shouldMockSeedance } from "@/lib/seedance/client";
+
+export const runtime = "nodejs";
+
+export async function GET() {
+  const evolink = getEvolinkImageSettings();
+  return jsonOk({
+    ok: true,
+    appId: "zhinian-web-studio",
+    webOnly: true,
+    visualApiMode: shouldMockVisualApi() ? "mock" : "volcengine",
+    evolinkMode: shouldMockEvolinkApi() ? "mock" : "evolink",
+    seedanceMode: shouldMockSeedance() ? "mock" : "seedance",
+    capabilities: [
+      ...getVisibleImageCapabilities().map((capability) => {
+        const engine = getEffectiveImageEngine(capability.id);
+        return {
+          id: capability.id,
+          label: capability.label,
+          engine,
+          engineLabel: engine === "evolink" ? "EvoLink" : "即梦",
+          reqKey: engine === "evolink" ? evolink.model : capability.reqKey
+        };
+      }),
+      {
+        id: "video.generate",
+        label: "Seedance 视频生成",
+        engine: "seedance",
+        engineLabel: "Seedance",
+        reqKey: getSeedanceConfig().model
+      }
+    ]
+  });
+}

app/api/prompt/assemble/route.ts

@@ -0,0 +1,13 @@
+import { assemblePrompt, type PromptAssemblyInput } from "@/lib/prompt/assembler";
+import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
+
+export const runtime = "nodejs";
+
+export async function POST(request: Request) {
+  try {
+    const body = await readJsonBody<PromptAssemblyInput & Record<string, unknown>>(request);
+    return jsonOk(assemblePrompt(body));
+  } catch (error) {
+    return jsonError(error);
+  }
+}

app/api/settings/route.ts

@@ -0,0 +1,21 @@
+import { getApiSettings, saveApiSettings } from "@/lib/server/app-settings";
+import { jsonError, jsonOk, readJsonBody } from "@/lib/server/api";
+
+export const runtime = "nodejs";
+
+export async function GET() {
+  try {
+    return jsonOk(await getApiSettings());
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}
+
+export async function POST(request: Request) {
+  try {
+    const body = await readJsonBody<{ values?: Record<string, unknown> }>(request);
+    return jsonOk(await saveApiSettings(body.values || {}));
+  } catch (error) {
+    return jsonError(error, 500);
+  }
+}