@controlflow-ai/daemon 0.1.1 → 0.1.3

package/src/args.ts

@@ -4,10 +4,45 @@ export interface ParsedArgs {
   flags: Record<string, string | boolean>;
 }
 
+const booleanFlags = new Set([
+  'apply',
+  'dry-run',
+  'force',
+  'help',
+  'hold-if-stale',
+  'interactive',
+  'json',
+  'infer-mentions',
+  'literal-mentions',
+  'no-reload',
+  'no-mention-lint',
+  'once',
+  'rebind-lark',
+  'send-anyway-if-stale',
+  'show-hidden',
+  'stdin',
+  'unbind-lark',
+  'verbose',
+  'yes',
+]);
+
 export function parseArgs(argv = process.argv.slice(2)): ParsedArgs {
-  const [command = 'help', ...rest] = argv;
+  const [first = 'help', ...tail] = argv;
+  const command = first.startsWith('--') ? 'help' : first;
+  const rest = first.startsWith('--') ? argv : tail;
   const values: string[] = [];
-  const flags: Record<string, string | boolean> = {};
+  const flags: Record<string, string | boolean | string[]> = {};
+
+  const setFlag = (name: string, value: string | boolean): void => {
+    const existing = flags[name];
+    if (existing === undefined) {
+      flags[name] = value;
+    } else if (Array.isArray(existing)) {
+      existing.push(String(value));
+    } else {
+      flags[name] = [String(existing), String(value)];
+    }
+  };
 
   for (let index = 0; index < rest.length; index += 1) {
     const arg = rest[index];
@@ -19,28 +54,40 @@ export function parseArgs(argv = process.argv.slice(2)): ParsedArgs {
     const raw = arg.slice(2);
     const eq = raw.indexOf('=');
     if (eq >= 0) {
-      flags[raw.slice(0, eq)] = raw.slice(eq + 1);
+      setFlag(raw.slice(0, eq), raw.slice(eq + 1));
+      continue;
+    }
+    if (booleanFlags.has(raw)) {
+      setFlag(raw, true);
       continue;
     }
 
     const next = rest[index + 1];
     if (next && !next.startsWith('--')) {
-      flags[raw] = next;
+      setFlag(raw, next);
       index += 1;
     } else {
-      flags[raw] = true;
+      setFlag(raw, true);
     }
   }
 
-  return { command, values, flags };
+  return { command, values, flags: flags as Record<string, string | boolean> };
 }
 
 export function flag(flags: Record<string, string | boolean>, name: string, fallback?: string): string | undefined {
-  const value = flags[name];
+  const value = flags[name] as string | boolean | string[] | undefined;
+  if (Array.isArray(value)) return value[value.length - 1] ?? fallback;
   if (typeof value === 'string') return value;
   return fallback;
 }
 
+export function flagValues(flags: Record<string, string | boolean>, name: string): string[] {
+  const value = flags[name] as string | boolean | string[] | undefined;
+  if (Array.isArray(value)) return value.filter((entry) => entry !== 'true');
+  if (typeof value === 'string') return [value];
+  return [];
+}
+
 export function boolFlag(flags: Record<string, string | boolean>, name: string): boolean {
   return flags[name] === true || flags[name] === 'true';
 }