@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
77 lines (76 loc) • 3.15 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 task';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
exports.default = async ({ regions, config, }) => new Promise(async (resolve) => {
const ecsTasks = [];
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 arns of all the tasks
*/
let ecsTaskArns = await Promise.all(ecsClusters.map(async ({ clusterName: cluster, region }) => new Promise(resolveEcsData => new ecs_1.default({ ...config, region, endpoint }).listTasks({ cluster }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ecs:listTasks',
err,
});
}
if (isEmpty_1.default(data)) {
return resolveEcsData([]);
}
const { taskArns = [] } = data;
resolveEcsData({ region, cluster, taskArns });
}))));
/**
* Check to make sure there are tasks before we try and search for them
*/
ecsTaskArns = ecsTaskArns
.flat()
.filter(({ taskArns }) => !isEmpty_1.default(taskArns));
/**
* Get all the details for each task
*/
const ecsTaskPromises = ecsTaskArns.map(async ({ region, taskArns: tasks, cluster }) => new Promise(resolveEcsData => new ecs_1.default({ ...config, region, endpoint }).describeTasks({ tasks, cluster }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ecs:describeTasks',
err,
});
}
if (isEmpty_1.default(data)) {
return resolveEcsData();
}
const { tasks = [] } = data;
logger.debug(lt.fetchedEcsTasks(tasks.length));
ecsTasks.push(...tasks.map(task => ({
region,
...task,
Tags: format_1.convertAwsTagsToTagMap(task.tags),
})));
resolveEcsData();
})));
await Promise.all(ecsTaskPromises);
errorLog.reset();
resolve(groupBy_1.default(ecsTasks, 'region'));
});