feat: add --file, --help, --version

Switch to yargs

Author: azz

README.md

@@ -26,6 +26,22 @@ npm install --global ast-grep
 
 ## Usage
 
+```shellsession
+$ ast-grep --help
+Options:
+  --anonymous, -a  Ignore all names in the AST                         [boolean]
+  --file, -f       Load pattern from a file                             [string]
+  --help           Show help                                           [boolean]
+  --version        Show version number                                 [boolean]
+
+Examples:
+  ast-grep.js -a 'fn()' file.js        Find all no-arg function calls in
+                                       'file.js'.
+  ast-grep.js -f pattern.js '**/*.js'  Match the pattern in 'pattern.js' across
+                                       all JS files.
+  echo 'foo' | ast-grep.js 'pattern'   Match 'pattern' on standard input.
+```
+
 On standard in:
 
 ```shellsession
@@ -41,24 +57,6 @@ On a set of files:
 $ ast-grep 'yield* foo();' '**/*.js'
 ```
 
-## Flags
-
-### `--anonymous`
-
-**Alias**: `-a`
-
-Ignore names, this includes identifiers, types, etc.
-
-## Examples
-
-### Find all no-arg function calls:
-
-```shellsession
-$ echo -e 'foo();\nbar();' | ast-grep 'fn()' -a
-foo();
-bar();
-```
-
 ## FAQ
 
 ### Q. But @azz, `grep` stands for Global Regular Expression Print, this tool doesn't use Regular Expressions!

package.json

@@ -19,8 +19,8 @@
     "get-stream": "^3.0.0",
     "globby": "^7.1.1",
     "mem": "^3.0.0",
-    "mri": "^1.1.0",
-    "omit-deep-lodash": "^1.0.0"
+    "omit-deep-lodash": "^1.0.0",
+    "yargs": "^10.1.1"
   },
   "scripts": {
     "prepublishOnly": "yarn build",

src/bin/ast-grep.js

@@ -2,27 +2,46 @@
 'use strict';
 
 import fs from 'fs';
-import mri from 'mri';
+import yargs from 'yargs';
 import globby from 'globby';
 import getStream from 'get-stream';
 
 import astGrep from '..';
 
-const { _: [grepPattern, ...filePatterns], ...args } = mri(
-  process.argv.slice(2),
-  {
-    alias: {
-      a: 'anonymous',
-    },
-    boolean: ['anonymous'],
-  }
-);
+yargs
+  .option('anonymous', {
+    alias: 'a',
+    boolean: true,
+    describe: 'Ignore all names in the AST',
+  })
+  .option('file', {
+    alias: 'f',
+    string: true,
+    describe: 'Load pattern from a file',
+  })
+  .option('debug', { string: true, hidden: true })
+  .help()
+  .version()
+  .example(
+    `$0 -a 'fn()' file.js`,
+    `Find all no-arg function calls in 'file.js'.`
+  )
+  .example(
+    `$0 -f pattern.js '**/*.js'`,
+    `Match the pattern in 'pattern.js' across all JS files.`
+  )
+  .example(`echo 'foo' | $0 'pattern'`, `Match 'pattern' on standard input.`);
 
-async function run() {
-  if (!grepPattern) {
-    console.error('usage: ast-grep pattern [glob ...]');
-    return 1;
+async function run({ _: [grepPattern, ...filePatterns], ...args }) {
+  if (!grepPattern && !args.file) {
+    yargs.showHelp();
+    return 2;
+  }
+  if (args.file && grepPattern) {
+    filePatterns.unshift(grepPattern);
+    grepPattern = fs.readFileSync(args.file, 'utf8');
   }
+
   if (filePatterns.length) {
     const files = globby.sync(filePatterns);
     if (!files.length) {
@@ -62,10 +81,10 @@ function displayMatches(matches, file) {
   }
 }
 
-run()
+run(yargs.argv)
   .then(process.exit)
   .catch(error => {
-    if (args.debug) {
+    if (yargs.argv.debug) {
       console.error(error);
     } else {
       console.error(`ast-grep: ${error.toString()}`);