Add support for appliedDirectives in extendSchema

Author: IvanGoncharov

src/utilities/extendSchema.js

@@ -28,6 +28,7 @@ import {
   GraphQLScalarType,
   GraphQLEnumType,
   GraphQLInputObjectType,
+  GraphQLAppliedDirectives,
   assertInputType,
   assertOutputType,
 } from '../type/definition';
@@ -82,6 +83,7 @@ import type {
 
 import type {
   DocumentNode,
+  DirectiveNode,
   InputValueDefinitionNode,
   TypeNode,
   NamedTypeNode,
@@ -95,6 +97,7 @@ import type {
   DirectiveDefinitionNode,
 } from '../language/ast';
 
+import { getArgumentValues } from '../execution/values';
 
 /**
  * Produces a new schema given an existing schema and a document which may
@@ -245,15 +248,38 @@ export function extendSchema(
     types.push(getTypeFromAST(typeDefinitionMap[typeName]));
   });
 
+  const directives = getMergedDirectives();
+  const innerDirectivesMap = keyValMap(
+      directives,
+      directive => directive.name,
+      directive => directive
+  );
+
   // Then produce and return a Schema with these types.
   return new GraphQLSchema({
     query: queryType,
     mutation: mutationType,
     subscription: subscriptionType,
     types,
-    directives: getMergedDirectives(),
+    directives,
+    appliedDirectives: schema.getAppliedDirectives(),
   });
 
+  function makeAppliedDirectives(appliedDirectives: ?Array<DirectiveNode>) {
+    return appliedDirectives && new GraphQLAppliedDirectives(keyValMap(
+      appliedDirectives,
+      directive => directive.name.value,
+      directive => (() => {
+        const directiveName = directive.name.value;
+        if (!innerDirectivesMap[directiveName]) {
+          throw new Error(
+            `Directive "${directiveName}" not found in document.`);
+        }
+        return getArgumentValues(innerDirectivesMap[directiveName], directive);
+      })
+    ));
+  }
+
   // Below are functions used for producing this schema that have closed over
   // this scope and have access to the schema, cache, and newly defined types.
 
@@ -349,6 +375,7 @@ export function extendSchema(
       description: type.description,
       interfaces: () => extendImplementedInterfaces(type),
       fields: () => extendFieldMap(type),
+      appliedDirectives: type.appliedDirectives,
       isTypeOf: type.isTypeOf,
     });
   }
@@ -360,6 +387,7 @@ export function extendSchema(
       name: type.name,
       description: type.description,
       fields: () => extendFieldMap(type),
+      appliedDirectives: type.appliedDirectives,
       resolveType: type.resolveType,
     });
   }
@@ -369,6 +397,7 @@ export function extendSchema(
       name: type.name,
       description: type.description,
       types: type.getTypes().map(getTypeFromDef),
+      appliedDirectives: type.appliedDirectives,
       resolveType: type.resolveType,
     });
   }
@@ -409,6 +438,7 @@ export function extendSchema(
         deprecationReason: field.deprecationReason,
         type: extendFieldType(field.type),
         args: keyMap(field.args, arg => arg.name),
+        appliedDirectives: field.appliedDirectives,
         resolve: field.resolve,
       };
     });
@@ -431,6 +461,7 @@ export function extendSchema(
             type: buildOutputFieldType(field.type),
             args: buildInputValues(field.arguments),
             deprecationReason: getDeprecationReason(field.directives),
+            appliedDirectives: makeAppliedDirectives(field.directives),
           };
         });
       });
@@ -467,6 +498,7 @@ export function extendSchema(
       description: getDescription(typeNode),
       interfaces: () => buildImplementedInterfaces(typeNode),
       fields: () => buildFieldMap(typeNode),
+      appliedDirectives: makeAppliedDirectives(typeNode.directives),
     });
   }
 
@@ -475,6 +507,7 @@ export function extendSchema(
       name: typeNode.name.value,
       description: getDescription(typeNode),
       fields: () => buildFieldMap(typeNode),
+      appliedDirectives: makeAppliedDirectives(typeNode.directives),
       resolveType: cannotExecuteExtendedSchema,
     });
   }
@@ -484,6 +517,7 @@ export function extendSchema(
       name: typeNode.name.value,
       description: getDescription(typeNode),
       types: typeNode.types.map(getObjectTypeFromAST),
+      appliedDirectives: makeAppliedDirectives(typeNode.directives),
       resolveType: cannotExecuteExtendedSchema,
     });
   }
@@ -492,6 +526,7 @@ export function extendSchema(
     return new GraphQLScalarType({
       name: typeNode.name.value,
       description: getDescription(typeNode),
+      appliedDirectives: makeAppliedDirectives(typeNode.directives),
       serialize: id => id,
       // Note: validation calls the parse functions to determine if a
       // literal value is correct. Returning null would cause use of custom
@@ -512,8 +547,10 @@ export function extendSchema(
         enumValue => ({
           description: getDescription(enumValue),
           deprecationReason: getDeprecationReason(enumValue.directives),
+          appliedDirectives: makeAppliedDirectives(enumValue.directives),
         }),
       ),
+      appliedDirectives: makeAppliedDirectives(typeNode.directives),
     });
   }
 
@@ -522,6 +559,7 @@ export function extendSchema(
       name: typeNode.name.value,
       description: getDescription(typeNode),
       fields: () => buildInputValues(typeNode.fields),
+      appliedDirectives: makeAppliedDirectives(typeNode.directives),
     });
   }
 
@@ -552,6 +590,7 @@ export function extendSchema(
         description: getDescription(field),
         args: buildInputValues(field.arguments),
         deprecationReason: getDeprecationReason(field.directives),
+        appliedDirectives: makeAppliedDirectives(field.directives),
       })
     );
   }
@@ -565,7 +604,8 @@ export function extendSchema(
         return {
           type,
           description: getDescription(value),
-          defaultValue: valueFromAST(value.defaultValue, type)
+          defaultValue: valueFromAST(value.defaultValue, type),
+          appliedDirectives: makeAppliedDirectives(value.directives),
         };
       }
     );