btrtracks/src/download.js

75 lines
2.0 KiB
JavaScript

'use strict'
import readline from 'readline'
import path from 'path'
import sanitize from 'sanitize-filename'
import * as asn from './common/async'
import * as dl from './common/download'
import { dbPromise } from './database'
const values = require(path.join(process.cwd(), 'values.json'))
const musicdir = path.resolve(values.directory)
const rl = readline.createInterface({
input: process.stdin,
output: process.stdout
})
async function download (furl) {
console.log('=> Getting information..')
const data = await dl.getVideoInfo(furl)
console.log('=> Downloading file..')
const file = await dl.fetchVideo(data)
console.log('=> Cleaning up..')
const clean = dl.parseTitle(file)
let filename = clean.artist + ' - ' + clean.title + '.mp3'
console.log('=> Original Title: ' + file.title + '\n')
console.log('== Determined Title: ' + clean.title)
console.log('== Determined Artist: ' + clean.artist)
console.log('== Determined File Name: ' + filename)
const titleAnsw = await asn.askAsync(rl, `Title [${clean.title}] ? `)
const artistAnsw = await asn.askAsync(rl, `Artist [${clean.artist}] ? `)
const fileAnsw = await asn.askAsync(rl, `File [${filename}] ? `)
if (titleAnsw && titleAnsw.trim() !== '') {
clean.title = titleAnsw
}
if (artistAnsw && artistAnsw.trim() !== '') {
clean.artist = artistAnsw
}
if (fileAnsw && fileAnsw.trim() !== '') {
filename = fileAnsw
}
filename = sanitize(filename)
const fn = path.join(musicdir, filename)
await asn.setMetadata(fn, clean, file.source, true)
const addAnsw = await asn.askAsync(rl, 'Would you like to add it to the database now? [N/y] ? ')
if (addAnsw && addAnsw.trim().toLowerCase().indexOf('y') === 0) {
// Add to database
try {
const verify = await asn.getInfos(fn)
await asn.insertDB(await dbPromise, verify)
} catch (e) {
console.warn('=!= Add to database failed!')
console.error(e.stack)
}
}
console.log('=> Done.')
rl.close()
}
download(process.argv[2]).catch((e) => {
console.error(e.message)
rl.close()
})