haste-server/lib/document_stores/file.js

64 lines
1.5 KiB
JavaScript
Raw Normal View History

2011-11-18 23:54:57 +01:00
var fs = require('fs');
var crypto = require('crypto');
2011-11-18 23:54:57 +01:00
var winston = require('winston');
// For storing in files
2011-11-22 04:03:50 +01:00
// options[type] = file
// options[path] - Where to store
2011-11-18 23:54:57 +01:00
2011-11-19 00:04:24 +01:00
var FileDocumentStore = function(options) {
this.basePath = options.path || './data';
2011-11-28 07:15:52 +01:00
this.expire = options.expire;
2011-11-18 23:54:57 +01:00
};
// 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
2011-11-28 07:15:52 +01:00
FileDocumentStore.prototype.set = function(key, data, callback, skipExpire) {
2011-11-19 00:08:04 +01:00
try {
var _this = this;
fs.mkdir(this.basePath, '700', function() {
var fn = _this.basePath + '/' + _this.md5(key);
fs.writeFile(fn, data, 'utf8', function(err) {
2011-11-19 00:08:04 +01:00
if (err) {
callback(false);
}
else {
callback(true);
2011-11-28 07:15:52 +01:00
if (_this.expire && !skipExpire) {
winston.warn('file store cannot set expirations on keys');
2011-11-28 07:15:52 +01:00
}
2011-11-19 00:08:04 +01:00
}
});
2011-11-18 23:54:57 +01:00
});
2011-11-19 00:08:04 +01:00
} catch(err) {
callback(false);
}
2011-11-18 23:54:57 +01:00
};
// Get data from a file from key
2011-11-28 07:15:52 +01:00
FileDocumentStore.prototype.get = function(key, callback, skipExpire) {
var _this = this;
var fn = this.basePath + '/' + this.md5(key);
fs.readFile(fn, 'utf8', function(err, data) {
2011-11-18 23:54:57 +01:00
if (err) {
callback(false);
}
else {
callback(data);
2011-11-28 07:15:52 +01:00
if (_this.expire && !skipExpire) {
winston.warn('file store cannot set expirations on keys');
2011-11-28 07:15:52 +01:00
}
2011-11-18 23:54:57 +01:00
}
});
};
module.exports = FileDocumentStore;