@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
102 lines (101 loc) • 3.82 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 utils_1 = require("../../utils");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const logger_1 = __importDefault(require("../../properties/logger"));
const format_1 = require("../../utils/format");
const { logger } = sdk_1.default;
const lt = { ...logger_1.default };
const serviceName = 'RDS DB instance';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
const listDBInstancesForRegion = async (rds) => new Promise(resolve => {
const dbInstanceList = [];
const descDBInstancesOpts = {};
const listAllDBInstances = (token) => {
if (token) {
descDBInstancesOpts.Marker = token;
}
try {
rds.describeDBInstances(descDBInstancesOpts, (err, data) => {
const { Marker, DBInstances = [] } = data || {};
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'rds:describeDBInstances',
err,
});
}
dbInstanceList.push(...DBInstances);
if (Marker) {
listAllDBInstances(Marker);
}
else {
resolve(dbInstanceList);
}
});
}
catch (error) {
resolve([]);
}
};
listAllDBInstances();
});
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 instances for the region
regions.split(',').map(region => {
const regionPromise = new Promise(async (resolveRegion) => {
const rds = new rds_1.default({ ...config, region, endpoint });
const instances = await listDBInstancesForRegion(rds);
if (!isEmpty_1.default(instances)) {
rdsData.push(...instances.map(instance => ({
...instance,
region,
})));
}
resolveRegion();
});
regionPromises.push(regionPromise);
});
await Promise.all(regionPromises);
logger.debug(lt.fetchedRdsInstances(rdsData.length));
// get all tags for each instance
rdsData.map(({ DBInstanceArn, region }, idx) => {
const rds = new rds_1.default({ ...config, region, endpoint });
const tagsPromise = new Promise(async (resolveTags) => {
rdsData[idx].Tags = await getResourceTags(rds, DBInstanceArn);
resolveTags();
});
tagsPromises.push(tagsPromise);
});
await Promise.all(tagsPromises);
errorLog.reset();
resolve(groupBy_1.default(rdsData, 'region'));
});