smueller/youtube-music
1
0
Fork 0
mirror of https://github.com/th-ch/youtube-music.git synced 2026-01-10 10:11:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Download plugin - advanced options (folder, FFmpeg args)

Browse Source
This commit is contained in:
TC
2020-12-03 18:55:27 +01:00
parent 4d4a1f038b
commit 7b20b9339d
2 changed files with 29 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
plugins/downloader/front.js
View File

@ -7,6 +7,7 @@ let progress = null;
const downloadButton = ElementFromFile(
templatePath(__dirname, "download.html")
);
let pluginOptions = {};
const observer = new MutationObserver((mutations, observer) => {
if (!menu) {
@ -43,11 +44,13 @@ global.download = () => {
triggerAction(CHANNEL, ACTIONS.ERROR, error);
reinit();
},
reinit
reinit,
pluginOptions
);
};
function observeMenu() {
function observeMenu(options) {
pluginOptions = { ...pluginOptions, ...options };
observer.observe(document, {
childList: true,
subtree: true,

29
plugins/downloader/youtube-dl.js
View File

@ -19,7 +19,13 @@ const ffmpeg = createFFmpeg({
progress: () => {}, // console.log,
});
const downloadVideoToMP3 = (videoUrl, sendFeedback, sendError, reinit) => {
const downloadVideoToMP3 = (
videoUrl,
sendFeedback,
sendError,
reinit,
options
) => {
sendFeedback("Downloading…");
let videoName = "YouTube Music - Unknown title";
@ -54,11 +60,18 @@ const downloadVideoToMP3 = (videoUrl, sendFeedback, sendError, reinit) => {
.on("error", sendError)
.on("end", () => {
const buffer = Buffer.concat(chunks);
toMP3(videoName, buffer, sendFeedback, sendError, reinit);
toMP3(videoName, buffer, sendFeedback, sendError, reinit, options);
});
};
const toMP3 = async (videoName, buffer, sendFeedback, sendError, reinit) => {
const toMP3 = async (
videoName,
buffer,
sendFeedback,
sendError,
reinit,
options
) => {
const safeVideoName = randomBytes(32).toString("hex");
try {
@ -71,11 +84,17 @@ const toMP3 = async (videoName, buffer, sendFeedback, sendError, reinit) => {
ffmpeg.FS("writeFile", safeVideoName, buffer);
sendFeedback("Converting…");
await ffmpeg.run("-i", safeVideoName, safeVideoName + ".mp3");
await ffmpeg.run(
"-i",
safeVideoName,
...options.ffmpegArgs,
safeVideoName + ".mp3"
);
const folder = options.downloadFolder || downloadsFolder();
const filename = filenamify(videoName + ".mp3", { replacement: "_" });
writeFileSync(
join(downloadsFolder(), filename),
join(folder, filename),
ffmpeg.FS("readFile", safeVideoName + ".mp3")
);