Mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/wukko/cobalt.git synced 2024-11-15 12:50:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

api: return correct extension for gif downloads in api response

Browse source
This commit is contained in:
dumbmoron 2024-09-18 18:03:04 +00:00
parent 29f967a3ec
commit d560c0d34a
No known key found for this signature in database
3 changed files with 18 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
api/src/misc/utils.js
View file

@ -65,3 +65,14 @@ export function merge(a, b) {
return a; return a;
} }
export function splitFilenameExtension(filename) {
const parts = filename.split('.');
const ext = parts.pop();
if (!parts.length) {
return [ ext, "" ]
} else {
return [ parts.join('.'), ext ]
}
}

10
api/src/processing/match-action.js
View file

@ -3,6 +3,7 @@ import createFilename from "./create-filename.js";
import { createResponse } from "./request.js"; import { createResponse } from "./request.js";
import { audioIgnore } from "./service-config.js"; import { audioIgnore } from "./service-config.js";
import { createStream } from "../stream/manage.js"; import { createStream } from "../stream/manage.js";
import { splitFilenameExtension } from "../misc/utils.js";
export default function({ r, host, audioFormat, isAudioOnly, isAudioMuted, disableMetadata, filenameStyle, twitterGif, requestIP, audioBitrate, alwaysProxy }) { export default function({ r, host, audioFormat, isAudioOnly, isAudioMuted, disableMetadata, filenameStyle, twitterGif, requestIP, audioBitrate, alwaysProxy }) {
let action, let action,
@ -32,10 +33,11 @@ export default function({ r, host, audioFormat, isAudioOnly, isAudioMuted, disab
} }
if (action === "muteVideo" && isAudioMuted && !r.filenameAttributes) { if (action === "muteVideo" && isAudioMuted && !r.filenameAttributes) {
const parts = r.filename.split("."); const [ name, ext ] = splitFilenameExtension(r.filename);
const ext = parts.pop(); defaultParams.filename = `${name}_mute.${ext}`;
} else if (action === "gif") {
defaultParams.filename = `${parts.join(".")}_mute.${ext}`; const [ name ] = splitFilenameExtension(r.filename);
defaultParams.filename = `${name}.gif`;
} }
switch (action) { switch (action) {

2
api/src/stream/types.js
View file

@ -291,7 +291,7 @@ const convertGif = (streamInfo, res) => {
const [,,, muxOutput] = process.stdio; const [,,, muxOutput] = process.stdio;
res.setHeader('Connection', 'keep-alive'); res.setHeader('Connection', 'keep-alive');
res.setHeader('Content-Disposition', contentDisposition(streamInfo.filename.split('.')[0] + ".gif")); res.setHeader('Content-Disposition', contentDisposition(streamInfo.filename));
pipe(muxOutput, res, shutdown); pipe(muxOutput, res, shutdown);