mirror of
https://github.com/coderaiser/cloudcmd.git
synced 2026-01-23 18:55:26 +00:00
92 lines
3.3 KiB
JavaScript
92 lines
3.3 KiB
JavaScript
/* inspired by http://procbits.com/2011/10/29/a-node-js-experiment-thinking-asynchronously-recursion-calculate-file-size-directory */
|
|
(function(){
|
|
'use strict';
|
|
|
|
if(!global.cloudcmd)
|
|
return console.log(
|
|
'# dir.js' + '\n' +
|
|
'# -----------' + '\n' +
|
|
'# Module is part of Cloud Commander,' + '\n' +
|
|
'# used for getting dir size.' + '\n' +
|
|
'# If you wont to see at work' + '\n' +
|
|
'# try GET /api/v1/fs/etc?size or' + '\n' +
|
|
'# dir.getSize(\'/etc, function(err, size) {' + '\n' +
|
|
'# });' + '\n' +
|
|
'# http://cloudcmd.io' + '\n');
|
|
|
|
var main = global.cloudcmd.main,
|
|
fs = main.fs,
|
|
Util = main.util,
|
|
path = main.path;
|
|
|
|
exports.getSize = function(dir, callback) {
|
|
var total = 0;
|
|
|
|
function calcSize(stat) {
|
|
var size = stat && stat.size || 0;
|
|
|
|
total += size;
|
|
}
|
|
|
|
processDir(dir, calcSize, function() {
|
|
Util.exec(callback, null, total);
|
|
});
|
|
};
|
|
|
|
function processDir(dir, func, callback) {
|
|
var asyncRunning = 0,
|
|
fileCounter = 1;
|
|
|
|
function getDirInfo(dir) {
|
|
/* The lstat() function shall be equivalent to stat(),
|
|
except when path refers to a symbolic link. In that case lstat()
|
|
shall return information about the link, while stat() shall return
|
|
information about the file the link references.
|
|
*/
|
|
|
|
fs.lstat(dir, getStat.bind(null, dir));
|
|
}
|
|
|
|
function getStat(dir, error, stat) {
|
|
--fileCounter;
|
|
|
|
if (!error) {
|
|
if (stat.isFile())
|
|
Util.exec(func, stat);
|
|
else if (stat.isDirectory()) {
|
|
++asyncRunning;
|
|
|
|
fs.readdir(dir, function(error, files) {
|
|
var dirPath, file, n, i;
|
|
|
|
asyncRunning--;
|
|
|
|
if (!error) {
|
|
n = files.length;
|
|
fileCounter += n;
|
|
|
|
for (i = 0; i < n; i++) {
|
|
file = files[i];
|
|
dirPath = path.join(dir, file);
|
|
|
|
process.nextTick(getDirInfo.bind(null, dirPath));
|
|
}
|
|
}
|
|
|
|
if(!n)
|
|
execCallBack();
|
|
});
|
|
}
|
|
}
|
|
|
|
execCallBack();
|
|
}
|
|
|
|
function execCallBack() {
|
|
if (!fileCounter && !asyncRunning)
|
|
Util.exec(callback);
|
|
}
|
|
|
|
getDirInfo(dir);
|
|
}
|
|
})();
|