1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 |
- import { visit } from '../language/visitor';
- /**
- * separateOperations accepts a single AST document which may contain many
- * operations and fragments and returns a collection of AST documents each of
- * which contains a single operation as well the fragment definitions it
- * refers to.
- */
- export function separateOperations(documentAST) {
- var operations = [];
- var fragments = Object.create(null);
- var positions = new Map();
- var depGraph = Object.create(null);
- var fromName;
- var idx = 0; // Populate metadata and build a dependency graph.
- visit(documentAST, {
- OperationDefinition: function OperationDefinition(node) {
- fromName = opName(node);
- operations.push(node);
- positions.set(node, idx++);
- },
- FragmentDefinition: function FragmentDefinition(node) {
- fromName = node.name.value;
- fragments[fromName] = node;
- positions.set(node, idx++);
- },
- FragmentSpread: function FragmentSpread(node) {
- var toName = node.name.value;
- (depGraph[fromName] || (depGraph[fromName] = Object.create(null)))[toName] = true;
- }
- }); // For each operation, produce a new synthesized AST which includes only what
- // is necessary for completing that operation.
- var separatedDocumentASTs = Object.create(null);
- for (var _i2 = 0; _i2 < operations.length; _i2++) {
- var operation = operations[_i2];
- var operationName = opName(operation);
- var dependencies = Object.create(null);
- collectTransitiveDependencies(dependencies, depGraph, operationName); // The list of definition nodes to be included for this operation, sorted
- // to retain the same order as the original document.
- var definitions = [operation];
- for (var _i4 = 0, _Object$keys2 = Object.keys(dependencies); _i4 < _Object$keys2.length; _i4++) {
- var name = _Object$keys2[_i4];
- definitions.push(fragments[name]);
- }
- definitions.sort(function (n1, n2) {
- return (positions.get(n1) || 0) - (positions.get(n2) || 0);
- });
- separatedDocumentASTs[operationName] = {
- kind: 'Document',
- definitions: definitions
- };
- }
- return separatedDocumentASTs;
- }
- // Provides the empty string for anonymous operations.
- function opName(operation) {
- return operation.name ? operation.name.value : '';
- } // From a dependency graph, collects a list of transitive dependencies by
- // recursing through a dependency graph.
- function collectTransitiveDependencies(collected, depGraph, fromName) {
- var immediateDeps = depGraph[fromName];
- if (immediateDeps) {
- for (var _i6 = 0, _Object$keys4 = Object.keys(immediateDeps); _i6 < _Object$keys4.length; _i6++) {
- var toName = _Object$keys4[_i6];
- if (!collected[toName]) {
- collected[toName] = true;
- collectTransitiveDependencies(collected, depGraph, toName);
- }
- }
- }
- }
|