finished basic sorting functionality
This commit is contained in:
parent
5f220204b3
commit
3eabcbee73
1 changed files with 94 additions and 61 deletions
155
badger-am.js
155
badger-am.js
|
|
@ -12,6 +12,9 @@ const metadata = require('musicmetadata');
|
||||||
const progress = require('progress');
|
const progress = require('progress');
|
||||||
const app = require('./package.json');
|
const app = require('./package.json');
|
||||||
|
|
||||||
|
// generate timestamp
|
||||||
|
const start = process.hrtime();
|
||||||
|
|
||||||
// general options
|
// general options
|
||||||
commander
|
commander
|
||||||
.version(app.version)
|
.version(app.version)
|
||||||
|
|
@ -41,33 +44,35 @@ commander
|
||||||
// parse command line arguments
|
// parse command line arguments
|
||||||
commander.parse(process.argv);
|
commander.parse(process.argv);
|
||||||
|
|
||||||
// functions
|
// sort files
|
||||||
function sort(input, output, options) {
|
function sort(input, output, options) {
|
||||||
const start = process.hrtime();
|
|
||||||
const concurrency = commander.concurrency;
|
const concurrency = commander.concurrency;
|
||||||
console.log('--- SORT MODE ---');
|
async.waterfall([
|
||||||
console.log('concurrency: ' + concurrency);
|
function (asyncCallback) {
|
||||||
// get files in input directory
|
recursive(input, [ignoreFilter], asyncCallback);
|
||||||
recursive(input, [ignoreFilter], function (err, files) {
|
},
|
||||||
if (err) {
|
function (files, asyncCallback) {
|
||||||
exit(err);
|
console.log(files.length + ' \'' + options.format + '\' files found');
|
||||||
}
|
// display progressbar
|
||||||
// display progressbar
|
const bar = createProgressBar(files.length);
|
||||||
const bar = createProgressBar(files.length);
|
// handle each file
|
||||||
// handle each file
|
async.eachLimit(files, concurrency, function (file, eachCallback) {
|
||||||
async.eachLimit(files, concurrency, function (file, eachCallback) {
|
processFile(output, file, function (err) {
|
||||||
handleSource(output, file, function (err) {
|
bar.tick();
|
||||||
bar.tick();
|
if (err) {
|
||||||
|
return eachCallback(err);
|
||||||
|
}
|
||||||
|
eachCallback();
|
||||||
|
});
|
||||||
|
}, function (err) {
|
||||||
if (err) {
|
if (err) {
|
||||||
return eachCallback(err);
|
return asyncCallback(err);
|
||||||
}
|
}
|
||||||
eachCallback();
|
asyncCallback();
|
||||||
});
|
});
|
||||||
}, function (err) {
|
}
|
||||||
const diff = process.hrtime(start);
|
], function (err, result) {
|
||||||
console.log('done after ' + (diff[0] * 1000 + diff[1] / 1000000) + ' milliseconds');
|
exit(err);
|
||||||
exit(err);
|
|
||||||
});
|
|
||||||
});
|
});
|
||||||
|
|
||||||
function ignoreFilter(file, stats) {
|
function ignoreFilter(file, stats) {
|
||||||
|
|
@ -75,24 +80,18 @@ function sort(input, output, options) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function exit(err) {
|
// move file to location defined by its metadata
|
||||||
if (err) {
|
function processFile(output, sourceFile, callback) {
|
||||||
console.error(err);
|
|
||||||
process.exit(1);
|
|
||||||
}
|
|
||||||
console.log('DONE');
|
|
||||||
process.exit(0);
|
|
||||||
}
|
|
||||||
|
|
||||||
function handleSource(output, file, callback) {
|
|
||||||
async.waterfall([
|
async.waterfall([
|
||||||
function (asyncCallback) {
|
function (asyncCallback) {
|
||||||
extractMetadata(file, asyncCallback)
|
extractMetadata(sourceFile, asyncCallback)
|
||||||
},
|
},
|
||||||
function (metadata, asyncCallback) {
|
function (sourceFile, metadata, asyncCallback) {
|
||||||
pathFromMetadata(output, metadata, asyncCallback);
|
pathFromMetadata(sourceFile, output, metadata, asyncCallback);
|
||||||
|
},
|
||||||
|
function (targetFile, asyncCallback) {
|
||||||
|
moveFile(sourceFile, targetFile, asyncCallback);
|
||||||
}
|
}
|
||||||
// TODO: create output directory, append file extension and move file
|
|
||||||
], function (err, results) {
|
], function (err, results) {
|
||||||
if (err) {
|
if (err) {
|
||||||
return callback(err);
|
return callback(err);
|
||||||
|
|
@ -101,20 +100,22 @@ function handleSource(output, file, callback) {
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
function extractMetadata(file, callback) {
|
// extract metadata for further processing
|
||||||
const stream = fs.createReadStream(file);
|
function extractMetadata(sourceFile, callback) {
|
||||||
|
const stream = fs.createReadStream(sourceFile);
|
||||||
metadata(stream, function (err, metadata) {
|
metadata(stream, function (err, metadata) {
|
||||||
if (err) {
|
if (err) {
|
||||||
return callback(err);
|
return callback(err);
|
||||||
}
|
}
|
||||||
stream.close();
|
stream.close();
|
||||||
callback(null, metadata)
|
callback(null, sourceFile, metadata)
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
function pathFromMetadata(output, metadata, callback) {
|
// create path for target file
|
||||||
|
function pathFromMetadata(file, output, metadata, callback) {
|
||||||
// define directory
|
// define directory
|
||||||
let directory = path.normalize(path.join(output));
|
let filePath = path.normalize(output);
|
||||||
if (metadata.albumartist && metadata.albumartist.length > 0) {
|
if (metadata.albumartist && metadata.albumartist.length > 0) {
|
||||||
let tmp;
|
let tmp;
|
||||||
const artistCount = metadata.albumartist.length;
|
const artistCount = metadata.albumartist.length;
|
||||||
|
|
@ -125,45 +126,77 @@ function pathFromMetadata(output, metadata, callback) {
|
||||||
tmp = metadata.albumartist[counter];
|
tmp = metadata.albumartist[counter];
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
directory = path.join(directory, tmp);
|
filePath = path.join(filePath, tmp);
|
||||||
} else {
|
} else {
|
||||||
directory = path.join(directory, metadata.artist[0]);
|
filePath = path.join(filePath, metadata.artist[0]);
|
||||||
}
|
}
|
||||||
if (metadata.album) {
|
if (metadata.album) {
|
||||||
directory = path.join(directory, metadata.album);
|
filePath = path.join(filePath, metadata.album);
|
||||||
}
|
}
|
||||||
// define filename
|
// define filename
|
||||||
let file = '';
|
let fileName = '';
|
||||||
if (metadata.disk.no) {
|
if (metadata.disk.no) {
|
||||||
file += frontFill(metadata.disk.no, '0', 2);
|
fileName += frontFill(metadata.disk.no, '0', 2);
|
||||||
}
|
}
|
||||||
if (metadata.track.no) {
|
if (metadata.track.no) {
|
||||||
if (file) {
|
if (fileName) {
|
||||||
file += '-' + frontFill(metadata.track.no, '0', 2) + ' ';
|
fileName += '-' + frontFill(metadata.track.no, '0', 2) + ' ';
|
||||||
} else {
|
} else {
|
||||||
file += frontFill(metadata.track.no, '0', 2) + ' ';
|
fileName += frontFill(metadata.track.no, '0', 2) + ' ';
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (metadata.artist && metadata.albumartist.length > 0) {
|
if (metadata.artist) {
|
||||||
file += metadata.artist[0] + ' - ';
|
fileName += metadata.artist[0] + ' - ';
|
||||||
}
|
}
|
||||||
if (metadata.title) {
|
if (metadata.title) {
|
||||||
file += metadata.title;
|
fileName += metadata.title;
|
||||||
}
|
}
|
||||||
|
// append extension
|
||||||
callback(null, directory, file);
|
fileName += path.extname(file);
|
||||||
}
|
// join directory and name
|
||||||
|
callback(null, path.join(filePath, fileName));
|
||||||
function frontFill(string, fill, length) {
|
|
||||||
while (string.length > length) {
|
|
||||||
string = fill + string;
|
|
||||||
}
|
|
||||||
return string;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// create a ascii progressbar
|
||||||
function createProgressBar(total) {
|
function createProgressBar(total) {
|
||||||
return new progress(':bar | progress: :current/:total (:percent) | elapsed: :elapseds | eta: :etas', {
|
return new progress(':bar | progress: :current/:total (:percent) | elapsed: :elapseds | eta: :etas', {
|
||||||
total: total,
|
total: total,
|
||||||
width: 32
|
width: 32
|
||||||
});
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// create target directory and move the file
|
||||||
|
function moveFile(source, target, callback) {
|
||||||
|
async.series([
|
||||||
|
function (asyncCallback) {
|
||||||
|
fse.mkdirs(path.dirname(target), asyncCallback);
|
||||||
|
},
|
||||||
|
function (asyncCallback) {
|
||||||
|
fse.move(source, target, asyncCallback);
|
||||||
|
}
|
||||||
|
], function (err, results) {
|
||||||
|
if (err) {
|
||||||
|
return callback(err);
|
||||||
|
}
|
||||||
|
callback();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// fill a string beginning from the front
|
||||||
|
function frontFill(string, fill, length) {
|
||||||
|
while (string.toString().length < length) {
|
||||||
|
string = fill + string;
|
||||||
|
}
|
||||||
|
return string;
|
||||||
|
}
|
||||||
|
|
||||||
|
// shutdown
|
||||||
|
function exit(err) {
|
||||||
|
if (err) {
|
||||||
|
console.error(err);
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
const diff = process.hrtime(start);
|
||||||
|
console.log('exiting after ' + ((diff[0] + (diff[1] / 1000000)) / 1000).toFixed(2) + ' seconds');
|
||||||
|
process.exit(0);
|
||||||
}
|
}
|
||||||
Loading…
Reference in a new issue