如何让含Promise的Node.js脚本同时支持代码调用与终端命令执行？

我来帮你搞定这个需求！要让你的Node.js脚本同时支持模块导入链式调用和终端直接执行，核心思路就是区分脚本的运行环境，分情况处理逻辑，同时保留Promise的链式能力。下面是具体实现方案：

核心思路拆解 #

• 先写一个返回Promise的核心业务函数，这是两种使用场景的共用逻辑
• 通过require.main === module判断脚本是被直接执行还是被require导入
• 直接执行时：解析命令行参数，调用核心函数并处理Promise结果（成功输出、失败报错）
• 被导入时：导出核心函数，让调用者可以直接链式调用

完整代码示例（foobar.js） #

const fs = require('fs').promises; // 用fs的Promise版本更简洁

// 核心业务函数：接收配置，返回Promise
function foobar(options) {
  // 先做参数校验
  if (!options?.file) {
    return Promise.reject(new Error('必须传入file参数'));
  }

  // 这里写你的实际业务逻辑，比如读取并处理文件
  return fs.readFile(options.file, 'utf8')
    .then(data => {
      // 模拟处理逻辑
      return `处理完成：文件内容长度为 ${data.length} 字符`;
    });
}

// 判断当前脚本的运行环境
if (require.main === module) {
  // 场景1：直接通过node命令执行脚本
  // 解析命令行参数，这里推荐用minimist简化解析（需先安装：npm install minimist）
  const argv = require('minimist')(process.argv.slice(2));
  
  // 调用核心函数并处理结果
  foobar({ file: argv.file })
    .then(result => {
      console.log(result);
      process.exit(0); // 成功退出
    })
    .catch(err => {
      console.error('执行失败：', err.message);
      process.exit(1); // 错误退出
    });
} else {
  // 场景2：作为模块被其他脚本require导入
  module.exports = foobar;
}

两种场景的使用方式 #

1. 作为模块链式调用 #

在其他Node.js脚本中导入并使用：

const foobar = require('./foobar');

// 直接链式调用Promise
foobar({ file: 'index.html' })
  .then(result => {
    console.log('模块调用结果：', result);
    // 可以继续链式调用其他逻辑
    return Promise.resolve('下一步操作');
  })
  .then(nextResult => console.log(nextResult))
  .catch(err => console.error('模块调用出错：', err));

2. 终端直接执行 #

在命令行中运行脚本，传入参数：

# 直接执行，指定file参数
node ./foobar.js --file=index.html

# 如果没有传参数，会输出错误提示
node ./foobar.js

优化小贴士 #

• 命令行参数解析：用minimist比原生process.argv处理更方便，支持短参数、默认值等
• 业务逻辑：尽量用原生Promise化的API（比如fs.promises）或者用util.promisify包装回调函数，让代码更简洁
• 参数校验：在核心函数开头加入参数校验，能提前抛出错误，提升脚本的健壮性

这样写出来的脚本，既能独立运行完成任务，又能作为模块无缝嵌入到其他项目中，完美满足你的需求！

内容的提问来源于stack exchange，提问作者Flame