chore: let one pass a z.object

Author: gagik

packages/arg-parser/src/arg-parser.spec.ts

@@ -1450,13 +1450,13 @@ describe('arg-parser', function () {
           '--deprecatedField',
           '100',
         ],
-        schema: {
+        schema: z.object({
           extendedField: z.number(),
           replacedField: z.number(),
           deprecatedField: z.number().register(argMetadata, {
             deprecationReplacement: 'replacedField',
           }),
-        },
+        }),
       });
 
       expect(options).to.deep.equal({
@@ -1486,9 +1486,9 @@ describe('arg-parser', function () {
             '--unknownField',
             '100',
           ],
-          schema: {
+          schema: z.object({
             extendedField: z.enum(['90', '100']),
-          },
+          }),
         })
       ).to.throw(UnknownCliArgumentError, 'Unknown argument: --unknownField');
     });

packages/arg-parser/src/arg-parser.ts

@@ -228,11 +228,14 @@ export function parseArgsWithCliOptions({
 }: {
   args: string[];
   /** Schema to extend the CLI options schema with. */
-  schema?: Record<string, z.ZodTypeAny>;
+  schema?: z.ZodObject;
 }): ReturnType<typeof parseArgs<CliOptions>> {
   const schema =
     schemaToExtend !== undefined
-      ? CliOptionsSchema.extend(schemaToExtend)
+      ? z.object({
+          ...CliOptionsSchema.shape,
+          ...schemaToExtend.shape,
+        })
       : CliOptionsSchema;
   const { parsed, positional, deprecated } = parseArgs<ParsedCliOptions>({
     args,