85 lines
2.2 KiB
JavaScript
85 lines
2.2 KiB
JavaScript
|
const moment = require('moment'),
|
||
|
|
path = require('path');
|
||
|
|
|
||
|
|
class StorageBase {
|
||
|
|
constructor() {
|
||
|
|
Object.defineProperty(this, 'requiredFns', {
|
||
|
|
value: ['exists', 'save', 'serve', 'delete', 'read'],
|
||
|
|
writable: false
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
getTargetDir(baseDir) {
|
||
|
|
const date = moment(),
|
||
|
|
month = date.format('MM'),
|
||
|
|
year = date.format('YYYY');
|
||
|
|
|
||
|
|
if (baseDir) {
|
||
|
|
return path.join(baseDir, year, month);
|
||
|
|
}
|
||
|
|
|
||
|
|
return path.join(year, month);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
*
|
||
|
|
* @param {String} dir
|
||
|
|
* @param {String} name
|
||
|
|
* @param {String} ext
|
||
|
|
* @param {Number} i index
|
||
|
|
* @returns {Promise<String>}
|
||
|
|
*/
|
||
|
|
generateUnique(dir, name, ext, i) {
|
||
|
|
let filename,
|
||
|
|
append = '';
|
||
|
|
|
||
|
|
if (i) {
|
||
|
|
append = '-' + i;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (ext) {
|
||
|
|
filename = name + append + ext;
|
||
|
|
} else {
|
||
|
|
filename = name + append;
|
||
|
|
}
|
||
|
|
|
||
|
|
return this.exists(filename, dir).then((exists) => {
|
||
|
|
if (exists) {
|
||
|
|
i = i + 1;
|
||
|
|
return this.generateUnique(dir, name, ext, i);
|
||
|
|
} else {
|
||
|
|
return path.join(dir, filename);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @param {Object} file
|
||
|
|
* @param {String} file.name
|
||
|
|
* @param {String} targetDir
|
||
|
|
*
|
||
|
|
* @returns {Promise<String>} unique file path
|
||
|
|
*/
|
||
|
|
getUniqueFileName(file, targetDir) {
|
||
|
|
var ext = path.extname(file.name), name;
|
||
|
|
|
||
|
|
// poor extension validation
|
||
|
|
// .1 or .342 is not a valid extension, .mp4 is though!
|
||
|
|
if (!ext.match(/\.\d+$/)) {
|
||
|
|
name = this.getSanitizedFileName(path.basename(file.name, ext));
|
||
|
|
return this.generateUnique(targetDir, name, ext, 0);
|
||
|
|
} else {
|
||
|
|
name = this.getSanitizedFileName(path.basename(file.name));
|
||
|
|
return this.generateUnique(targetDir, name, null, 0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
getSanitizedFileName(fileName) {
|
||
|
|
// below only matches ascii characters, @, and .
|
||
|
|
// unicode filenames like город.zip would therefore resolve to ----.zip
|
||
|
|
return fileName.replace(/[^\w@.]/gi, '-');
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = StorageBase;
|