nodejs实现选择文件夹

Node.js是一种流行的后端JavaScript编程语言，被广泛应用于构建Web应用、API、命令行工具等。在Node.js中，有时候需要让用户选择文件夹，以便在程序中进行读取、写入和修改等操作。本文将介绍如何使用Node.js实现选择文件夹。

一、使用原生Node.js实现选择文件夹

Node.js提供了fs模块来操作文件系统。在fs模块中，有一个方法叫做readdir，可以读取一个目录下的所有文件和子目录。下面是一个使用原生Node.js实现选择文件夹的代码示例：

const fs = require('fs');
const path = require('path');

let dir = process.argv[2] || '.';

fs.readdir(dir, (err, files) => {
  if (err) throw err;
  console.log('请选择一个文件夹：');
  files.forEach(file => {
    let filePath = path.join(dir, file);
    fs.stat(filePath, (err, stats) => {
      if (err) throw err;
      if (stats.isDirectory()) {
        console.log(`[${file}]`);
      } else {
        console.log(file);
      }
    });
  });
  process.stdin.resume();
  process.stdin.setEncoding('utf8');
  process.stdin.on('data', data => {
    data = data.trim();
    let filePath = path.join(dir, data);
    fs.stat(filePath, (err, stats) => {
      if (err) throw err;
      if (stats.isDirectory()) {
        process.stdin.pause();
        console.log(`已选择[${data}]文件夹！`);
      } else {
        console.log(`${data}不是一个有效的文件夹！`);
      }
    });
  });
});

该代码先使用fs.readdir方法读取用户指定的文件夹，然后使用fs.stat方法判断每个文件是不是文件夹。最后使用process.stdin来获取用户的输入，如果用户输入的是文件夹，则表示用户选择了该文件夹。

二、使用第三方模块实现选择文件夹

除了使用原生Node.js，还可以使用第三方模块来实现选择文件夹。下面介绍两种常见的选择文件夹的第三方模块。

1. inquirer

inquirer是一个强大的交互式命令行用户界面库，可以像开发web表单一样在命令行上创建交互式选择。下面是一个使用inquirer选择文件夹的代码示例：

const inquirer = require('inquirer');
const fs = require('fs');
const path = require('path');

let dirs = fs
  .readdirSync('.')
  .filter(file => {
    return fs.statSync(file).isDirectory();
  })
  .map(dir => {
    return { name: `[${dir}]`, value: dir };
  });

inquirer
  .prompt([
    {
      type: 'list',
      name: 'dir',
      message: '请选择一个文件夹：',
      choices: dirs,
    },
  ])
  .then(answer => {
    console.log(`已选择[${answer.dir}]文件夹！`);
  });

该代码使用fs.readdirSync方法读取当前目录下的所有文件夹，并使用filter方法筛选出所有文件夹。然后通过map方法将每个文件夹封装成一个对象，使得可以在inquirer的选择列表中显示出文件夹名字。最后，使用inquirer.prompt方法弹出一个选择列表，让用户选择文件夹，并使用answer.dir来获取用户选择的文件夹名字。

2. file-tree-select

file-tree-select是一个构建在inquirer和node-file-tree上的文件选择器插件。它可以在命令行中展示当前目录下的文件树，用户可以通过键盘的上下箭头和空格键来选择文件夹。下面是一个使用file-tree-select选择文件夹的代码示例：

const fileTreeSelect = require('file-tree-select');
const path = require('path');

let dir = process.argv[2] || '.';
let absDir = path.resolve(dir);

fileTreeSelect(absDir, { hideFiles: true }).then(result => {
  console.log(`已选择[${result}]文件夹！`);
});

该代码先使用path.resolve方法将用户输入的文件夹名字解析为绝对路径。然后使用fileTreeSelect方法来展示当前目录下的文件树，用户可以使用键盘操作来选择文件夹。最后使用.then方法获取用户选择的文件夹路径。

三、总结

选择文件夹是Node.js中比较常见的操作。本文介绍了使用原生Node.js、inquirer和file-tree-select三种方式来实现选择文件夹。对于小型项目来说，使用原生Node.js的方式可以满足需求；对于大型项目，使用第三方模块可以提高开发效率和代码质量。