Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/content/__tests__/buildFilterComplex.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ describe("buildFilterComplex", () => {
});

expect(result).toContain("[0:v]crop=ih*9/16:ih,scale=720:1280[video_base]");
expect(result).toContain("[1:v]scale=150:150[ovr_0]");
expect(result).toContain("[1:v]scale=250:250[ovr_0]");
expect(result).toContain("overlay=30:30");
expect(result).toContain("[out]");
});
Expand All @@ -22,7 +22,7 @@ describe("buildFilterComplex", () => {
});

expect(result).toMatch(/overlay=30:30/);
expect(result).toMatch(/overlay=30:200/);
expect(result).toMatch(/overlay=30:300/);
});

it("chains multiple overlays correctly", () => {
Expand Down
77 changes: 77 additions & 0 deletions src/content/__tests__/buildStaticImageArgs.test.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,77 @@
import { describe, it, expect } from "vitest";

import { buildStaticImageArgs } from "../buildStaticImageArgs";

describe("buildStaticImageArgs", () => {
it("builds ffmpeg args with image input and overlay images", () => {
const args = buildStaticImageArgs({
imagePath: "/tmp/base.png",
overlayImagePaths: ["/tmp/ovr-0.png", "/tmp/ovr-1.png"],
outputPath: "/tmp/output.png",
});

expect(args).toContain("-y");
expect(args).toContain("/tmp/base.png");
expect(args).toContain("/tmp/ovr-0.png");
expect(args).toContain("/tmp/ovr-1.png");
expect(args).toContain("/tmp/output.png");
});

it("includes filter_complex with crop, scale, and overlays", () => {
const args = buildStaticImageArgs({
imagePath: "/tmp/base.png",
overlayImagePaths: ["/tmp/ovr-0.png"],
outputPath: "/tmp/output.png",
});

const filterIdx = args.indexOf("-filter_complex");
expect(filterIdx).toBeGreaterThan(-1);

const filter = args[filterIdx + 1];
expect(filter).toContain("crop=ih*9/16:ih");
expect(filter).toContain("scale=720:1280");
expect(filter).toContain("scale=250:250");
expect(filter).toContain("overlay=30:30");
});

it("stacks multiple overlays vertically", () => {
const args = buildStaticImageArgs({
imagePath: "/tmp/base.png",
overlayImagePaths: ["/tmp/a.png", "/tmp/b.png"],
outputPath: "/tmp/output.png",
});

const filter = args[args.indexOf("-filter_complex") + 1];
expect(filter).toContain("overlay=30:30");
expect(filter).toContain("overlay=30:300");
});

it("handles no overlay images — just crops and scales", () => {
const args = buildStaticImageArgs({
imagePath: "/tmp/base.png",
overlayImagePaths: [],
outputPath: "/tmp/output.png",
});

expect(args).toContain("-vf");
expect(args).not.toContain("-filter_complex");

const vfIdx = args.indexOf("-vf");
const vf = args[vfIdx + 1];
expect(vf).toContain("crop=ih*9/16:ih");
expect(vf).toContain("scale=720:1280");
});

it("maps [out] and outputs single frame", () => {
const args = buildStaticImageArgs({
imagePath: "/tmp/base.png",
overlayImagePaths: ["/tmp/ovr.png"],
outputPath: "/tmp/output.png",
});

expect(args).toContain("-map");
expect(args).toContain("[out]");
expect(args).toContain("-frames:v");
expect(args).toContain("1");
});
});
10 changes: 5 additions & 5 deletions src/content/__tests__/pollContentRuns.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -23,14 +23,14 @@ describe("pollContentRuns", () => {
it("returns completed results when all runs finish", async () => {
mockPoll.mockResolvedValue({
status: "COMPLETED",
output: { videoSourceUrl: "https://v.mp4", captionText: "caption" },
output: { videoSourceUrl: "https://v.mp4", captionText: "caption", staticImageUrl: "https://img.png" },
} as any);

const results = await pollContentRuns(["run-1", "run-2"]);

expect(results).toEqual([
{ runId: "run-1", status: "completed", videoUrl: "https://v.mp4", captionText: "caption" },
{ runId: "run-2", status: "completed", videoUrl: "https://v.mp4", captionText: "caption" },
{ runId: "run-1", status: "completed", videoUrl: "https://v.mp4", captionText: "caption", imageUrl: "https://img.png" },
{ runId: "run-2", status: "completed", videoUrl: "https://v.mp4", captionText: "caption", imageUrl: "https://img.png" },
]);
expect(mockPoll).toHaveBeenCalledTimes(2);
expect(mockPoll).toHaveBeenCalledWith("run-1", { pollIntervalMs: 30_000 });
Expand Down Expand Up @@ -76,7 +76,7 @@ describe("pollContentRuns", () => {
const results = await pollContentRuns(["run-1"]);

expect(results).toEqual([
{ runId: "run-1", status: "completed", videoUrl: undefined, captionText: undefined },
{ runId: "run-1", status: "completed", videoUrl: undefined, captionText: undefined, imageUrl: undefined },
]);
});

Expand Down Expand Up @@ -105,7 +105,7 @@ describe("pollContentRuns", () => {
const results = await pollContentRuns(["run-1", "run-2", "run-3"]);

expect(results).toEqual([
{ runId: "run-1", status: "completed", videoUrl: "https://v.mp4", captionText: undefined },
{ runId: "run-1", status: "completed", videoUrl: "https://v.mp4", captionText: undefined, imageUrl: undefined },
{ runId: "run-2", status: "failed", error: "Run failed" },
{ runId: "run-3", status: "failed", error: "Network error" },
]);
Expand Down
104 changes: 104 additions & 0 deletions src/content/__tests__/renderStaticImage.test.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,104 @@
import { describe, it, expect, vi, beforeEach } from "vitest";

vi.mock("node:fs/promises", () => ({
writeFile: vi.fn().mockResolvedValue(undefined),
unlink: vi.fn().mockResolvedValue(undefined),
mkdir: vi.fn().mockResolvedValue(undefined),
}));

vi.mock("../downloadMediaToFile", () => ({
downloadMediaToFile: vi.fn().mockResolvedValue(undefined),
}));

vi.mock("../downloadOverlayImages", () => ({
downloadOverlayImages: vi.fn().mockResolvedValue(["/tmp/ovr-0.png"]),
}));

vi.mock("../runFfmpeg", () => ({
runFfmpeg: vi.fn().mockResolvedValue(undefined),
}));

vi.mock("../uploadToFalStorage", () => ({
uploadToFalStorage: vi.fn().mockResolvedValue({
url: "https://fal.ai/static-image.png",
mimeType: "image/png",
sizeBytes: 12345,
}),
}));

vi.mock("../../sandboxes/logStep", () => ({
logStep: vi.fn(),
}));

import { renderStaticImage } from "../renderStaticImage";
import { runFfmpeg } from "../runFfmpeg";
import { uploadToFalStorage } from "../uploadToFalStorage";
import { downloadOverlayImages } from "../downloadOverlayImages";
import { downloadMediaToFile } from "../downloadMediaToFile";

beforeEach(() => {
vi.clearAllMocks();
});

describe("renderStaticImage", () => {
it("returns the uploaded image URL and size", async () => {
const result = await renderStaticImage({
imageUrl: "https://example.com/base.png",
overlayImageUrls: ["https://example.com/cover.png"],
});

expect(result).toEqual({
imageUrl: "https://fal.ai/static-image.png",
mimeType: "image/png",
sizeBytes: 12345,
});
});

it("downloads the base image to a temp file", async () => {
await renderStaticImage({
imageUrl: "https://example.com/base.png",
overlayImageUrls: [],
});

expect(downloadMediaToFile).toHaveBeenCalledWith(
"https://example.com/base.png",
expect.stringContaining("base-image.png"),
);
});

it("downloads overlay images", async () => {
await renderStaticImage({
imageUrl: "https://example.com/base.png",
overlayImageUrls: ["https://example.com/a.png", "https://example.com/b.png"],
});

expect(downloadOverlayImages).toHaveBeenCalledWith(
["https://example.com/a.png", "https://example.com/b.png"],
expect.any(String),
);
});

it("calls runFfmpeg with args", async () => {
await renderStaticImage({
imageUrl: "https://example.com/base.png",
overlayImageUrls: ["https://example.com/cover.png"],
});

expect(runFfmpeg).toHaveBeenCalledTimes(1);
const args = vi.mocked(runFfmpeg).mock.calls[0][0];
expect(args).toContain("-y");
});

it("uploads the output as image/png", async () => {
await renderStaticImage({
imageUrl: "https://example.com/base.png",
overlayImageUrls: [],
});

expect(uploadToFalStorage).toHaveBeenCalledWith(
expect.stringContaining("static-image.png"),
"static-image.png",
"image/png",
);
});
});
2 changes: 1 addition & 1 deletion src/content/buildFilterComplex.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ const FRAME_WIDTH = 720;
const FRAME_HEIGHT = 1280;

/** Overlay image size */
const OVERLAY_SIZE = 150;
const OVERLAY_SIZE = 250;
const EDGE_PADDING = 30;
const OVERLAY_GAP = 20;

Expand Down
63 changes: 63 additions & 0 deletions src/content/buildStaticImageArgs.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,63 @@
/** Overlay image size */
const OVERLAY_SIZE = 250;
const EDGE_PADDING = 30;
const OVERLAY_GAP = 20;

/**
* Builds ffmpeg arguments to render a static image with overlay images.
*
* Pipeline: crop 16:9 → 9:16, scale to 720×1280, overlay images top-left stacked vertically.
* Outputs a single PNG frame.
*/
export function buildStaticImageArgs({
imagePath,
overlayImagePaths,
outputPath,
}: {
imagePath: string;
overlayImagePaths: string[];
outputPath: string;
}): string[] {
const hasOverlays = overlayImagePaths.length > 0;
const args = ["-y"];

if (hasOverlays) {
args.push("-i", imagePath);
for (const p of overlayImagePaths) {
args.push("-i", p);
}

const parts: string[] = [];

// Crop + scale base image
parts.push("[0:v]crop=ih*9/16:ih,scale=720:1280[img_base]");

// Scale each overlay
for (let i = 0; i < overlayImagePaths.length; i++) {
parts.push(`[${1 + i}:v]scale=${OVERLAY_SIZE}:${OVERLAY_SIZE}[ovr_${i}]`);
}

// Chain overlays — stacked vertically from top-left
let prevLabel = "img_base";
for (let i = 0; i < overlayImagePaths.length; i++) {
const x = EDGE_PADDING;
const y = EDGE_PADDING + i * (OVERLAY_SIZE + OVERLAY_GAP);
const outLabel = i < overlayImagePaths.length - 1 ? `ovr_out_${i}` : "out";
parts.push(`[${prevLabel}][ovr_${i}]overlay=${x}:${y}[${outLabel}]`);
prevLabel = outLabel;
}

args.push("-filter_complex", parts.join(";"));
args.push("-map", "[out]");
args.push("-frames:v", "1", outputPath);
} else {
args.push(
"-i", imagePath,
"-vf", "crop=ih*9/16:ih,scale=720:1280",
"-frames:v", "1",
outputPath,
);
}

return args;
}
3 changes: 3 additions & 0 deletions src/content/pollContentRuns.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ export type ContentRunResult = {
runId: string;
status: "completed" | "failed" | "timeout";
videoUrl?: string;
imageUrl?: string;
captionText?: string;
error?: string;
};
Expand Down Expand Up @@ -41,6 +42,7 @@ export async function pollContentRuns(
if (run.status === "COMPLETED") {
const output = run.output as {
videoSourceUrl?: string;
staticImageUrl?: string;
captionText?: string;
} | null;

Expand All @@ -49,6 +51,7 @@ export async function pollContentRuns(
runId,
status: "completed" as const,
videoUrl: output?.videoSourceUrl,
imageUrl: output?.staticImageUrl,
captionText: output?.captionText,
};
}
Expand Down
64 changes: 64 additions & 0 deletions src/content/renderStaticImage.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,64 @@
import { randomUUID } from "node:crypto";
import { unlink, mkdir } from "node:fs/promises";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { logStep } from "../sandboxes/logStep";
import { buildStaticImageArgs } from "./buildStaticImageArgs";
import { downloadOverlayImages } from "./downloadOverlayImages";
import { downloadMediaToFile } from "./downloadMediaToFile";
import { runFfmpeg } from "./runFfmpeg";
import { uploadToFalStorage } from "./uploadToFalStorage";

export interface RenderStaticImageInput {
imageUrl: string;
overlayImageUrls: string[];
}

export interface RenderStaticImageOutput {
imageUrl: string;
mimeType: string;
sizeBytes: number;
}

/**
* Renders a static image by cropping to 9:16, scaling to 720×1280,
* and overlaying playlist cover images. Uploads the result to fal.ai storage.
*/
export async function renderStaticImage(
input: RenderStaticImageInput,
): Promise<RenderStaticImageOutput> {
const tempDir = join(tmpdir(), `content-static-${randomUUID()}`);
await mkdir(tempDir, { recursive: true });

const imagePath = join(tempDir, "base-image.png");
const outputPath = join(tempDir, "static-image.png");
let overlayPaths: string[] = [];

try {
logStep("Downloading base image for static render");
await downloadMediaToFile(input.imageUrl, imagePath);

overlayPaths = await downloadOverlayImages(input.overlayImageUrls, tempDir);

const ffmpegArgs = buildStaticImageArgs({
imagePath,
overlayImagePaths: overlayPaths,
outputPath,
});

logStep("Running ffmpeg static image render", true, {
overlayCount: overlayPaths.length,
});

await runFfmpeg(ffmpegArgs);

logStep("Static image rendered, uploading to fal.ai storage");
const result = await uploadToFalStorage(outputPath, "static-image.png", "image/png");
logStep("Static image uploaded", false, { imageUrl: result.url, sizeBytes: result.sizeBytes });

return { imageUrl: result.url, mimeType: result.mimeType, sizeBytes: result.sizeBytes };
} finally {
const cleanupPaths = [imagePath, outputPath, ...overlayPaths];
await Promise.all(cleanupPaths.map(p => unlink(p).catch(() => undefined)));
}
}
Loading
Loading