Implement nargs

ryangoree · Apr 22, 2024 · 47bf81f · 47bf81f
1 parent 9742d9d
commit 47bf81f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 10 deletions.
diff --git a/packages/clide-js/src/core/options/types.ts b/packages/clide-js/src/core/options/types.ts
@@ -1,4 +1,3 @@
-import { PromptOptions } from 'src/core/client';
 import { MaybeReadonly } from 'src/utils/types';
 
 /**
@@ -21,7 +20,10 @@ export interface OptionConfig<
   alias?: MaybeReadonly<TAlias[]>;
   /** The type of the option. */
   type: T;
-  /** Whether the option is a string (optional, inferred from `type`). */
+  /**
+   * Whether the option is a string (optional, inferred from `type`). Useful
+   * for array options to specify the type of the array values.
+   */
   string?: boolean;
   /** The number of arguments the option accepts (optional). */
   nargs?: number;
@@ -69,9 +71,9 @@ export type OptionPrimitiveType<T extends OptionType = OptionType> = T extends
   | 'secret'
   ? string
   : T extends 'number'
-  ? number
-  : T extends 'boolean'
-  ? boolean
-  : T extends 'array'
-  ? string[]
-  : never;
+    ? number
+    : T extends 'boolean'
+      ? boolean
+      : T extends 'array'
+        ? string[]
+        : never;
diff --git a/packages/clide-js/src/core/parse.ts b/packages/clide-js/src/core/parse.ts
@@ -54,13 +54,18 @@ export function parseCommand(
     boolean: [],
     number: [],
     string: [],
+    narg: {},
   };
 
   for (const [key, option] of Object.entries(optionsConfig)) {
     if (option.alias) {
       parseOptions.alias![key] = option.alias as string[];
     }
 
+    if (option.nargs) {
+      parseOptions.narg![key] = option.nargs;
+    }
+
     switch (option.type) {
       case 'array':
         parseOptions.array!.push(key as any);
@@ -81,8 +86,6 @@ export function parseCommand(
     }
   }
 
-  // TODO:  const nargs: Record<string, number> = {};
-
   // Parse the command string with yargs-parser
   let { _: tokens, ...options } = parse(commandString, {
     ...parseOptions,