feat(modality)!: SnapOtter 2.0 phase 3 modality framework: media/doc engines, pool routing, display modes (#218)

2026-08-03 07:46:42 +02:00 · 2026-06-13 10:18:39 +08:00
parent c451b939c7
commit d647d8ed19
99 changed files with 4380 additions and 1813 deletions
@@ -0,0 +1,72 @@
+import { mkdtempSync, rmSync } from "node:fs";
+import { readFile } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { qpdfAvailable } from "@snapotter/doc-engine";
+import { ffmpegAvailable } from "@snapotter/media-engine";
+import { afterAll, describe, expect, it } from "vitest";
+import { InputValidationError } from "../../apps/api/src/modality/contract.js";
+import { DocumentInputHandler } from "../../apps/api/src/modality/document-input.js";
+import { MediaInputHandler } from "../../apps/api/src/modality/media-input.js";
+
+const scratchDir = mkdtempSync(join(tmpdir(), "modality-input-"));
+afterAll(() => rmSync(scratchDir, { recursive: true, force: true }));
+
+describe.skipIf(!ffmpegAvailable())("MediaInputHandler (requires ffmpeg)", () => {
+  it("accepts the mp4 fixture as video", async () => {
+    const buf = await readFile(join(process.cwd(), "tests/fixtures/media/tiny.mp4"));
+    const out = await new MediaInputHandler("video").prepare(buf, "tiny.mp4", { scratchDir });
+    expect(out.buffer).toBe(buf);
+  });
+
+  it("rejects an audio-only file as video", async () => {
+    const buf = await readFile(join(process.cwd(), "tests/fixtures/media/tiny.mp3"));
+    await expect(
+      new MediaInputHandler("video").prepare(buf, "fake.mp4", { scratchDir }),
+    ).rejects.toThrow(InputValidationError);
+  });
+
+  it("enforces the duration cap", async () => {
+    const { env } = await import("../../apps/api/src/config.js");
+    const original = env.MAX_AUDIO_DURATION_S;
+    (env as Record<string, unknown>).MAX_AUDIO_DURATION_S = 0.5;
+    try {
+      const buf = await readFile(join(process.cwd(), "tests/fixtures/media/tiny.mp3"));
+      await expect(
+        new MediaInputHandler("audio").prepare(buf, "tiny.mp3", { scratchDir }),
+      ).rejects.toThrow(/exceeds the maximum/);
+    } finally {
+      (env as Record<string, unknown>).MAX_AUDIO_DURATION_S = original;
+    }
+  });
+});
+
+describe("DocumentInputHandler", () => {
+  it("rejects a non-pdf with pdf extension", async () => {
+    await expect(
+      new DocumentInputHandler().prepare(Buffer.from("hello"), "fake.pdf", { scratchDir }),
+    ).rejects.toThrow(/PDF header/);
+  });
+
+  it.skipIf(!qpdfAvailable())("accepts the 3-page fixture and enforces page caps", async () => {
+    const buf = await readFile(join(process.cwd(), "tests/fixtures/test-3page.pdf"));
+    const out = await new DocumentInputHandler().prepare(buf, "test.pdf", { scratchDir });
+    expect(out.buffer).toBe(buf);
+    const { env } = await import("../../apps/api/src/config.js");
+    const original = env.MAX_PDF_PAGES;
+    (env as Record<string, unknown>).MAX_PDF_PAGES = 2;
+    try {
+      await expect(
+        new DocumentInputHandler().prepare(buf, "test.pdf", { scratchDir }),
+      ).rejects.toThrow(/exceeding the maximum/);
+    } finally {
+      (env as Record<string, unknown>).MAX_PDF_PAGES = original;
+    }
+  });
+
+  it("accepts the docx fixture container", async () => {
+    const buf = await readFile(join(process.cwd(), "tests/fixtures/documents/tiny.docx"));
+    const out = await new DocumentInputHandler().prepare(buf, "tiny.docx", { scratchDir });
+    expect(out.buffer).toBe(buf);
+  });
+});