mirror of
https://github.com/musix-org/musix-oss
synced 2024-09-20 17:11:57 +00:00
43 lines
1.3 KiB
JavaScript
43 lines
1.3 KiB
JavaScript
|
const ytdl = require('ytdl-core');
|
||
|
|
const prism = require('prism-media');
|
||
|
|
|
||
|
|
function filter(format) {
|
||
|
|
return format.audioEncoding === 'opus' &&
|
||
|
|
format.container === 'webm' &&
|
||
|
|
format.audio_sample_rate == 48000;
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = function download(url, options = {}) {
|
||
|
|
return new Promise((resolve, reject) => {
|
||
|
|
ytdl.getInfo(url, (err, info) => {
|
||
|
|
if (err) return reject(err);
|
||
|
|
// Prefer opus
|
||
|
|
const canDemux = info.formats.find(filter) && info.length_seconds != 0;
|
||
|
|
if (canDemux) Object.assign(options, { filter });
|
||
|
|
else if (info.length_seconds != 0) Object.assign(options, { filter: 'audioonly' });
|
||
|
|
const ytdlStream = ytdl.downloadFromInfo(info, options);
|
||
|
|
if (canDemux) {
|
||
|
|
const demuxer = new prism.opus.WebmDemuxer();
|
||
|
|
return resolve(ytdlStream.pipe(demuxer).on('end', () => demuxer.destroy()));
|
||
|
|
} else {
|
||
|
|
const transcoder = new prism.FFmpeg({
|
||
|
|
args: [
|
||
|
|
'-analyzeduration', '0',
|
||
|
|
'-loglevel', '0',
|
||
|
|
'-f', 's16le',
|
||
|
|
'-ar', '48000',
|
||
|
|
'-ac', '2',
|
||
|
|
],
|
||
|
|
});
|
||
|
|
const opus = new prism.opus.Encoder({ rate: 48000, channels: 2, frameSize: 960 });
|
||
|
|
const stream = ytdlStream.pipe(transcoder).pipe(opus);
|
||
|
|
stream.on('close', () => {
|
||
|
|
transcoder.destroy();
|
||
|
|
opus.destroy();
|
||
|
|
});
|
||
|
|
return resolve(stream);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
});
|
||
|
|
};
|