mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/imputnet/cobalt.git synced 2025-06-28 01:18:27 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

api/stream: split types.js into proxy.js and ffmpeg.js

Browse Source
This commit is contained in:
wukko 2025-06-24 20:09:41 +06:00
parent aa376d76f6
commit 14657e51d3
No known key found for this signature in database
GPG Key ID: 3E30B3F26C7B4AA2
3 changed files with 50 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
api/src/stream/types.js → api/src/stream/ffmpeg.js
View File

@ -1,12 +1,11 @@
import ffmpeg from "ffmpeg-static";
import { spawn } from "child_process";
import { Agent, request } from "undici";
import { create as contentDisposition } from "content-disposition-header";
import { env } from "../config.js";
import { destroyInternalStream } from "./manage.js";
import { hlsExceptions } from "../processing/service-config.js";
import { getHeaders, closeRequest, closeResponse, pipe, estimateTunnelLength, estimateAudioMultiplier } from "./shared.js";
import { closeResponse, pipe, estimateTunnelLength, estimateAudioMultiplier } from "./shared.js";
const metadataTags = [
"album",
@ -55,44 +54,6 @@ const getCommand = (args) => {
return [ffmpeg, args]
}
const defaultAgent = new Agent();
const proxy = async (streamInfo, res) => {
const abortController = new AbortController();
const shutdown = () => (
closeRequest(abortController),
closeResponse(res),
destroyInternalStream(streamInfo.urls)
);
try {
res.setHeader('Cross-Origin-Resource-Policy', 'cross-origin');
res.setHeader('Content-disposition', contentDisposition(streamInfo.filename));
const { body: stream, headers, statusCode } = await request(streamInfo.urls, {
headers: {
...getHeaders(streamInfo.service),
Range: streamInfo.range
},
signal: abortController.signal,
maxRedirections: 16,
dispatcher: defaultAgent,
});
res.status(statusCode);
for (const headerName of ['accept-ranges', 'content-type', 'content-length']) {
if (headers[headerName]) {
res.setHeader(headerName, headers[headerName]);
}
}
pipe(stream, res, shutdown);
} catch {
shutdown();
}
}
const render = async (res, streamInfo, ffargs, multiplier) => {
let process;
const urls = Array.isArray(streamInfo.urls) ? streamInfo.urls : [streamInfo.urls];
@ -245,7 +206,6 @@ const convertGif = async (streamInfo, res) => {
}
export default {
proxy,
remux,
convertAudio,
convertGif,

43
api/src/stream/proxy.js Normal file
View File

@ -0,0 +1,43 @@
import { Agent, request } from "undici";
import { create as contentDisposition } from "content-disposition-header";
import { destroyInternalStream } from "./manage.js";
import { getHeaders, closeRequest, closeResponse, pipe } from "./shared.js";
const defaultAgent = new Agent();
export default async function (streamInfo, res) {
const abortController = new AbortController();
const shutdown = () => (
closeRequest(abortController),
closeResponse(res),
destroyInternalStream(streamInfo.urls)
);
try {
res.setHeader('Cross-Origin-Resource-Policy', 'cross-origin');
res.setHeader('Content-disposition', contentDisposition(streamInfo.filename));
const { body: stream, headers, statusCode } = await request(streamInfo.urls, {
headers: {
...getHeaders(streamInfo.service),
Range: streamInfo.range
},
signal: abortController.signal,
maxRedirections: 16,
dispatcher: defaultAgent,
});
res.status(statusCode);
for (const headerName of ['accept-ranges', 'content-type', 'content-length']) {
if (headers[headerName]) {
res.setHeader(headerName, headers[headerName]);
}
}
pipe(stream, res, shutdown);
} catch {
shutdown();
}
}

11
api/src/stream/stream.js
View File

@ -1,4 +1,5 @@
import stream from "./types.js";
import proxy from "./proxy.js";
import ffmpeg from "./ffmpeg.js";
import { closeResponse } from "./shared.js";
import { internalStream } from "./internal.js";
@ -7,7 +8,7 @@ export default async function(res, streamInfo) {
try {
switch (streamInfo.type) {
case "proxy":
return await stream.proxy(streamInfo, res);
return await proxy(streamInfo, res);
case "internal":
return await internalStream(streamInfo.data, res);
@ -15,13 +16,13 @@ export default async function(res, streamInfo) {
case "merge":
case "remux":
case "mute":
return await stream.remux(streamInfo, res);
return await ffmpeg.remux(streamInfo, res);
case "audio":
return await stream.convertAudio(streamInfo, res);
return await ffmpeg.convertAudio(streamInfo, res);
case "gif":
return await stream.convertGif(streamInfo, res);
return await ffmpeg.convertGif(streamInfo, res);
}
closeResponse(res);