mirror of
https://github.com/wukko/cobalt.git
synced 2025-02-15 15:26:34 +01:00
web/queue: add a remux worker to saving pipeline, use pipelineResults
This commit is contained in:
parent
f2325bdc24
commit
1590490db2
3 changed files with 33 additions and 1 deletions
|
@ -1,3 +1,5 @@
|
||||||
|
import mime from "mime";
|
||||||
|
|
||||||
import { addItem } from "$lib/state/queen-bee/queue";
|
import { addItem } from "$lib/state/queen-bee/queue";
|
||||||
import type { CobaltPipelineItem } from "$lib/types/workers";
|
import type { CobaltPipelineItem } from "$lib/types/workers";
|
||||||
import type { CobaltLocalProcessingResponse } from "$lib/types/api";
|
import type { CobaltLocalProcessingResponse } from "$lib/types/api";
|
||||||
|
@ -49,7 +51,10 @@ export const createSavePipeline = (info: CobaltLocalProcessingResponse) => {
|
||||||
const parentId = crypto.randomUUID();
|
const parentId = crypto.randomUUID();
|
||||||
const pipeline: CobaltPipelineItem[] = [];
|
const pipeline: CobaltPipelineItem[] = [];
|
||||||
|
|
||||||
for (const tunnel of info.tunnel) {
|
// reverse is needed for audio (second item) to be downloaded first
|
||||||
|
const tunnels = info.tunnel.reverse();
|
||||||
|
|
||||||
|
for (const tunnel of tunnels) {
|
||||||
pipeline.push({
|
pipeline.push({
|
||||||
worker: "fetch",
|
worker: "fetch",
|
||||||
workerId: crypto.randomUUID(),
|
workerId: crypto.randomUUID(),
|
||||||
|
@ -60,6 +65,24 @@ export const createSavePipeline = (info: CobaltLocalProcessingResponse) => {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pipeline.push({
|
||||||
|
worker: "remux",
|
||||||
|
workerId: crypto.randomUUID(),
|
||||||
|
parentId,
|
||||||
|
workerArgs: {
|
||||||
|
ffargs: [
|
||||||
|
"-c:v", "copy",
|
||||||
|
"-c:a", "copy"
|
||||||
|
],
|
||||||
|
output: {
|
||||||
|
// TODO: return mime type from api to avoid dragging a big ass package into web build
|
||||||
|
type: mime.getType(info.filename) || undefined,
|
||||||
|
extension: info.filename.split(".").pop(),
|
||||||
|
},
|
||||||
|
filename: info.filename,
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
addItem({
|
addItem({
|
||||||
id: parentId,
|
id: parentId,
|
||||||
state: "waiting",
|
state: "waiting",
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
import RemuxWorker from "$lib/workers/remux?worker";
|
import RemuxWorker from "$lib/workers/remux?worker";
|
||||||
import FetchWorker from "$lib/workers/fetch?worker";
|
import FetchWorker from "$lib/workers/fetch?worker";
|
||||||
|
|
||||||
|
import { get } from "svelte/store";
|
||||||
import { updateWorkerProgress } from "$lib/state/queen-bee/current-tasks";
|
import { updateWorkerProgress } from "$lib/state/queen-bee/current-tasks";
|
||||||
import { pipelineTaskDone, itemError, queue } from "$lib/state/queen-bee/queue";
|
import { pipelineTaskDone, itemError, queue } from "$lib/state/queen-bee/queue";
|
||||||
|
|
||||||
|
@ -151,6 +152,13 @@ export const startWorker = async ({ worker, workerId, parentId, workerArgs }: Co
|
||||||
files = workerArgs.files;
|
files = workerArgs.files;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (files?.length === 0) {
|
||||||
|
const parent = get(queue)[parentId];
|
||||||
|
if (parent.state === "running" && parent.pipelineResults) {
|
||||||
|
files = parent.pipelineResults;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (files.length > 0 && workerArgs.ffargs && workerArgs.output && workerArgs.filename) {
|
if (files.length > 0 && workerArgs.ffargs && workerArgs.output && workerArgs.filename) {
|
||||||
await runRemuxWorker(
|
await runRemuxWorker(
|
||||||
workerId,
|
workerId,
|
||||||
|
|
|
@ -31,6 +31,7 @@ const remux = async (files: File[], args: string[], output: FileInfo, filename:
|
||||||
await ff.init();
|
await ff.init();
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
// probing just the first file in files array (usually audio) for duration progress
|
||||||
const file_info = await ff.probe(files[0]).catch((e) => {
|
const file_info = await ff.probe(files[0]).catch((e) => {
|
||||||
if (e?.message?.toLowerCase().includes("out of memory")) {
|
if (e?.message?.toLowerCase().includes("out of memory")) {
|
||||||
console.error("uh oh! out of memory");
|
console.error("uh oh! out of memory");
|
||||||
|
|
Loading…
Reference in a new issue