mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/imputnet/cobalt.git synced 2025-07-17 18:58:33 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

all: initial clip scaffolding

Browse Source
This commit is contained in:
dumbmoron 2024-02-10 21:04:29 +00:00
parent 6835bc4bdb
commit 24dcea3490
No known key found for this signature in database
GPG Key ID: C59997C76C6A8E5F
4 changed files with 49 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/modules/processing/matchActionDecider.js
View File

@ -18,6 +18,7 @@ export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, di
if (r.isPhoto) action = "photo";
else if (r.picker) action = "picker"
else if (r.isClip) action = "clip"; // TODO: user-specified clips
else if (r.isGif && toGif) action = "gif";
else if (isAudioMuted) action = "muteVideo";
else if (isAudioOnly) action = "audio";
@ -40,7 +41,11 @@ export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, di
case "photo":
responseType = 1;
break;
case "clip":
params = { type: "clip" }
break;
case "gif":
params = { type: "gif" }
break;

4
src/modules/stream/manage.js
View File

@ -34,6 +34,10 @@ export function createStream(obj) {
isAudioOnly: !!obj.isAudioOnly,
audioFormat: obj.audioFormat,
time: obj.time ? obj.time : false,
clip: obj.clip ? {
start: parseFloat(obj.clip.start).toFixed(6),
end: parseFloat(obj.clip.end).toFixed(6),
} : false,
copy: !!obj.copy,
mute: !!obj.mute,
metadata: obj.fileMetadata ? obj.fileMetadata : false

5
src/modules/stream/stream.js
View File

@ -1,4 +1,4 @@
import { streamAudioOnly, streamDefault, streamLiveRender, streamVideoOnly, convertToGif } from "./types.js";
import { streamAudioOnly, streamDefault, streamLiveRender, streamVideoOnly, streamClip, convertToGif } from "./types.js";
export default async function(res, streamInfo) {
try {
@ -17,6 +17,9 @@ export default async function(res, streamInfo) {
case "mute":
streamVideoOnly(streamInfo, res);
break;
case "clip":
streamClip(streamInfo, res);
break;
default:
await streamDefault(streamInfo, res);
break;

41
src/modules/stream/types.js
View File

@ -1,10 +1,12 @@
import ffmpeg from "ffmpeg-static";
import { ffmpegArgs, genericUserAgent } from "../config.js";
import { spawn, pipe, killProcess } from "./shared.js";
import { metadataManager } from "../sub/utils.js";
import { request } from "undici";
import { create as contentDisposition } from "content-disposition-header";
import { request } from "undici"
import ffmpeg from "ffmpeg-static"
import { AbortController } from "abort-controller"
import { create as contentDisposition } from "content-disposition-header"
import { makeCut } from "./cut.js"
import { metadataManager } from "../sub/utils.js"
import { spawn, pipe, killProcess } from "./shared.js"
import { ffmpegArgs, genericUserAgent } from "../config.js"
function closeRequest(controller) {
try { controller.abort() } catch {}
@ -226,3 +228,30 @@ export function convertToGif(streamInfo, res) {
shutdown();
}
}
export async function streamClip(streamInfo, res) {
if (typeof streamInfo.urls === 'string')
streamInfo.urls = [streamInfo.urls];
const shutdown = () => (killProcess(process), closeResponse(res));
try {
const [ video, audio ] = streamInfo.urls;
const { start, end } = streamInfo.clip;
if (!start || !end || start === 'NaN' || end === 'NaN')
return shutdown();
const stream = await makeCut(video, { start, end }, audio);
process = stream.process;
res.setHeader('Connection', 'keep-alive');
res.setHeader('Content-Disposition', contentDisposition(streamInfo.filename));
pipe(stream, res, shutdown);
process.on('close', shutdown);
res.on('finish', shutdown);
} catch {
shutdown();
}
}