技术指南：读指定文件夹，轻松输出该文件夹的文件树

前言

相信大家都遇到过这样的需求：打开某个文件夹，查看其子文件和子目录，然后再逐层打开子目录，查看其子文件和子目录，如此循环往复，直到打开所有子目录，查看所有文件。那么这种需求应该怎么用代码实现呢？文件树这种数据结构会不会派上用场呢？

正文

使用fs.readdir(dir)读取目录

在Node.js中，我们可以使用fs.readdir(dir)函数来读取指定目录下的所有文件和文件夹。这个函数会返回一个包含所有文件和文件夹名称的数组。例如，以下代码将读取./data目录下的所有文件和文件夹：

const fs = require('fs');

fs.readdir('./data', (err, files) => {
  if (err) {
    console.error(err);
    return;
  }

  console.log(files);
});

运行以上代码后，将在控制台输出./data目录下的所有文件和文件夹的名称。

使用fs.stat(dir).isDirectory()判断是否为目录

为了构建文件树，我们需要知道哪些是文件，哪些是目录。我们可以使用fs.stat(dir)函数来获取指定目录或文件的详细信息。这个函数会返回一个包含文件或目录详细信息的对象。例如，以下代码将获取./data/file.txt文件的详细信息：

fs.stat('./data/file.txt', (err, stats) => {
  if (err) {
    console.error(err);
    return;
  }

  console.log(stats);
});

运行以上代码后，将在控制台输出./data/file.txt文件的详细信息，其中包括文件的类型。如果文件的类型为isDirectory()，则表示该文件是一个目录。

使用递归构建文件树

现在我们已经知道如何读取目录和判断文件类型，接下来就可以使用递归来构建文件树了。递归是一种编程技术，它允许函数调用自身。在构建文件树时，我们可以使用递归来遍历目录，并为每个目录创建一个子树。子树的根节点是目录本身，子节点是目录下的文件和子目录。

以下代码是一个递归函数，它可以用来构建文件树：

function buildTree(dir) {
  const files = fs.readdirSync(dir);
  const tree = { name: dir, children: [] };

  for (const file of files) {
    const filePath = path.join(dir, file);
    const stats = fs.statSync(filePath);

    if (stats.isDirectory()) {
      tree.children.push(buildTree(filePath));
    } else {
      tree.children.push({ name: file });
    }
  }

  return tree;
}

这个函数首先读取指定目录下的所有文件和文件夹，然后为每个文件或文件夹创建一个节点。如果文件或文件夹是目录，则递归调用该函数来构建子树。

输出文件树

构建好文件树后，我们就可以将其输出到控制台或文件了。以下代码将文件树输出到控制台：

const tree = buildTree('./data');
console.log(JSON.stringify(tree, null, 2));

运行以上代码后，将在控制台输出文件树的JSON表示。

结语

以上就是如何使用Node.js读指定文件夹，并输出该文件夹的文件树的方法。通过本文，我们不仅学习了Node.js的文件系统API，还学习了如何使用递归来构建数据结构。希望本文对大家有所帮助。