mirror of
https://github.com/TryGhost/Ghost.git
synced 2025-02-03 23:00:14 -05:00
07e59cf27e
no issue - getUniqueFileName does not replace . by - - added poor extensions validation
58 lines
1.4 KiB
JavaScript
58 lines
1.4 KiB
JavaScript
var moment = require('moment'),
|
|
path = require('path');
|
|
|
|
function StorageBase() {
|
|
}
|
|
|
|
StorageBase.prototype.getTargetDir = function (baseDir) {
|
|
var m = moment(),
|
|
month = m.format('MM'),
|
|
year = m.format('YYYY');
|
|
|
|
if (baseDir) {
|
|
return path.join(baseDir, year, month);
|
|
}
|
|
|
|
return path.join(year, month);
|
|
};
|
|
|
|
StorageBase.prototype.generateUnique = function (store, dir, name, ext, i) {
|
|
var self = this,
|
|
filename,
|
|
append = '';
|
|
|
|
if (i) {
|
|
append = '-' + i;
|
|
}
|
|
|
|
if (ext) {
|
|
filename = path.join(dir, name + append + ext);
|
|
} else {
|
|
filename = path.join(dir, name + append);
|
|
}
|
|
|
|
return store.exists(filename).then(function (exists) {
|
|
if (exists) {
|
|
i = i + 1;
|
|
return self.generateUnique(store, dir, name, ext, i);
|
|
} else {
|
|
return filename;
|
|
}
|
|
});
|
|
};
|
|
|
|
StorageBase.prototype.getUniqueFileName = function (store, image, targetDir) {
|
|
var ext = path.extname(image.name), name;
|
|
|
|
// poor extension validation
|
|
// .1 is not a valid extension
|
|
if (!ext.match(/.\d/)) {
|
|
name = path.basename(image.name, ext).replace(/[^\w@.]/gi, '-');
|
|
return this.generateUnique(store, targetDir, name, ext, 0);
|
|
} else {
|
|
name = path.basename(image.name).replace(/[^\w@.]/gi, '-');
|
|
return this.generateUnique(store, targetDir, name, null, 0);
|
|
}
|
|
};
|
|
|
|
module.exports = StorageBase;
|