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

api/stream/types: refactor, support mkv, don't duplicate args

Browse Source
This commit is contained in:
wukko 2025-06-20 14:39:17 +06:00
parent 33c801f66b
commit c4e910dd29
No known key found for this signature in database
GPG Key ID: 3E30B3F26C7B4AA2
1 changed files with 23 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
api/src/stream/types.js
View File

@ -1,6 +1,6 @@
import { Agent, request } from "undici";
import ffmpeg from "ffmpeg-static"; import ffmpeg from "ffmpeg-static";
import { spawn } from "child_process"; import { spawn } from "child_process";
import { Agent, request } from "undici";
import { create as contentDisposition } from "content-disposition-header"; import { create as contentDisposition } from "content-disposition-header";
import { env } from "../config.js"; import { env } from "../config.js";
@ -8,13 +8,6 @@ import { destroyInternalStream } from "./manage.js";
import { hlsExceptions } from "../processing/service-config.js"; import { hlsExceptions } from "../processing/service-config.js";
import { getHeaders, closeRequest, closeResponse, pipe, estimateTunnelLength, estimateAudioMultiplier } from "./shared.js"; import { getHeaders, closeRequest, closeResponse, pipe, estimateTunnelLength, estimateAudioMultiplier } from "./shared.js";
const ffmpegArgs = {
webm: ["-c:v", "copy", "-c:a", "copy"],
mp4: ["-c:v", "copy", "-c:a", "copy", "-movflags", "faststart+frag_keyframe+empty_moov"],
m4a: ["-movflags", "frag_keyframe+empty_moov"],
gif: ["-vf", "scale=-1:-1:flags=lanczos,split[s0][s1];[s0]palettegen[p];[s1][p]paletteuse", "-loop", "0"]
}
const metadataTags = [ const metadataTags = [
"album", "album",
"composer", "composer",
@ -129,16 +122,23 @@ const merge = async (streamInfo, res) => {
args.push( args.push(
'-i', streamInfo.subtitles, '-i', streamInfo.subtitles,
'-map', '2:s', '-map', '2:s',
'-c:s', format === 'mp4' ? 'mov_text' : 'webvtt' '-c:s', format === 'mp4' ? 'mov_text' : 'webvtt',
); );
}; };
args.push( args.push(
'-map', '0:v', '-map', '0:v',
'-map', '1:a', '-map', '1:a',
'-c:v', 'copy',
'-c:a', 'copy',
); );
args = args.concat(ffmpegArgs[format]); if (format === "mp4") {
args.push(
'-movflags',
'faststart+frag_keyframe+empty_moov',
)
}
if (hlsExceptions.includes(streamInfo.service) && streamInfo.isHLS) { if (hlsExceptions.includes(streamInfo.service) && streamInfo.isHLS) {
if (streamInfo.service === "youtube" && format === "webm") { if (streamInfo.service === "youtube" && format === "webm") {
@ -152,7 +152,10 @@ const merge = async (streamInfo, res) => {
args = args.concat(convertMetadataToFFmpeg(streamInfo.metadata)); args = args.concat(convertMetadataToFFmpeg(streamInfo.metadata));
} }
args.push('-f', format, 'pipe:3'); args.push(
'-f', format === "mkv" ? "matroska" : format,
'pipe:3'
);
process = spawn(...getCommand(args), { process = spawn(...getCommand(args), {
windowsHide: true, windowsHide: true,
@ -206,7 +209,7 @@ const remux = async (streamInfo, res) => {
if (hlsExceptions.includes(streamInfo.service)) { if (hlsExceptions.includes(streamInfo.service)) {
if (streamInfo.type !== "mute") { if (streamInfo.type !== "mute") {
args.push('-c:a', 'aac') args.push('-c:a', 'aac');
} }
args.push('-bsf:a', 'aac_adtstoasc'); args.push('-bsf:a', 'aac_adtstoasc');
} }
@ -275,11 +278,11 @@ const convertAudio = async (streamInfo, res) => {
} }
if (streamInfo.audioFormat === "opus") { if (streamInfo.audioFormat === "opus") {
args.push("-vbr", "off") args.push("-vbr", "off");
} }
if (ffmpegArgs[streamInfo.audioFormat]) { if (streamInfo.audioFormat === "mp4a") {
args = args.concat(ffmpegArgs[streamInfo.audioFormat]) args.push("-movflags", "frag_keyframe+empty_moov");
} }
if (streamInfo.metadata) { if (streamInfo.metadata) {
@ -329,7 +332,11 @@ const convertGif = async (streamInfo, res) => {
} }
args.push('-i', streamInfo.urls); args.push('-i', streamInfo.urls);
args = args.concat(ffmpegArgs.gif); args.push(
'-vf',
'scale=-1:-1:flags=lanczos,split[s0][s1];[s0]palettegen[p];[s1][p]paletteuse',
'-loop', '0'
);
args.push('-f', "gif", 'pipe:3'); args.push('-f', "gif", 'pipe:3');
process = spawn(...getCommand(args), { process = spawn(...getCommand(args), {