1 |
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 | import { visit } from '../language/visitor';
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 | export function separateOperations(documentAST) {
|
18 | var operations = [];
|
19 | var fragments = Object.create(null);
|
20 | var positions = new Map();
|
21 | var depGraph = Object.create(null);
|
22 | var fromName;
|
23 | var idx = 0;
|
24 |
|
25 | visit(documentAST, {
|
26 | OperationDefinition: function OperationDefinition(node) {
|
27 | fromName = opName(node);
|
28 | operations.push(node);
|
29 | positions.set(node, idx++);
|
30 | },
|
31 | FragmentDefinition: function FragmentDefinition(node) {
|
32 | fromName = node.name.value;
|
33 | fragments[fromName] = node;
|
34 | positions.set(node, idx++);
|
35 | },
|
36 | FragmentSpread: function FragmentSpread(node) {
|
37 | var toName = node.name.value;
|
38 | (depGraph[fromName] || (depGraph[fromName] = Object.create(null)))[toName] = true;
|
39 | }
|
40 | });
|
41 |
|
42 |
|
43 | var separatedDocumentASTs = Object.create(null);
|
44 |
|
45 | for (var _i = 0; _i < operations.length; _i++) {
|
46 | var operation = operations[_i];
|
47 | var operationName = opName(operation);
|
48 | var dependencies = Object.create(null);
|
49 | collectTransitiveDependencies(dependencies, depGraph, operationName);
|
50 |
|
51 |
|
52 | var definitions = [operation];
|
53 |
|
54 | var _arr = Object.keys(dependencies);
|
55 |
|
56 | for (var _i2 = 0; _i2 < _arr.length; _i2++) {
|
57 | var name = _arr[_i2];
|
58 | definitions.push(fragments[name]);
|
59 | }
|
60 |
|
61 | definitions.sort(function (n1, n2) {
|
62 | return (positions.get(n1) || 0) - (positions.get(n2) || 0);
|
63 | });
|
64 | separatedDocumentASTs[operationName] = {
|
65 | kind: 'Document',
|
66 | definitions: definitions
|
67 | };
|
68 | }
|
69 |
|
70 | return separatedDocumentASTs;
|
71 | }
|
72 |
|
73 |
|
74 | function opName(operation) {
|
75 | return operation.name ? operation.name.value : '';
|
76 | }
|
77 |
|
78 |
|
79 |
|
80 | function collectTransitiveDependencies(collected, depGraph, fromName) {
|
81 | var immediateDeps = depGraph[fromName];
|
82 |
|
83 | if (immediateDeps) {
|
84 | var _arr2 = Object.keys(immediateDeps);
|
85 |
|
86 | for (var _i3 = 0; _i3 < _arr2.length; _i3++) {
|
87 | var toName = _arr2[_i3];
|
88 |
|
89 | if (!collected[toName]) {
|
90 | collected[toName] = true;
|
91 | collectTransitiveDependencies(collected, depGraph, toName);
|
92 | }
|
93 | }
|
94 | }
|
95 | } |
\ | No newline at end of file |