@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
140 lines (139 loc) • 5.42 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 ec2_1 = __importDefault(require("aws-sdk/clients/ec2"));
const format_1 = require("../../utils/format");
const logger_1 = __importDefault(require("../../properties/logger"));
const utils_1 = require("../../utils");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const lt = { ...logger_1.default };
const { logger } = sdk_1.default;
const serviceName = 'Managed Prefix List';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
const listManagedPrefixListData = async ({ ec2, resolveRegion, }) => new Promise(resolve => {
const managedPrefixLists = [];
let args = {};
const listManagedPrefixLists = (token) => {
if (token) {
args = { ...args, NextToken: token };
}
try {
ec2.describeManagedPrefixLists(args, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ec2:describeManagedPrefixLists',
err,
});
}
if (isEmpty_1.default(data)) {
return resolveRegion();
}
const { NextToken: nextToken, PrefixLists: prefixLists = [] } = data;
if (isEmpty_1.default(prefixLists)) {
return resolveRegion();
}
managedPrefixLists.push(...prefixLists);
logger.debug(lt.fetchedManagedPrefixLists(prefixLists.length));
if (nextToken) {
listManagedPrefixLists(nextToken);
}
else {
resolve(managedPrefixLists);
}
});
}
catch (error) {
resolve([]);
}
};
listManagedPrefixLists();
});
const getManagedPrefixListEntries = async ({ ec2, prefixListId, resolveEntry, }) => new Promise(resolve => {
const managedPrefixListEntries = [];
let args = {
PrefixListId: prefixListId,
};
const listManagedPrefixListEntries = (token) => {
if (token) {
args = { ...args, NextToken: token };
}
try {
ec2.getManagedPrefixListEntries(args, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ec2:getManagedPrefixListEntries',
err,
});
}
if (isEmpty_1.default(data)) {
return resolveEntry();
}
const { NextToken: nextToken, Entries: entries = [] } = data;
if (isEmpty_1.default(entries)) {
return resolveEntry();
}
managedPrefixListEntries.push(...entries);
logger.debug(lt.fetchedManagedPrefixListEntries(entries.length));
if (nextToken) {
listManagedPrefixListEntries(nextToken);
}
else {
resolve(managedPrefixListEntries);
}
});
}
catch (error) {
resolve([]);
}
};
listManagedPrefixListEntries();
});
exports.default = async ({ regions, config, }) => new Promise(async (resolve) => {
const managedPrefixListsResult = [];
const entriesPromises = [];
const regionPromises = regions.split(',').map(region => {
const ec2 = new ec2_1.default({ ...config, region, endpoint });
return new Promise(async (resolveRegion) => {
const managedPrefixLists = await listManagedPrefixListData({
ec2,
resolveRegion,
});
if (!isEmpty_1.default(managedPrefixLists)) {
for (const managedPrefixList of managedPrefixLists) {
managedPrefixListsResult.push({
...managedPrefixList,
region,
Tags: format_1.convertAwsTagsToTagMap(managedPrefixList.Tags),
});
}
}
resolveRegion();
});
});
await Promise.all(regionPromises);
managedPrefixListsResult.map(({ PrefixListId: prefixListId, region }, idx) => {
const ec2 = new ec2_1.default({ ...config, region, endpoint });
const entryPromise = new Promise(async (resolveEntry) => {
const entries = await getManagedPrefixListEntries({
ec2,
prefixListId,
resolveEntry,
});
managedPrefixListsResult[idx] = {
...managedPrefixListsResult[idx],
Entries: entries || [],
};
resolveEntry();
});
entriesPromises.push(entryPromise);
});
await Promise.all(entriesPromises);
errorLog.reset();
resolve(groupBy_1.default(managedPrefixListsResult, 'region'));
});