UNPKG

@csermet/multiprovider

Version:

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

80 lines (79 loc) 3.16 kB
"use strict"; 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 logger_1 = __importDefault(require("../../properties/logger")); const utils_1 = require("../../utils"); const format_1 = require("../../utils/format"); const errorLog_1 = __importDefault(require("../../utils/errorLog")); const lt = { ...logger_1.default }; const { logger } = sdk_1.default; const serviceName = 'TransitGateway'; const errorLog = new errorLog_1.default(serviceName); const endpoint = utils_1.initTestEndpoint(serviceName); const listTransitGatewaysData = async ({ ec2, resolveRegion, }) => new Promise(resolve => { const transitGatewayList = []; let args = {}; const listTransitGateways = (token) => { if (token) { args = { ...args, NextToken: token }; } try { ec2.describeTransitGateways(args, (err, data) => { if (err) { errorLog.generateAwsErrorLog({ functionName: 'ec2:describeTransitGateways', err, }); } if (isEmpty_1.default(data)) { return resolveRegion(); } const { NextToken: nextToken, TransitGateways: transitGateways = [], } = data; transitGatewayList.push(...transitGateways); logger.debug(lt.fetchedTransitGateways(transitGateways.length)); if (nextToken) { listTransitGateways(nextToken); } else { resolve(transitGatewayList); } }); } catch (error) { resolve([]); } }; listTransitGateways(); }); exports.default = async ({ regions, config, }) => new Promise(async (resolve) => { const transitGatewaysResult = []; const regionPromises = regions.split(',').map(region => { const ec2 = new ec2_1.default({ ...config, region, endpoint }); return new Promise(async (resolveRegion) => { // Get Transit Gateway Data const transitGateways = await listTransitGatewaysData({ ec2, resolveRegion, }); if (!isEmpty_1.default(transitGateways)) { for (const gateway of transitGateways) { transitGatewaysResult.push({ ...gateway, region, Tags: format_1.convertAwsTagsToTagMap(gateway.Tags), }); } } resolveRegion(); }); }); await Promise.all(regionPromises); errorLog.reset(); resolve(groupBy_1.default(transitGatewaysResult, 'region')); });