relay-compiler/lib/FilterDirectivesTransform.js

/**
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *
 *  strict-local
 * @format
 */
'use strict';

/**
 * A transform that removes any directives that were not present in the
 * server schema.
 */
function filterDirectivesTransform(context) {
  return require("./GraphQLIRTransformer").transform(context, {
    Directive: visitDirective
  });
}
/**
 * @internal
 *
 * Skip directives not defined in the original schema.
 */


function visitDirective(directive) {
  if (this.getContext().serverSchema.getDirectives().some(function (schemaDirective) {
    return schemaDirective.name === directive.name;
  })) {
    return directive;
  }

  return null;
}

module.exports = {
  transform: filterDirectivesTransform
};