visit-related-shape-names.js
1.15 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
/**
*
* @param {string} startingShape
* @param {{[key: string]: any}} shapeMap
*/
function visitRelatedShapeNames(startingShape, shapeMap) {
var shape = shapeMap[startingShape];
if (shape.visited) {
// exit early if the shape has been visited
return;
}
shape.visited = true;
if (['structure', 'map', 'list'].indexOf(shape.type) < 0) {
// not a complex shape, so it's a terminal shape
return;
}
if (shape.type === 'structure') {
var members = shape.members;
for (var memberName of Object.keys(members)) {
var memberShapeName = members[memberName].shape;
visitRelatedShapeNames(memberShapeName, shapeMap);
}
} else if (shape.type === 'map') {
var keyShape = shape.key.shape;
var valueShape = shape.value.shape;
visitRelatedShapeNames(keyShape, shapeMap);
visitRelatedShapeNames(valueShape, shapeMap);
} else if (shape.type === 'list') {
var memberShape = shape.member.shape;
visitRelatedShapeNames(memberShape, shapeMap);
}
}
module.exports = {
visitRelatedShapeNames: visitRelatedShapeNames
};