feat: support enum

Author: MH4GF

src/valibot/index.ts

@@ -1,5 +1,6 @@
 import { DeclarationBlock, indent } from '@graphql-codegen/visitor-plugin-common';
 import type {
+  EnumTypeDefinitionNode,
   FieldDefinitionNode,
   GraphQLSchema,
   InputObjectTypeDefinitionNode,
@@ -28,7 +29,11 @@ export class ValibotSchemaVisitor extends BaseSchemaVisitor {
   }
 
   initialEmit(): string {
-    return '';
+    return (
+      `\n${[
+        ...this.enumDeclarations,
+      ].join('\n')}`
+    );
   }
 
   get InputObjectTypeDefinition() {
@@ -42,6 +47,25 @@ export class ValibotSchemaVisitor extends BaseSchemaVisitor {
     };
   }
 
+  get EnumTypeDefinition() {
+    return {
+      leave: (node: EnumTypeDefinitionNode) => {
+        const visitor = this.createVisitor('both');
+        const enumname = visitor.convertName(node.name.value);
+        this.importTypes.push(enumname);
+
+        // hoist enum declarations
+        this.enumDeclarations.push(
+          new DeclarationBlock({})
+            .export()
+            .asKind('const')
+            .withName(`${enumname}Schema`)
+            .withContent(`v.enum_(${enumname})`).string,
+        );
+      },
+    };
+  }
+
   protected buildInputFields(
     fields: readonly (FieldDefinitionNode | InputValueDefinitionNode)[],
     visitor: Visitor,
@@ -102,6 +126,8 @@ function generateNameNodeValibotSchema(config: ValidationSchemaPluginConfig, vis
         default:
           return `${converter.convertName()}Schema()`;
       }
+    case 'EnumTypeDefinition':
+      return `${converter.convertName()}Schema`;
     default:
       if (converter?.targetKind)
         console.warn('Unknown targetKind', converter?.targetKind);

tests/valibot.spec.ts

@@ -19,6 +19,7 @@ describe('valibot', () => {
     const result = await plugin(schema, [], { schema: 'valibot', scalars }, {});
     expect(result.content).toMatchInlineSnapshot(`
       "
+
       export function PrimitiveInputSchema(): v.GenericSchema<PrimitiveInput> {
         return v.object({
           a: v.string(),
@@ -48,6 +49,7 @@ describe('valibot', () => {
     const result = await plugin(schema, [], { schema: 'valibot', scalars }, {});
     expect(result.content).toMatchInlineSnapshot(`
       "
+
       export function PrimitiveInputSchema(): v.GenericSchema<PrimitiveInput> {
         return v.object({
           a: v.nullish(v.string()),
@@ -76,6 +78,7 @@ describe('valibot', () => {
     const result = await plugin(schema, [], { schema: 'valibot', scalars }, {});
     expect(result.content).toMatchInlineSnapshot(`
       "
+
       export function PrimitiveInputSchema(): v.GenericSchema<PrimitiveInput> {
         return v.object({
           a: v.nullish(v.array(v.nullable(v.string()))),
@@ -105,6 +108,7 @@ describe('valibot', () => {
     const result = await plugin(schema, [], { schema: 'valibot', scalars }, {});
     expect(result.content).toMatchInlineSnapshot(`
       "
+
       export function AInputSchema(): v.GenericSchema<AInput> {
         return v.object({
           b: v.lazy(() => BInputSchema())
@@ -125,4 +129,29 @@ describe('valibot', () => {
       "
     `);
   })
-})
+  it.todo('nested input object')
+  it('enum', async () => {
+    const schema = buildSchema(/* GraphQL */ `
+      enum PageType {
+        PUBLIC
+        BASIC_AUTH
+      }
+      input PageInput {
+        pageType: PageType!
+      }
+    `);
+    const scalars = undefined
+    const result = await plugin(schema, [], { schema: 'valibot', scalars }, {});
+    expect(result.content).toMatchInlineSnapshot(`
+      "
+      export const PageTypeSchema = v.enum_(PageType);
+
+      export function PageInputSchema(): v.GenericSchema<PageInput> {
+        return v.object({
+          pageType: PageTypeSchema
+        })
+      }
+      "
+    `);
+  })
+})