快捷導(dǎo)航

Node.js折騰記一：讀指定文件夾，輸出該文件夾的文件樹(shù)詳解

更新時(shí)間：2019年04月20日 14:13:44 作者：在足各上

這篇文章主要介紹了Node.js讀指定文件夾輸出該文件夾文件樹(shù)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

前言

用來(lái)干什么：想干嘛干嘛
為什么寫(xiě)：寫(xiě)來(lái)玩，學(xué)習(xí)node.js文件系統(tǒng)相關(guān)api；樹(shù)結(jié)構(gòu)這種東西還是挺不錯(cuò)的，會(huì)用會(huì)造才是真的會(huì)
用了什么： fs.readdir(dir), fs.stat(dir).isFile(), path處理路徑等

思路：

讀取當(dāng)前文件夾（不是文件夾的另作處理），獲得其下所有文件和目錄組成的數(shù)組；
循環(huán)該數(shù)組，判斷是文件夾還是文件，文件的話(huà)直接push到childFiles（對(duì)象有兩個(gè)屬性：short文件名，full完整文件路徑）
文件夾的話(huà)，先把當(dāng)前文件夾作為key，存到父級(jí)文件夾的childDir屬性下，然后自調(diào)用傳當(dāng)前文件夾路徑
每一層文件夾都包含三個(gè)屬性：dir文件夾路徑，childFiles子文件，childDir子文件夾，存儲(chǔ)為對(duì)象結(jié)構(gòu)
以上步驟重復(fù)，直到達(dá)到最底層空文件夾或該文件夾只有文件

輸出的樣子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
          }
        }
      }
    }
  }
}

。

目錄結(jié)構(gòu)(僅components)

...
|-- components
    -- index.js
    -- file.js
    -- components-dir-tree.json // 生成的文件樹(shù)對(duì)象的輸出文件，方便查看
    -- no
    -- test
       -- aa
        -- cc

使用

將輸出結(jié)果格式化寫(xiě)入到j(luò)son文件，看起來(lái)一目了然

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`;
  // 寫(xiě)入文件
  fs.writeFile(outdir, JSON.stringify(treeObj, '', '\t'), 'utf8', (err) => {
   if (err) throw err;
   console.log(`目錄樹(shù)已輸出為文件保存: ${outdir}`.greenBG);
  });
 }
 return init;
})();

module.exports = ComponentInit;

主函數(shù) getDirTree：

/components/file.js
const fs = require('fs');

/**
 * 獲取目錄下的文件樹(shù)
 * @param {讀取的路徑} dir
 * @returns 返回 dir 目錄下的文件樹(shù)
 */
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);
   // 在當(dāng)前文件夾的對(duì)象下 childDir 屬性(1)，以文件夾名作為key(2)，
   // (2)的值是該目錄下 路徑dir、childFiles子文件、childDir子文件夾組成的對(duì)象或null
   obj.childDir[dirname] = getDirTree(tempdir);
  }
 });
 return JSON.stringify(obj) === objStr ? null : obj;
}

工具函數(shù) 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
}

完結(jié)，撒花，撒花！

以上所述是小編給大家介紹的Node.js讀指定文件夾輸出該文件夾文件樹(shù)詳解整合，希望對(duì)大家有所幫助，如果大家有任何疑問(wèn)請(qǐng)給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)腳本之家網(wǎng)站的支持！

您可能感興趣的文章: