fix(wb): retry command

WillBooster · Nov 9, 2023 · 1a274dd · 1a274dd
1 parent 728c414
commit 1a274dd
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/packages/wb/src/commands/retry.ts b/packages/wb/src/commands/retry.ts
@@ -18,20 +18,18 @@ const argumentsBuilder = {
   command: {
     description: 'A command to retry',
     type: 'string',
-    demand: true,
   },
   args: {
     description: 'Arguments for the command',
     type: 'array',
-    default: [],
   },
 } as const;
 
 export const retryCommand: CommandModule<
   unknown,
   InferredOptionTypes<typeof builder & typeof sharedOptionsBuilder & typeof argumentsBuilder>
 > = {
-  command: 'retry <command> [args...]',
+  command: 'retry [command] [args...]',
   describe: 'Retry the given command until it succeeds',
   builder,
   async handler(argv) {
@@ -41,7 +39,7 @@ export const retryCommand: CommandModule<
       process.exit(1);
     }
 
-    const cmdAndArgs = [argv.command, ...argv.args];
+    const cmdAndArgs = [argv.command, ...(argv.args ?? []), ...argv._.slice(1)].filter(Boolean);
     let lastStatus = 0;
     for (let i = 0; i < argv.retry; i++) {
       if (i > 0) {