yargs 简易指南：撸一个CLI

const yargs = require('yargs/yargs');
const process = require('process');
const { hideBin } = require('yargs/helpers');
const path = require('path');
const { measureRun, measureRunAllModule } = require('./run'); // 功能实现
const genData2File = require('./gen'); // 功能实现
const clean = require('./clean'); // 功能实现

// 检测终端输入的中断快捷键信号【cmd+c】，强行退出进程
process.on('SIGINT', () => {
  process.kill(process.pid);
});

const argv = yargs(hideBin(process.argv))
  .strict()// 严格模式，参数错误直接抛出异常
  .command({
    // 提供子命令
    command: 'measure', // 子命令全称
    aliases: ['m'], // 子命令别名
    desc: '测量文件', // 子命令描述
    builder: function (yargs) {
      // 此处返回命令参数组合
      return yargs
        .check((argv) => {
          // .check支持手动校验接受的参数
          if (argv.all) return true;
          if (typeof argv.sourceFile !== 'string' || !argv.sourceFile) {
            throw new Error('CLI出错啦，源文件路径必须是字符串且不能为空！\n');
          }
          return true;
        })
        .options({
          // 子命令参数
          sourceFile: {
            alias: 'path', // 子命令参数别名
            describe: '测量路径', // 子命令参数描述
            string: true, // 接受类型是字符串
          },
          recursive: {
            alias: 'r',
            describe: '递归查找文件,一般用于测试整个模块使用',
            boolean: true, // 接受类型是布尔值，默认会隐形转换
          },
          docker: {
            alias: 'd',
            describe: '该参数启用的话，测量是基于docker运行【无头浏览器运行】，数据包含视频录制',
            boolean: false,
          },
          open: {
            alias: 'o',
            describe: '该参数启用的话，会尝试调用本地浏览器打开测量后的数据网页',
            boolean: false,
          },
          'allow-ignore': {
            alias: 'keep',
            describe: '该参数启用的话，忽略文件机制会生效',
            boolean: true,
            default: true,
          },
          'all-module': {
            alias: 'all',
            describe: '该参数启用的话，会依次执行所有模块，最终输出所有模块报告',
            boolean: false,
          },
          'result-dir-name': {
            describe: '该参数可以把报告数据源的归纳到[resultDirName参数]目录，一般结合all参数使用',
            string: '',
          },
          number: {
            alias: 'n',
            describe: '配置测试走几轮，精度会高一些，但是每轮测试的时长会增加',
            number: true,
            default: 1,
          },
        })
        .usage('$0 measure <path>') // 辅助指南，终端输出的可以看到
        .usage('$0 measure [--path=<path>] [--docker=<true|false>]')
        .example([
          // 辅助指南，终端输出的可以看到
          ['$0 measure  "src/pages/desk/measures.js"', '测试该测量文件'],
          ['$0 m src/pages/desk/measures.js -r', '测试该测量文件及下层文件【递归往下找】'],
          ['$0 m src/pages/project/components/sprint/plan/measure.js --allow-ignore=false', '强行测量处于忽略清单的文件'],
        ]);
    },
    handler: function (argv) {
      // 响应句柄，这里处理参数通过校验后接收到的对象，然后你自己丢到你自己实现的功能函数引用即可！
      // 支持async await , promise这类异步调用
      if (argv.all) {
        measureRunAllModule();
      } else {
        const entryPath = path.resolve(process.cwd(), argv?.sourceFile);
        const arg = {
          sourceFile: entryPath,
          r: argv?.r ?? false,
          n: argv?.n,
          d: argv?.d,
          open: argv?.o,
          docker: argv?.docker,
          allowIgnore: argv?.keep,
          resultDirName: argv?.resultDirName,
        };
        measureRun(arg);
      }
    },
  })
  .command({
    command: 'generate',
    aliases: ['g'],
    desc: '生成数据',
    builder: function (yargs) {
      return yargs
        .options({
          format: {
            describe: '生成报表的数据格式',
            array: true,
            choices: ['json', 'excel'], // 数组支持多个值
            default: 'excel', // 也能设置默认值
          },
          date: {
            describe: '指定日期的报表【检索数据源】返回，日期格式【2022-08-21 , 2022/08-21】都可以',
            string: true,
            default: null,
          },
          dateRangeType: {
            array: true,
            choices: ['d', 'M', 'w', 'y'],
            describe: '指定日期的报表检索区间，默认是传递日期的当天内\n M 是月份，w 是周，y是年，d是当天',
            default: 'd',
          },
          all: {
            alias: 'a',
            describe: '检索所有数据源并生成报表[优先级比指定日期低，全量资源开销大]',
            boolean: true,
            default: false,
          },
          list: {
            alias: 'l',
            describe: '检索所有数据源并在终端展示',
            boolean: true,
            default: false,
          },
        })
        .usage('$0 g [--date] [--format=<json|excel>]')
        .example([
          ['$0 generate', '执行数据源生成报表,默认就是格式就是excel,基于最近一周内最新的一份有效数据作为数据源'],
          ['$0 g --list', '检索所有数据源并在终端展示'],
          ['$0 g --format json  ', '执行数据源生成报表,格式为json'],
          ['$0 g --date 2022-08-22 ', '检索特定日期当天内的数据源并生成报表'],
          ['$0 g --date 2022-08-22 --dateRangeType m', '检索特定日期当月内的数据源并生成报表'],
          ['$0 g --all ', '检索所有数据源并生成报表'],
        ]);
    },
    handler: (argv) => {
      genData2File({
        date: argv?.date,
        all: argv?.all,
        format: argv?.format,
        dateRangeType: argv?.dateRangeType,
        list: argv?.l,
      });
    },
  })
  .command({
    command: 'clean',
    aliases: ['c'],
    desc: '删除数据源文件',
    builder: function (yargs) {
      return yargs
        .options({
          src: {
            alias: 'path',
            describe: '删除的资源路径',
            string: true,
          },
        })
        .demandOption(['src', 'path'], '请传入要删除的路径！相对于项目根路径，比如: sitespeed-result/docker-test/2022-08-22') // 强制某个参数为必填，不如check灵活
        .usage('$0 clean --path ')
        .example([
          ['$0 clean --path sitespeed-result/docker-test/2022-08-18/12-25-46_1580d24f-4d01-4155-ad22-bd3b8305f73d', '清除指定的数据源'],
        ]);
    },
    handler: async (argv) => {
      clean(argv?.path);
    },
  })
  .fail(function (msg, err, yargs) {
    // 此处捕获yargs执行异常或者抛出的异常【throw new Error】
    console.error(yargs.help());
    console.error('\n\n\n=====命令执行错误，信息如下=====\n\n', msg);
    process.exit(1);
  })
  .showHelpOnFail(false, '命令指定 --help 查看有效的选项') // 当命令执行错误的时候，自动调用一下帮助命令并输出到终端
  .version(false) // cli版本设置，此处关闭
  .wrap(null) // 设置为null，就是自适应，固宽的话，
  .locale('zh_CN') // yargs提供多语言支持，配置对应地区，核心错误这些有对应的语言文本
  .help('help', '查看命令行帮助').argv;

module.exports = argv;

"scripts": {
  "prepare": "husky install",
  "commit": "git-cz",
  "help": "npm run cli -- --help",
  "cli": "node ./node-scripts/cli.js",
  "m:all": "npm run cli --  m --all",
  "m:dev": "npm run cli -- m --open ",
  "m:docker": "npm run cli -- m --open -d",
  "g": "npm run cli -- g ",
  "clean": "npm run cli -- c"
},