@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
149 lines (148 loc) • 5.22 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const eks_1 = __importDefault(require("aws-sdk/clients/eks"));
const sdk_1 = __importDefault(require("@cloudgraph/sdk"));
const groupBy_1 = __importDefault(require("lodash/groupBy"));
const isEmpty_1 = __importDefault(require("lodash/isEmpty"));
const logger_1 = __importDefault(require("../../properties/logger"));
const utils_1 = require("../../utils");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const lt = { ...logger_1.default };
const { logger } = sdk_1.default;
const serviceName = 'EKS cluster';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
const MAX_ITEMS = 100;
const listClustersForRegion = async ({ eks, resolveRegion, }) => new Promise(resolve => {
const clusterList = [];
const listClustersOpts = {};
const listAllClusters = (token) => {
listClustersOpts.maxResults = MAX_ITEMS;
if (token) {
listClustersOpts.nextToken = token;
}
try {
eks.listClusters(listClustersOpts, (err, data) => {
const { nextToken, clusters = [] } = data || {};
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'eks:listClusters',
err,
});
}
/**
* No clusters for this region
*/
if (isEmpty_1.default(data)) {
return resolveRegion();
}
clusterList.push(...clusters);
if (nextToken) {
logger.debug(lt.foundMoreEKSClusters(clusters.length));
listAllClusters(nextToken);
}
else {
resolve(clusterList);
}
});
}
catch (error) {
resolve([]);
}
};
listAllClusters();
});
const describeCluster = async (eks, name) => new Promise(resolve => {
const descClusterOpts = {
name,
};
try {
eks.describeCluster(descClusterOpts, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'eks:describeCluster',
err,
});
return resolve({});
}
const { cluster = {} } = data || {};
resolve(cluster);
});
}
catch (error) {
resolve({});
}
});
const getResourceTags = async (eks, arn) => new Promise(resolve => {
try {
eks.listTagsForResource({ resourceArn: arn }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'eks:listTagsForResource',
err,
});
return resolve({});
}
const { tags = {} } = data || {};
resolve(tags);
});
}
catch (error) {
resolve({});
}
});
exports.default = async ({ regions, config, }) => new Promise(async (resolve) => {
const eksData = [];
const regionPromises = [];
const clusterPromises = [];
const tagsPromises = [];
// get all clusters for all regions
regions.split(',').map(region => {
const eks = new eks_1.default({ ...config, region, endpoint });
const regionPromise = new Promise(async (resolveRegion) => {
const clusterList = await listClustersForRegion({
eks,
resolveRegion,
});
if (!isEmpty_1.default(clusterList)) {
eksData.push(...clusterList.map(cluster => ({
name: cluster,
region,
})));
}
resolveRegion();
});
regionPromises.push(regionPromise);
});
await Promise.all(regionPromises);
// get all attributes for each cluster
eksData.map(({ name, region }, idx) => {
const eks = new eks_1.default({ ...config, region, endpoint });
const clusterPromise = new Promise(async (resolveCluster) => {
const clusterAttributes = await describeCluster(eks, name);
eksData[idx] = {
...eksData[idx],
...clusterAttributes,
};
resolveCluster();
});
clusterPromises.push(clusterPromise);
});
await Promise.all(clusterPromises);
// get all tags for each cluster
eksData.map(({ arn, region }, idx) => {
const eks = new eks_1.default({ ...config, region, endpoint });
const tagsPromise = new Promise(async (resolveTags) => {
const envTags = await getResourceTags(eks, arn);
eksData[idx].Tags = envTags;
resolveTags();
});
tagsPromises.push(tagsPromise);
});
await Promise.all(tagsPromises);
errorLog.reset();
resolve(groupBy_1.default(eksData, 'region'));
});