@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
122 lines (121 loc) • 4.47 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 rds_1 = __importDefault(require("aws-sdk/clients/rds"));
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 = 'RDS DB cluster';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
const listClustersForRegion = async (rds) => new Promise(resolve => {
const clusterList = [];
const descClustersOpts = {};
const listAllClusters = (token) => {
if (token) {
descClustersOpts.Marker = token;
}
try {
rds.describeDBClusters(descClustersOpts, (err, data) => {
const { Marker, DBClusters = [] } = data || {};
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'rds:describeDBClusters',
err,
});
}
clusterList.push(...DBClusters);
if (Marker) {
listAllClusters(Marker);
}
else {
resolve(clusterList);
}
});
}
catch (error) {
resolve([]);
}
};
listAllClusters();
});
const describeDBSubnetGroups = async (rds, DBSubnetGroupName) => new Promise(resolve => {
try {
rds.describeDBSubnetGroups({ DBSubnetGroupName }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'rds:describeDBSubnetGroups',
err,
});
return resolve([]);
}
if (!isEmpty_1.default(data)) {
resolve(data.DBSubnetGroups);
}
});
}
catch (error) {
resolve([]);
}
});
const getResourceTags = async (rds, arn) => new Promise(resolve => {
try {
rds.listTagsForResource({ ResourceName: arn }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'rds:listTagsForResource',
err,
});
return resolve({});
}
const { TagList: tags = [] } = data || {};
resolve(format_1.convertAwsTagsToTagMap(tags));
});
}
catch (error) {
resolve({});
}
});
exports.default = async ({ regions, config, }) => new Promise(async (resolve) => {
const rdsData = [];
const regionPromises = [];
const tagsPromises = [];
// Get all the clusters for the region
regions.split(',').map(region => {
const regionPromise = new Promise(async (resolveRegion) => {
const rds = new rds_1.default({ ...config, region, endpoint });
const clusters = await listClustersForRegion(rds);
if (!isEmpty_1.default(clusters)) {
rdsData.push(...(await Promise.all(clusters.map(async (cluster) => ({
...cluster,
dbSubnetGroups: await describeDBSubnetGroups(rds, cluster.DBSubnetGroup),
region,
})))));
}
resolveRegion();
});
regionPromises.push(regionPromise);
});
await Promise.all(regionPromises);
logger.debug(lt.fetchedRdsClusters(rdsData.length));
// get all tags for each cluster
rdsData.map(({ DBClusterArn, region }, idx) => {
const rds = new rds_1.default({ ...config, region, endpoint });
const tagsPromise = new Promise(async (resolveTags) => {
rdsData[idx].Tags = await getResourceTags(rds, DBClusterArn);
resolveTags();
});
tagsPromises.push(tagsPromise);
});
await Promise.all(tagsPromises);
errorLog.reset();
resolve(groupBy_1.default(rdsData, 'region'));
});