-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
143 lines (109 loc) · 3.59 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
const request = require('request')
const ffmpeg = require('fluent-ffmpeg')
const fs = require('fs')
const ytdl = require('ytdl-core')
const id3 = require('node-id3')
const { promisify } = require('util')
const { getAlbum } = require('./youtube-music')
const { initProgress } = require('./progress')
const unlink = promisify(fs.unlink)
const writeId3 = promisify((...args) => id3.write(...args))
async function fetchThumbnail (album, file) {
if (fs.existsSync(file)) {
return file
}
const { thumbnails } = album.thumbnailDetails
const thumbnail = thumbnails[thumbnails.length - 1]
return new Promise((resolve, reject) => {
request(thumbnail.url)
.on('error', reject)
.pipe(fs.createWriteStream(file, { flags: 'wx' }))
.on('error', reject)
.on('finish', () => resolve(file))
})
}
function fetchRawTrack (bar, baseFile, url) {
const youtubeStream = ytdl(url, {
quality: 'highestaudio'
})
youtubeStream.on('progress', (chunk, downloaded, total) => {
bar.update(downloaded, total)
})
return new Promise((resolve, reject) => {
youtubeStream.on('error', reject)
youtubeStream.on('info', (info, format) => {
const sourceFile = `${baseFile}.${format.container}`
bar.tag('name', sourceFile)
youtubeStream
.pipe(fs.createWriteStream(sourceFile))
.on('error', reject)
.on('finish', () => resolve(sourceFile))
})
})
}
function convert (bar, format, baseFile, destFile) {
return new Promise((resolve, reject) => {
ffmpeg(baseFile)
.format(format)
.on('progress', progress => {
bar.update(progress.percent, 100)
})
.on('error', reject)
.on('end', resolve)
.save(destFile)
})
}
async function tagTrack (track, image, file) {
const tags = {
title: track.title,
artist: track.artistNames,
album: track.albumRelease.title,
image,
trackNumber: track.albumTrackIndex
}
return writeId3(tags, file)
}
async function fetchTrack (track, bar, thumbnailPromise, baseFile, url) {
const format = 'mp3'
const destFile = `${baseFile}.${format}`
if (fs.existsSync(destFile)) {
return
}
bar.reset()
bar.tag('action', 'downloading')
const sourceFile = await fetchRawTrack(bar, baseFile, url)
bar.reset()
bar.tag('name', destFile)
bar.tag('action', 'converting')
await convert(bar, format, sourceFile, destFile)
bar.tag('action', 'tagging')
bar.setSchema(':name :action')
const unlinkPromise = unlink(sourceFile)
const tagPromise = tagTrack(track, await thumbnailPromise, destFile)
await Promise.all([unlinkPromise, tagPromise])
bar.tag('action', 'done')
bar.setSchema(':name :action')
}
async function fetchAlbum (url) {
const album = await getAlbum(url)
for (const track of album.details.tracks) {
track.albumRelease = album
}
const thumbnailFile = 'thumbnail.jpg'
const thumbnailPromise = fetchThumbnail(album, thumbnailFile)
const { tracks } = album.details
const baseTitles = tracks.map(track => `${track.artistNames} - ${track.title}`)
const progress = initProgress({
name: baseTitles.reduce((size, title) => title.length > size ? title.length : size, 0) + 6,
action: 12
})
const promises = tracks.map(track => {
const bar = progress.add({ schema: ':name :action [:bar] :percent' })
const baseFile = `${track.artistNames} - ${track.title}`
const url = `https://www.youtube.com/watch?v=${encodeURIComponent(track.videoId)}`
return fetchTrack(track, bar, thumbnailPromise, baseFile, url)
})
await Promise.all(promises)
await unlink(thumbnailFile)
}
exports.fetchAlbum = fetchAlbum