84 lines
2.8 KiB
JavaScript
84 lines
2.8 KiB
JavaScript
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);
|
|
}
|
|
}
|
|
}
|
|
}
|