File size: 3,887 Bytes
bc20498 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 |
var fs = require('fs'),
randomstring = require('./randomstring');
/**
* create unique name file.
*
* @param {String} template template string for filename.
* @param {Function} callback callback function.
* @throws {TypeError} cannot use Promise and callback function is not found.
* @return {Promise} Promise.
*/
function createFile(template, callback) {
var filename = randomstring.generate(template);
// callback is not a Function
if (typeof callback !== 'function') {
// can use Promise
if (typeof Promise === 'function') {
// return Promise
return new Promise(function(resolve, reject) {
createFile(template, function(err, filename) {
(err) ? reject(err) : resolve(filename);
});
});
} else {
// throw TypeError when cannot use Promise
throw new TypeError('callback function required');
}
}
fs.open(filename, 'ax+', 384 /*=0600*/, function(err, fd) {
if (err) {
if (err.code === 'EEXIST') {
// FIXME: infinite loop
setImmediate(function(tmpl, cb) {
createFile(tmpl, cb);
}, template, callback);
return;
}
// filename set to null if throws error
filename = null;
}
if (fd) {
fs.close(fd, function(err) {
callback(err, filename);
});
} else {
callback(err, filename);
}
});
}
/**
* sync version createFile.
*
* @param {String} template template string for filename.
* @throws {Error} error of fs.openSync or fs.closeSync.
* @return {String} created filename.
*/
function createFileSync(template) {
var isExist, filename, fd;
// FIXME: infinite loop
do {
isExist = false;
filename = randomstring.generate(template);
try {
fd = fs.openSync(filename, 'ax+', 384 /*=0600*/);
} catch (e) {
if (e.code === 'EEXIST') {
isExist = true;
} else {
throw e;
}
} finally {
fd && fs.closeSync(fd);
}
} while (isExist);
return filename;
}
/**
* create unique name dir.
*
* @param {String} template template string for dirname.
* @param {Function} callback callback function.
* @throws {TypeError} cannot use Promise and callback function is not found.
* @return {Promise} Promise.
*/
function createDir(template, callback) {
var dirname = randomstring.generate(template);
// callback is not a Function
if (typeof callback !== 'function') {
// can use Promise
if (typeof Promise === 'function') {
// return Promise
return new Promise(function(resolve, reject) {
createDir(template, function(err, dirname) {
(err) ? reject(err) : resolve(dirname);
});
});
} else {
// throw TypeError when cannot use Promise
throw new TypeError('callback function required');
}
}
fs.mkdir(dirname, 448 /*=0700*/, function(err) {
if (err) {
if (err.code === 'EEXIST') {
// FIXME: infinite loop
setImmediate(function(tmpl, cb) {
createDir(tmpl, cb);
}, template, callback);
return;
}
// dirname set to null if throws error
dirname = null;
}
callback(err, dirname);
});
}
/**
* sync version createDir.
*
* @param {String} template template string for dirname.
* @return {String} created dirname.
*/
function createDirSync(template) {
var isExist, dirname;
// FIXME: infinite loop
do {
isExist = false;
dirname = randomstring.generate(template);
try {
fs.mkdirSync(dirname, 448 /*=0700*/);
} catch (e) {
if (e.code === 'EEXIST') {
isExist = true;
} else {
throw e;
}
}
} while (isExist);
return dirname;
}
/**
* export.
*/
module.exports = {
createFile: createFile,
createFileSync: createFileSync,
createDir: createDir,
createDirSync: createDirSync
};
|