feat: support cli

package.json

@@ -13,6 +13,7 @@
   "license": "MIT",
   "publisher": "JayFong",
   "main": "./out/extension.js",
+  "bin": "./out/cli.js",
   "scripts": {
     "compile": "rimraf out && tsc -p tsconfig.build.json",
     "postinstall": "node ./node_modules/vscode/bin/install",
@@ -75,7 +76,8 @@
   "dependencies": {
     "change-case": "^3.1.0",
     "globby": "^10.0.0",
-    "vtils": "^2.7.1"
+    "vtils": "^2.7.1",
+    "yargs": "^15.3.1"
   },
   "devDependencies": {
     "@types/jest": "^25.2.1",

src/IndexGenerator.ts

@@ -86,7 +86,7 @@ export class IndexGenerator {
     }
   }
 
-  private static findMarkers(
+  public static findMarkers(
     text: string,
     onInvalid?: (msg: string) => any,
   ): Marker[] {
@@ -140,11 +140,11 @@ export class IndexGenerator {
     return markers
   }
 
-  private static normalizePath(path: string): string {
+  public static normalizePath(path: string): string {
     return path.replace(/[/\\]+/g, '/')
   }
 
-  private static getRelativePath(from: string, to: string): string {
+  public static getRelativePath(from: string, to: string): string {
     const relativePath = IndexGenerator.normalizePath(relative(from, to))
     return relativePath.startsWith('../') ? relativePath : `./${relativePath}`
   }

src/cli.ts

@@ -0,0 +1,35 @@
+#!/usr/bin/env node
+import globby from 'globby'
+import yargs from 'yargs'
+import { generateIndex } from './generateIndex'
+import { IndexGenerator } from './IndexGenerator'
+
+const argv = yargs
+  .usage('Usage: $0 <pattern> [<pattern> ...] [options]')
+  .help('help', 'Show help')
+  .alias('help', 'h')
+  .option('cwd', {
+    type: 'string',
+    describe: 'Current working directory',
+    default: process.cwd(),
+  })
+  // Show help if no args
+  // ref: https://github.com/yargs/yargs/issues/895
+  .demandCommand(1, '').argv
+
+const filePaths = !argv._.length
+  ? []
+  : globby.sync(argv._, {
+      absolute: true,
+      cwd: argv.cwd,
+      dot: true,
+      onlyFiles: true,
+      unique: true,
+      gitignore: false,
+    })
+
+for (const filePath of filePaths) {
+  generateIndex(filePath, true).then(() => {
+    console.log(`✔️ ${IndexGenerator.getRelativePath(argv.cwd, filePath)}`)
+  })
+}

src/generateIndex.ts

@@ -4,6 +4,7 @@ import { IndexGenerator } from './IndexGenerator'
 export async function generateIndex(
   filePath: string,
   replaceFile = false,
+  onWarning?: (msg: string) => any,
 ): Promise<string> {
   if (!existsSync(filePath)) {
     throw new Error('File not found.')
@@ -13,7 +14,7 @@ export async function generateIndex(
     filePath: filePath,
     fileContent: fileContent,
     onWarning: msg => {
-      console.warn(msg)
+      onWarning?.(msg)
     },
     onGenerate: ({ code, marker }) => {
       fileContent =