Refactor schema building for field arguments

- Abstract the building of arguments schema block into `buildObjectTypeDefinitionArguments` method in `BaseSchemaVisitor`
- Change visibility of `buildInputFields` method from private to protected
- Update necessary imports in `schema_visitor.ts`

Author: Code-Hex

src/myzod/index.ts

@@ -58,10 +58,7 @@ export class MyZodSchemaVisitor extends BaseSchemaVisitor {
         this.importTypes.push(name);
 
         // Building schema for field arguments.
-        const argumentBlocks = visitor.buildArgumentsSchemaBlock(node, (typeName, field) => {
-          this.importTypes.push(typeName);
-          return this.buildInputFields(field.arguments ?? [], visitor, typeName);
-        });
+        const argumentBlocks = this.buildObjectTypeDefinitionArguments(node, visitor);
         const appendArguments = argumentBlocks ? '\n' + argumentBlocks : '';
 
         // Building schema for fields.
@@ -176,7 +173,7 @@ export class MyZodSchemaVisitor extends BaseSchemaVisitor {
     };
   }
 
-  private buildInputFields(
+  protected buildInputFields(
     fields: readonly (FieldDefinitionNode | InputValueDefinitionNode)[],
     visitor: Visitor,
     name: string

src/schema_visitor.ts

@@ -1,4 +1,4 @@
-import { GraphQLSchema } from 'graphql';
+import { FieldDefinitionNode, GraphQLSchema, InputValueDefinitionNode, ObjectTypeDefinitionNode } from 'graphql';
 
 import { ValidationSchemaPluginConfig } from './config';
 import { SchemaVisitor } from './types';
@@ -32,4 +32,17 @@ export abstract class BaseSchemaVisitor implements SchemaVisitor {
   createVisitor(scalarDirection: 'input' | 'output' | 'both'): Visitor {
     return new Visitor(scalarDirection, this.schema, this.config);
   }
+
+  protected abstract buildInputFields(
+    fields: readonly (FieldDefinitionNode | InputValueDefinitionNode)[],
+    visitor: Visitor,
+    name: string
+  ): string;
+
+  protected buildObjectTypeDefinitionArguments(node: ObjectTypeDefinitionNode, visitor: Visitor) {
+    return visitor.buildArgumentsSchemaBlock(node, (typeName, field) => {
+      this.importTypes.push(typeName);
+      return this.buildInputFields(field.arguments ?? [], visitor, typeName);
+    });
+  }
 }

src/yup/index.ts

@@ -64,10 +64,7 @@ export class YupSchemaVisitor extends BaseSchemaVisitor {
         this.importTypes.push(name);
 
         // Building schema for field arguments.
-        const argumentBlocks = visitor.buildArgumentsSchemaBlock(node, (typeName, field) => {
-          this.importTypes.push(typeName);
-          return this.buildInputFields(field.arguments ?? [], visitor, typeName);
-        });
+        const argumentBlocks = this.buildObjectTypeDefinitionArguments(node, visitor);
         const appendArguments = argumentBlocks ? '\n' + argumentBlocks : '';
 
         // Building schema for fields.
@@ -201,7 +198,7 @@ export class YupSchemaVisitor extends BaseSchemaVisitor {
     };
   }
 
-  private buildInputFields(
+  protected buildInputFields(
     fields: readonly (FieldDefinitionNode | InputValueDefinitionNode)[],
     visitor: Visitor,
     name: string

src/zod/index.ts

@@ -74,10 +74,7 @@ export class ZodSchemaVisitor extends BaseSchemaVisitor {
         this.importTypes.push(name);
 
         // Building schema for field arguments.
-        const argumentBlocks = visitor.buildArgumentsSchemaBlock(node, (typeName, field) => {
-          this.importTypes.push(typeName);
-          return this.buildInputFields(field.arguments ?? [], visitor, typeName);
-        });
+        const argumentBlocks = this.buildObjectTypeDefinitionArguments(node, visitor);
         const appendArguments = argumentBlocks ? '\n' + argumentBlocks : '';
 
         // Building schema for fields.
@@ -189,7 +186,7 @@ export class ZodSchemaVisitor extends BaseSchemaVisitor {
     };
   }
 
-  private buildInputFields(
+  protected buildInputFields(
     fields: readonly (FieldDefinitionNode | InputValueDefinitionNode)[],
     visitor: Visitor,
     name: string