Upload video-io.ts

web/src/lib/video-io.ts

@@ -11,6 +11,12 @@ import { embedWatermark } from '@core/embedder.js';
 let ffmpegInstance: FFmpeg | null = null;
 let ffmpegLoaded = false;
 
+/** Reset the FFmpeg singleton so the next getFFmpeg() call creates a fresh instance */
+function resetFFmpeg() {
+  ffmpegInstance = null;
+  ffmpegLoaded = false;
+}
+
 /** Get or initialize the shared FFmpeg instance */
 async function getFFmpeg(onLog?: (msg: string) => void): Promise<FFmpeg> {
   if (ffmpegInstance && ffmpegLoaded) return ffmpegInstance;
@@ -86,6 +92,22 @@ export async function streamExtractAndEmbed(
   config: WatermarkConfig,
   comparisonSample: number,
   onProgress?: (phase: string, current: number, total: number) => void
+): Promise<StreamEmbedResult> {
+  try {
+    return await _streamExtractAndEmbed(videoUrl, payload, key, config, comparisonSample, onProgress);
+  } catch (e) {
+    resetFFmpeg();
+    throw e;
+  }
+}
+
+async function _streamExtractAndEmbed(
+  videoUrl: string,
+  payload: Uint8Array,
+  key: string,
+  config: WatermarkConfig,
+  comparisonSample: number,
+  onProgress?: (phase: string, current: number, total: number) => void
 ): Promise<StreamEmbedResult> {
   const t0 = performance.now();
 
@@ -368,7 +390,7 @@ export interface AttackYPlanes {
  * Write blob to ffmpeg FS, run attack filter, decode output to Y planes.
  * Only processes `maxFrames` frames to keep WASM encoding fast.
  */
-async function runAttackPipeline(
+async function runAttackPipelineOnce(
   blob: Blob,
   filterArgs: string[],
   outputWidth: number,
@@ -420,6 +442,23 @@ async function runAttackPipeline(
   return { yPlanes, width: outputWidth, height: outputHeight };
 }
 
+/** Run attack pipeline with one retry on WASM crash */
+async function runAttackPipeline(
+  blob: Blob,
+  filterArgs: string[],
+  outputWidth: number,
+  outputHeight: number,
+  maxFrames: number = 30,
+): Promise<AttackYPlanes> {
+  try {
+    return await runAttackPipelineOnce(blob, filterArgs, outputWidth, outputHeight, maxFrames);
+  } catch (e) {
+    console.warn('ffmpeg.wasm crashed, reinitializing and retrying:', e);
+    resetFFmpeg();
+    return await runAttackPipelineOnce(blob, filterArgs, outputWidth, outputHeight, maxFrames);
+  }
+}
+
 /** Attack: re-encode at a given CRF quality level */
 export async function attackReencode(blob: Blob, crf: number, width: number, height: number): Promise<AttackYPlanes> {
   return runAttackPipeline(blob, ['-crf', String(crf)], width, height);