@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
130 lines (129 loc) • 4.71 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const sdk_1 = __importDefault(require("@cloudgraph/sdk"));
const groupBy_1 = __importDefault(require("lodash/groupBy"));
const isEmpty_1 = __importDefault(require("lodash/isEmpty"));
const cloudwatch_1 = __importDefault(require("aws-sdk/clients/cloudwatch"));
const logger_1 = __importDefault(require("../../properties/logger"));
const utils_1 = require("../../utils");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const format_1 = require("../../utils/format");
const constants_1 = require("../../config/constants");
/**
* Cloudwatch
*/
const lt = { ...logger_1.default };
const MAX_ITEMS = 100;
const { logger } = sdk_1.default;
const serviceName = 'Cloudwatch';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
const customRetrySettings = utils_1.setAwsRetryOptions({
baseDelay: constants_1.CLOUDWATCH_CUSTOM_DELAY,
});
const listMetricAlarmsForRegion = async ({ cloudwatch, resolveRegion, }) => new Promise(resolve => {
const metricAlarmsList = [];
const listMetricAlarmsOpts = {};
const listAllAlarms = (token) => {
listMetricAlarmsOpts.MaxRecords = MAX_ITEMS;
if (token) {
listMetricAlarmsOpts.NextToken = token;
}
try {
cloudwatch.describeAlarms(listMetricAlarmsOpts, (err, data) => {
const { NextToken: nextToken, MetricAlarms: metricAlarms } = data || {};
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'cloudwatch:describeAlarms',
err,
});
}
/**
* No metrics for this region
*/
if (isEmpty_1.default(data)) {
return resolveRegion();
}
metricAlarmsList.push(...metricAlarms);
if (nextToken) {
logger.debug(lt.foundMoreCloudwatchAlarms(metricAlarms.length));
listAllAlarms(nextToken);
}
else {
resolve(metricAlarmsList);
}
});
}
catch (error) {
resolve([]);
}
};
listAllAlarms();
});
const getResourceTags = async (cloudwatch, arn) => new Promise(resolve => {
try {
cloudwatch.listTagsForResource({ ResourceARN: arn }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'cloudwatch:listTagsForResource',
err,
});
return resolve({});
}
const { Tags = [] } = data || {};
resolve(format_1.convertAwsTagsToTagMap(Tags));
});
}
catch (error) {
resolve({});
}
});
exports.default = async ({ regions, config, }) => new Promise(async (resolve) => {
const cloudwatchData = [];
const regionPromises = [];
const tagsPromises = [];
// get all metrics for all regions
regions.split(',').map(region => {
const cloudwatch = new cloudwatch_1.default({
...config,
region,
endpoint,
});
const regionPromise = new Promise(async (resolveRegion) => {
const metricAlarmsList = await listMetricAlarmsForRegion({
cloudwatch,
resolveRegion,
});
cloudwatchData.push(...metricAlarmsList.map(metricAlarm => ({
...metricAlarm,
region,
})));
resolveRegion();
});
regionPromises.push(regionPromise);
});
logger.debug(lt.gettingCloudwatchAlarms);
await Promise.all(regionPromises);
// get all tags for each environment
cloudwatchData.map(({ AlarmArn, region }, idx) => {
const cloudwatch = new cloudwatch_1.default({
...config,
region,
endpoint,
...customRetrySettings,
});
const tagsPromise = new Promise(async (resolveTags) => {
const envTags = await getResourceTags(cloudwatch, AlarmArn);
cloudwatchData[idx].Tags = envTags;
resolveTags();
});
tagsPromises.push(tagsPromise);
});
logger.debug(lt.gettingCloudwatchAlarmTags);
await Promise.all(tagsPromises);
errorLog.reset();
resolve(groupBy_1.default(cloudwatchData, 'region'));
});