UNPKG

@csermet/multiprovider

Version:

cloud-graph provider plugin for AWS used to fetch AWS cloud data.

79 lines (78 loc) 3.07 kB
"use strict"; var __importDefault = (this && this.__importDefault) || function (mod) { return (mod && mod.__esModule) ? mod : { "default": mod }; }; Object.defineProperty(exports, "__esModule", { value: true }); const ec2_1 = __importDefault(require("aws-sdk/clients/ec2")); const sdk_1 = __importDefault(require("@cloudgraph/sdk")); 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 = 'NAT Gateway'; const errorLog = new errorLog_1.default(serviceName); const endpoint = utils_1.initTestEndpoint(serviceName); exports.default = async ({ regions, config, }) => new Promise(async (resolve) => { const natGatewayData = []; const regionPromises = []; const listNatGatewayData = async ({ ec2, region, token: NextToken = '', resolveRegion, }) => { let args = {}; if (NextToken) { args = { ...args, NextToken }; } return ec2.describeNatGateways(args, (err, data) => { if (err) { errorLog.generateAwsErrorLog({ functionName: 'ec2:describeNatGateways', err, }); } /** * No Nat Gateway data for this region */ if (isEmpty_1.default(data)) { return resolveRegion(); } const { NatGateways: natGateways, NextToken: token } = data; logger.debug(lt.fetchedNatGateways(natGateways.length)); /** * No Nat Gateways Found */ if (isEmpty_1.default(natGateways)) { return resolveRegion(); } /** * Check to see if there are more */ if (token) { listNatGatewayData({ region, token, ec2, resolveRegion }); } /** * Add the found Nat Gateways to the natGatewayData */ natGatewayData.push(...natGateways.map(({ Tags, ...nat }) => ({ ...nat, region, Tags: format_1.convertAwsTagsToTagMap(Tags), }))); /** * If this is the last page of data then return */ if (!token) { resolveRegion(); } }); }; regions.split(',').map(region => { const ec2 = new ec2_1.default({ ...config, region, endpoint }); const regionPromise = new Promise(resolveRegion => listNatGatewayData({ ec2, region, resolveRegion })); regionPromises.push(regionPromise); }); await Promise.all(regionPromises); errorLog.reset(); resolve(groupBy_1.default(natGatewayData, 'region')); });