Added support for enumSuffix (#9304)

esfomeado · saihaj · beerose · web-flow · commit e1dc75f3c598 · 2023-05-23T15:25:26.000+02:00
Co-authored-by: Saihajpreet Singh &lt;saihajpreet.singh@gmail.com&gt;
Co-authored-by: Aleksandra &lt;alexsandra.sikora@gmail.com&gt;
diff --git a/.changeset/swift-hornets-camp.md b/.changeset/swift-hornets-camp.md
@@ -0,0 +1,8 @@
+---
+"@graphql-codegen/typescript-operations": minor
+"@graphql-codegen/typescript-resolvers": minor
+"@graphql-codegen/typescript": minor
+"@graphql-codegen/visitor-plugin-common": minor
+---
+
+Added support for disabling suffixes on Enums.
diff --git a/packages/plugins/other/visitor-plugin-common/src/base-documents-visitor.ts b/packages/plugins/other/visitor-plugin-common/src/base-documents-visitor.ts
@@ -168,6 +168,7 @@ export class BaseDocumentsVisitor<
     super(rawConfig, {
       exportFragmentSpreadSubTypes: getConfigValue(rawConfig.exportFragmentSpreadSubTypes, false),
       enumPrefix: getConfigValue(rawConfig.enumPrefix, true),
+      enumSuffix: getConfigValue(rawConfig.enumSuffix, true),
       preResolveTypes: getConfigValue(rawConfig.preResolveTypes, true),
       dedupeOperationSuffix: getConfigValue(rawConfig.dedupeOperationSuffix, false),
       omitOperationSuffix: getConfigValue(rawConfig.omitOperationSuffix, false),
diff --git a/packages/plugins/other/visitor-plugin-common/src/base-resolvers-visitor.ts b/packages/plugins/other/visitor-plugin-common/src/base-resolvers-visitor.ts
@@ -66,6 +66,7 @@ export interface ParsedResolversConfig extends ParsedConfig {
   resolverTypeWrapperSignature: string;
   federation: boolean;
   enumPrefix: boolean;
+  enumSuffix: boolean;
   optionalResolveType: boolean;
   immutableTypes: boolean;
   namespacedImportName: string;
@@ -500,6 +501,33 @@ export interface RawResolversConfig extends RawConfig {
    * ```
    */
   enumPrefix?: boolean;
+
+  /**
+   * @default true
+   * @description Allow you to disable suffixing for generated enums, works in combination with `typesSuffix`.
+   *
+   * @exampleMarkdown
+   * ## Disable enum suffixes
+   *
+   * ```ts filename="codegen.ts"
+   *  import type { CodegenConfig } from '@graphql-codegen/cli';
+   *
+   *  const config: CodegenConfig = {
+   *    // ...
+   *    generates: {
+   *      'path/to/file': {
+   *        plugins: ['typescript', 'typescript-resolver'],
+   *        config: {
+   *          typesSuffix: 'I',
+   *          enumSuffix: false
+   *        },
+   *      },
+   *    },
+   *  };
+   *  export default config;
+   * ```
+   */
+  enumSuffix?: boolean;
   /**
    * @default false
    * @description Sets the `__resolveType` field as optional field.
@@ -635,6 +663,7 @@ export class BaseResolversVisitor<
       immutableTypes: getConfigValue(rawConfig.immutableTypes, false),
       optionalResolveType: getConfigValue(rawConfig.optionalResolveType, false),
       enumPrefix: getConfigValue(rawConfig.enumPrefix, true),
+      enumSuffix: getConfigValue(rawConfig.enumSuffix, true),
       federation: getConfigValue(rawConfig.federation, false),
       resolverTypeWrapperSignature: getConfigValue(rawConfig.resolverTypeWrapperSignature, 'Promise<T> | T'),
       enumValues: parseEnumValues({
@@ -816,7 +845,14 @@ export class BaseResolversVisitor<
         const generic = this.convertName(currentType);
         prev[typeName] = applyWrapper(`${type}<${generic}>['${typeName}']`);
       } else if (isEnumType(schemaType)) {
-        prev[typeName] = this.convertName(typeName, { useTypesPrefix: this.config.enumPrefix }, true);
+        prev[typeName] = this.convertName(
+          typeName,
+          {
+            useTypesPrefix: this.config.enumPrefix,
+            useTypesSuffix: this.config.enumSuffix,
+          },
+          true
+        );
       } else {
         prev[typeName] = this.convertName(typeName, {}, true);
 
diff --git a/packages/plugins/other/visitor-plugin-common/src/base-types-visitor.ts b/packages/plugins/other/visitor-plugin-common/src/base-types-visitor.ts
@@ -53,6 +53,7 @@ export interface ParsedTypesConfig extends ParsedConfig {
   onlyEnums: boolean;
   onlyOperationTypes: boolean;
   enumPrefix: boolean;
+  enumSuffix: boolean;
   fieldWrapperValue: string;
   wrapFieldDefinitions: boolean;
   entireFieldWrapperValue: string;
@@ -226,6 +227,32 @@ export interface RawTypesConfig extends RawConfig {
    * ```
    */
   enumPrefix?: boolean;
+  /**
+   * @default true
+   * @description Allow you to disable suffixing for generated enums, works in combination with `typesSuffix`.
+   *
+   * @exampleMarkdown
+   * ## Disable enum suffixes
+   *
+   * ```ts filename="codegen.ts"
+   *  import type { CodegenConfig } from '@graphql-codegen/cli';
+   *
+   *  const config: CodegenConfig = {
+   *    // ...
+   *    generates: {
+   *      'path/to/file': {
+   *        // plugins...
+   *        config: {
+   *          typesSuffix: 'I',
+   *          enumSuffix: false
+   *        },
+   *      },
+   *    },
+   *  };
+   *  export default config;
+   * ```
+   */
+  enumSuffix?: boolean;
   /**
    * @description Allow you to add wrapper for field type, use T as the generic value. Make sure to set `wrapFieldDefinitions` to `true` in order to make this flag work.
    * @default T
@@ -482,6 +509,7 @@ export class BaseTypesVisitor<
   ) {
     super(rawConfig, {
       enumPrefix: getConfigValue(rawConfig.enumPrefix, true),
+      enumSuffix: getConfigValue(rawConfig.enumSuffix, true),
       onlyEnums: getConfigValue(rawConfig.onlyEnums, false),
       onlyOperationTypes: getConfigValue(rawConfig.onlyOperationTypes, false),
       addUnderscoreToArgsType: getConfigValue(rawConfig.addUnderscoreToArgsType, false),
@@ -855,7 +883,12 @@ export class BaseTypesVisitor<
     return new DeclarationBlock(this._declarationBlockConfig)
       .export()
       .asKind('enum')
-      .withName(this.convertName(node, { useTypesPrefix: this.config.enumPrefix }))
+      .withName(
+        this.convertName(node, {
+          useTypesPrefix: this.config.enumPrefix,
+          useTypesSuffix: this.config.enumSuffix,
+        })
+      )
       .withComment(node.description as any as string)
       .withBlock(this.buildEnumValuesBlock(enumName, node.values)).string;
   }
@@ -993,7 +1026,10 @@ export class BaseTypesVisitor<
     const schemaType = this._schema.getType(node.name as any);
 
     if (schemaType && isEnumType(schemaType)) {
-      return this.convertName(node, { useTypesPrefix: this.config.enumPrefix });
+      return this.convertName(node, {
+        useTypesPrefix: this.config.enumPrefix,
+        useTypesSuffix: this.config.enumSuffix,
+      });
     }
 
     return this.convertName(node);
diff --git a/packages/plugins/other/visitor-plugin-common/src/selection-set-processor/base.ts b/packages/plugins/other/visitor-plugin-common/src/selection-set-processor/base.ts
@@ -11,6 +11,7 @@ export type SelectionSetProcessorConfig = {
   namespacedImportName: string | null;
   convertName: ConvertNameFn<any>;
   enumPrefix: boolean | null;
+  enumSuffix: boolean | null;
   scalars: NormalizedScalarsMap;
   formatNamedField(
     name: string,
diff --git a/packages/plugins/other/visitor-plugin-common/src/selection-set-processor/pre-resolve-types.ts b/packages/plugins/other/visitor-plugin-common/src/selection-set-processor/pre-resolve-types.ts
@@ -54,7 +54,10 @@ export class PreResolveTypesProcessor extends BaseSelectionSetProcessor<Selectio
       if (isEnumType(baseType)) {
         typeToUse =
           (this.config.namespacedImportName ? `${this.config.namespacedImportName}.` : '') +
-          this.config.convertName(baseType.name, { useTypesPrefix: this.config.enumPrefix });
+          this.config.convertName(baseType.name, {
+            useTypesPrefix: this.config.enumPrefix,
+            useTypesSuffix: this.config.enumSuffix,
+          });
       } else if (this.config.scalars[baseType.name]) {
         typeToUse = this.config.scalars[baseType.name].output;
       }
@@ -92,7 +95,10 @@ export class PreResolveTypesProcessor extends BaseSelectionSetProcessor<Selectio
       if (isEnumType(baseType)) {
         typeToUse =
           (this.config.namespacedImportName ? `${this.config.namespacedImportName}.` : '') +
-          this.config.convertName(baseType.name, { useTypesPrefix: this.config.enumPrefix });
+          this.config.convertName(baseType.name, {
+            useTypesPrefix: this.config.enumPrefix,
+            useTypesSuffix: this.config.enumSuffix,
+          });
       }
 
       const name = this.config.formatNamedField(aliasedField.alias, fieldObj.type, undefined, unsetTypes);
diff --git a/packages/plugins/other/visitor-plugin-common/src/variables-to-object.ts b/packages/plugins/other/visitor-plugin-common/src/variables-to-object.ts
@@ -24,6 +24,7 @@ export class OperationVariablesToObject {
     protected _namespacedImportName: string | null = null,
     protected _enumNames: string[] = [],
     protected _enumPrefix = true,
+    protected _enumSuffix = true,
     protected _enumValues: ParsedEnumValuesMap = {},
     protected _applyCoercion: Boolean = false,
     protected _directiveArgumentAndInputFieldMappings: ParsedDirectiveArgumentAndInputFieldMappings = {}
@@ -104,6 +105,7 @@ export class OperationVariablesToObject {
       } else {
         typeValue = `${prefix}${this._convertName(baseType, {
           useTypesPrefix: this._enumNames.includes(typeName) ? this._enumPrefix : true,
+          useTypesSuffix: this._enumNames.includes(typeName) ? this._enumSuffix : true,
         })}`;
       }
     }
diff --git a/packages/plugins/other/visitor-plugin-common/tests/create-resolvers-fields.spec.ts b/packages/plugins/other/visitor-plugin-common/tests/create-resolvers-fields.spec.ts
@@ -80,6 +80,31 @@ export type IResolversParentTypes = {
   Boolean: Scalars['Boolean']['output']
   String: Scalars['String']['output']
 };
+`
+    );
+  });
+
+  it('generates proper types when typesSuffix is used along with `enumSuffix: false`', () => {
+    const visitor = new BaseResolversVisitor(
+      {
+        mappers: {
+          B: './some-file#B',
+        },
+        typesSuffix: 'I',
+        enumSuffix: false,
+      },
+      {} as ParsedResolversConfig,
+      schema
+    );
+
+    expect(visitor.buildResolversParentTypes()).toEqual(
+      `/** Mapping between all available schema types and the resolvers parents */
+export type ResolversParentTypesI = {
+  Query: {}
+  A: AI
+  Boolean: Scalars['Boolean']['output']
+  String: Scalars['String']['output']
+};
 `
     );
   });
diff --git a/packages/plugins/typescript/operations/src/visitor.ts b/packages/plugins/typescript/operations/src/visitor.ts
@@ -78,6 +78,7 @@ export class TypeScriptDocumentsVisitor extends BaseDocumentsVisitor<
       namespacedImportName: this.config.namespacedImportName,
       convertName: this.convertName.bind(this),
       enumPrefix: this.config.enumPrefix,
+      enumSuffix: this.config.enumSuffix,
       scalars: this.scalars,
       formatNamedField,
       wrapTypeWithModifiers(baseType, type) {
@@ -109,6 +110,7 @@ export class TypeScriptDocumentsVisitor extends BaseDocumentsVisitor<
         this.config.namespacedImportName,
         enumsNames,
         this.config.enumPrefix,
+        this.config.enumSuffix,
         this.config.enumValues,
         this.config.arrayInputCoercion,
         undefined,
diff --git a/packages/plugins/typescript/operations/tests/ts-documents.spec.ts b/packages/plugins/typescript/operations/tests/ts-documents.spec.ts
@@ -2644,6 +2644,58 @@ describe('TypeScript Operations Plugin', () => {
       expect(o).toContain(`__typename?: 'Information_Entry', id: Information_EntryType,`);
     });
 
+    it('Should produce valid output with preResolveTypes=true and enums with no suffixes', async () => {
+      const ast = parse(/* GraphQL */ `
+        query test($e: Information_EntryType!) {
+          info {
+            ...information
+          }
+          infoArgTest(e: $e) {
+            ...information
+          }
+        }
+
+        fragment information on Information {
+          entries {
+            id
+            value
+          }
+        }
+      `);
+      const testSchema = buildSchema(/* GraphQL */ `
+        type Information {
+          entries: [Information_Entry!]!
+        }
+
+        enum Information_EntryType {
+          NAME
+          ADDRESS
+        }
+
+        type Information_Entry {
+          id: Information_EntryType!
+          value: String
+        }
+
+        type Query {
+          infoArgTest(e: Information_EntryType!): Information
+          info: Information
+        }
+      `);
+      const config = { preResolveTypes: true, typesSuffix: 'I', enumSuffix: false };
+      const { content } = await plugin(testSchema, [{ location: 'test-file.ts', document: ast }], config, {
+        outputFile: '',
+      });
+
+      const o = await validate(content, config, testSchema);
+      expect(o).toBeSimilarStringTo(` export type TestQueryVariablesI = Exact<{
+        e: Information_EntryType;
+      }>;`);
+      expect(o).toContain(`export type QueryI = {`);
+      expect(o).toContain(`export enum Information_EntryType {`);
+      expect(o).toContain(`__typename?: 'Information_Entry', id: Information_EntryType,`);
+    });
+
     it('Should build a basic selection set based on basic query', async () => {
       const ast = parse(/* GraphQL */ `
         query dummy {
diff --git a/packages/plugins/typescript/resolvers/src/visitor.ts b/packages/plugins/typescript/resolvers/src/visitor.ts
@@ -52,6 +52,7 @@ export class TypeScriptResolversVisitor extends BaseResolversVisitor<
         this.config.namespacedImportName,
         [],
         this.config.enumPrefix,
+        this.config.enumSuffix,
         this.config.enumValues
       )
     );
diff --git a/packages/plugins/typescript/typescript/src/typescript-variables-to-object.ts b/packages/plugins/typescript/typescript/src/typescript-variables-to-object.ts
@@ -18,6 +18,7 @@ export class TypeScriptOperationVariablesToObject extends OperationVariablesToOb
     _namespacedImportName: string | null = null,
     _enumNames: string[] = [],
     _enumPrefix = true,
+    _enumSuffix = true,
     _enumValues: ParsedEnumValuesMap = {},
     _applyCoercion: Boolean = false,
     _directiveArgumentAndInputFieldMappings: ParsedDirectiveArgumentAndInputFieldMappings = {},
@@ -29,6 +30,7 @@ export class TypeScriptOperationVariablesToObject extends OperationVariablesToOb
       _namespacedImportName,
       _enumNames,
       _enumPrefix,
+      _enumSuffix,
       _enumValues,
       _applyCoercion,
       _directiveArgumentAndInputFieldMappings
diff --git a/packages/plugins/typescript/typescript/src/visitor.ts b/packages/plugins/typescript/typescript/src/visitor.ts
@@ -93,6 +93,7 @@ export class TsVisitor<
         null,
         enumNames,
         pluginConfig.enumPrefix,
+        pluginConfig.enumSuffix,
         this.config.enumValues,
         false,
         this.config.directiveArgumentAndInputFieldMappings,
@@ -373,6 +374,7 @@ export class TsVisitor<
 
     const enumTypeName = this.convertName(node, {
       useTypesPrefix: this.config.enumPrefix,
+      useTypesSuffix: this.config.enumSuffix,
     });
 
     if (this.config.enumsAsTypes) {
diff --git a/packages/plugins/typescript/typescript/tests/typescript.spec.ts b/packages/plugins/typescript/typescript/tests/typescript.spec.ts
@@ -3085,6 +3085,21 @@ describe('TypeScript', () => {
       validateTs(result);
     });
 
+    it('Should allow to disable typesSuffix for enums', async () => {
+      const schema = buildSchema(`type T { f: String, e: E } enum E { A }`);
+      const result = (await plugin(
+        schema,
+        [],
+        { typesSuffix: 'I', enumSuffix: false },
+        { outputFile: '' }
+      )) as Types.ComplexPluginOutput;
+
+      expect(result.content).toContain(`export enum E {`);
+      expect(result.content).toContain(`e?: Maybe<E>;`);
+
+      validateTs(result);
+    });
+
     it('Should enable typesPrefix for enums by default', async () => {
       const schema = buildSchema(`type T { f: String, e: E } enum E { A }`);
       const result = await plugin(schema, [], { typesPrefix: 'I' }, { outputFile: '' });
diff --git a/website/public/config.schema.json b/website/public/config.schema.json
