Collect used enums

Author: eddeee888

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

@@ -1,6 +1,6 @@
 import { oldVisit, PluginFunction, Types } from '@graphql-codegen/plugin-helpers';
-import { LoadedFragment, optimizeOperations } from '@graphql-codegen/visitor-plugin-common';
-import { concatAST, FragmentDefinitionNode, GraphQLSchema, Kind } from 'graphql';
+import { optimizeOperations } from '@graphql-codegen/visitor-plugin-common';
+import { concatAST, GraphQLSchema } from 'graphql';
 import { TypeScriptDocumentsPluginConfig } from './config.js';
 import { TypeScriptDocumentsVisitor } from './visitor.js';
 
@@ -20,19 +20,7 @@ export const plugin: PluginFunction<TypeScriptDocumentsPluginConfig, Types.Compl
     : rawDocuments;
   const allAst = concatAST(documents.map(v => v.document));
 
-  const allFragments: LoadedFragment[] = [
-    ...(allAst.definitions.filter(d => d.kind === Kind.FRAGMENT_DEFINITION) as FragmentDefinitionNode[]).map(
-      fragmentDef => ({
-        node: fragmentDef,
-        name: fragmentDef.name.value,
-        onType: fragmentDef.typeCondition.name.value,
-        isExternal: false,
-      })
-    ),
-    ...(config.externalFragments || []),
-  ];
-
-  const visitor = new TypeScriptDocumentsVisitor(schema, config, allFragments);
+  const visitor = new TypeScriptDocumentsVisitor(schema, config, allAst);
 
   const visitorResult = oldVisit(allAst, {
     leave: visitor,

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

@@ -13,7 +13,21 @@ import {
   wrapTypeWithModifiers,
 } from '@graphql-codegen/visitor-plugin-common';
 import autoBind from 'auto-bind';
-import { GraphQLNamedType, GraphQLOutputType, GraphQLSchema, isEnumType, isNonNullType } from 'graphql';
+import {
+  type DocumentNode,
+  type FragmentDefinitionNode,
+  GraphQLEnumType,
+  GraphQLInputObjectType,
+  type GraphQLNamedInputType,
+  type GraphQLNamedType,
+  type GraphQLOutputType,
+  GraphQLScalarType,
+  type GraphQLSchema,
+  isEnumType,
+  isNonNullType,
+  Kind,
+  visit,
+} from 'graphql';
 import { TypeScriptDocumentsPluginConfig } from './config.js';
 import { TypeScriptOperationVariablesToObject } from './ts-operation-variables-to-object.js';
 import { TypeScriptSelectionSetProcessor } from './ts-selection-set-processor.js';
@@ -27,11 +41,14 @@ export interface TypeScriptDocumentsParsedConfig extends ParsedDocumentsConfig {
   allowUndefinedQueryVariables: boolean;
 }
 
+type UsedNamedInputTypes = Record<string, GraphQLNamedInputType>;
+
 export class TypeScriptDocumentsVisitor extends BaseDocumentsVisitor<
   TypeScriptDocumentsPluginConfig,
   TypeScriptDocumentsParsedConfig
 > {
-  constructor(schema: GraphQLSchema, config: TypeScriptDocumentsPluginConfig, allFragments: LoadedFragment[]) {
+  protected _usedNamedInputTypes: UsedNamedInputTypes = {};
+  constructor(schema: GraphQLSchema, config: TypeScriptDocumentsPluginConfig, documentNode: DocumentNode) {
     super(
       config,
       {
@@ -76,6 +93,20 @@ export class TypeScriptDocumentsVisitor extends BaseDocumentsVisitor<
       return (this.config.immutableTypes ? `readonly ${name}` : name) + (optional ? '?' : '');
     };
 
+    const allFragments: LoadedFragment[] = [
+      ...(documentNode.definitions.filter(d => d.kind === Kind.FRAGMENT_DEFINITION) as FragmentDefinitionNode[]).map(
+        fragmentDef => ({
+          node: fragmentDef,
+          name: fragmentDef.name.value,
+          onType: fragmentDef.typeCondition.name.value,
+          isExternal: false,
+        })
+      ),
+      ...(config.externalFragments || []),
+    ];
+
+    this._usedNamedInputTypes = this.collectUsedInputTypes({ schema, documentNode });
+
     const processorConfig: SelectionSetProcessorConfig = {
       namespacedImportName: this.config.namespacedImportName,
       convertName: this.convertName.bind(this),
@@ -142,4 +173,36 @@ export class TypeScriptDocumentsVisitor extends BaseDocumentsVisitor<
 
     return `${prefix}Exact<${variablesBlock === '{}' ? `{ [key: string]: never; }` : variablesBlock}>${extraType}`;
   }
+
+  private collectUsedInputTypes({
+    schema,
+    documentNode,
+  }: {
+    schema: GraphQLSchema;
+    documentNode: DocumentNode;
+  }): UsedNamedInputTypes {
+    const schemaTypes = schema.getTypeMap();
+
+    const usedInputTypes: UsedNamedInputTypes = {};
+
+    visit(documentNode, {
+      VariableDefinition: variableDefinitionNode => {
+        visit(variableDefinitionNode, {
+          NamedType: namedTypeNode => {
+            const foundInputType = schemaTypes[namedTypeNode.name.value];
+            if (
+              foundInputType &&
+              (foundInputType instanceof GraphQLInputObjectType ||
+                foundInputType instanceof GraphQLScalarType ||
+                foundInputType instanceof GraphQLEnumType)
+            ) {
+              usedInputTypes[namedTypeNode.name.value] = foundInputType;
+            }
+          },
+        });
+      },
+    });
+
+    return usedInputTypes;
+  }
 }