Add ability to getDirectives() on a type

Author: mjmahone

src/type/definition.js

@@ -35,6 +35,7 @@ import type {
   OperationDefinitionNode,
   FieldNode,
   FragmentDefinitionNode,
+  DirectiveNode,
   ValueNode,
 } from '../language/ast';
 import type { GraphQLSchema } from './schema';
@@ -541,6 +542,8 @@ export class GraphQLScalarType {
   astNode: ?ScalarTypeDefinitionNode;
   extensionASTNodes: ?$ReadOnlyArray<ScalarTypeExtensionNode>;
 
+  _directives: ?$ReadOnlyArray<DirectiveNode>;
+
   constructor(config: GraphQLScalarTypeConfig<*, *>): void {
     this.name = config.name;
     this.description = config.description;
@@ -566,6 +569,28 @@ export class GraphQLScalarType {
     }
   }
 
+  getDirectives(): $ReadOnlyArray<DirectiveNode> {
+    if (this._directives) {
+      return this._directives;
+    }
+
+    const directives = [];
+    if (this.astNode && this.astNode.directives) {
+      directives.push(...this.astNode.directives);
+    }
+    const extensionASTNodes = this.extensionASTNodes;
+    if (extensionASTNodes) {
+      for (let i = 0; i < extensionASTNodes.length; i++) {
+        const extensionNode = extensionASTNodes[i];
+        if (extensionNode.directives) {
+          directives.push(...extensionNode.directives);
+        }
+      }
+    }
+    this._directives = directives;
+    return directives;
+  }
+
   toString(): string {
     return this.name;
   }
@@ -641,6 +666,7 @@ export class GraphQLObjectType {
 
   _fields: Thunk<GraphQLFieldMap<*, *>>;
   _interfaces: Thunk<Array<GraphQLInterfaceType>>;
+  _directives: ?$ReadOnlyArray<DirectiveNode>;
 
   constructor(config: GraphQLObjectTypeConfig<*, *>): void {
     this.name = config.name;
@@ -659,6 +685,28 @@ export class GraphQLObjectType {
     }
   }
 
+  getDirectives(): $ReadOnlyArray<DirectiveNode> {
+    if (this._directives) {
+      return this._directives;
+    }
+
+    const directives = [];
+    if (this.astNode && this.astNode.directives) {
+      directives.push(...this.astNode.directives);
+    }
+    const extensionASTNodes = this.extensionASTNodes;
+    if (extensionASTNodes) {
+      for (let i = 0; i < extensionASTNodes.length; i++) {
+        const extensionNode = extensionASTNodes[i];
+        if (extensionNode.directives) {
+          directives.push(...extensionNode.directives);
+        }
+      }
+    }
+    this._directives = directives;
+    return directives;
+  }
+
   getFields(): GraphQLFieldMap<*, *> {
     if (typeof this._fields === 'function') {
       this._fields = this._fields();
@@ -894,6 +942,7 @@ export class GraphQLInterfaceType {
   resolveType: ?GraphQLTypeResolver<*, *>;
 
   _fields: Thunk<GraphQLFieldMap<*, *>>;
+  _directives: ?$ReadOnlyArray<DirectiveNode>;
 
   constructor(config: GraphQLInterfaceTypeConfig<*, *>): void {
     this.name = config.name;
@@ -911,6 +960,28 @@ export class GraphQLInterfaceType {
     }
   }
 
+  getDirectives(): $ReadOnlyArray<DirectiveNode> {
+    if (this._directives) {
+      return this._directives;
+    }
+
+    const directives = [];
+    if (this.astNode && this.astNode.directives) {
+      directives.push(...this.astNode.directives);
+    }
+    const extensionASTNodes = this.extensionASTNodes;
+    if (extensionASTNodes) {
+      for (let i = 0; i < extensionASTNodes.length; i++) {
+        const extensionNode = extensionASTNodes[i];
+        if (extensionNode.directives) {
+          directives.push(...extensionNode.directives);
+        }
+      }
+    }
+    this._directives = directives;
+    return directives;
+  }
+
   getFields(): GraphQLFieldMap<*, *> {
     if (typeof this._fields === 'function') {
       this._fields = this._fields();
@@ -972,6 +1043,7 @@ export class GraphQLUnionType {
   resolveType: ?GraphQLTypeResolver<*, *>;
 
   _types: Thunk<Array<GraphQLObjectType>>;
+  _directives: ?$ReadOnlyArray<DirectiveNode>;
 
   constructor(config: GraphQLUnionTypeConfig<*, *>): void {
     this.name = config.name;
@@ -989,6 +1061,28 @@ export class GraphQLUnionType {
     }
   }
 
+  getDirectives(): $ReadOnlyArray<DirectiveNode> {
+    if (this._directives) {
+      return this._directives;
+    }
+
+    const directives = [];
+    if (this.astNode && this.astNode.directives) {
+      directives.push(...this.astNode.directives);
+    }
+    const extensionASTNodes = this.extensionASTNodes;
+    if (extensionASTNodes) {
+      for (let i = 0; i < extensionASTNodes.length; i++) {
+        const extensionNode = extensionASTNodes[i];
+        if (extensionNode.directives) {
+          directives.push(...extensionNode.directives);
+        }
+      }
+    }
+    this._directives = directives;
+    return directives;
+  }
+
   getTypes(): Array<GraphQLObjectType> {
     if (typeof this._types === 'function') {
       this._types = this._types();
@@ -1058,6 +1152,7 @@ export class GraphQLEnumType /* <T> */ {
   astNode: ?EnumTypeDefinitionNode;
   extensionASTNodes: ?$ReadOnlyArray<EnumTypeExtensionNode>;
 
+  _directives: ?$ReadOnlyArray<DirectiveNode>;
   _values: Array<GraphQLEnumValue /* <T> */>;
   _valueLookup: Map<any /* T */, GraphQLEnumValue>;
   _nameLookup: ObjMap<GraphQLEnumValue>;
@@ -1076,6 +1171,28 @@ export class GraphQLEnumType /* <T> */ {
     invariant(typeof config.name === 'string', 'Must provide name.');
   }
 
+  getDirectives(): $ReadOnlyArray<DirectiveNode> {
+    if (this._directives) {
+      return this._directives;
+    }
+
+    const directives = [];
+    if (this.astNode && this.astNode.directives) {
+      directives.push(...this.astNode.directives);
+    }
+    const extensionASTNodes = this.extensionASTNodes;
+    if (extensionASTNodes) {
+      for (let i = 0; i < extensionASTNodes.length; i++) {
+        const extensionNode = extensionASTNodes[i];
+        if (extensionNode.directives) {
+          directives.push(...extensionNode.directives);
+        }
+      }
+    }
+    this._directives = directives;
+    return directives;
+  }
+
   getValues(): Array<GraphQLEnumValue /* <T> */> {
     return this._values;
   }
@@ -1204,6 +1321,7 @@ export class GraphQLInputObjectType {
   astNode: ?InputObjectTypeDefinitionNode;
   extensionASTNodes: ?$ReadOnlyArray<InputObjectTypeExtensionNode>;
 
+  _directives: ?$ReadOnlyArray<DirectiveNode>;
   _fields: Thunk<GraphQLInputFieldMap>;
 
   constructor(config: GraphQLInputObjectTypeConfig): void {
@@ -1215,6 +1333,28 @@ export class GraphQLInputObjectType {
     invariant(typeof config.name === 'string', 'Must provide name.');
   }
 
+  getDirectives(): $ReadOnlyArray<DirectiveNode> {
+    if (this._directives) {
+      return this._directives;
+    }
+
+    const directives = [];
+    if (this.astNode && this.astNode.directives) {
+      directives.push(...this.astNode.directives);
+    }
+    const extensionASTNodes = this.extensionASTNodes;
+    if (extensionASTNodes) {
+      for (let i = 0; i < extensionASTNodes.length; i++) {
+        const extensionNode = extensionASTNodes[i];
+        if (extensionNode.directives) {
+          directives.push(...extensionNode.directives);
+        }
+      }
+    }
+    this._directives = directives;
+    return directives;
+  }
+
   getFields(): GraphQLInputFieldMap {
     if (typeof this._fields === 'function') {
       this._fields = this._fields();