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

Pass (optional) existing metadata and subfolder in mp3 converter

Browse Source
This commit is contained in:
TC
2021-03-24 21:46:59 +01:00
parent 4146ae60bc
commit 951689c5ea
1 changed files with 31 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,5 @@
const { randomBytes } = require("crypto"); const { randomBytes } = require("crypto");
const { writeFileSync } = require("fs");
const { join } = require("path"); const { join } = require("path");
const Mutex = require("async-mutex").Mutex; const Mutex = require("async-mutex").Mutex;
@ -29,7 +30,9 @@ const downloadVideoToMP3 = (
sendFeedback, sendFeedback,
sendError, sendError,
reinit, reinit,
options options,
metadata = undefined,
subfolder = ""
) => { ) => {
sendFeedback("Downloading…"); sendFeedback("Downloading…");
@ -70,7 +73,16 @@ const downloadVideoToMP3 = (
.on("error", sendError) .on("error", sendError)
.on("end", async () => { .on("end", async () => {
const buffer = Buffer.concat(chunks); const buffer = Buffer.concat(chunks);
await toMP3(videoName, buffer, sendFeedback, sendError, reinit, options); await toMP3(
videoName,
buffer,
sendFeedback,
sendError,
reinit,
options,
metadata,
subfolder
);
}); });
}; };
@ -80,7 +92,9 @@ const toMP3 = async (
sendFeedback, sendFeedback,
sendError, sendError,
reinit, reinit,
options options,
existingMetadata = undefined,
subfolder = ""
) => { ) => {
const safeVideoName = randomBytes(32).toString("hex"); const safeVideoName = randomBytes(32).toString("hex");
const extension = options.extension || "mp3"; const extension = options.extension || "mp3";
@ -96,7 +110,7 @@ const toMP3 = async (
ffmpeg.FS("writeFile", safeVideoName, buffer); ffmpeg.FS("writeFile", safeVideoName, buffer);
sendFeedback("Converting…"); sendFeedback("Converting…");
const metadata = getMetadata(); const metadata = existingMetadata || getMetadata();
await ffmpeg.run( await ffmpeg.run(
"-i", "-i",
safeVideoName, safeVideoName,
@ -113,14 +127,19 @@ const toMP3 = async (
replacement: "_", replacement: "_",
}); });
// Add the metadata const filePath = join(folder, subfolder, filename);
sendFeedback("Adding metadata…"); const fileBuffer = ffmpeg.FS("readFile", safeVideoName + "." + extension);
ipcRenderer.send(
"add-metadata", if (existingMetadata) {
join(folder, filename), writeFileSync(filePath, fileBuffer);
ffmpeg.FS("readFile", safeVideoName + "." + extension) reinit();
); } else {
ipcRenderer.once("add-metadata-done", reinit); // Add the metadata
sendFeedback("Adding metadata…");
ipcRenderer.send("add-metadata", filePath, fileBuffer);
ipcRenderer.once("add-metadata-done", reinit);
sendFeedback("Finished converting", metadata);
}
} catch (e) { } catch (e) {
sendError(e); sendError(e);
} finally { } finally {