feat: add durable CMS-free image video subsystem and UI components

2026-09-15 01:04:09 +08:00
parent 9349317eb8
commit 40e03e393c
14 changed files with 795 additions and 0 deletions
@@ -0,0 +1,98 @@
+import { mkdtemp, mkdir, readFile, writeFile } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { Readable } from "node:stream";
+import sharp from "sharp";
+import { describe, expect, it } from "vitest";
+import * as harness from "../src/index.js";
+import type { ImageVideoRemote, ImageVideoRecord } from "../src/image-video/contracts.js";
+
+async function fixture() {
+  expect(harness).toHaveProperty("ImageVideoService");
+  const root = await mkdtemp(join(tmpdir(), "i2v-test-"));
+  await mkdir(join(root, "ses_owner"));
+  const attachments = new harness.FileAttachmentStore({ sessionsRoot: root });
+  const attachment = await attachments.add("ses_owner", { name: "test.png", mediaType: "image/png", stream: Readable.from(await sharp({ create: { width: 8, height: 8, channels: 3, background: "red" } }).png().toBuffer()) });
+  const calls: string[] = [];
+  const remote: ImageVideoRemote = {
+    preflight() { calls.push("identity"); return Promise.resolve({ tenantId: "tenant", platform: "tiktok" }); },
+    initialize() { calls.push("initialize"); return Promise.resolve({ uploadId: "upload_1", objectKey: "obj", uploadUrl: "https://public.example/put?token=private" }); },
+    put() { calls.push("put"); return Promise.resolve(); }, finalize() { calls.push("finalize"); return Promise.resolve(); },
+    importStatus() { calls.push("import"); return Promise.resolve({ state: "ready", assetId: "asset_1" }); },
+    submit() { calls.push("submit"); return Promise.resolve({ jobId: "remote_job_1" }); },
+    status() { calls.push("status"); return Promise.resolve({ state: "queued" }); },
+  };
+  const options = { root: join(root, "tasks"), attachments, remote, collect: () => Promise.reject(new Error("No fake video")) };
+  const service = new harness.ImageVideoService(options);
+  const draft = { attachmentId: attachment.attachmentId, title: "Motion", prompt: "Natural movement with ambient sound", durationSeconds: 10 };
+  return { service, options, draft, calls, remote, root };
+}
+
+describe("durable image video tasks", () => {
+  it("copies a local reference without transmitting it and hides private bindings", async () => {
+    const f = await fixture(); const task = await f.service.create("ses_owner", f.draft);
+    expect(f.calls).toEqual([]); expect(task.state).toBe("draft");
+    await f.options.attachments.removeAll("ses_owner");
+    expect((await f.service.media(task.id, "reference")).size).toBeGreaterThan(0);
+    expect(JSON.stringify(await f.service.list())).not.toMatch(/ownerSessionId|uploadUrl|objectKey|referenceHash|\/private\//);
+  });
+  it("rejects foreign owners before any external operation", async () => {
+    const f = await fixture(); const task = await f.service.create("ses_owner", f.draft);
+    await expect(f.service.upload("ses_other", task.id)).rejects.toMatchObject({ code: "I2V_TASK_NOT_FOUND" });
+    await expect(f.service.submit("ses_other", task.id)).rejects.toMatchObject({ code: "I2V_TASK_NOT_FOUND" });
+    expect(f.calls).toEqual([]);
+  });
+  it("requires upload first and coalesces concurrent submission, restoring only the known job", async () => {
+    const f = await fixture(); const task = await f.service.create("ses_owner", f.draft);
+    await expect(f.service.submit("ses_owner", task.id)).rejects.toMatchObject({ code: "I2V_NOT_READY" });
+    await f.service.upload("ses_owner", task.id); await f.service.refresh(task.id);
+    const [a, b] = await Promise.all([f.service.submit("ses_owner", task.id), f.service.submit("ses_owner", task.id)]);
+    expect(a.jobId).toBe("remote_job_1"); expect(b.jobId).toBe(a.jobId);
+    const restored = new harness.ImageVideoService(f.options);
+    expect((await restored.view(task.id)).jobId).toBe(a.jobId);
+    await restored.submit("ses_owner", task.id);
+    expect(f.calls.filter(call => call === "submit")).toHaveLength(1);
+    expect(JSON.stringify(await restored.view(task.id))).not.toContain("private");
+  });
+  it("isolates unknown submissions after timeout and after process interruption", async () => {
+    const f = await fixture(); const task = await f.service.create("ses_owner", f.draft);
+    await f.service.upload("ses_owner", task.id); await f.service.refresh(task.id);
+    f.remote.submit = () => { f.calls.push("submit"); return Promise.reject(new Error("token=private")); };
+    expect((await f.service.submit("ses_owner", task.id)).state).toBe("unknown_submission");
+    await expect(new harness.ImageVideoService(f.options).submit("ses_owner", task.id)).rejects.toMatchObject({ code: "I2V_UNKNOWN_SUBMISSION" });
+    expect(f.calls.filter(call => call === "submit")).toHaveLength(1);
+    const file = join(f.options.root, task.id, "record.json"); const record = JSON.parse(await readFile(file, "utf8")) as ImageVideoRecord;
+    record.state = "submitting"; await writeFile(file, JSON.stringify(record));
+    expect((await new harness.ImageVideoService(f.options).view(task.id)).state).toBe("unknown_submission");
+  });
+  it("never reinitializes uncertain uploads and blocks stale persisted revisions", async () => {
+    const f = await fixture(); const task = await f.service.create("ses_owner", f.draft);
+    f.remote.initialize = () => { f.calls.push("initialize"); return Promise.reject(new Error("timeout")); };
+    expect((await f.service.upload("ses_owner", task.id)).state).toBe("unknown_upload");
+    await expect(f.service.upload("ses_owner", task.id)).rejects.toMatchObject({ code: "I2V_UNKNOWN_UPLOAD" });
+    expect(f.calls.filter(call => call === "initialize")).toHaveLength(1);
+    const file = join(f.options.root, task.id, "record.json"); const record = JSON.parse(await readFile(file, "utf8")) as ImageVideoRecord;
+    record.requestHash = "invalid"; await writeFile(file, JSON.stringify(record));
+    await expect(f.service.view(task.id)).rejects.toMatchObject({ code: "I2V_RECORD_INVALID" });
+  });
+  it("throttles read-only polling even across instances and accepts no arbitrary identifiers", async () => {
+    const f = await fixture(); const task = await f.service.create("ses_owner", f.draft);
+    await f.service.upload("ses_owner", task.id); await f.service.refresh(task.id); await f.service.submit("ses_owner", task.id);
+    await Promise.all([f.service.refresh(task.id), new harness.ImageVideoService(f.options).refresh(task.id)]);
+    expect(f.calls.filter(call => call === "status")).toHaveLength(1);
+    await expect(f.service.view("../../etc/passwd")).rejects.toMatchObject({ code: "I2V_TASK_NOT_FOUND" });
+  });
+  it("rejects a late submit result from overwriting a newer revision, while retaining its known receipt", async () => {
+    const f = await fixture(); const task = await f.service.create("ses_owner", f.draft);
+    await f.service.upload("ses_owner", task.id); await f.service.refresh(task.id);
+    f.remote.submit = async () => {
+      const file = join(f.options.root, task.id, "record.json"); const newer = JSON.parse(await readFile(file, "utf8")) as ImageVideoRecord;
+      newer.revision += 1; newer.state = "unknown_submission"; await writeFile(file, JSON.stringify(newer));
+      return { jobId: "late_job" };
+    };
+    await expect(f.service.submit("ses_owner", task.id)).rejects.toMatchObject({ code: "I2V_STALE_REVISION" });
+    expect((await f.service.view(task.id)).state).toBe("unknown_submission");
+    const receipt = JSON.parse(await readFile(join(f.options.root, task.id, "submission-receipt.json"), "utf8")) as { jobId: string };
+    expect(receipt.jobId).toBe("late_job");
+  });
+});