feat: migrate to unjs/citty (#284)

package.json

@@ -35,6 +35,7 @@
     "@rollup/plugin-replace": "^5.0.2",
     "@rollup/pluginutils": "^5.0.2",
     "chalk": "^5.3.0",
+    "citty": "^0.1.2",
     "consola": "^3.2.3",
     "defu": "^6.1.2",
     "esbuild": "^0.18.14",
@@ -44,7 +45,6 @@
     "magic-string": "^0.30.1",
     "mkdist": "^1.3.0",
     "mlly": "^1.4.0",
-    "mri": "^1.2.0",
     "pathe": "^1.1.1",
     "pkg-types": "^1.0.3",
     "pretty-bytes": "^6.1.1",
@@ -54,7 +54,6 @@
     "untyped": "^1.3.2"
   },
   "devDependencies": {
-    "@types/mri": "^1.1.1",
     "@types/node": "^20.4.2",
     "@vitest/coverage-v8": "^0.33.0",
     "changelogen": "^0.5.4",

src/cli.ts

@@ -1,18 +1,33 @@
 #!/usr/bin/env node
+import { defineCommand, runMain } from "citty";
 import { resolve } from "pathe";
-import mri from "mri";
+import { name, version, description } from "../package.json";
 import { build } from "./build";
 
-async function main() {
-  const args = mri(process.argv.splice(2));
-  const rootDir = resolve(process.cwd(), args._[0] || ".");
-  await build(rootDir, args.stub).catch((error) => {
-    console.error(`Error building ${rootDir}: ${error}`);
-    throw error;
-  });
-}
-
-main().catch((error) => {
-  console.error(error);
-  process.exit(1);
+const main = defineCommand({
+  meta: {
+    name,
+    version,
+    description,
+  },
+  args: {
+    dir: {
+      type: "positional",
+      description: "The directory to build",
+      required: false,
+    },
+    stub: {
+      type: "boolean",
+      description: "Stub build",
+    },
+  },
+  async run({ args }) {
+    const rootDir = resolve(process.cwd(), args.dir || ".");
+    await build(rootDir, args.stub).catch((error) => {
+      console.error(`Error building ${rootDir}: ${error}`);
+      throw error;
+    });
+  },
 });
+
+runMain(main);