Mirror: The small sibling of the graphql package, slimmed down for client-side libraries.
Add initial alternative visitor implementation
+76
+76
alias/language/visitor.mjs
+76
alias/language/visitor.mjs
···
1
+
import { getVisitFn, visitInParallel, BREAK } from 'graphql/language/visitor';
2
+
3
+
export { getVisitFn, visitInParallel, BREAK };
4
+
5
+
export function visit(node, visitor) {
6
+
const path = [];
7
+
const ancestors = [];
8
+
9
+
function callback(node, key, parent, isLeaving) {
10
+
const visitFn = getVisitFn(visitor, node.kind, isLeaving)
11
+
if (visitFn) {
12
+
const result = visitFn.call(visitor, node, key, parent, path, ancestors)
13
+
if (result === BREAK) throw BREAK
14
+
return result
15
+
}
16
+
}
17
+
18
+
function traverse(node, key, parent) {
19
+
let result;
20
+
21
+
result = callback(node, key, parent, false);
22
+
if (result === false) {
23
+
return;
24
+
} else if (result && typeof result.kind === 'string') {
25
+
node = result;
26
+
}
27
+
28
+
ancestors.push(node);
29
+
const copy = {};
30
+
31
+
let hasEdited = false;
32
+
for (const nodeKey in node) {
33
+
const value = node[nodeKey];
34
+
35
+
let newValue;
36
+
path.push(nodeKey);
37
+
if (Array.isArray(value)) {
38
+
newValue = value.slice();
39
+
for (let index = 0; index < value.length; index++) {
40
+
if (value[index] != null && typeof value[index].kind === 'string') {
41
+
path.push(index);
42
+
result = traverse(value[index], index, node);
43
+
path.pop();
44
+
if (result !== undefined) {
45
+
newValue[index] = result;
46
+
hasEdited = true;
47
+
}
48
+
}
49
+
}
50
+
51
+
} else if (value != null && typeof value.kind === 'string') {
52
+
result = traverse(value, nodeKey, node);
53
+
if (result !== undefined) {
54
+
newValue = result;
55
+
hasEdited = true;
56
+
}
57
+
}
58
+
59
+
path.pop();
60
+
copy[nodeKey] = hasEdited ? newValue : value;
61
+
}
62
+
63
+
ancestors.pop();
64
+
65
+
if (hasEdited) node = copy;
66
+
return callback(node, key, parent, true);
67
+
}
68
+
69
+
try {
70
+
const result = traverse(node);
71
+
return result !== undefined ? result : node;
72
+
} catch (error) {
73
+
if (error !== BREAK) throw error;
74
+
return node;
75
+
}
76
+
}