web: rename queen-bee to task-manager

less corny and less cryptic

web/src/lib/task-manager/runners/remux.ts

@@ -0,0 +1,109 @@
+import RemuxWorker from "$lib/workers/remux?worker";
+
+import { killWorker } from "$lib/task-manager/run-worker";
+import { updateWorkerProgress } from "$lib/state/task-manager/current-tasks";
+import { pipelineTaskDone, itemError, queue } from "$lib/state/task-manager/queue";
+
+import type { FileInfo } from "$lib/types/libav";
+import type { CobaltQueue } from "$lib/types/queue";
+import type { CobaltFileReference } from "$lib/types/storage";
+
+let startAttempts = 0;
+
+export const runRemuxWorker = async (
+    workerId: string,
+    parentId: string,
+    files: CobaltFileReference[],
+    args: string[],
+    output: FileInfo,
+    resetStartCounter = false
+) => {
+    const worker = new RemuxWorker();
+
+    // sometimes chrome refuses to start libav wasm,
+    // so we check if it started, try 10 more times if not, and kill self if it still doesn't work
+    // TODO: fix the underlying issue because this is ridiculous
+
+    if (resetStartCounter) startAttempts = 0;
+
+    let bumpAttempts = 0;
+    const startCheck = setInterval(async () => {
+        bumpAttempts++;
+
+        if (bumpAttempts === 10) {
+            startAttempts++;
+            if (startAttempts <= 10) {
+                killWorker(worker, unsubscribe, startCheck);
+                console.error("worker didn't start after 5 seconds, so it was killed and started again");
+                return await runRemuxWorker(workerId, parentId, files, args, output);
+            } else {
+                killWorker(worker, unsubscribe, startCheck);
+                console.error("worker didn't start after 10 attempts, so we're giving up");
+
+                // TODO: proper error code
+                return itemError(parentId, workerId, "worker didn't start");
+            }
+        }
+    }, 500);
+
+    const unsubscribe = queue.subscribe((queue: CobaltQueue) => {
+        if (!queue[parentId]) {
+            // TODO: remove logging
+            console.log("worker's parent is gone, so it killed itself");
+            killWorker(worker, unsubscribe, startCheck);
+        }
+    });
+
+    worker.postMessage({
+        cobaltRemuxWorker: {
+            files,
+            args,
+            output,
+        }
+    });
+
+    worker.onerror = (e) => {
+        console.error("remux worker exploded:", e);
+        killWorker(worker, unsubscribe, startCheck);
+
+        // TODO: proper error code
+        return itemError(parentId, workerId, "internal error");
+    };
+
+    let totalDuration: number | null = null;
+
+    worker.onmessage = (event) => {
+        const eventData = event.data.cobaltRemuxWorker;
+        if (!eventData) return;
+
+        clearInterval(startCheck);
+
+        // temporary debug logging
+        console.log(JSON.stringify(eventData, null, 2));
+
+        if (eventData.progress) {
+            if (eventData.progress.duration) {
+                totalDuration = eventData.progress.duration;
+            }
+
+            updateWorkerProgress(workerId, {
+                percentage: totalDuration ? (eventData.progress.durationProcessed / totalDuration) * 100 : 0,
+                size: eventData.progress.size,
+            })
+        }
+
+        if (eventData.render) {
+            killWorker(worker, unsubscribe, startCheck);
+            return pipelineTaskDone(
+                parentId,
+                workerId,
+                eventData.render,
+            );
+        }
+
+        if (eventData.error) {
+            killWorker(worker, unsubscribe, startCheck);
+            return itemError(parentId, workerId, eventData.error);
+        }
+    };
+}