Use enum converter for Enum

Author: eddeee888

packages/plugins/typescript/operations/package.json

@@ -15,6 +15,7 @@
   "dependencies": {
     "@graphql-codegen/plugin-helpers": "^6.0.0",
     "@graphql-codegen/typescript": "^5.0.4",
+    "@graphql-codegen/schema-ast": "^5.0.0",
     "@graphql-codegen/visitor-plugin-common": "6.1.2",
     "auto-bind": "~4.0.0",
     "tslib": "~2.6.0"

packages/plugins/typescript/operations/src/config.ts

@@ -1,4 +1,4 @@
-import { AvoidOptionalsConfig, RawDocumentsConfig } from '@graphql-codegen/visitor-plugin-common';
+import { AvoidOptionalsConfig, type EnumValuesMap, RawDocumentsConfig } from '@graphql-codegen/visitor-plugin-common';
 
 /**
  * @description This plugin generates TypeScript types based on your GraphQLSchema _and_ your GraphQL operations and fragments.
@@ -336,4 +336,8 @@ export interface TypeScriptDocumentsPluginConfig extends RawDocumentsConfig {
   nullability?: {
     errorHandlingClient: boolean;
   };
+
+  enumType?: 'string-literal' | 'native-numeric' | 'const' | 'native-const' | 'native';
+  enumValues?: EnumValuesMap;
+  futureProofEnums?: boolean;
 }

packages/plugins/typescript/operations/src/index.ts

@@ -1,4 +1,5 @@
 import { oldVisit, PluginFunction, Types } from '@graphql-codegen/plugin-helpers';
+import { transformSchemaAST } from '@graphql-codegen/schema-ast';
 import { optimizeOperations } from '@graphql-codegen/visitor-plugin-common';
 import { concatAST, GraphQLSchema } from 'graphql';
 import { TypeScriptDocumentsPluginConfig } from './config.js';
@@ -22,11 +23,9 @@ export const plugin: PluginFunction<TypeScriptDocumentsPluginConfig, Types.Compl
 
   const visitor = new TypeScriptDocumentsVisitor(schema, config, allAst);
 
-  const visitorResult = oldVisit(allAst, {
-    leave: visitor,
-  });
+  const operationsResult = oldVisit(allAst, { leave: visitor });
 
-  let content = visitorResult.definitions.join('\n');
+  let operationsContent = operationsResult.definitions.join('\n');
 
   if (config.addOperationExport) {
     const exportConsts = [];
@@ -37,23 +36,37 @@ export const plugin: PluginFunction<TypeScriptDocumentsPluginConfig, Types.Compl
       }
     }
 
-    content = visitorResult.definitions.concat(exportConsts).join('\n');
+    operationsContent = operationsResult.definitions.concat(exportConsts).join('\n');
   }
 
   if (config.globalNamespace) {
-    content = `
+    operationsContent = `
     declare global {
-      ${content}
+      ${operationsContent}
     }`;
   }
 
+  const schemaTypes = oldVisit(transformSchemaAST(schema, config).ast, { leave: visitor });
+
+  // IMPORTANT: when a visitor leaves a node with no transformation logic,
+  // It will leave the node as an object.
+  // Here, we filter in nodes that have been turned into strings, i.e. they have been transformed
+  // This way, we do not have to explicitly declare a method for every node type to convert them to null
+  const schemaTypesContent = schemaTypes.definitions.filter(def => typeof def === 'string').join('\n');
+
+  const content: string[] = [];
+  if (schemaTypesContent) {
+    content.push(schemaTypesContent);
+  }
+  content.push(operationsContent);
+
   return {
     prepend: [
       ...visitor.getImports(),
       ...visitor.getGlobalDeclarations(visitor.config.noExport),
       'type Exact<T extends { [key: string]: unknown }> = { [K in keyof T]: T[K] };',
     ],
-    content,
+    content: content.join('\n'),
   };
 };
 

packages/plugins/typescript/operations/src/visitor.ts

@@ -1,12 +1,15 @@
 import {
   BaseDocumentsVisitor,
+  convertSchemaEnumToDeclarationBlockString,
   DeclarationKind,
   generateFragmentImportStatement,
   getConfigValue,
   LoadedFragment,
   normalizeAvoidOptionals,
   NormalizedAvoidOptionalsConfig,
   ParsedDocumentsConfig,
+  type ParsedEnumValuesMap,
+  parseEnumValues,
   PreResolveTypesProcessor,
   SelectionSetProcessorConfig,
   SelectionSetToObject,
@@ -15,6 +18,7 @@ import {
 import autoBind from 'auto-bind';
 import {
   type DocumentNode,
+  EnumTypeDefinitionNode,
   type FragmentDefinitionNode,
   GraphQLEnumType,
   GraphQLInputObjectType,
@@ -39,6 +43,9 @@ export interface TypeScriptDocumentsParsedConfig extends ParsedDocumentsConfig {
   noExport: boolean;
   maybeValue: string;
   allowUndefinedQueryVariables: boolean;
+  enumType: 'string-literal' | 'native-numeric' | 'const' | 'native-const' | 'native';
+  futureProofEnums: boolean;
+  enumValues: ParsedEnumValuesMap;
 }
 
 type UsedNamedInputTypes = Record<string, GraphQLNamedInputType>;
@@ -60,6 +67,13 @@ export class TypeScriptDocumentsVisitor extends BaseDocumentsVisitor<
         preResolveTypes: getConfigValue(config.preResolveTypes, true),
         mergeFragmentTypes: getConfigValue(config.mergeFragmentTypes, false),
         allowUndefinedQueryVariables: getConfigValue(config.allowUndefinedQueryVariables, false),
+        enumType: getConfigValue(config.enumType, 'string-literal'),
+        enumValues: parseEnumValues({
+          schema,
+          mapOrStr: config.enumValues,
+          ignoreEnumValuesFromSchema: config.ignoreEnumValuesFromSchema,
+        }),
+        futureProofEnums: getConfigValue(config.futureProofEnums, false),
       } as TypeScriptDocumentsParsedConfig,
       schema
     );
@@ -156,6 +170,31 @@ export class TypeScriptDocumentsVisitor extends BaseDocumentsVisitor<
     };
   }
 
+  EnumTypeDefinition(node: EnumTypeDefinitionNode): string {
+    const enumName = node.name.value;
+    if (!this._usedNamedInputTypes[enumName]) {
+      return null;
+    }
+
+    return convertSchemaEnumToDeclarationBlockString({
+      schema: this._schema,
+      node,
+      declarationBlockConfig: this._declarationBlockConfig,
+      enumName,
+      enumValues: this.config.enumValues,
+      futureProofEnums: this.config.futureProofEnums,
+      ignoreEnumValuesFromSchema: this.config.ignoreEnumValuesFromSchema,
+      outputType: this.config.enumType,
+      naming: {
+        convert: this.config.convert,
+        typesPrefix: this.config.typesPrefix,
+        typesSuffix: this.config.typesSuffix,
+        useTypesPrefix: this.config.enumPrefix,
+        useTypesSuffix: this.config.enumSuffix,
+      },
+    });
+  }
+
   public getImports(): Array<string> {
     return !this.config.globalNamespace &&
       (this.config.inlineFragmentTypes === 'combine' || this.config.inlineFragmentTypes === 'mask')

packages/plugins/typescript/operations/tests/ts-documents.standalone.spec.ts

@@ -37,6 +37,7 @@ describe('TypeScript Operations Plugin - Standalone', () => {
       input UsersInput {
         from: DateTime
         to: DateTime
+        role: UserRole
       }
 
       type UsersResponseOk {
@@ -69,8 +70,8 @@ describe('TypeScript Operations Plugin - Standalone', () => {
         }
       }
 
-      query UsersWithScalarInput($from: DateTime!, $to: DateTime) {
-        users(input: { from: $from, to: $to }) {
+      query UsersWithScalarInput($from: DateTime!, $to: DateTime, $role: UserRole) {
+        users(input: { from: $from, to: $to, role: $role }) {
           ... on UsersResponseOk {
             result {
               __typename
@@ -87,6 +88,10 @@ describe('TypeScript Operations Plugin - Standalone', () => {
 
     expect(result).toMatchInlineSnapshot(`
       "type Exact<T extends { [key: string]: unknown }> = { [K in keyof T]: T[K] };
+      export type UserRole =
+        | 'ADMIN'
+        | 'CUSTOMER';
+
       export type UserQueryVariables = Exact<{
         id: string;
       }>;
@@ -107,6 +112,7 @@ describe('TypeScript Operations Plugin - Standalone', () => {
       export type UsersWithScalarInputQueryVariables = Exact<{
         from: any;
         to?: any | null;
+        role?: UserRole | null;
       }>;
 
 
@@ -157,3 +163,13 @@ describe('TypeScript Operations Plugin - Standalone', () => {
     `);
   });
 });
+
+describe('TypeScript Operations Plugin - Enum', () => {
+  it.todo('does not generate unused enum in variables and result');
+  it.todo('handles native numeric enum correctly');
+  it.todo('handles const enum correctly');
+  it.todo('handles native const enum correctly');
+  it.todo('handles native enum correctly');
+  it.todo('handles EnumValues correctly');
+  // Bring over tests from https://github.com/dotansimha/graphql-code-generator/blob/accdab69106605241933e9d66d64dc7077656f30/packages/plugins/typescript/typescript/tests/typescript.spec.ts
+});