@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
75 lines (74 loc) • 3.38 kB
JavaScript
;
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 flatMap_1 = __importDefault(require("lodash/flatMap"));
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 ecsCluster_1 = __importDefault(require("../ecsCluster"));
const lt = { ...logger_1.default };
const { logger } = sdk_1.default;
const serviceName = 'ECS container';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
exports.default = async ({ regions, config, }) => new Promise(async (resolve) => {
const ecsContainers = [];
const ecsClusterClass = new ecsCluster_1.default({ logger: sdk_1.default.logger });
const clusterResult = await ecsClusterClass.getData({
...config,
regions,
});
const ecsClusters = flatMap_1.default(clusterResult);
/**
* Get the instance arns of all the containers in each cluster
*/
let containerInstanceArns = await Promise.all(ecsClusters.map(async ({ clusterName: cluster, region }) => new Promise(resolveEcsData => new ecs_1.default({ ...config, region, endpoint }).listContainerInstances({ cluster }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ecs:listContainerInstances',
err,
});
}
if (isEmpty_1.default(data)) {
return resolveEcsData([]);
}
const { containerInstanceArns: containerInstances = [] } = data;
resolveEcsData({ cluster, containerInstances, region });
}))));
/**
* Get all of the containers for each instance arn
*/
const ecsContainerPromises = containerInstanceArns.map(async ({ cluster, containerInstances, region }) => new Promise(resolveEcsData => {
if (isEmpty_1.default(containerInstances))
return resolveEcsData();
new ecs_1.default({ ...config, region, endpoint }).describeContainerInstances({ cluster, containerInstances }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ecs:describeContainerInstances',
err,
});
}
if (isEmpty_1.default(data)) {
return resolveEcsData();
}
const { containerInstances = [] } = data;
logger.debug(lt.fetchedEcsContainers(containerInstances.length));
ecsContainers.push(...containerInstances.map(container => ({
region,
...container,
Tags: format_1.convertAwsTagsToTagMap(container.tags),
})));
resolveEcsData();
});
}));
await Promise.all(ecsContainerPromises);
errorLog.reset();
resolve(groupBy_1.default(ecsContainers, 'region'));
});