@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
102 lines (101 loc) • 3.85 kB
JavaScript
"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const ecs_1 = __importDefault(require("aws-sdk/clients/ecs"));
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 format_1 = require("../../utils/format");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const utils_1 = require("../../utils");
const lt = { ...logger_1.default };
const { logger } = sdk_1.default;
const serviceName = 'ECS cluster';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
exports.default = async ({ regions, config, }) => new Promise(async (resolve) => {
/**
* Get the arns of all the ECS Clusters
*/
const ecsClusterData = [];
const ecsClusterArns = [];
const regionPromises = [];
const listClusterArns = async ({ ecs, region, token: nextToken = '', resolveRegion, }) => {
let args = {};
if (nextToken) {
args = { ...args, nextToken };
}
return ecs.listClusters(args, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ecs:listClusters',
err,
});
}
/**
* No Cluster data for this region
*/
if (isEmpty_1.default(data)) {
return resolveRegion();
}
const { clusterArns, nextToken: token } = data;
logger.debug(lt.fetchedEcsClusters(clusterArns.length));
/**
* No Clusters Found
*/
if (isEmpty_1.default(clusterArns)) {
return resolveRegion();
}
/**
* Check to see if there are more
*/
if (token) {
listClusterArns({ region, token, ecs, resolveRegion });
}
/**
* Add the found Clusters to the ecsClusterArns
*/
ecsClusterArns.push({ region, clusterArns });
/**
* If this is the last page of data then return
*/
if (!token) {
resolveRegion();
}
});
};
regions.split(',').map(region => {
const ecs = new ecs_1.default({ ...config, region, endpoint });
const regionPromise = new Promise(resolveRegion => listClusterArns({ ecs, region, resolveRegion }));
regionPromises.push(regionPromise);
});
await Promise.all(regionPromises);
/**
* Get the details for each cluster
*/
const clusterPromises = ecsClusterArns.map(async ({ region, clusterArns: clusters }) => new Promise(resolveEcsData => new ecs_1.default({ ...config, region, endpoint }).describeClusters({ clusters }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ecs:describeClusters',
err,
});
}
if (isEmpty_1.default(data)) {
return resolveEcsData();
}
const { clusters = [] } = data;
logger.debug(lt.fetchedEcsClusters(clusters.length));
ecsClusterData.push(...clusters.map(cluster => ({
...cluster,
Tags: format_1.convertAwsTagsToTagMap(cluster.tags),
region,
})));
resolveEcsData();
})));
await Promise.all(clusterPromises);
errorLog.reset();
resolve(groupBy_1.default(ecsClusterData, 'region'));
});