fixed some code

Author: Code-Hex

src/graphql.ts

@@ -3,14 +3,7 @@ import {
   NonNullTypeNode,
   NamedTypeNode,
   TypeNode,
-  GraphQLSchema,
-  InputObjectTypeDefinitionNode,
-  EnumTypeDefinitionNode,
-  ScalarTypeDefinitionNode,
-  visit,
 } from "graphql";
-import { transformSchemaAST } from "@graphql-codegen/schema-ast";
-import { Nodes, ValidationSchemaPluginConfig } from "./types";
 
 export const isListType = (typ: TypeNode): typ is ListTypeNode =>
   typ.kind === "ListType";
@@ -19,37 +12,4 @@ export const isNonNullType = (typ: TypeNode): typ is NonNullTypeNode =>
 export const isNamedType = (typ: TypeNode): typ is NamedTypeNode =>
   typ.kind === "NamedType";
 
-export const isRef = (kind: string) => kind.includes("Input");
-export const isBoolean = (kind: string) => kind === "Boolean";
-export const isString = (kind: string) => ["ID", "String"].includes(kind);
-export const isNumber = (kind: string) => ["Int", "Float"].includes(kind);
-
-
-export const retrieveSchema = (
-  schema: GraphQLSchema,
-  config: ValidationSchemaPluginConfig
-): Nodes => {
-  const { ast } = transformSchemaAST(schema, config);
-
-  const inputObjects: InputObjectTypeDefinitionNode[] = [];
-  const enums: Record<string, EnumTypeDefinitionNode> = {};
-  const scalars: Record<string, ScalarTypeDefinitionNode> = {};
-
-  visit(ast, {
-    leave: {
-      InputObjectTypeDefinition: (node) => {
-        inputObjects.unshift(node);
-      },
-      EnumTypeDefinition: (node) => {
-        if (node.values) {
-          enums[node.name.value] = node;
-        }
-      },
-      ScalarTypeDefinition: (node) => {
-        scalars[node.name.value] = node;
-      },
-    },
-  });
-
-  return { inputObjects, enums, scalars };
-};
+export const isInput = (kind: string) => kind.includes("Input");

src/index.ts

@@ -1,46 +1,26 @@
-import { ImportYup, YupGenerator } from "./yup/index";
-import { ValidationSchema, ValidationSchemaPluginConfig } from "./types";
+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 } from "graphql";
-import { retrieveSchema } from "./graphql";
+import { GraphQLSchema, visit } from "graphql";
 
 export const plugin: PluginFunction<ValidationSchemaPluginConfig> = async (
   schema: GraphQLSchema,
   _documents: Types.DocumentFile[],
   config: ValidationSchemaPluginConfig
 ): Promise<Types.PluginOutput> => {
-  const { inputObjects, enums, scalars } = retrieveSchema(schema, config);
+  const { schema: _schema, ast } = transformSchemaAST(schema, config);
+  const { buildImports, ...visitor } = YupSchemaVisitor(_schema, config);
 
-  const prepend = [importSchema(config.schema)];
-  if (config.importFrom) {
-    prepend.push(
-      buildImportStatement(
-        [
-          // Should put on Scalar here?
-          ...Object.values(enums).map((enumdef) => enumdef.name.value),
-          ...inputObjects.map((inputObject) => inputObject.name.value),
-        ],
-        config.importFrom
-      )
-    );
-  }
-  return {
-    prepend,
-    content:
-      "\n" +
-      [new YupGenerator({ inputObjects, enums, scalars }).generate()].join(
-        "\n"
-      ),
-  };
-};
+  const result = visit(ast, {
+    leave: visitor,
+  });
 
-const buildImportStatement = (types: string[], importFrom: string): string =>
-  `import { ${types.join(", ")} } from "${importFrom}";`;
+  // @ts-ignore
+  const generated = result.definitions.filter((def) => typeof def === "string");
 
-const importSchema = (schema?: ValidationSchema): string => {
-  if (schema === "yup") {
-    return ImportYup();
-  }
-  // TODO(codehex): support zod
-  return ImportYup();
+  return {
+    prepend: buildImports(),
+    content: "\n" + [...generated].join("\n"),
+  };
 };

src/types.ts

@@ -1,8 +1,11 @@
-import { TypeScriptPluginConfig } from "@graphql-codegen/typescript";
+import { TsVisitor, TypeScriptPluginConfig } from "@graphql-codegen/typescript";
 import {
+  ASTKindToNode,
+  ASTNode,
   EnumTypeDefinitionNode,
   InputObjectTypeDefinitionNode,
   ScalarTypeDefinitionNode,
+  VisitFn,
 } from "graphql";
 
 export type ValidationSchema = "yup";
@@ -24,7 +27,6 @@ export interface ValidationSchemaPluginConfig extends TypeScriptPluginConfig {
    * ```
    */
   schema?: ValidationSchema;
-
   /**
    * @description import types from generated typescript type path
    * if not given, omit import statement.
@@ -44,10 +46,51 @@ export interface ValidationSchemaPluginConfig extends TypeScriptPluginConfig {
    * ```
    */
   importFrom?: string;
+  /**
+   * @description Generates validation schema for generated `const enum`, you can read more about const enums here: https://www.typescriptlang.org/docs/handbook/enums.html.
+   * @default false
+   *
+   * @exampleMarkdown
+   * ```yml
+   * generates:
+   *   path/to/file.ts:
+   *     plugins:
+   *       - typescript
+   *       - graphql-codegen-validation-schema
+   *     config:
+   *       constEnums: true
+   * ```
+   */
+  constEnums?: boolean;
+  /**
+   * @description Generates validation schema for enum as TypeScript `type`
+   * @default false
+   *
+   * @exampleMarkdown
+   * ```yml
+   * generates:
+   *   path/to/file.ts:
+   *     plugins:
+   *       - graphql-codegen-validation-schema
+   *     config:
+   *       enumsAsTypes: true
+   * ```
+   *
+   * ```yml
+   * generates:
+   *   path/to/file.ts:
+   *     plugins:
+   *       - typescript
+   *       - graphql-codegen-validation-schema
+   *     config:
+   *       enumsAsTypes: true
+   * ```
+   */
+  enumsAsTypes?: boolean;
 }
 
-export interface Nodes {
-  inputObjects: InputObjectTypeDefinitionNode[];
-  enums: Record<string, EnumTypeDefinitionNode>;
-  scalars: Record<string, ScalarTypeDefinitionNode>;
-}
+export type ValidationSchemaVisitor = {
+  [K in keyof ASTKindToNode]?: VisitFn<ASTNode, ASTKindToNode[K]>;
+} & {
+  buildImports: () => string[]
+};