490 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			490 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
/*!
 | 
						|
 * Tmp
 | 
						|
 *
 | 
						|
 * Copyright (c) 2011-2015 KARASZI Istvan <github@spam.raszi.hu>
 | 
						|
 *
 | 
						|
 * MIT Licensed
 | 
						|
 */
 | 
						|
 | 
						|
/**
 | 
						|
 * Module dependencies.
 | 
						|
 */
 | 
						|
var
 | 
						|
  fs     = require('fs'),
 | 
						|
  path   = require('path'),
 | 
						|
  crypto = require('crypto'),
 | 
						|
  tmpDir = require('os-tmpdir'),
 | 
						|
  _c     = process.binding('constants');
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * The working inner variables.
 | 
						|
 */
 | 
						|
var
 | 
						|
  // store the actual TMP directory
 | 
						|
  _TMP = tmpDir(),
 | 
						|
 | 
						|
  // the random characters to choose from
 | 
						|
  RANDOM_CHARS = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz',
 | 
						|
 | 
						|
  TEMPLATE_PATTERN = /XXXXXX/,
 | 
						|
 | 
						|
  DEFAULT_TRIES = 3,
 | 
						|
 | 
						|
  CREATE_FLAGS = (_c.O_CREAT || _c.fs.O_CREAT) | (_c.O_EXCL || _c.fs.O_EXCL) | (_c.O_RDWR || _c.fs.O_RDWR),
 | 
						|
 | 
						|
  EBADF = _c.EBADF || _c.os.errno.EBADF,
 | 
						|
  ENOENT = _c.ENOENT || _c.os.errno.ENOENT,
 | 
						|
 | 
						|
  DIR_MODE = 448 /* 0700 */,
 | 
						|
  FILE_MODE = 384 /* 0600 */,
 | 
						|
 | 
						|
  // this will hold the objects need to be removed on exit
 | 
						|
  _removeObjects = [],
 | 
						|
 | 
						|
  _gracefulCleanup = false,
 | 
						|
  _uncaughtException = false;
 | 
						|
 | 
						|
/**
 | 
						|
 * Random name generator based on crypto.
 | 
						|
 * Adapted from http://blog.tompawlak.org/how-to-generate-random-values-nodejs-javascript
 | 
						|
 *
 | 
						|
 * @param {Number} howMany
 | 
						|
 * @return {String}
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _randomChars(howMany) {
 | 
						|
  var
 | 
						|
    value = [],
 | 
						|
    rnd = null;
 | 
						|
 | 
						|
  // make sure that we do not fail because we ran out of entropy
 | 
						|
  try {
 | 
						|
    rnd = crypto.randomBytes(howMany);
 | 
						|
  } catch (e) {
 | 
						|
    rnd = crypto.pseudoRandomBytes(howMany);
 | 
						|
  }
 | 
						|
 | 
						|
  for (var i = 0; i < howMany; i++) {
 | 
						|
    value.push(RANDOM_CHARS[rnd[i] % RANDOM_CHARS.length]);
 | 
						|
  }
 | 
						|
 | 
						|
  return value.join('');
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Checks whether the `obj` parameter is defined or not.
 | 
						|
 *
 | 
						|
 * @param {Object} obj
 | 
						|
 * @return {Boolean}
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _isUndefined(obj) {
 | 
						|
  return typeof obj === 'undefined';
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Parses the function arguments.
 | 
						|
 *
 | 
						|
 * This function helps to have optional arguments.
 | 
						|
 *
 | 
						|
 * @param {Object} options
 | 
						|
 * @param {Function} callback
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _parseArguments(options, callback) {
 | 
						|
  if (typeof options == 'function') {
 | 
						|
    var
 | 
						|
      tmp = options,
 | 
						|
      options = callback || {},
 | 
						|
      callback = tmp;
 | 
						|
  } else if (typeof options == 'undefined') {
 | 
						|
    options = {};
 | 
						|
  }
 | 
						|
 | 
						|
  return [options, callback];
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Generates a new temporary name.
 | 
						|
 *
 | 
						|
 * @param {Object} opts
 | 
						|
 * @returns {String}
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _generateTmpName(opts) {
 | 
						|
  if (opts.name) {
 | 
						|
    return path.join(opts.dir || _TMP, opts.name);
 | 
						|
  }
 | 
						|
 | 
						|
  // mkstemps like template
 | 
						|
  if (opts.template) {
 | 
						|
    return opts.template.replace(TEMPLATE_PATTERN, _randomChars(6));
 | 
						|
  }
 | 
						|
 | 
						|
  // prefix and postfix
 | 
						|
  var name = [
 | 
						|
    opts.prefix || 'tmp-',
 | 
						|
    process.pid,
 | 
						|
    _randomChars(12),
 | 
						|
    opts.postfix || ''
 | 
						|
  ].join('');
 | 
						|
 | 
						|
  return path.join(opts.dir || _TMP, name);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Gets a temporary file name.
 | 
						|
 *
 | 
						|
 * @param {Object} options
 | 
						|
 * @param {Function} callback
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _getTmpName(options, callback) {
 | 
						|
  var
 | 
						|
    args = _parseArguments(options, callback),
 | 
						|
    opts = args[0],
 | 
						|
    cb = args[1],
 | 
						|
    tries = opts.tries || DEFAULT_TRIES;
 | 
						|
 | 
						|
  if (isNaN(tries) || tries < 0)
 | 
						|
    return cb(new Error('Invalid tries'));
 | 
						|
 | 
						|
  if (opts.template && !opts.template.match(TEMPLATE_PATTERN))
 | 
						|
    return cb(new Error('Invalid template provided'));
 | 
						|
 | 
						|
  (function _getUniqueName() {
 | 
						|
    var name = _generateTmpName(opts);
 | 
						|
 | 
						|
    // check whether the path exists then retry if needed
 | 
						|
    fs.stat(name, function (err) {
 | 
						|
      if (!err) {
 | 
						|
        if (tries-- > 0) return _getUniqueName();
 | 
						|
 | 
						|
        return cb(new Error('Could not get a unique tmp filename, max tries reached ' + name));
 | 
						|
      }
 | 
						|
 | 
						|
      cb(null, name);
 | 
						|
    });
 | 
						|
  }());
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Synchronous version of _getTmpName.
 | 
						|
 *
 | 
						|
 * @param {Object} options
 | 
						|
 * @returns {String}
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _getTmpNameSync(options) {
 | 
						|
  var
 | 
						|
    args = _parseArguments(options),
 | 
						|
    opts = args[0],
 | 
						|
    tries = opts.tries || DEFAULT_TRIES;
 | 
						|
 | 
						|
  if (isNaN(tries) || tries < 0)
 | 
						|
    throw new Error('Invalid tries');
 | 
						|
 | 
						|
  if (opts.template && !opts.template.match(TEMPLATE_PATTERN))
 | 
						|
    throw new Error('Invalid template provided');
 | 
						|
 | 
						|
  do {
 | 
						|
    var name = _generateTmpName(opts);
 | 
						|
    try {
 | 
						|
      fs.statSync(name);
 | 
						|
    } catch (e) {
 | 
						|
      return name;
 | 
						|
    }
 | 
						|
  } while (tries-- > 0);
 | 
						|
 | 
						|
  throw new Error('Could not get a unique tmp filename, max tries reached');
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Creates and opens a temporary file.
 | 
						|
 *
 | 
						|
 * @param {Object} options
 | 
						|
 * @param {Function} callback
 | 
						|
 * @api public
 | 
						|
 */
 | 
						|
function _createTmpFile(options, callback) {
 | 
						|
  var
 | 
						|
    args = _parseArguments(options, callback),
 | 
						|
    opts = args[0],
 | 
						|
    cb = args[1];
 | 
						|
 | 
						|
  opts.postfix = (_isUndefined(opts.postfix)) ? '.tmp' : opts.postfix;
 | 
						|
 | 
						|
  // gets a temporary filename
 | 
						|
  _getTmpName(opts, function _tmpNameCreated(err, name) {
 | 
						|
    if (err) return cb(err);
 | 
						|
 | 
						|
    // create and open the file
 | 
						|
    fs.open(name, CREATE_FLAGS, opts.mode || FILE_MODE, function _fileCreated(err, fd) {
 | 
						|
      if (err) return cb(err);
 | 
						|
 | 
						|
      if (opts.discardDescriptor) {
 | 
						|
        return fs.close(fd, function _discardCallback(err) {
 | 
						|
          if (err) {
 | 
						|
            // Low probability, and the file exists, so this could be
 | 
						|
            // ignored.  If it isn't we certainly need to unlink the
 | 
						|
            // file, and if that fails too its error is more
 | 
						|
            // important.
 | 
						|
            try {
 | 
						|
              fs.unlinkSync(name);
 | 
						|
            } catch (e) {
 | 
						|
              err = e;
 | 
						|
            }
 | 
						|
            return cb(err);
 | 
						|
          }
 | 
						|
          cb(null, name, undefined, _prepareTmpFileRemoveCallback(name, -1, opts));
 | 
						|
        });
 | 
						|
      }
 | 
						|
      if (opts.detachDescriptor) {
 | 
						|
        return cb(null, name, fd, _prepareTmpFileRemoveCallback(name, -1, opts));
 | 
						|
      }
 | 
						|
      cb(null, name, fd, _prepareTmpFileRemoveCallback(name, fd, opts));
 | 
						|
    });
 | 
						|
  });
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Synchronous version of _createTmpFile.
 | 
						|
 *
 | 
						|
 * @param {Object} options
 | 
						|
 * @returns {Object} object consists of name, fd and removeCallback
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _createTmpFileSync(options) {
 | 
						|
  var
 | 
						|
    args = _parseArguments(options),
 | 
						|
    opts = args[0];
 | 
						|
 | 
						|
  opts.postfix = opts.postfix || '.tmp';
 | 
						|
 | 
						|
  var name = _getTmpNameSync(opts);
 | 
						|
  var fd = fs.openSync(name, CREATE_FLAGS, opts.mode || FILE_MODE);
 | 
						|
 | 
						|
  return {
 | 
						|
    name : name,
 | 
						|
    fd : fd,
 | 
						|
    removeCallback : _prepareTmpFileRemoveCallback(name, fd, opts)
 | 
						|
  };
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Removes files and folders in a directory recursively.
 | 
						|
 *
 | 
						|
 * @param {String} root
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _rmdirRecursiveSync(root) {
 | 
						|
  var dirs = [root];
 | 
						|
 | 
						|
  do {
 | 
						|
    var
 | 
						|
      dir = dirs.pop(),
 | 
						|
      deferred = false,
 | 
						|
      files = fs.readdirSync(dir);
 | 
						|
 | 
						|
    for (var i = 0, length = files.length; i < length; i++) {
 | 
						|
      var
 | 
						|
        file = path.join(dir, files[i]),
 | 
						|
        stat = fs.lstatSync(file); // lstat so we don't recurse into symlinked directories
 | 
						|
 | 
						|
      if (stat.isDirectory()) {
 | 
						|
        if (!deferred) {
 | 
						|
          deferred = true;
 | 
						|
          dirs.push(dir);
 | 
						|
        }
 | 
						|
        dirs.push(file);
 | 
						|
      } else {
 | 
						|
        fs.unlinkSync(file);
 | 
						|
      }
 | 
						|
    }
 | 
						|
 | 
						|
    if (!deferred) {
 | 
						|
      fs.rmdirSync(dir);
 | 
						|
    }
 | 
						|
  } while (dirs.length !== 0);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Creates a temporary directory.
 | 
						|
 *
 | 
						|
 * @param {Object} options
 | 
						|
 * @param {Function} callback
 | 
						|
 * @api public
 | 
						|
 */
 | 
						|
function _createTmpDir(options, callback) {
 | 
						|
  var
 | 
						|
    args = _parseArguments(options, callback),
 | 
						|
    opts = args[0],
 | 
						|
    cb = args[1];
 | 
						|
 | 
						|
  // gets a temporary filename
 | 
						|
  _getTmpName(opts, function _tmpNameCreated(err, name) {
 | 
						|
    if (err) return cb(err);
 | 
						|
 | 
						|
    // create the directory
 | 
						|
    fs.mkdir(name, opts.mode || DIR_MODE, function _dirCreated(err) {
 | 
						|
      if (err) return cb(err);
 | 
						|
 | 
						|
      cb(null, name, _prepareTmpDirRemoveCallback(name, opts));
 | 
						|
    });
 | 
						|
  });
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Synchronous version of _createTmpDir.
 | 
						|
 *
 | 
						|
 * @param {Object} options
 | 
						|
 * @returns {Object} object consists of name and removeCallback
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _createTmpDirSync(options) {
 | 
						|
  var
 | 
						|
    args = _parseArguments(options),
 | 
						|
    opts = args[0];
 | 
						|
 | 
						|
  var name = _getTmpNameSync(opts);
 | 
						|
  fs.mkdirSync(name, opts.mode || DIR_MODE);
 | 
						|
 | 
						|
  return {
 | 
						|
    name : name,
 | 
						|
    removeCallback : _prepareTmpDirRemoveCallback(name, opts)
 | 
						|
  };
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Prepares the callback for removal of the temporary file.
 | 
						|
 *
 | 
						|
 * @param {String} name
 | 
						|
 * @param {int} fd
 | 
						|
 * @param {Object} opts
 | 
						|
 * @api private
 | 
						|
 * @returns {Function} the callback
 | 
						|
 */
 | 
						|
function _prepareTmpFileRemoveCallback(name, fd, opts) {
 | 
						|
  var removeCallback = _prepareRemoveCallback(function _removeCallback(fdPath) {
 | 
						|
    try {
 | 
						|
      if (0 <= fdPath[0]) {
 | 
						|
        fs.closeSync(fdPath[0]);
 | 
						|
      }
 | 
						|
    }
 | 
						|
    catch (e) {
 | 
						|
      // under some node/windows related circumstances, a temporary file
 | 
						|
      // may have not be created as expected or the file was already closed
 | 
						|
      // by the user, in which case we will simply ignore the error
 | 
						|
      if (e.errno != -EBADF && e.errno != -ENOENT) {
 | 
						|
        // reraise any unanticipated error
 | 
						|
        throw e;
 | 
						|
      }
 | 
						|
    }
 | 
						|
    fs.unlinkSync(fdPath[1]);
 | 
						|
  }, [fd, name]);
 | 
						|
 | 
						|
  if (!opts.keep) {
 | 
						|
    _removeObjects.unshift(removeCallback);
 | 
						|
  }
 | 
						|
 | 
						|
  return removeCallback;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Prepares the callback for removal of the temporary directory.
 | 
						|
 *
 | 
						|
 * @param {String} name
 | 
						|
 * @param {Object} opts
 | 
						|
 * @returns {Function} the callback
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _prepareTmpDirRemoveCallback(name, opts) {
 | 
						|
  var removeFunction = opts.unsafeCleanup ? _rmdirRecursiveSync : fs.rmdirSync.bind(fs);
 | 
						|
  var removeCallback = _prepareRemoveCallback(removeFunction, name);
 | 
						|
 | 
						|
  if (!opts.keep) {
 | 
						|
    _removeObjects.unshift(removeCallback);
 | 
						|
  }
 | 
						|
 | 
						|
  return removeCallback;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Creates a guarded function wrapping the removeFunction call.
 | 
						|
 *
 | 
						|
 * @param {Function} removeFunction
 | 
						|
 * @param {Object} arg
 | 
						|
 * @returns {Function}
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _prepareRemoveCallback(removeFunction, arg) {
 | 
						|
  var called = false;
 | 
						|
 | 
						|
  return function _cleanupCallback(next) {
 | 
						|
    if (!called) {
 | 
						|
        var index = _removeObjects.indexOf(_cleanupCallback);
 | 
						|
        if (index >= 0) {
 | 
						|
          _removeObjects.splice(index, 1);
 | 
						|
        }
 | 
						|
 | 
						|
        called = true;
 | 
						|
        removeFunction(arg);
 | 
						|
    }
 | 
						|
    if (next) next(null);
 | 
						|
  };
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * The garbage collector.
 | 
						|
 *
 | 
						|
 * @api private
 | 
						|
 */
 | 
						|
function _garbageCollector() {
 | 
						|
  if (_uncaughtException && !_gracefulCleanup) {
 | 
						|
    return;
 | 
						|
  }
 | 
						|
 | 
						|
  // the function being called removes itself from _removeObjects,
 | 
						|
  // loop until _removeObjects is empty
 | 
						|
  while (_removeObjects.length) {
 | 
						|
    try {
 | 
						|
      _removeObjects[0].call(null);
 | 
						|
    } catch (e) {
 | 
						|
      // already removed?
 | 
						|
    }
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
function _setGracefulCleanup() {
 | 
						|
  _gracefulCleanup = true;
 | 
						|
}
 | 
						|
 | 
						|
var version = process.versions.node.split('.').map(function (value) {
 | 
						|
  return parseInt(value, 10);
 | 
						|
});
 | 
						|
 | 
						|
if (version[0] === 0 && (version[1] < 9 || version[1] === 9 && version[2] < 5)) {
 | 
						|
  process.addListener('uncaughtException', function _uncaughtExceptionThrown(err) {
 | 
						|
    _uncaughtException = true;
 | 
						|
    _garbageCollector();
 | 
						|
 | 
						|
    throw err;
 | 
						|
  });
 | 
						|
}
 | 
						|
 | 
						|
process.addListener('exit', function _exit(code) {
 | 
						|
  if (code) _uncaughtException = true;
 | 
						|
  _garbageCollector();
 | 
						|
});
 | 
						|
 | 
						|
// exporting all the needed methods
 | 
						|
module.exports.tmpdir = _TMP;
 | 
						|
module.exports.dir = _createTmpDir;
 | 
						|
module.exports.dirSync = _createTmpDirSync;
 | 
						|
module.exports.file = _createTmpFile;
 | 
						|
module.exports.fileSync = _createTmpFileSync;
 | 
						|
module.exports.tmpName = _getTmpName;
 | 
						|
module.exports.tmpNameSync = _getTmpNameSync;
 | 
						|
module.exports.setGracefulCleanup = _setGracefulCleanup;
 |