@ardatan/relay-compiler/lib/transforms/SkipHandleFieldTransform.js

Fork of `relay-compiler`

/**
 * 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.
 *
 *
 * @format
 */
// flowlint ambiguous-object-type:error
'use strict';

var _interopRequireDefault = require('@babel/runtime/helpers/interopRequireDefault');

var _objectSpread2 = _interopRequireDefault(require('@babel/runtime/helpers/objectSpread2'));

var IRTransformer = require('../core/IRTransformer');

/**
 * A transform that removes field `handles`. Intended for use when e.g.
 * printing queries to send to a GraphQL server.
 */
function skipHandleFieldTransform(context) {
  return IRTransformer.transform(context, {
    LinkedField: visitField,
    ScalarField: visitField,
  });
}

function visitField(field) {
  // $FlowFixMe[incompatible-use]
  var transformedNode = this.traverse(field);

  if (transformedNode.handles) {
    return (0, _objectSpread2['default'])(
      (0, _objectSpread2['default'])({}, transformedNode),
      {},
      {
        handles: null,
      },
    );
  }

  return transformedNode;
}

module.exports = {
  transform: skipHandleFieldTransform,
};