@epic-web/workshop-utils 6.56.0 → 6.57.0

package/dist/offline-videos.server.d.ts

@@ -1,3 +1,4 @@
+import { EventEmitter } from 'node:events';
 import { Readable } from 'node:stream';
 type OfflineVideoEntryStatus = 'ready' | 'downloading' | 'error';
 export type OfflineVideoDownloadState = {
@@ -58,6 +59,18 @@ export type OfflineVideoAsset = {
         end: number;
     }) => Readable;
 };
+export type VideoDownloadProgress = {
+    playbackId: string;
+    bytesDownloaded: number;
+    totalBytes: number | null;
+    status: 'downloading' | 'complete' | 'error';
+};
+export declare const DOWNLOAD_PROGRESS_EVENTS: {
+    readonly PROGRESS: "progress";
+};
+declare class DownloadProgressEmitter extends EventEmitter {
+}
+export declare const downloadProgressEmitter: DownloadProgressEmitter;
 export declare function getOfflineVideoDownloadState(): OfflineVideoDownloadState;
 export declare function getOfflineVideoSummary({ request, }?: {
     request?: Request;

package/dist/offline-videos.server.js

@@ -1,4 +1,5 @@
 import { createHash, randomUUID } from 'node:crypto';
+import { EventEmitter } from 'node:events';
 import { createReadStream, createWriteStream, promises as fs } from 'node:fs';
 import path from 'node:path';
 import { Readable, Transform } from 'node:stream';
@@ -25,6 +26,15 @@ const log = logger('epic:offline-videos');
 const offlineVideoDirectoryName = 'offline-videos';
 const offlineVideoIndexFileName = 'index.json';
 const offlineVideoConfigFileName = 'offline-video-config.json';
+export const DOWNLOAD_PROGRESS_EVENTS = {
+    PROGRESS: 'progress',
+};
+class DownloadProgressEmitter extends EventEmitter {
+}
+export const downloadProgressEmitter = new DownloadProgressEmitter();
+function emitDownloadProgress(progress) {
+    downloadProgressEmitter.emit(DOWNLOAD_PROGRESS_EVENTS.PROGRESS, progress);
+}
 let downloadState = {
     status: 'idle',
     startedAt: null,
@@ -213,6 +223,16 @@ function createSliceTransform({ skipBytes, takeBytes, }) {
         },
     });
 }
+function createProgressTrackingTransform({ onProgress, }) {
+    let totalBytes = 0;
+    return new Transform({
+        transform(chunk, _encoding, callback) {
+            totalBytes += chunk.length;
+            onProgress(totalBytes);
+            callback(null, chunk);
+        },
+    });
+}
 function createOfflineVideoReadStream({ filePath, size, key, iv, range, }) {
     if (!range) {
         const decipher = createOfflineVideoDecipher({ key, iv });
@@ -331,6 +351,12 @@ async function isOfflineVideoReady(index, playbackId, keyId, cryptoVersion, work
 async function downloadMuxVideo({ playbackId, filePath, key, iv, resolution, }) {
     const urls = getMuxMp4Urls(playbackId, resolution);
     let lastError = null;
+    emitDownloadProgress({
+        playbackId,
+        bytesDownloaded: 0,
+        totalBytes: null,
+        status: 'downloading',
+    });
     for (const url of urls) {
         const response = await fetch(url).catch((error) => {
             lastError = error;
@@ -342,16 +368,49 @@ async function downloadMuxVideo({ playbackId, filePath, key, iv, resolution, })
             lastError = new Error(`Failed to download ${playbackId} from ${url} (${response.status})`);
             continue;
         }
+        // Get content-length if available for progress tracking
+        const contentLengthHeader = response.headers.get('content-length');
+        const totalBytes = contentLengthHeader
+            ? parseInt(contentLengthHeader, 10)
+            : null;
+        emitDownloadProgress({
+            playbackId,
+            bytesDownloaded: 0,
+            totalBytes,
+            status: 'downloading',
+        });
         await ensureOfflineVideoDir();
         const tmpPath = `${filePath}.tmp-${randomUUID()}`;
         const stream = createWriteStream(tmpPath, { mode: 0o600 });
         const cipher = createOfflineVideoCipher({ key, iv });
+        const progressTracker = createProgressTrackingTransform({
+            onProgress: (bytesDownloaded) => {
+                emitDownloadProgress({
+                    playbackId,
+                    bytesDownloaded,
+                    totalBytes,
+                    status: 'downloading',
+                });
+            },
+        });
         const webStream = response.body;
-        await pipeline(Readable.from(webStream), cipher, stream);
+        await pipeline(Readable.from(webStream), progressTracker, cipher, stream);
         await fs.rename(tmpPath, filePath);
         const stat = await fs.stat(filePath);
+        emitDownloadProgress({
+            playbackId,
+            bytesDownloaded: stat.size,
+            totalBytes: stat.size,
+            status: 'complete',
+        });
         return { size: stat.size };
     }
+    emitDownloadProgress({
+        playbackId,
+        bytesDownloaded: 0,
+        totalBytes: null,
+        status: 'error',
+    });
     throw lastError ?? new Error(`Unable to download video ${playbackId}`);
 }
 async function runOfflineVideoDownloads({ videos, index, keyInfo, workshop, resolution, }) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@epic-web/workshop-utils",
-  "version": "6.56.0",
+  "version": "6.57.0",
   "publishConfig": {
     "access": "public"
   },