UNPKG

@csermet/multiprovider

Version:

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

129 lines (128 loc) 4.97 kB
"use strict"; var __importDefault = (this && this.__importDefault) || function (mod) { return (mod && mod.__esModule) ? mod : { "default": mod }; }; Object.defineProperty(exports, "__esModule", { value: true }); exports.getCertificatesForRegion = void 0; const sdk_1 = __importDefault(require("@cloudgraph/sdk")); const acm_1 = __importDefault(require("aws-sdk/clients/acm")); const isEmpty_1 = __importDefault(require("lodash/isEmpty")); const groupBy_1 = __importDefault(require("lodash/groupBy")); const logger_1 = __importDefault(require("../../properties/logger")); const utils_1 = require("../../utils"); const errorLog_1 = __importDefault(require("../../utils/errorLog")); const constants_1 = require("../../config/constants"); const lt = { ...logger_1.default }; const { logger } = sdk_1.default; const MAX_CERTIFICATES = 500; const serviceName = 'ACM'; const errorLog = new errorLog_1.default(serviceName); const endpoint = utils_1.initTestEndpoint(serviceName); const customRetrySettings = utils_1.setAwsRetryOptions({ baseDelay: constants_1.API_GATEWAY_CUSTOM_DELAY, }); const getCertificatesForRegion = async (acm) => new Promise(async (resolve) => { const certificateSummaryList = []; const listCertificatesOpts = {}; const listAllCertificates = (token) => { listCertificatesOpts.MaxItems = MAX_CERTIFICATES; if (token) { listCertificatesOpts.NextToken = token; } try { acm.listCertificates(listCertificatesOpts, (err, data) => { if (err) { errorLog.generateAwsErrorLog({ functionName: 'acm:listCertificates', err, }); } if (isEmpty_1.default(data)) { return resolve([]); } const { NextToken: nextToken, CertificateSummaryList: items = [] } = data || {}; if (isEmpty_1.default(items)) { return resolve([]); } logger.debug(lt.fetchedAcmCertificates(items.length)); certificateSummaryList.push(...items); if (nextToken) { listAllCertificates(nextToken); } else { resolve(certificateSummaryList); } }); } catch (error) { resolve([]); } }; listAllCertificates(); }); exports.getCertificatesForRegion = getCertificatesForRegion; const getTagsForCertificate = (acm, certificateArn) => new Promise(resolve => { const args = { CertificateArn: certificateArn }; const listTags = () => { try { acm.listTagsForCertificate(args, (err, data) => { if (err) { errorLog.generateAwsErrorLog({ functionName: 'acm:listTagsForCertificate', err, }); } if (isEmpty_1.default(data)) { return resolve({ certificateArn, tags: [], }); } const { Tags: tags = [] } = data || {}; resolve({ certificateArn, tags }); }); } catch (error) { resolve({ certificateArn, tags: [], }); } }; listTags(); }); exports.default = async ({ regions, config, account, }) => new Promise(async (resolve) => { const acmResult = []; const regionPromises = regions.split(',').map(region => { const acm = new acm_1.default({ ...config, region, endpoint, ...customRetrySettings, }); return new Promise(async (resolveAcmData) => { // Get ACM certificate summaries const certificates = await exports.getCertificatesForRegion(acm); const tagsPromises = certificates.map(({ CertificateArn: certificateArn }) => getTagsForCertificate(acm, certificateArn)); const tagsData = await Promise.all(tagsPromises); if (!isEmpty_1.default(certificates)) { for (const certificate of certificates) { acmResult.push({ ...certificate, Tags: tagsData?.find(t => t.certificateArn === certificate.CertificateArn) ?.tags.reduce((tagMap, { Key, Value }) => { tagMap[Key] = Value; return tagMap; }, {}), region, account, }); } } resolveAcmData(); }); }); await Promise.all(regionPromises); errorLog.reset(); resolve(groupBy_1.default(acmResult, 'region')); });