Support InputObjectTypeDefinition & InputObjectTypeExtension for getFieldsWithDirectives

packages/utils/src/get-fields-with-directives.ts

@@ -1,15 +1,29 @@
-import { DocumentNode, ObjectTypeDefinitionNode, ObjectTypeExtensionNode, ValueNode, Kind } from 'graphql';
+import {
+  DocumentNode,
+  ObjectTypeDefinitionNode,
+  ObjectTypeExtensionNode,
+  InputObjectTypeDefinitionNode,
+  InputObjectTypeExtensionNode,
+  ValueNode,
+  Kind,
+} from 'graphql';
 
 export type DirectiveArgs = { [name: string]: any };
 export type DirectiveUsage = { name: string; args: DirectiveArgs };
 export type TypeAndFieldToDirectives = {
   [typeAndField: string]: DirectiveUsage[];
 };
 
-function isObjectTypeDefinitionOrExtension(obj: any): obj is ObjectTypeDefinitionNode | ObjectTypeDefinitionNode {
-  return obj && (obj.kind === 'ObjectTypeDefinition' || obj.kind === 'ObjectTypeExtension');
+interface Options {
+  includeInputTypes?: boolean;
 }
 
+type SelectedNodes =
+  | ObjectTypeDefinitionNode
+  | ObjectTypeExtensionNode
+  | InputObjectTypeDefinitionNode
+  | InputObjectTypeExtensionNode;
+
 function parseDirectiveValue(value: ValueNode): any {
   switch (value.kind) {
     case Kind.INT:
@@ -32,11 +46,16 @@ function parseDirectiveValue(value: ValueNode): any {
   }
 }
 
-export function getFieldsWithDirectives(documentNode: DocumentNode): TypeAndFieldToDirectives {
+export function getFieldsWithDirectives(documentNode: DocumentNode, options: Options = {}): TypeAndFieldToDirectives {
   const result: TypeAndFieldToDirectives = {};
-  const allTypes = documentNode.definitions.filter<ObjectTypeDefinitionNode | ObjectTypeExtensionNode>(
-    isObjectTypeDefinitionOrExtension
-  );
+
+  let selected = ['ObjectTypeDefinition', 'ObjectTypeExtension'];
+
+  if (options.includeInputTypes) {
+    selected = [...selected, 'InputObjectTypeDefinition', 'InputObjectTypeExtension'];
+  }
+
+  const allTypes = documentNode.definitions.filter(obj => selected.includes(obj.kind)) as SelectedNodes[];
 
   for (const type of allTypes) {
     const typeName = type.name.value;

packages/utils/tests/get-fields-with-directives.spec.ts

@@ -136,4 +136,36 @@ describe('getFieldsWithDirectives', () => {
     const result = getFieldsWithDirectives(node);
     expect(Object.keys(result).length).toBe(2);
   });
+
+  it('Should detect multiple input types', () => {
+    const node = parse(`
+        input A {
+          f1: String @a
+        }
+
+        input B {
+          f2: String @a
+        }
+    `);
+
+    const result = getFieldsWithDirectives(node, {includeInputTypes: true});
+    expect(result['A.f1']).toEqual([{ name: 'a', args: {} }]);
+    expect(result['B.f2']).toEqual([{ name: 'a', args: {} }]);
+  });
+
+  it('Should detect multiple extend input types', () => {
+    const node = parse(`
+        extend input A {
+          f1: String @a
+        }
+
+        extend input B {
+          f2: String @a
+        }
+    `);
+
+    const result = getFieldsWithDirectives(node, {includeInputTypes: true});
+    expect(result['A.f1']).toEqual([{ name: 'a', args: {} }]);
+    expect(result['B.f2']).toEqual([{ name: 'a', args: {} }]);
+  });
 });