Add custom menu in downloader plugin ("download playlist" item)

This commit is contained in:
TC
2021-03-24 21:52:12 +01:00
parent bee2da567b
commit e72915c5d0

View File

@ -0,0 +1,63 @@
const { existsSync, mkdirSync } = require("fs");
const { join } = require("path");
const { URL } = require("url");
const { ipcMain } = require("electron");
const is = require("electron-is");
const ytpl = require("ytpl");
const { sendError } = require("./back");
const { defaultMenuDownloadLabel, getFolder } = require("./utils");
let downloadLabel = defaultMenuDownloadLabel;
module.exports = (win, options, refreshMenu) => [
{
label: downloadLabel,
click: async () => {
const currentURL = win.webContents.getURL();
const playlistID = new URL(currentURL).searchParams.get("list");
if (!playlistID) {
sendError(win, new Error("No playlist ID found"));
return;
}
const playlist = await ytpl(playlistID);
const playlistTitle = playlist.title;
const folder = getFolder(options.downloadFolder);
const playlistFolder = join(folder, playlistTitle);
if (existsSync(playlistFolder)) {
sendError(
win,
new Error(`The folder ${playlistFolder} already exists`)
);
return;
}
mkdirSync(playlistFolder, { recursive: true });
ipcMain.on("downloader-feedback", (_, feedback) => {
downloadLabel = feedback;
refreshMenu();
});
downloadLabel = `Downloading "${playlistTitle}"`;
refreshMenu();
if (is.dev()) {
console.log(
`Downloading playlist "${playlistTitle}" (${playlist.items.length} songs)`
);
}
playlist.items.slice(0, options.playlistMaxItems).forEach((song) => {
win.webContents.send(
"downloader-download-playlist",
song,
playlistTitle,
options
);
});
},
},
];