flatten zod errors (#789)

* flatten zod errors * fix error
AcalaNetwork · Jul 10, 2024 · 524a935 · 524a935
1 parent 74a8042
commit 524a935
Showing 1 changed file with 17 additions and 8 deletions.
diff --git a/packages/chopsticks/src/cli.ts b/packages/chopsticks/src/cli.ts
@@ -1,5 +1,6 @@
 import { config as dotenvConfig } from 'dotenv'
 import { hideBin } from 'yargs/helpers'
+import { z } from 'zod'
 import _ from 'lodash'
 import yargs from 'yargs'
 import type { MiddlewareFunction } from 'yargs'
@@ -12,14 +13,22 @@ import { setupWithServer } from './index.js'
 dotenvConfig()
 
 const processArgv: MiddlewareFunction<{ config?: string; port?: number; unsafeRpcMethods?: string }> = async (argv) => {
-  if (argv.unsafeRpcMethods) {
-    await loadRpcMethodsByScripts(argv.unsafeRpcMethods)
-  }
-  if (argv.config) {
-    Object.assign(argv, _.defaults(argv, await fetchConfig(argv.config)))
-  }
-  if (environment.PORT) {
-    argv.port = Number(environment.PORT)
+  try {
+    if (argv.unsafeRpcMethods) {
+      await loadRpcMethodsByScripts(argv.unsafeRpcMethods)
+    }
+    if (argv.config) {
+      Object.assign(argv, _.defaults(argv, await fetchConfig(argv.config)))
+    }
+    if (environment.PORT) {
+      argv.port = Number(environment.PORT)
+    }
+  } catch (error) {
+    if (error instanceof z.ZodError) {
+      throw new Error('Bad argv', { cause: error.flatten().fieldErrors })
+    } else {
+      throw error
+    }
   }
 }