fixed some code to use oldVisit function

Author: Code-Hex

src/index.ts

@@ -1,8 +1,8 @@
 import { transformSchemaAST } from "@graphql-codegen/schema-ast";
 import { YupSchemaVisitor } from "./yup/index";
 import { ValidationSchemaPluginConfig } from "./types";
-import { PluginFunction, Types } from "@graphql-codegen/plugin-helpers";
-import { GraphQLSchema, visit } from "graphql";
+import { oldVisit, PluginFunction, Types } from "@graphql-codegen/plugin-helpers";
+import { GraphQLSchema } from "graphql";
 
 export const plugin: PluginFunction<ValidationSchemaPluginConfig> = async (
   schema: GraphQLSchema,
@@ -12,7 +12,9 @@ export const plugin: PluginFunction<ValidationSchemaPluginConfig> = async (
   const { schema: _schema, ast } = transformSchemaAST(schema, config);
   const { buildImports, ...visitor } = YupSchemaVisitor(_schema, config);
 
-  const result = visit(ast, visitor);
+  const result = oldVisit(ast, {
+    leave: visitor
+  });
 
   // @ts-ignore
   const generated = result.definitions.filter((def) => typeof def === "string");

src/types.ts

@@ -1,9 +1,4 @@
 import { TypeScriptPluginConfig } from "@graphql-codegen/typescript";
-import {
-  ASTKindToNode,
-  ASTNode,
-  ASTVisitor,
-} from "graphql";
 
 export type ValidationSchema = "yup";
 
@@ -85,7 +80,3 @@ export interface ValidationSchemaPluginConfig extends TypeScriptPluginConfig {
    */
   enumsAsTypes?: boolean;
 }
-
-export type ValidationSchemaVisitor = ASTVisitor & {
-  buildImports: () => string[]
-};

src/yup/index.ts

@@ -1,13 +1,12 @@
 import { isInput, isNonNullType, isListType, isNamedType } from "./../graphql";
-import {
-  ValidationSchemaPluginConfig,
-  ValidationSchemaVisitor,
-} from "./../types";
+import { ValidationSchemaPluginConfig } from "./../types";
 import {
   InputValueDefinitionNode,
   NameNode,
   TypeNode,
   GraphQLSchema,
+  InputObjectTypeDefinitionNode,
+  EnumTypeDefinitionNode,
 } from "graphql";
 import {
   DeclarationBlock,
@@ -20,7 +19,7 @@ const importYup = `import * as yup from 'yup'`;
 export const YupSchemaVisitor = (
   schema: GraphQLSchema,
   config: ValidationSchemaPluginConfig
-): ValidationSchemaVisitor => {
+) => {
   const tsVisitor = new TsVisitor(schema, config);
 
   const importTypes: string[] = [];
@@ -35,52 +34,48 @@ export const YupSchemaVisitor = (
       }
       return [importYup];
     },
-    InputObjectTypeDefinition: {
-      leave(node) {
-        const name = node.name.value;
-        importTypes.push(name);
-
-        const shape = node.fields
-          ?.map((field) =>
-            generateInputObjectFieldYupSchema(tsVisitor, schema, field, 2)
-          )
-          .join(",\n");
-
-        return new DeclarationBlock({})
-          .export()
-          .asKind("function")
-          .withName(`${name}Schema(): yup.SchemaOf<${name}>`)
-          .withBlock(
-            [indent(`return yup.object({`), shape, indent("})")].join("\n")
-          ).string;
-      },
+    InputObjectTypeDefinition: (node: InputObjectTypeDefinitionNode) => {
+      const name = node.name.value;
+      importTypes.push(name);
+
+      const shape = node.fields
+        ?.map((field) =>
+          generateInputObjectFieldYupSchema(tsVisitor, schema, field, 2)
+        )
+        .join(",\n");
+
+      return new DeclarationBlock({})
+        .export()
+        .asKind("function")
+        .withName(`${name}Schema(): yup.SchemaOf<${name}>`)
+        .withBlock(
+          [indent(`return yup.object({`), shape, indent("})")].join("\n")
+        ).string;
     },
-    EnumTypeDefinition: {
-      leave(node) {
-        const enumname = node.name.value;
-        importTypes.push(enumname);
-
-        if (config.enumsAsTypes) {
-          return new DeclarationBlock({})
-            .export()
-            .asKind("const")
-            .withName(`${enumname}Schema`)
-            .withContent(
-              `yup.mixed().oneOf([${node.values
-                ?.map((v) => `'${tsVisitor.convertName(v.name.value)}'`)
-                .join(", ")}])`
-            ).string;
-        }
-
-        const values = node.values
-          ?.map((v) => `${enumname}.${tsVisitor.convertName(v.name.value)}`)
-          .join(", ");
+    EnumTypeDefinition: (node: EnumTypeDefinitionNode) => {
+      const enumname = node.name.value;
+      importTypes.push(enumname);
+
+      if (config.enumsAsTypes) {
         return new DeclarationBlock({})
           .export()
           .asKind("const")
           .withName(`${enumname}Schema`)
-          .withContent(`yup.mixed().oneOf([${values}])`).string;
-      },
+          .withContent(
+            `yup.mixed().oneOf([${node.values
+              ?.map((v) => `'${tsVisitor.convertName(v.name.value)}'`)
+              .join(", ")}])`
+          ).string;
+      }
+
+      const values = node.values
+        ?.map((v) => `${enumname}.${tsVisitor.convertName(v.name.value)}`)
+        .join(", ");
+      return new DeclarationBlock({})
+        .export()
+        .asKind("const")
+        .withName(`${enumname}Schema`)
+        .withContent(`yup.mixed().oneOf([${values}])`).string;
     },
     // ScalarTypeDefinition: (node) => {
     //   const decl = new DeclarationBlock({})