首页 > 编程 > JavaScript > 正文

Node.js折腾记一:读指定文件夹,输出该文件夹的文件树详解

2019-11-19 11:44:34
字体:
来源:转载
供稿:网友

前言

用来干什么:想干嘛干嘛
为什么写:写来玩,学习node.js文件系统相关api;树结构这种东西还是挺不错的,会用会造才是真的会
用了什么: fs.readdir(dir), fs.stat(dir).isFile(), path处理路径等

思路:

  1. 读取当前文件夹(不是文件夹的另作处理),获得其下所有文件和目录组成的数组;
  2. 循环该数组,判断是文件夹还是文件,文件的话直接push到childFiles(对象有两个属性:short文件名,full完整文件路径)
  3. 文件夹的话,先把当前文件夹作为key,存到父级文件夹的childDir属性下,然后自调用传当前文件夹路径
  4. 每一层文件夹都包含三个属性:dir文件夹路径,childFiles子文件,childDir子文件夹,存储为对象结构
  5. 以上步骤重复,直到达到最底层空文件夹或该文件夹只有文件

输出的样子components-dir-tree.json

{  "dir": "D://node-test//components",  "childFiles": [    {      "short": "components-dir-tree.json",      "full": "D://node-test//components//components-dir-tree.json"    },    {      "short": "file.js",      "full": "D://node-test//components//file.js"    },    {      "short": "index.js",      "full": "D://node-test//components//index.js"    }  ],  "childDir": {    "no": null,    "test": {      "dir": "D://node-test//components//test",      "childFiles": [],      "childDir": {        "aa": {          "dir": "D://node-test//components//test//aa",          "childFiles": [            {              "short": "bb.js",              "full": "D://node-test//components//test//aa//bb.js"            }          ],          "childDir": {            "cc": null          }        }      }    }  }}。

目录结构(仅components)

...
|-- components
    -- index.js
    -- file.js
    -- components-dir-tree.json  // 生成的文件树对象的输出文件,方便查看
    -- no
    -- test
       -- aa
        -- cc

使用

将输出结果格式化写入到json文件,看起来一目了然

components/index.js:/** * init */require('console-color-mr'); // 命令行样式const fs = require('fs');const path = require('path');const { getDirTree, getDirName } = require('./file.js');const componentDir = path.resolve(__dirname, './');// console.log('componentDir: ', componentDir);const ComponentInit = (function init() { console.log('______ init ______'.blueBG, '/n'); let treeObj = getDirTree(componentDir); // console.log('treeObj: ',treeObj); if (treeObj) {  let outdir = `${__dirname}//${getDirName(componentDir)}-dir-tree.json`;  // 写入文件  fs.writeFile(outdir, JSON.stringify(treeObj, '', '/t'), 'utf8', (err) => {   if (err) throw err;   console.log(`目录树已输出为文件保存: ${outdir}`.greenBG);  }); } return init;})();module.exports = ComponentInit;

主函数 getDirTree:

/components/file.jsconst fs = require('fs');/** * 获取目录下的文件树 * @param {读取的路径} dir * @returns 返回 dir 目录下的文件树 */function getDirTree(dir) { let obj = {  dir: dir, // 文件夹路径  childFiles: [], // 子文件  childDir: {} // 子目录 }; let objStr = JSON.stringify(obj); if (isFile(dir)) return console.log(`${dir}: 不是文件夹`.redBG);  // 读取目录 let files = readDir(dir); if (!files.length) console.log(`${dir}: 文件夹为空`.redBG);  // 遍历文件 files.forEach(file => {  let tempdir = `${dir}//${file}`;  if (isFile(tempdir)) {   obj.childFiles.push({    short: file, // 文件名    full: tempdir // 完整路径   });     } else {   // console.log('tempdir: ',tempdir);   let dirname = getDirName(tempdir);   // 在当前文件夹的对象下 childDir 属性(1),以文件夹名作为key(2),   // (2)的值是该目录下 路径dir、childFiles子文件、childDir子文件夹组成的对象或null   obj.childDir[dirname] = getDirTree(tempdir);  } }); return JSON.stringify(obj) === objStr ? null : obj;}

工具函数 readDir/isFile

// 读取路径下的文件、文件夹function readDir(dir) { return fs.readdirSync(dir, (err, files) => {  if (err) throw err;  // console.log(`${dir}, files: `.green, files);  // if (!files.length) console.log(`${dir}: 文件夹为空`.redBG);  return files; })}// 判断制定路径是否是文件function isFile(dir) { return fs.statSync(dir).isFile();}// 获取目录名function getDirName(dir) { let tempdir = dir.substr(dir.lastIndexOf('//')+1, dir.length); return tempdir;}// const components_out = readFile(path.resolve(__dirname, './components-dir-tree.json'));// console.log('components-dir-tree: ', components_out);// 读取指定目录的文件function readFile(dir) { let result = fs.readFileSync(dir, 'utf-8'); return (  result   ? {   dir: dir,   result: result  }   : null );}module.exports = { getDirTree, readDir, isFile, readFile}

完结,撒花,撒花!

以上所述是小编给大家介绍的Node.js读指定文件夹输出该文件夹文件树详解整合,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对武林网网站的支持!

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表