feat: support directives

Author: MH4GF

src/valibot/index.ts

@@ -18,6 +18,7 @@ import {
   isNamedType,
   isNonNullType,
 } from './../graphql';
+import { buildApi, buildApiForValibot, formatDirectiveConfig } from 'src/directive';
 
 export class ValibotSchemaVisitor extends BaseSchemaVisitor {
   constructor(schema: GraphQLSchema, config: ValidationSchemaPluginConfig) {
@@ -118,28 +119,39 @@ function generateFieldTypeValibotSchema(config: ValidationSchemaPluginConfig, vi
     if (isListType(parentType))
       return `v.nullable(${gen})`;
 
+    let appliedDirectivesGen = applyDirectives(config, field, gen);
+
     if (field.kind === Kind.INPUT_VALUE_DEFINITION) {
       const { defaultValue } = field;
       if (defaultValue?.kind === Kind.INT || defaultValue?.kind === Kind.FLOAT || defaultValue?.kind === Kind.BOOLEAN)
-        gen = `v.optional(${gen}, ${defaultValue.value})`;
+        appliedDirectivesGen = `v.optional(${appliedDirectivesGen}, ${defaultValue.value})`;
 
       if (defaultValue?.kind === Kind.STRING || defaultValue?.kind === Kind.ENUM)
-        gen = `v.optional(${gen}, "${defaultValue.value}")`;
+        appliedDirectivesGen = `v.optional(${appliedDirectivesGen}, "${defaultValue.value}")`;
 
     }
     if (isNonNullType(parentType)) {
       if (visitor.shouldEmitAsNotAllowEmptyString(type.name.value))
         return "v.string([v.minLength(1)])"; // TODO
 
-      return gen;
+      return appliedDirectivesGen;
     }
 
-    return `v.nullish(${gen})`;
+    return `v.nullish(${appliedDirectivesGen})`;
   }
   console.warn('unhandled type:', type);
   return '';
 }
 
+function applyDirectives(config: ValidationSchemaPluginConfig, field: InputValueDefinitionNode | FieldDefinitionNode, gen: string): string {
+  if (config.directives && field.directives) {
+    const formatted = formatDirectiveConfig(config.directives);
+    // TODO: handle other scalar
+    return `v.string([${buildApiForValibot(formatted, field.directives).join('')}])`;
+  }
+  return gen;
+}
+
 function generateNameNodeValibotSchema(config: ValidationSchemaPluginConfig, visitor: Visitor, node: NameNode): string {
   const converter = visitor.getNameNodeConverter(node);
 

tests/valibot.spec.ts

@@ -366,4 +366,77 @@ describe('valibot', () => {
       "
     `)
   });
+
+  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() {
+          return v.object({
+            profile: v.nullish(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() {
+          return v.object({
+            profile: v.string([v.minLength(1, "Please input more than 1"), v.maxLength(5000, "Please input less than 5000")])
+          })
+        }
+        "
+      `)
+    });
+
+    it.todo('list field');
+  })
 })