UNPKG

@csermet/multiprovider

Version:

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

92 lines (91 loc) 3.26 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 errorLog_1 = __importDefault(require("../../utils/errorLog")); const format_1 = require("../../utils/format"); const lt = { ...logger_1.default }; const { logger } = sdk_1.default; const serviceName = 'ClientVpnEndpoint'; const errorLog = new errorLog_1.default(serviceName); const endpoint = utils_1.initTestEndpoint(serviceName); /** * Client Vpn Endpoint */ const listClientVpnEndpoints = async ({ ec2, region, nextToken: NextToken = '', clientVpnEndpointData, resolveRegion, }) => { let args = {}; if (NextToken) { args = { ...args, NextToken }; } ec2.describeClientVpnEndpoints(args, (err, data) => { if (err) { errorLog.generateAwsErrorLog({ functionName: 'ec2:describeClientVpnEndpoints', err, }); } /** * No Client Vpn Endpoint data for this region */ if (isEmpty_1.default(data)) { return resolveRegion(); } const { ClientVpnEndpoints: endpoints, NextToken: nextToken } = data || {}; logger.debug(lt.fetchedClientVpnEndpoints(endpoints.length)); /** * No Client Vpn Endpoints Found */ if (isEmpty_1.default(endpoints)) { return resolveRegion(); } /** * Check to see if there are more */ if (nextToken) { listClientVpnEndpoints({ ec2, region, nextToken, clientVpnEndpointData, resolveRegion, }); } /** * Add the found Client Vpn Endpoints to the clientVpnEndpointData */ clientVpnEndpointData.push(...endpoints.map(clientVpnEndpoint => ({ ...clientVpnEndpoint, region, Tags: format_1.convertAwsTagsToTagMap(clientVpnEndpoint.Tags), }))); /** * If this is the last page of data then return */ if (!nextToken) { resolveRegion(); } }); }; exports.default = async ({ regions, config, }) => new Promise(async (resolve) => { const clientVpnEndpointData = []; // Get all the Client Vpn Endpoint data for each region const regionPromises = regions.split(',').map(region => { const ec2 = new ec2_1.default({ ...config, region, endpoint }); return new Promise(resolveRegion => listClientVpnEndpoints({ ec2, region, clientVpnEndpointData, resolveRegion, })); }); await Promise.all(regionPromises); errorLog.reset(); resolve(groupBy_1.default(clientVpnEndpointData, 'region')); });