UNPKG

@csermet/multiprovider

Version:

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

80 lines (79 loc) 3.04 kB
"use strict"; var __importDefault = (this && this.__importDefault) || function (mod) { return (mod && mod.__esModule) ? mod : { "default": mod }; }; Object.defineProperty(exports, "__esModule", { value: true }); const groupBy_1 = __importDefault(require("lodash/groupBy")); const isEmpty_1 = __importDefault(require("lodash/isEmpty")); const ec2_1 = __importDefault(require("aws-sdk/clients/ec2")); const sdk_1 = __importDefault(require("@cloudgraph/sdk")); const logger_1 = __importDefault(require("../../properties/logger")); // import { Tag } from '../../types/generated' 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 = 'Subnet'; const errorLog = new errorLog_1.default(serviceName); const endpoint = utils_1.initTestEndpoint(serviceName); exports.default = ({ regions, config, }) => new Promise(async (resolve) => { const subnetData = []; const regionPromises = []; const listSubnetData = async ({ ec2, region, token: NextToken = '', resolveSubnet, }) => { let args = {}; if (NextToken) { args = { ...args, NextToken }; } return ec2.describeSubnets(args, (err, data) => { if (err) { errorLog.generateAwsErrorLog({ functionName: 'ec2:describeSubnets', err, }); } /** * No Subnet data for this region */ if (isEmpty_1.default(data)) { return resolveSubnet(); } const { Subnets: subnets, NextToken: token } = data; logger.debug(lt.fetchedSubnets(subnets.length)); /** * No subnets Found */ if (isEmpty_1.default(subnets)) { return resolveSubnet(); } /** * Check to see if there are more */ if (token) { listSubnetData({ region, token, ec2, resolveSubnet }); } /** * Add the found subnets to the subnetData */ subnetData.push(...subnets.map(({ Tags, ...subnet }) => ({ ...subnet, region, Tags: format_1.convertAwsTagsToTagMap(Tags), }))); /** * If this is the last page of data then return */ if (!token) { resolveSubnet(); } }); }; regions.split(',').map(region => { const ec2 = new ec2_1.default({ ...config, region, endpoint }); const regionPromise = new Promise(resolveSubnet => listSubnetData({ region, ec2, resolveSubnet })); regionPromises.push(regionPromise); }); await Promise.all(regionPromises); errorLog.reset(); resolve(groupBy_1.default(subnetData, 'region')); });