94 lines
2.4 KiB
JavaScript
94 lines
2.4 KiB
JavaScript
var fs = require('fs');
|
|
var crypto = require('crypto');
|
|
|
|
var winston = require('winston');
|
|
|
|
// For storing in files
|
|
// options[type] = file
|
|
// options[path] - Where to store
|
|
|
|
var FileDocumentStore = function(options) {
|
|
this.basePath = options.path || './data';
|
|
this.expire = options.expire;
|
|
};
|
|
|
|
// Generate secret (10 chars)
|
|
FileDocumentStore.secret = function() {
|
|
return Math.random().toString(24).slice(-10);
|
|
};
|
|
|
|
// Generate md5 of a string
|
|
FileDocumentStore.md5 = function(str) {
|
|
var md5sum = crypto.createHash('md5');
|
|
md5sum.update(str);
|
|
return md5sum.digest('hex');
|
|
};
|
|
|
|
// Save data in a file, key as md5 - since we don't know what we could
|
|
// be passed here
|
|
FileDocumentStore.prototype.set = function(key, data, callback, skipExpire) {
|
|
try {
|
|
var _this = this;
|
|
fs.mkdir(this.basePath, '700', function() {
|
|
var sc = FileDocumentStore.secret();
|
|
var fn = _this.basePath + '/' + FileDocumentStore.md5(key);
|
|
var sfn = fn + "-" + sc;
|
|
fs.writeFile(fn, data, 'utf8', function(err) {
|
|
if (err) {
|
|
callback(false);
|
|
}
|
|
else {
|
|
fs.writeFile(sfn, "A validation file for " + fn, 'utf8', function (err) {
|
|
if (!err) callback(true, sc);
|
|
});
|
|
if (_this.expire && !skipExpire) {
|
|
winston.warn('file store cannot set expirations on keys');
|
|
}
|
|
}
|
|
});
|
|
});
|
|
} catch(err) {
|
|
callback(false);
|
|
}
|
|
};
|
|
|
|
FileDocumentStore.prototype.remove = function(str, key, callback) {
|
|
var file = FileDocumentStore.md5(str);
|
|
var kfn = key;
|
|
fs.exists(this.basePath + "/" + file + "-" + kfn, function(exists) {
|
|
if (exists) {
|
|
try {
|
|
fs.unlink(this.basePath + "/" + file, function() {
|
|
fs.unlink(this.basePath + "/" + file + "-" + kfn, function() {
|
|
callback(true);
|
|
});
|
|
});
|
|
} catch(err) {
|
|
callback(false);
|
|
}
|
|
}
|
|
else {
|
|
callback(false);
|
|
}
|
|
});
|
|
};
|
|
|
|
// Get data from a file from key
|
|
FileDocumentStore.prototype.get = function(key, callback, skipExpire) {
|
|
var _this = this;
|
|
var fn = this.basePath + '/' + FileDocumentStore.md5(key);
|
|
fs.readFile(fn, 'utf8', function(err, data) {
|
|
if (err) {
|
|
callback(false);
|
|
}
|
|
else {
|
|
callback(data);
|
|
if (_this.expire && !skipExpire) {
|
|
winston.warn('file store cannot set expirations on keys');
|
|
}
|
|
}
|
|
});
|
|
};
|
|
|
|
module.exports = FileDocumentStore;
|