feat: support directives

Author: MH4GF

src/valibot/index.ts

@@ -12,6 +12,7 @@ import type {
 import type { ValidationSchemaPluginConfig } from '../config';
 import { BaseSchemaVisitor } from '../schema_visitor';
 import type { Visitor } from '../visitor';
+import { buildApiForValibot, formatDirectiveConfig } from '../directive';
 import {
   isInput,
   isListType,
@@ -114,15 +115,33 @@ function generateFieldTypeValibotSchema(config: ValidationSchemaPluginConfig, vi
     if (isListType(parentType))
       return `v.nullable(${gen})`;
 
+    const actions = actionsFromDirectives(config, field);
+
     if (isNonNullType(parentType))
-      return gen;
+      return pipeSchemaAndActions(gen, actions); ;
 
-    return `v.nullish(${gen})`;
+    return `v.nullish(${pipeSchemaAndActions(gen, actions)})`;
   }
   console.warn('unhandled type:', type);
   return '';
 }
 
+function actionsFromDirectives(config: ValidationSchemaPluginConfig, field: InputValueDefinitionNode | FieldDefinitionNode): string[] {
+  if (config.directives && field.directives) {
+    const formatted = formatDirectiveConfig(config.directives);
+    return buildApiForValibot(formatted, field.directives);
+  }
+
+  return [];
+}
+
+function pipeSchemaAndActions(schema: string, actions: string[]): string {
+  if (actions.length === 0)
+    return schema;
+
+  return `v.pipe(${schema}, ${actions.join(', ')})`;
+}
+
 function generateNameNodeValibotSchema(config: ValidationSchemaPluginConfig, visitor: Visitor, node: NameNode): string {
   const converter = visitor.getNameNodeConverter(node);
 

tests/valibot.spec.ts

@@ -338,4 +338,75 @@ describe('valibot', () => {
       "
     `)
   });
+  it.todo('with typesPrefix')
+  it.todo('with typesSuffix')
+  it.todo('with default input values')
+  describe('issues #19', () => {
+    it('string field', async () => {
+      const schema = buildSchema(/* GraphQL */ `
+        input UserCreateInput {
+          profile: String @constraint(minLength: 1, maxLength: 5000)
+        }
+        directive @constraint(minLength: Int!, maxLength: Int!) on INPUT_FIELD_DEFINITION
+      `);
+      const result = await plugin(
+        schema,
+        [],
+        {
+          schema: 'valibot',
+          directives: {
+            constraint: {
+              minLength: ['minLength', '$1', 'Please input more than $1'],
+              maxLength: ['maxLength', '$1', 'Please input less than $1'],
+            },
+          },
+        },
+        {},
+      );
+      expect(result.content).toMatchInlineSnapshot(`
+        "
+
+        export function UserCreateInputSchema(): v.GenericSchema<UserCreateInput> {
+          return v.object({
+            profile: v.nullish(v.pipe(v.string(), v.minLength(1, "Please input more than 1"), v.maxLength(5000, "Please input less than 5000")))
+          })
+        }
+        "
+      `)
+    });
+
+    it('not null field', async () => {
+      const schema = buildSchema(/* GraphQL */ `
+        input UserCreateInput {
+          profile: String! @constraint(minLength: 1, maxLength: 5000)
+        }
+        directive @constraint(minLength: Int!, maxLength: Int!) on INPUT_FIELD_DEFINITION
+      `);
+      const result = await plugin(
+        schema,
+        [],
+        {
+          schema: 'valibot',
+          directives: {
+            constraint: {
+              minLength: ['minLength', '$1', 'Please input more than $1'],
+              maxLength: ['maxLength', '$1', 'Please input less than $1'],
+            },
+          },
+        },
+        {},
+      );
+
+      expect(result.content).toMatchInlineSnapshot(`
+        "
+
+        export function UserCreateInputSchema(): v.GenericSchema<UserCreateInput> {
+          return v.object({
+            profile: v.pipe(v.string(), v.minLength(1, "Please input more than 1"), v.maxLength(5000, "Please input less than 5000"))
+          })
+        }
+        "
+      `)
+    });
+  })
 })