腳本之家服務器常用軟件

快捷導航

Node.js 中如何收集和解析命令行參數(shù)

更新時間：2021年01月08日 10:54:05 作者：descire

這篇文章主要介紹了Node.js 中如何收集和解析命令行參數(shù)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

前言

在開發(fā) CLI（Command Line Interface）工具的業(yè)務場景下，離不開命令行參數(shù)的收集和解析。

接下來，本文介紹如何收集和解析命令行參數(shù)。

收集命令行參數(shù)

在 Node.js 中，可以通過 process.argv 屬性收集進程被啟動時傳入的命令行參數(shù)：

 // ./example/demo.js
 process.argv.slice(2);

 // 命令行執(zhí)行如下命令
 node ./example/demo.js --name=xiaoming --age=20 man

 // 得到的結(jié)果
 [ '--name=xiaoming', '--age=20', 'man' ]

由上述示例可以發(fā)現(xiàn)，Node.js 在處理命令行參數(shù)時，只是簡單地通過空格來分割字符串。

對于這樣的參數(shù)數(shù)組，無法很方便地獲取到每個參數(shù)對應的值，所以需要再進行一次解析操作。

命令行參數(shù)風格

在解析命令行參數(shù)之前，需要了解一些常見的命令行參數(shù)風格：

Unix 風格：參數(shù)以「-」（連字符）開頭
GNU 風格：參數(shù)以「--」（雙連字符）開頭
BSD 風格：參數(shù)以空格分割

Unix 參數(shù)風格有一個特殊的注意事項：「「-」后面緊鄰的每一個字母都表示一個參數(shù)名」。

ls -al

上述命令用來顯示當前目錄下所有的文件、文件夾并且顯示它們的詳細信息，等同于：

ls -a -l

GNU 風格的參數(shù)以「--」開頭，一般后面會跟上一個單詞或者短語，例如熟悉的 npm 安裝依賴的命令：

 npm install --save koa

對于兩個單詞的情況，在 GNU 參數(shù)風格中，會通過「-」來連接，例如 npm 安裝僅用于開發(fā)環(huán)境的依賴：

npm install --save-dev webpack

BSD 是加州大學伯克利分校開發(fā)的一個 Unix 版本。其與 Unix 的區(qū)別主要在于參數(shù)前面沒有「-」，個人感覺這樣很難區(qū)別參數(shù)和參數(shù)值。

注意事項：-- 后面緊鄰空格時，表示后面的字符串不需要解析。

解析命令行參數(shù)

function parse(args = []) {
 // _ 屬性用來保留不需要處理的參數(shù)字符串
 const output = { _: [] };

 for (let index = 0; index < args.length; index++) {
  const arg = args[index];
  
  if (isIgnoreFollowingParameters(output, args, index, arg)) {
   break;
  }
  
  if (!isParameter(arg)) {
   output._.push(arg);
   continue;
  }

  ...
 }

 return output;
}

parse(process.argv.slice(2));

接收到命令行參數(shù)數(shù)組之后，需要遍歷數(shù)組，處理每一個參數(shù)字符串。

isIgnoreFollowingParameters 方法主要用來判斷單個「--」的場景，后續(xù)的參數(shù)字符串不再需要處理：

function isIgnoreFollowingParameters(output, args, index, arg) {
 if (arg !== '--') {
  return false;
 }
 output._ = output._.concat(args.slice(++index));
 return true;
}

接下來，如果參數(shù)字符串不以「-」開頭，同樣也不需要處理，參數(shù)的形式以 Unix 和 GNU 風格為主：

function isParameter(arg) {
 return arg.startsWith('-');
}

參數(shù)的表現(xiàn)形式主要分為以下幾種：

"--name=xiaoming": 參數(shù)名為 name，參數(shù)值為 xiaoming
"-abc=10": 參數(shù)名為 a，參數(shù)值為 true；參數(shù)名為 b，參數(shù)值為 true；參數(shù)名為 c，參數(shù)值為 10
"--save-dev": 參數(shù)名為 save-dev，參數(shù)值為 true
"--age 20"：參數(shù)名為 age，參數(shù)值為 20

 let hyphensIndex;
 for (hyphensIndex = 0; hyphensIndex < arg.length; hyphensIndex++) {
  if (arg.charCodeAt(hyphensIndex) !== 45) {
   break;
  }
 }

 let assignmentIndex;
 for (assignmentIndex = hyphensIndex + 1; assignmentIndex < arg.length; assignmentIndex++) {
  if (arg[assignmentIndex].charCodeAt(0) === 61) {
   break;
  }
 }

利用 Unicode 碼點值找出連字符和等號的下標值，從而根據(jù)下標分割出參數(shù)名和參數(shù)值：

 const name = arg.substring(hyphensIndex, assignmentIndex);

 let value;
 const assignmentValue = arg.substring(++assignmentIndex);

處理參數(shù)值時，需要考慮參數(shù)賦值的四種場景：

 if (assignmentValue) {
  value = assignmentValue; // --name=xiaoming or -abc=10
 } else if (index + 1 === args.length) {
  value = true; // --save-dev
 } else if (('' + args[index + 1]).charCodeAt(0) !== 45) {
  value = args[++index]; // --age 20
 } else {
  value = true; // 缺省情況
 }

由于 Unix 風格中每一個字母都代表一個參數(shù)，并且「手動傳遞的參數(shù)值應該賦值給最后一個參數(shù)」，所以還需針對該場景進行適配：

 // 「-」or「--」
 const arr = hyphensIndex === 2 ? [name] : name;
 for (let keyIndex = 0; keyIndex < arr.length; keyIndex++) {
  const _key = arr[keyIndex];
  const _value = keyIndex + 1 < arr.length || value;
  handleKeyValue(output, _key, _value);
 }

最后針對參數(shù)的賦值操作，需要考慮到「多次賦值」的情況：

function handleKeyValue(output, key, value) {
 const oldValue = output[key];
 if (Array.isArray(oldValue)) {
  output[key] = oldValue.concat(value);
  return;
 }

 if (oldValue) {
  output[key] = [oldValue, value];
  return;
 }

 output[key] = value;
}

到此，命令行參數(shù)的解析功能就完成了，上述方法執(zhí)行的效果如下：

 # 命令行執(zhí)行
 node ./example/step1.js --name=xiaoming --age 20 --save-dev -abc=10 -c=20 -- --ignore

 # 解析結(jié)果
 {
  _: [ '--ignore' ],
  name: 'xiaoming',
  age: '20',
  'save-dev': true,
  a: true,
  b: true,
  c: [ '10', '20' ]
 }

別名機制

比較優(yōu)秀的 CLI 工具在參數(shù)的解析上都支持參數(shù)的別名設置，例如使用 npm 安裝開發(fā)環(huán)境依賴時，你可以選擇這種完整的寫法：

npm install --save-dev webpack

你也可以使用下面這種別名方式：

npm install -D webpack

從使用上來說 -D 和 --save-dev 是兩種方式，但是從 CLI 工具的開發(fā)者來說，最終處理邏輯時只能以一個參數(shù)名為標準，所以對于一個命令行參數(shù)解析庫來說，其結(jié)果需要包含所有的情況：

npm install --save-dev webpack
# 解析的結(jié)果
{ 'save-dev': true, 'D': true }

以上文的解析方法為例，需要添加額外的選項參數(shù)，加入 alias 屬性來聲明別名屬性的對應關系：

 parse(process.argv.slice(2), {
  alias: {
   'save-dev': 'S'
  }
 })

上述方式符合正常的理解：設置參數(shù)對應的別名。但這是一個「單向查找關系」，需要轉(zhuǎn)化為：

 "alias": {
  "save-dev": ["s"],
  "s": ["save-dev"]
 }

因為對于使用者來說，只會選擇一種方式傳遞參數(shù)。對于開發(fā)者的話需要根據(jù)任意一個別名找到其相關聯(lián)的別名：

function parse(args = [], options = {}) {
 const output = { _: [] };

 const { alias } = options;

 const hasAlias = alias !== void 666;

 if (hasAlias) {
  Object.keys(alias).forEach(key => {
   alias[key] = toArr(alias[key]);
   alias[key].forEach((item, index) => {
    (alias[item] = alias[key].concat(key)).splice(index, 1);
   })
  })
 }

 // 省略解析代碼
 ...

 if (hasAlias) {
  Object.keys(output).forEach(key => {
   const arr = alias[key] || [];
   arr.forEach(sub => output[sub] = output[key])
  })
 }

 return output;
}

除了別名之外，還可以在參數(shù)解析之后做如下優(yōu)化：

參數(shù)值的類型約束
參數(shù)的默認值設定

成熟的解析庫

針對一些成熟的命令行參數(shù)解析庫可以采用基準測試查看它們的解析效率：

const nopt = require('nopt');
const mri = require('mri');
const yargs = require('yargs-parser');
const minimist = require('minimist');
const { Suite } = require('benchmark');

const bench = new Suite();
const args = ['--name=xiaoming', '-abc', '10', '--save-dev', '--age', '20'];

bench
 .add('minimist   ', () => minimist(args))
 .add('mri     ', () => mri(args))
 .add('nopt     ', () => nopt(args))
 .add('yargs-parser ', () => yargs(args))
 .on('cycle', e => console.log(String(e.target)))
 .run();