forked from GoogleCloudPlatform/professional-services
-
Notifications
You must be signed in to change notification settings - Fork 0
/
gke.js
99 lines (91 loc) · 2.97 KB
/
gke.js
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
var containerResourceTypes = [
"cloudresourcemanager.googleapis.com/Organization",
"cloudresourcemanager.googleapis.com/Folder",
"cloudresourcemanager.googleapis.com/Project",
];
var wantedResourceTypes = [
"container.googleapis.com/Cluster",
"container.googleapis.com/NodePool",
"k8s.io/Node",
"k8s.io/Namespace",
"k8s.io/Pod",
"k8s.io/Service",
];
var resourceTypes = containerResourceTypes.concat(wantedResourceTypes);
var nodes = [];
var follow = function (n, depth) {
var out = n.tag("parent").labelContext(resourceTypes, "type").out("child");
if (out.count() == 0) {
return;
}
nodes = nodes.concat(out.tagArray());
follow(out, depth + 1);
};
// Filters disconnected vertexes from results
var filterEmptyNodes = function (nodes) {
var filteredNodes = [];
var m = g.Morphism().labelContext(resourceTypes, "type").in(["child", "uses"]);
nodes.forEach(function (node) {
if (wantedResourceTypes.indexOf(node.type) > -1) {
if (g.V(node.id).follow(m).count() > 0) {
filteredNodes.push(node);
}
} else {
filteredNodes.push(node);
}
});
return filteredNodes;
}
// Filters empty projects from results
var filterEmptyProjects = function (nodes) {
var filteredNodes = [];
var projectM = g.Morphism().labelContext(resourceTypes, "type").in(["child", "uses"]);
nodes.forEach(function (node) {
if (node.type == "cloudresourcemanager.googleapis.com/Project") {
if (g.V(node.id).follow(projectM).count() > 1) {
filteredNodes.push(node);
}
} else {
filteredNodes.push(node);
}
});
return filteredNodes;
}
// Filters empty folders from results
var filterEmptyFolders = function (nodes) {
var folderMap = {};
var folderItemCount = {};
var filteredNodes = [];
nodes.forEach(function (node) {
if (containerResourceTypes.indexOf(node.type) > -1) {
folderMap[node.id] = node;
if (node.type == "cloudresourcemanager.googleapis.com/Folder") {
folderItemCount[node.id] = 0;
}
}
});
nodes.forEach(function (node) {
if (node.type == "cloudresourcemanager.googleapis.com/Project") {
var iNode = node;
while (iNode && iNode.parent in folderMap) {
folderItemCount[iNode.parent] += 1;
iNode = folderMap[iNode.parent];
}
}
});
nodes.forEach(function (node) {
if (node.type == "cloudresourcemanager.googleapis.com/Folder") {
if (folderItemCount[node.id] > 0) {
filteredNodes.push(node);
}
} else {
filteredNodes.push(node);
}
});
return filteredNodes;
}
var root = g.V("{{ index .Organizations 0 }}");
follow(root, 1);
filterEmptyFolders(filterEmptyProjects(root.tagArray().concat(nodes))).forEach(function (node) {
g.emit(node);
});