-
Notifications
You must be signed in to change notification settings - Fork 559
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
download using youtubei,js instead of ytdl-core
- Loading branch information
Showing
9 changed files
with
308 additions
and
329 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,193 @@ | ||
| const { existsSync, mkdirSync, createWriteStream, writeFileSync } = require('fs'); | ||
| const { ipcMain, app } = require("electron"); | ||
| const { join } = require("path"); | ||
|
|
||
| const { Innertube, UniversalCache, Utils } = require('youtubei.js'); | ||
| const filenamify = require("filenamify"); | ||
| const id3 = require('node-id3').Promise; | ||
|
|
||
| const { sendError } = require("./back"); | ||
| const { presets } = require('./utils'); | ||
|
|
||
| ffmpegWriteTags | ||
| /** @type {Innertube} */ | ||
| let yt; | ||
| let options; | ||
|
|
||
| module.exports = async (options_) => { | ||
| options = options_; | ||
| yt = await Innertube.create({ cache: new UniversalCache(false), generate_session_locally: true }); | ||
| ipcMain.handle("download-song", (_, url) => downloadSong(url)); | ||
| }; | ||
|
|
||
| async function downloadSong(url, playlistFolder = undefined) { | ||
| const metadata = await getMetadata(url); | ||
|
|
||
| const stream = await yt.download(metadata.id, { | ||
| type: 'audio', // audio, video or video+audio | ||
| quality: 'best', // best, bestefficiency, 144p, 240p, 480p, 720p and so on. | ||
| format: 'any' // media container format | ||
| }); | ||
|
|
||
| console.info(`Downloading ${metadata.artist} - ${metadata.title} {${metadata.id}}...`); | ||
|
|
||
| const iterableStream = Utils.streamToIterable(stream); | ||
|
|
||
| const dir = playlistFolder || options.downloadFolder || app.getPath("downloads"); | ||
| const name = `${metadata.artist ? `${metadata.artist} - ` : ""}${metadata.title}`; | ||
|
|
||
| const extension = presets[options.preset]?.extension || 'mp3'; | ||
|
|
||
| const filename = filenamify(`${name}.${extension}`, { | ||
| replacement: "_", | ||
| maxLength: 255, | ||
| }); | ||
| const filePath = join(dir, filename); | ||
|
|
||
| if (!existsSync(dir)) { | ||
| mkdirSync(dir); | ||
| } | ||
|
|
||
| if (!presets[options.preset]) { | ||
| await toMP3(iterableStream, filePath, metadata); | ||
| console.info('writing id3 tags...'); // DELETE | ||
| await writeID3(filePath, metadata).then(() => console.info('done writing id3 tags!')); // DELETE | ||
| } else { | ||
| const file = createWriteStream(filePath); | ||
| //stream.pipeTo(file); | ||
| for await (const chunk of iterableStream) { | ||
| file.write(chunk); | ||
| } | ||
| ffmpegWriteTags(filePath, metadata, presets[options.preset]?.ffmpegArgs); | ||
| } | ||
|
|
||
| console.info(`${filePath} - Done!`, '\n'); | ||
| } | ||
| module.exports.downloadSong = downloadSong; | ||
|
|
||
| function getIdFromUrl(url) { | ||
| const match = url.match(/v=([^&]+)/); | ||
| return match ? match[1] : null; | ||
| } | ||
|
|
||
| async function getMetadata(url) { | ||
| const id = getIdFromUrl(url); | ||
| const info = await yt.music.getInfo(id); | ||
| //console.log('got info:' + JSON.stringify(info, null, 2)); // DELETE | ||
|
|
||
| return { | ||
| id: info.basic_info.id, | ||
| title: info.basic_info.title, | ||
| artist: info.basic_info.author, | ||
| album: info.player_overlays?.browser_media_session?.album?.text, | ||
| image: info.basic_info.thumbnail[0].url, | ||
| }; | ||
| } | ||
|
|
||
| const { getImage } = require("../../providers/song-info"); | ||
| const { cropMaxWidth } = require("./utils"); | ||
|
|
||
| async function writeID3(filePath, metadata) { | ||
| const tags = { | ||
| title: metadata.title, | ||
| artist: metadata.artist, | ||
| album: metadata.album, | ||
| image: { | ||
| mime: "image/png", | ||
| type: { | ||
| id: 3, | ||
| name: "front cover" | ||
| }, | ||
| description: "", | ||
| imageBuffer: cropMaxWidth(await getImage(metadata.image))?.toPNG(), | ||
| } | ||
| // TODO: lyrics | ||
| }; | ||
|
|
||
| await id3.write(tags, filePath); | ||
| } | ||
|
|
||
| const { randomBytes } = require("crypto"); | ||
| const Mutex = require("async-mutex").Mutex; | ||
| const ffmpeg = require("@ffmpeg/ffmpeg").createFFmpeg({ | ||
| log: false, | ||
| logger: () => { }, // console.log, | ||
| progress: () => { }, // console.log, | ||
| }); | ||
|
|
||
| const ffmpegMutex = new Mutex(); | ||
|
|
||
| async function toMP3(stream, filePath, metadata, extension = "mp3") { | ||
| const chunks = []; | ||
| for await (const chunk of stream) { | ||
| chunks.push(chunk); | ||
| } | ||
| const buffer = Buffer.concat(chunks); | ||
| const safeVideoName = randomBytes(32).toString("hex"); | ||
| const releaseFFmpegMutex = await ffmpegMutex.acquire(); | ||
|
|
||
| try { | ||
| if (!ffmpeg.isLoaded()) { | ||
| // sendFeedback("Loading…", 2); // indefinite progress bar after download | ||
| await ffmpeg.load(); | ||
| } | ||
|
|
||
| // sendFeedback("Preparing file…"); | ||
| ffmpeg.FS("writeFile", safeVideoName, buffer); | ||
|
|
||
| // sendFeedback("Converting…"); | ||
|
|
||
| await ffmpeg.run( | ||
| "-i", | ||
| safeVideoName, | ||
| ...getFFmpegMetadataArgs(metadata), | ||
| safeVideoName + "." + extension | ||
| ); | ||
|
|
||
| const fileBuffer = ffmpeg.FS("readFile", safeVideoName + "." + extension); | ||
|
|
||
| await writeID3(fileBuffer, metadata); | ||
|
|
||
| // sendFeedback("Saving…"); | ||
|
|
||
| writeFileSync(filePath, fileBuffer); | ||
| } catch (e) { | ||
| sendError(e); | ||
| } finally { | ||
| releaseFFmpegMutex(); | ||
| } | ||
| } | ||
|
|
||
| async function ffmpegWriteTags(filePath, metadata, ffmpegArgs = []) { | ||
| const releaseFFmpegMutex = await ffmpegMutex.acquire(); | ||
|
|
||
| try { | ||
| if (!ffmpeg.isLoaded()) { | ||
| await ffmpeg.load(); | ||
| } | ||
|
|
||
| await ffmpeg.run( | ||
| "-i", | ||
| filePath, | ||
| ...getFFmpegMetadataArgs(metadata), | ||
| //...ffmpegArgs, | ||
| filePath | ||
| ); | ||
| } catch (e) { | ||
| sendError(e); | ||
| } finally { | ||
| releaseFFmpegMutex(); | ||
| } | ||
| } | ||
|
|
||
| function getFFmpegMetadataArgs(metadata) { | ||
| if (!metadata) { | ||
| return; | ||
| } | ||
|
|
||
| return [ | ||
| ...(metadata.title ? ["-metadata", `title=${metadata.title}`] : []), | ||
| ...(metadata.artist ? ["-metadata", `artist=${metadata.artist}`] : []), | ||
| ...(metadata.album ? ["-metadata", `album=${metadata.album}`] : []), | ||
| ]; | ||
| }; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.