Add support for directives applied on IDL&Schema

Author: IvanGoncharov

src/type/__tests__/definition-test.js

@@ -180,7 +180,8 @@ describe('Type System: Example', () => {
       description: undefined,
       isDeprecated: true,
       deprecationReason: 'Just because',
-      value: 'foo'
+      value: 'foo',
+      appliedDirectives: undefined,
     });
   });
 
@@ -200,13 +201,15 @@ describe('Type System: Example', () => {
         isDeprecated: false,
         deprecationReason: undefined,
         value: null,
+        appliedDirectives: undefined,
       },
       {
         name: 'UNDEFINED',
         description: undefined,
         isDeprecated: false,
         deprecationReason: undefined,
         value: undefined,
+        appliedDirectives: undefined,
       },
     ]);
   });

src/type/definition.js

@@ -272,6 +272,54 @@ function resolveThunk<T>(thunk: Thunk<T>): T {
   return typeof thunk === 'function' ? thunk() : thunk;
 }
 
+export class GraphQLAppliedDirectives {
+  _appliedDirectivesConfig: GraphQLAppliedDirectivesConfig;
+  _argsMap: GraphQLDirectiveArgsMap;
+
+  constructor(config: GraphQLAppliedDirectivesConfig) {
+    invariant(
+      isPlainObj(config),
+      'Applied directives must be an object with directive names as keys.'
+    );
+
+    this._appliedDirectivesConfig = config;
+    this._argsMap = {};
+  }
+
+  isApplied(directiveName: string): boolean {
+    return Boolean(this._appliedDirectivesConfig[directiveName]);
+  }
+
+  getAppliedDirectives(): Array<string> {
+    return Object.keys(this._appliedDirectivesConfig);
+  }
+
+  getDirectiveArgs(directiveName: string): ?{ [argName: string]: any } {
+    if (this._argsMap[directiveName]) {
+      return this._argsMap[directiveName];
+    }
+    const argsThunk = this._appliedDirectivesConfig[directiveName];
+    const args = resolveThunk(argsThunk);
+    if (!args) {
+      return null;
+    }
+    invariant(
+      isPlainObj(args),
+      `@${directiveName} args must be an object with argument names as keys ` +
+      'or a function which returns such an object.'
+    );
+    this._argsMap[directiveName] = args;
+    return args;
+  }
+}
+
+type GraphQLDirectiveArgsMap = {
+  [key: string]: { [argName: string]: any }
+};
+
+type GraphQLAppliedDirectivesConfig = {
+  [key: string]: Thunk<*>
+};
 
 /**
  * Scalar Type Definition
@@ -293,13 +341,15 @@ function resolveThunk<T>(thunk: Thunk<T>): T {
 export class GraphQLScalarType {
   name: string;
   description: ?string;
+  appliedDirectives: ?GraphQLAppliedDirectives;
 
   _scalarConfig: GraphQLScalarTypeConfig<*, *>;
 
   constructor(config: GraphQLScalarTypeConfig<*, *>) {
     assertValidName(config.name);
     this.name = config.name;
     this.description = config.description;
+    this.appliedDirectives = config.appliedDirectives;
     invariant(
       typeof config.serialize === 'function',
       `${this.name} must provide "serialize" function. If this custom Scalar ` +
@@ -351,6 +401,7 @@ GraphQLScalarType.prototype.toJSON =
 export type GraphQLScalarTypeConfig<TInternal, TExternal> = {
   name: string;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
   serialize: (value: mixed) => ?TExternal;
   parseValue?: (value: mixed) => ?TInternal;
   parseLiteral?: (valueNode: ValueNode) => ?TInternal;
@@ -398,6 +449,7 @@ export type GraphQLScalarTypeConfig<TInternal, TExternal> = {
 export class GraphQLObjectType {
   name: string;
   description: ?string;
+  appliedDirectives: ?GraphQLAppliedDirectives;
   isTypeOf: ?GraphQLIsTypeOfFn<*, *>;
 
   _typeConfig: GraphQLObjectTypeConfig<*, *>;
@@ -408,6 +460,7 @@ export class GraphQLObjectType {
     assertValidName(config.name, config.isIntrospection);
     this.name = config.name;
     this.description = config.description;
+    this.appliedDirectives = config.appliedDirectives;
     if (config.isTypeOf) {
       invariant(
         typeof config.isTypeOf === 'function',
@@ -549,7 +602,8 @@ function defineFieldMap<TSource, TContext>(
           name: argName,
           description: arg.description === undefined ? null : arg.description,
           type: arg.type,
-          defaultValue: arg.defaultValue
+          defaultValue: arg.defaultValue,
+          appliedDirectives: arg.appliedDirectives,
         };
       });
     }
@@ -574,6 +628,7 @@ export type GraphQLObjectTypeConfig<TSource, TContext> = {
   isTypeOf?: ?GraphQLIsTypeOfFn<TSource, TContext>;
   description?: ?string;
   isIntrospection?: boolean;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 export type GraphQLTypeResolver<TSource, TContext> = (
@@ -616,6 +671,7 @@ export type GraphQLFieldConfig<TSource, TContext> = {
   resolve?: GraphQLFieldResolver<TSource, TContext>;
   deprecationReason?: ?string;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 export type GraphQLFieldConfigArgumentMap = {
@@ -626,6 +682,7 @@ export type GraphQLArgumentConfig = {
   type: GraphQLInputType;
   defaultValue?: mixed;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 export type GraphQLFieldConfigMap<TSource, TContext> = {
@@ -640,13 +697,15 @@ export type GraphQLField<TSource, TContext> = {
   resolve?: GraphQLFieldResolver<TSource, TContext>;
   isDeprecated?: boolean;
   deprecationReason?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 export type GraphQLArgument = {
   name: string;
   type: GraphQLInputType;
   defaultValue?: mixed;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 export type GraphQLFieldMap<TSource, TContext> = {
@@ -676,6 +735,7 @@ export type GraphQLFieldMap<TSource, TContext> = {
 export class GraphQLInterfaceType {
   name: string;
   description: ?string;
+  appliedDirectives: ?GraphQLAppliedDirectives;
   resolveType: ?GraphQLTypeResolver<*, *>;
 
   _typeConfig: GraphQLInterfaceTypeConfig<*, *>;
@@ -685,6 +745,7 @@ export class GraphQLInterfaceType {
     assertValidName(config.name);
     this.name = config.name;
     this.description = config.description;
+    this.appliedDirectives = config.appliedDirectives;
     if (config.resolveType) {
       invariant(
         typeof config.resolveType === 'function',
@@ -722,7 +783,8 @@ export type GraphQLInterfaceTypeConfig<TSource, TContext> = {
    * Object type.
    */
   resolveType?: ?GraphQLTypeResolver<TSource, TContext>,
-  description?: ?string
+  description?: ?string,
+  appliedDirectives?: ?GraphQLAppliedDirectives,
 };
 
 
@@ -753,6 +815,7 @@ export type GraphQLInterfaceTypeConfig<TSource, TContext> = {
 export class GraphQLUnionType {
   name: string;
   description: ?string;
+  appliedDirectives: ?GraphQLAppliedDirectives;
   resolveType: ?GraphQLTypeResolver<*, *>;
 
   _typeConfig: GraphQLUnionTypeConfig<*, *>;
@@ -763,6 +826,7 @@ export class GraphQLUnionType {
     assertValidName(config.name);
     this.name = config.name;
     this.description = config.description;
+    this.appliedDirectives = config.appliedDirectives;
     if (config.resolveType) {
       invariant(
         typeof config.resolveType === 'function',
@@ -839,6 +903,7 @@ export type GraphQLUnionTypeConfig<TSource, TContext> = {
    */
   resolveType?: ?GraphQLTypeResolver<TSource, TContext>;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 
@@ -867,6 +932,7 @@ export type GraphQLUnionTypeConfig<TSource, TContext> = {
 export class GraphQLEnumType/* <T> */ {
   name: string;
   description: ?string;
+  appliedDirectives: ?GraphQLAppliedDirectives;
 
   _enumConfig: GraphQLEnumTypeConfig/* <T> */;
   _values: Array<GraphQLEnumValue/* <T> */>;
@@ -877,6 +943,7 @@ export class GraphQLEnumType/* <T> */ {
     this.name = config.name;
     assertValidName(config.name, config.isIntrospection);
     this.description = config.description;
+    this.appliedDirectives = config.appliedDirectives;
     this._values = defineEnumValues(this, config.values);
     this._enumConfig = config;
   }
@@ -983,6 +1050,7 @@ function defineEnumValues(
       description: value.description,
       isDeprecated: Boolean(value.deprecationReason),
       deprecationReason: value.deprecationReason,
+      appliedDirectives: value.appliedDirectives,
       value: value.hasOwnProperty('value') ? value.value : valueName,
     };
   });
@@ -992,6 +1060,7 @@ export type GraphQLEnumTypeConfig/* <T> */ = {
   name: string;
   values: GraphQLEnumValueConfigMap/* <T> */;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
   isIntrospection?: boolean;
 };
 
@@ -1003,13 +1072,15 @@ export type GraphQLEnumValueConfig/* <T> */ = {
   value?: any/* T */;
   deprecationReason?: ?string;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 export type GraphQLEnumValue/* <T> */ = {
   name: string;
   description: ?string;
   isDeprecated?: boolean;
   deprecationReason: ?string;
+  appliedDirectives: ?GraphQLAppliedDirectives;
   value: any/* T */;
 };
 
@@ -1038,6 +1109,7 @@ export type GraphQLEnumValue/* <T> */ = {
 export class GraphQLInputObjectType {
   name: string;
   description: ?string;
+  appliedDirectives: ?GraphQLAppliedDirectives;
 
   _typeConfig: GraphQLInputObjectTypeConfig;
   _fields: GraphQLInputFieldMap;
@@ -1046,6 +1118,7 @@ export class GraphQLInputObjectType {
     assertValidName(config.name);
     this.name = config.name;
     this.description = config.description;
+    this.appliedDirectives = config.appliedDirectives;
     this._typeConfig = config;
   }
 
@@ -1105,12 +1178,14 @@ export type GraphQLInputObjectTypeConfig = {
   name: string;
   fields: Thunk<GraphQLInputFieldConfigMap>;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 export type GraphQLInputFieldConfig = {
   type: GraphQLInputType;
   defaultValue?: mixed;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 export type GraphQLInputFieldConfigMap = {
@@ -1122,6 +1197,7 @@ export type GraphQLInputField = {
   type: GraphQLInputType;
   defaultValue?: mixed;
   description?: ?string;
+  appliedDirectives?: ?GraphQLAppliedDirectives;
 };
 
 export type GraphQLInputFieldMap = {