feat: support for enumsAsTypes

Author: MH4GF

src/valibot/index.ts

@@ -56,11 +56,18 @@ export class ValibotSchemaVisitor extends BaseSchemaVisitor {
 
         // hoist enum declarations
         this.enumDeclarations.push(
-          new DeclarationBlock({})
-            .export()
-            .asKind('const')
-            .withName(`${enumname}Schema`)
-            .withContent(`v.enum_(${enumname})`).string,
+          this.config.enumsAsTypes
+            ? new DeclarationBlock({})
+              .export()
+              .asKind('const')
+              .withName(`${enumname}Schema`)
+              .withContent(`v.picklist([${node.values?.map(enumOption => `'${enumOption.name.value}'`).join(', ')}])`)
+              .string
+            : new DeclarationBlock({})
+              .export()
+              .asKind('const')
+              .withName(`${enumname}Schema`)
+              .withContent(`v.enum_(${enumname})`).string,
         );
       },
     };

tests/valibot.spec.ts

@@ -279,4 +279,26 @@ describe('valibot', () => {
       "
     `);
   });
+  it('with enumsAsTypes', async () => {
+    const schema = buildSchema(/* GraphQL */ `
+      enum PageType {
+        PUBLIC
+        BASIC_AUTH
+      }
+    `);
+    const result = await plugin(
+      schema,
+      [],
+      {
+        schema: 'valibot',
+        enumsAsTypes: true,
+      },
+      {},
+    );
+    expect(result.content).toMatchInlineSnapshot(`
+      "
+      export const PageTypeSchema = v.picklist([\'PUBLIC\', \'BASIC_AUTH\']);
+      "
+    `);
+  });
 })