mirror of
https://github.com/heygen-com/hyperframes.git
synced 2026-09-05 00:56:23 +00:00
initial code (#2)
* feat: initial code port from hyperframes-internal Port all OSS-ready packages from the internal monorepo: - @hyperframes/core — shared types, HTML generation, GSAP utilities, runtime - @hyperframes/cli — CLI for creating, previewing, and rendering compositions - @hyperframes/engine — framework-agnostic rendering engine (BeginFrame + FFmpeg) - @hyperframes/producer — video rendering pipeline (Puppeteer + FFmpeg) - @hyperframes/ui-player — browser-based video player component - @hyperframes/studio — composition editor (React frontend + Hono backend) Includes regression test suite with Docker-based test harness. All HeyGen-internal references, deployment infrastructure, and proprietary assets have been removed. Package names migrated from @app/* to @hyperframes/*. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com> * fix: scrub internal codenames and stale references from OSS port - Replace static.heygen.ai runtime URLs in test fixtures - Remove internal CDN publish script (publish-hyperframe-runtime.ts) - Replace sandbox-studio, sandbox-interceptor, __magicEditRuntime with neutral names (studio, hyperframe-runtime, __hyperframeRuntime) - Fix stale Vault API / localhost references in docs - Remove broken deprecated_studio link Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com> * fix: remove remaining internal codenames and stale references - Delete stale producer README.md and PIPELINE.md (referenced nonexistent files) - Replace "Cerberus" codename with "HyperFrames" in test design reviews - Replace magic-edit postMessage identifiers with hf-preview/hf-parent - Rename debug-magic-edit-timeline.ts to debug-timeline.ts - Replace "Motion Cut" with "HyperFrames" in Timeline comments - Fix studio/CLI references to nonexistent archive package (use local data/projects/ dir, stub render proxy) Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com> --------- Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Vance Ingalls
co-authored by
Claude Opus 4.6
Claude Opus 4.6
parent
10621e7903
commit
9f8e5ba5a1
@@ -0,0 +1,399 @@
|
||||
/**
|
||||
* Chunk Encoder Service
|
||||
*
|
||||
* Encodes captured frames into video using FFmpeg.
|
||||
* Supports CPU (libx264) and GPU encoding.
|
||||
*/
|
||||
|
||||
import { spawn } from "child_process";
|
||||
import { existsSync, mkdirSync, readdirSync, statSync, writeFileSync } from "fs";
|
||||
import { join, dirname } from "path";
|
||||
import { DEFAULT_CONFIG, type EngineConfig } from "../config.js";
|
||||
import { type GpuEncoder, getCachedGpuEncoder, getGpuEncoderName } from "../utils/gpuEncoder.js";
|
||||
import { runFfmpeg } from "../utils/runFfmpeg.js";
|
||||
import type { EncoderOptions, EncodeResult, MuxResult } from "./chunkEncoder.types.js";
|
||||
|
||||
export type { EncoderOptions, EncodeResult, MuxResult } from "./chunkEncoder.types.js";
|
||||
|
||||
export const ENCODER_PRESETS = {
|
||||
draft: { preset: "ultrafast", quality: 28, codec: "h264" as const },
|
||||
standard: { preset: "medium", quality: 23, codec: "h264" as const },
|
||||
high: { preset: "slow", quality: 18, codec: "h264" as const },
|
||||
};
|
||||
|
||||
// Re-export GPU utilities so existing consumers that import from chunkEncoder still work.
|
||||
export { detectGpuEncoder, type GpuEncoder } from "../utils/gpuEncoder.js";
|
||||
|
||||
function buildEncoderArgs(
|
||||
options: EncoderOptions,
|
||||
inputArgs: string[],
|
||||
outputPath: string,
|
||||
gpuEncoder: GpuEncoder = null,
|
||||
): string[] {
|
||||
const {
|
||||
fps,
|
||||
codec = "h264",
|
||||
preset = "medium",
|
||||
quality = 23,
|
||||
bitrate,
|
||||
pixelFormat = "yuv420p",
|
||||
useGpu = false,
|
||||
} = options;
|
||||
|
||||
const args: string[] = [...inputArgs, "-r", String(fps)];
|
||||
const shouldUseGpu = useGpu && gpuEncoder !== null;
|
||||
|
||||
if (codec === "h264" || codec === "h265") {
|
||||
if (shouldUseGpu) {
|
||||
const encoderName = getGpuEncoderName(gpuEncoder, codec);
|
||||
args.push("-c:v", encoderName);
|
||||
|
||||
switch (gpuEncoder) {
|
||||
case "nvenc":
|
||||
args.push("-preset", preset);
|
||||
if (bitrate) args.push("-b:v", bitrate);
|
||||
else args.push("-cq", String(quality));
|
||||
break;
|
||||
case "videotoolbox":
|
||||
if (bitrate) args.push("-b:v", bitrate);
|
||||
else {
|
||||
const vtQuality = Math.max(0, Math.min(100, 100 - quality * 2));
|
||||
args.push("-q:v", String(vtQuality));
|
||||
}
|
||||
args.push("-allow_sw", "1");
|
||||
break;
|
||||
case "vaapi":
|
||||
args.unshift("-vaapi_device", "/dev/dri/renderD128");
|
||||
args.push("-vf", "format=nv12,hwupload");
|
||||
if (bitrate) args.push("-b:v", bitrate);
|
||||
else args.push("-qp", String(quality));
|
||||
break;
|
||||
case "qsv":
|
||||
args.push("-preset", preset);
|
||||
if (bitrate) args.push("-b:v", bitrate);
|
||||
else args.push("-global_quality", String(quality));
|
||||
break;
|
||||
}
|
||||
} else {
|
||||
const encoderName = codec === "h264" ? "libx264" : "libx265";
|
||||
args.push("-c:v", encoderName, "-preset", preset);
|
||||
if (bitrate) args.push("-b:v", bitrate);
|
||||
else args.push("-crf", String(quality));
|
||||
}
|
||||
} else if (codec === "vp9") {
|
||||
args.push("-c:v", "libvpx-vp9", "-b:v", bitrate || "0", "-crf", String(quality));
|
||||
args.push("-deadline", preset === "ultrafast" ? "realtime" : "good");
|
||||
} else if (codec === "prores") {
|
||||
args.push("-c:v", "prores_ks", "-profile:v", preset, "-vendor", "apl0");
|
||||
return [...args, "-y", outputPath];
|
||||
}
|
||||
|
||||
if (gpuEncoder !== "vaapi") {
|
||||
args.push("-pix_fmt", pixelFormat);
|
||||
}
|
||||
|
||||
args.push("-y", outputPath);
|
||||
return args;
|
||||
}
|
||||
|
||||
export async function encodeFramesFromDir(
|
||||
framesDir: string,
|
||||
framePattern: string,
|
||||
outputPath: string,
|
||||
options: EncoderOptions,
|
||||
signal?: AbortSignal,
|
||||
config?: Partial<Pick<EngineConfig, "ffmpegEncodeTimeout">>,
|
||||
): Promise<EncodeResult> {
|
||||
const startTime = Date.now();
|
||||
|
||||
const outputDir = dirname(outputPath);
|
||||
if (!existsSync(outputDir)) mkdirSync(outputDir, { recursive: true });
|
||||
|
||||
const files = readdirSync(framesDir).filter((f) => f.match(/\.(jpg|jpeg|png)$/i));
|
||||
const frameCount = files.length;
|
||||
|
||||
if (frameCount === 0) {
|
||||
return {
|
||||
success: false,
|
||||
outputPath,
|
||||
durationMs: Date.now() - startTime,
|
||||
framesEncoded: 0,
|
||||
fileSize: 0,
|
||||
error: "[FFmpeg] No frame files found in directory",
|
||||
};
|
||||
}
|
||||
|
||||
let gpuEncoder: GpuEncoder = null;
|
||||
if (options.useGpu) {
|
||||
gpuEncoder = await getCachedGpuEncoder();
|
||||
}
|
||||
|
||||
const inputPath = join(framesDir, framePattern);
|
||||
const inputArgs = ["-framerate", String(options.fps), "-i", inputPath];
|
||||
const args = buildEncoderArgs(options, inputArgs, outputPath, gpuEncoder);
|
||||
|
||||
return new Promise((resolve) => {
|
||||
const ffmpeg = spawn("ffmpeg", args);
|
||||
let stderr = "";
|
||||
const onAbort = () => {
|
||||
ffmpeg.kill("SIGTERM");
|
||||
};
|
||||
if (signal) {
|
||||
if (signal.aborted) {
|
||||
ffmpeg.kill("SIGTERM");
|
||||
} else {
|
||||
signal.addEventListener("abort", onAbort, { once: true });
|
||||
}
|
||||
}
|
||||
|
||||
const encodeTimeout = config?.ffmpegEncodeTimeout ?? DEFAULT_CONFIG.ffmpegEncodeTimeout;
|
||||
const timer = setTimeout(() => {
|
||||
ffmpeg.kill("SIGTERM");
|
||||
}, encodeTimeout);
|
||||
|
||||
ffmpeg.stderr.on("data", (data) => {
|
||||
stderr += data.toString();
|
||||
});
|
||||
|
||||
ffmpeg.on("close", (code) => {
|
||||
clearTimeout(timer);
|
||||
if (signal) signal.removeEventListener("abort", onAbort);
|
||||
const durationMs = Date.now() - startTime;
|
||||
if (signal?.aborted) {
|
||||
resolve({
|
||||
success: false,
|
||||
outputPath,
|
||||
durationMs,
|
||||
framesEncoded: 0,
|
||||
fileSize: 0,
|
||||
error: "FFmpeg encode cancelled",
|
||||
});
|
||||
return;
|
||||
}
|
||||
|
||||
if (code !== 0) {
|
||||
resolve({
|
||||
success: false,
|
||||
outputPath,
|
||||
durationMs,
|
||||
framesEncoded: 0,
|
||||
fileSize: 0,
|
||||
error: `FFmpeg exited with code ${code}`,
|
||||
});
|
||||
return;
|
||||
}
|
||||
|
||||
const fileSize = existsSync(outputPath) ? statSync(outputPath).size : 0;
|
||||
resolve({ success: true, outputPath, durationMs, framesEncoded: frameCount, fileSize });
|
||||
});
|
||||
|
||||
ffmpeg.on("error", (err) => {
|
||||
clearTimeout(timer);
|
||||
if (signal) signal.removeEventListener("abort", onAbort);
|
||||
resolve({
|
||||
success: false,
|
||||
outputPath,
|
||||
durationMs: Date.now() - startTime,
|
||||
framesEncoded: 0,
|
||||
fileSize: 0,
|
||||
error: `[FFmpeg] ${err.message}`,
|
||||
});
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
export async function encodeFramesChunkedConcat(
|
||||
framesDir: string,
|
||||
framePattern: string,
|
||||
outputPath: string,
|
||||
options: EncoderOptions,
|
||||
chunkSizeFrames: number,
|
||||
signal?: AbortSignal,
|
||||
): Promise<EncodeResult> {
|
||||
const start = Date.now();
|
||||
const files = readdirSync(framesDir)
|
||||
.filter((f) => f.match(/\.(jpg|jpeg|png)$/i))
|
||||
.sort();
|
||||
if (files.length === 0) {
|
||||
return {
|
||||
success: false,
|
||||
outputPath,
|
||||
durationMs: Date.now() - start,
|
||||
framesEncoded: 0,
|
||||
fileSize: 0,
|
||||
error: "[FFmpeg] No frame files found in directory",
|
||||
};
|
||||
}
|
||||
const chunkSize = Math.max(30, Math.floor(chunkSizeFrames));
|
||||
const chunkCount = Math.ceil(files.length / chunkSize);
|
||||
const chunkDir = join(dirname(outputPath), "chunk-encode");
|
||||
if (!existsSync(chunkDir)) mkdirSync(chunkDir, { recursive: true });
|
||||
const chunkPaths: string[] = [];
|
||||
|
||||
for (let i = 0; i < chunkCount; i++) {
|
||||
if (signal?.aborted) {
|
||||
return {
|
||||
success: false,
|
||||
outputPath,
|
||||
durationMs: Date.now() - start,
|
||||
framesEncoded: 0,
|
||||
fileSize: 0,
|
||||
error: "Chunked encode cancelled",
|
||||
};
|
||||
}
|
||||
const startNumber = i * chunkSize;
|
||||
const framesInChunk = Math.min(chunkSize, files.length - startNumber);
|
||||
const chunkPath = join(chunkDir, `chunk_${String(i).padStart(4, "0")}.mp4`);
|
||||
const inputPath = join(framesDir, framePattern);
|
||||
const inputArgs = [
|
||||
"-framerate",
|
||||
String(options.fps),
|
||||
"-start_number",
|
||||
String(startNumber),
|
||||
"-i",
|
||||
inputPath,
|
||||
"-frames:v",
|
||||
String(framesInChunk),
|
||||
];
|
||||
let gpuEncoder: GpuEncoder = null;
|
||||
if (options.useGpu) gpuEncoder = await getCachedGpuEncoder();
|
||||
const args = buildEncoderArgs(options, inputArgs, chunkPath, gpuEncoder);
|
||||
const chunkResult = await new Promise<{ success: boolean; error?: string }>((resolve) => {
|
||||
const ffmpeg = spawn("ffmpeg", args);
|
||||
let stderr = "";
|
||||
ffmpeg.stderr.on("data", (d) => {
|
||||
stderr += d.toString();
|
||||
});
|
||||
ffmpeg.on("close", (code) => {
|
||||
if (code === 0) resolve({ success: true });
|
||||
else resolve({ success: false, error: `Chunk ${i} encode failed: ${stderr.slice(-400)}` });
|
||||
});
|
||||
ffmpeg.on("error", (err) => {
|
||||
resolve({ success: false, error: `Chunk ${i} encode error: ${err.message}` });
|
||||
});
|
||||
});
|
||||
if (!chunkResult.success) {
|
||||
return {
|
||||
success: false,
|
||||
outputPath,
|
||||
durationMs: Date.now() - start,
|
||||
framesEncoded: 0,
|
||||
fileSize: 0,
|
||||
error: chunkResult.error,
|
||||
};
|
||||
}
|
||||
chunkPaths.push(chunkPath);
|
||||
}
|
||||
|
||||
const concatListPath = join(chunkDir, "concat-list.txt");
|
||||
const concatInput = chunkPaths.map((path) => `file '${path.replace(/'/g, "'\\''")}'`).join("\n");
|
||||
writeFileSync(concatListPath, concatInput, "utf-8");
|
||||
|
||||
const concatArgs = ["-f", "concat", "-safe", "0", "-i", concatListPath, "-c", "copy", "-y", outputPath];
|
||||
const concatResult = await new Promise<{ success: boolean; error?: string }>((resolve) => {
|
||||
const ffmpeg = spawn("ffmpeg", concatArgs);
|
||||
let stderr = "";
|
||||
ffmpeg.stderr.on("data", (d) => {
|
||||
stderr += d.toString();
|
||||
});
|
||||
ffmpeg.on("close", (code) => {
|
||||
if (code === 0) resolve({ success: true });
|
||||
else resolve({ success: false, error: `Chunk concat failed: ${stderr.slice(-400)}` });
|
||||
});
|
||||
ffmpeg.on("error", (err) => {
|
||||
resolve({ success: false, error: `Chunk concat error: ${err.message}` });
|
||||
});
|
||||
});
|
||||
|
||||
if (!concatResult.success) {
|
||||
return {
|
||||
success: false,
|
||||
outputPath,
|
||||
durationMs: Date.now() - start,
|
||||
framesEncoded: 0,
|
||||
fileSize: 0,
|
||||
error: concatResult.error,
|
||||
};
|
||||
}
|
||||
|
||||
const fileSize = existsSync(outputPath) ? statSync(outputPath).size : 0;
|
||||
return {
|
||||
success: true,
|
||||
outputPath,
|
||||
durationMs: Date.now() - start,
|
||||
framesEncoded: files.length,
|
||||
fileSize,
|
||||
};
|
||||
}
|
||||
|
||||
export async function muxVideoWithAudio(
|
||||
videoPath: string,
|
||||
audioPath: string,
|
||||
outputPath: string,
|
||||
signal?: AbortSignal,
|
||||
config?: Partial<Pick<EngineConfig, "ffmpegProcessTimeout">>,
|
||||
): Promise<MuxResult> {
|
||||
const outputDir = dirname(outputPath);
|
||||
if (!existsSync(outputDir)) mkdirSync(outputDir, { recursive: true });
|
||||
|
||||
const args = [
|
||||
"-i",
|
||||
videoPath,
|
||||
"-i",
|
||||
audioPath,
|
||||
"-c:v",
|
||||
"copy",
|
||||
"-c:a",
|
||||
"aac",
|
||||
"-b:a",
|
||||
"192k",
|
||||
"-shortest",
|
||||
"-movflags",
|
||||
"+faststart",
|
||||
"-y",
|
||||
outputPath,
|
||||
];
|
||||
|
||||
const processTimeout = config?.ffmpegProcessTimeout ?? DEFAULT_CONFIG.ffmpegProcessTimeout;
|
||||
const result = await runFfmpeg(args, { signal, timeout: processTimeout });
|
||||
|
||||
if (signal?.aborted) {
|
||||
return { success: false, outputPath, durationMs: result.durationMs, error: "FFmpeg mux cancelled" };
|
||||
}
|
||||
return {
|
||||
success: result.success,
|
||||
outputPath,
|
||||
durationMs: result.durationMs,
|
||||
error: !result.success
|
||||
? result.exitCode !== null
|
||||
? `FFmpeg exited with code ${result.exitCode}`
|
||||
: `[FFmpeg] ${result.stderr}`
|
||||
: undefined,
|
||||
};
|
||||
}
|
||||
|
||||
export async function applyFaststart(
|
||||
inputPath: string,
|
||||
outputPath: string,
|
||||
signal?: AbortSignal,
|
||||
config?: Partial<Pick<EngineConfig, "ffmpegProcessTimeout">>,
|
||||
): Promise<MuxResult> {
|
||||
const args = ["-i", inputPath, "-c", "copy", "-movflags", "+faststart", "-y", outputPath];
|
||||
|
||||
const processTimeout = config?.ffmpegProcessTimeout ?? DEFAULT_CONFIG.ffmpegProcessTimeout;
|
||||
const result = await runFfmpeg(args, { signal, timeout: processTimeout });
|
||||
|
||||
if (signal?.aborted) {
|
||||
return { success: false, outputPath, durationMs: result.durationMs, error: "FFmpeg faststart cancelled" };
|
||||
}
|
||||
return {
|
||||
success: result.success,
|
||||
outputPath,
|
||||
durationMs: result.durationMs,
|
||||
error: !result.success
|
||||
? result.exitCode !== null
|
||||
? `FFmpeg exited with code ${result.exitCode}`
|
||||
: `[FFmpeg] ${result.stderr}`
|
||||
: undefined,
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user