haste-server/lib/document_stores/file.js

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;