用node列出特定文件夹内的文件

1 篇文章 0 订阅
// description: list files in specific folder   
var fs = require('fs');
var readdir = promisify(fs.readdir);
var stat = promisify(fs.stat);
var readFile = promisify(fs.readFile);
var filesObj = {}, inPathArr = ["src", "src/utils", "src/i18n"];

function promisify(fn) {
	return function () {
		var args = arguments;
		return new Promise(function (resolve, reject) {
			[].push.call(args, function (err, result) {
				if (err) {
					reject(err);
				} else {
					resolve(result);
				}
			});
			fn.apply(null, args);
		});
	}
}

function addFileToObj(obj, pArr, name) {
	const key = pArr.shift();
	if (pArr.length) {
		if (!obj[key] || obj[key].constructor != Object) {
			obj[key] = {};
			addFileToObj(obj[key], pArr, name);
		} else {
			obj[key] = addFileToObj(obj[key], pArr, name);
		};
	} else {
		obj[key] = name;
	};
	return obj;
};

function getFilePath(file, callback) {
	return readdir(file).then(function (files) {
		files = files.map(function (item) {
			var fPath = file + '/' + item;
			return stat(fPath).then(function (stats) {
				if (stats.isDirectory()) {
					callback('folder', item, fPath);
					return getFilePath(fPath, callback);
				} else {
					if (!/^(\.github|\.vscode|node_modules|dist|fileList\.json)$/.test(item)) {
						return readFile(fPath, 'utf8').then(function (content) {
							var des, firstLine;
							if (content && content.split(/\n|\r\n/).length > 0) {
								firstLine = content.split(/\n|\r\n/)[0];
								if (/description:/.test(firstLine)) {
									des = firstLine.replace(/^.*description:\s?/, "").replace(/\s*$/, "");
								};
							};
							callback("file", item, fPath, des);
						});
					};
				};
			});
		});
		return Promise.all(files);
	});
}

function listFiles(arr) {
	arr.forEach(path => {
		getFilePath(path, function (type, item, fPath, des) {
			var tmpArr = fPath.split("/");
			addFileToObj(filesObj, tmpArr, des ? des : item);
		}).then(function () {
			var str = JSON.stringify(filesObj);
			fs.writeFile(`${path}/fileList.json`, str, function (err) {
				if (err) {
					return console.log(err);
				};
				console.log("success");
			});
		}).catch(function (err) {
			console.log("error",err);
		});
	});
}

listFiles(inPathArr.slice(0, 1));

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值