@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
123 lines (122 loc) • 4.39 kB
JavaScript
"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 flatMap_1 = __importDefault(require("lodash/flatMap"));
const route53_1 = __importDefault(require("aws-sdk/clients/route53"));
const logger_1 = __importDefault(require("../../properties/logger"));
const utils_1 = require("../../utils");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const data_1 = require("../route53HostedZone/data");
const constants_1 = require("../../config/constants");
const services_1 = __importDefault(require("../../enums/services"));
const regions_1 = require("../../enums/regions");
const lt = { ...logger_1.default };
const { logger } = sdk_1.default;
const serviceName = 'Route53 Records';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
const customRetrySettings = utils_1.setAwsRetryOptions({
baseDelay: constants_1.ROUTE_53_CUSTOM_DELAY,
});
const listRecordsForHostedZone = async ({ route53, recordData, HostedZoneId, resolveRecords, StartRecordName = '', StartRecordType = '', }) => {
let args = { HostedZoneId };
if (StartRecordName) {
args = {
...args,
StartRecordName,
StartRecordType,
};
}
return route53.listResourceRecordSets(args, async (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'route53:listResourceRecordSets',
err,
});
}
/**
* No records
*/
if (isEmpty_1.default(data)) {
return resolveRecords();
}
const { ResourceRecordSets: records = [], IsTruncated: truncated, NextRecordName: StartRecordName, NextRecordType: StartRecordType, } = data;
logger.debug(lt.fetchedRoute53ZonesRecords(records.length, HostedZoneId));
/**
* No records found
*/
if (isEmpty_1.default(records)) {
return resolveRecords();
}
/**
* Check to see if there are more
*/
if (truncated) {
listRecordsForHostedZone({
route53,
recordData,
HostedZoneId,
resolveRecords,
StartRecordName,
StartRecordType,
});
}
/**
* If there are not, then add the records to the zone's records
*/
for (const record of records) {
recordData.push({ ...record, HostedZoneId, region: regions_1.globalRegionName });
}
/**
* If this is the last page of data then return
*/
if (!truncated) {
return resolveRecords();
}
});
};
/**
* Route53 Records
*/
exports.default = async ({ config, rawData, }) => new Promise(async (resolve) => {
const zoneIds = [];
let hostedZonesData = [];
const recordData = [];
const route53 = new route53_1.default({
...config,
endpoint,
...customRetrySettings,
});
const existingData = flatMap_1.default(rawData.find(({ name }) => name === services_1.default.route53HostedZone)?.data) || [];
if (isEmpty_1.default(existingData)) {
// Refresh data
await data_1.listHostedZones(route53, zoneIds);
await data_1.getHostedZoneData(route53, zoneIds, hostedZonesData);
}
else {
// Uses existing data
hostedZonesData = existingData;
}
/**
* Step 3, get all the records for each zone
*/
const recordPromises = hostedZonesData.map(zone => {
const { Id: HostedZoneId } = zone;
const recordsPromise = new Promise(resolveRecords => listRecordsForHostedZone({
route53,
recordData,
HostedZoneId,
resolveRecords,
}));
return recordsPromise;
});
await Promise.all(recordPromises);
errorLog.reset();
logger.debug(lt.doneFetchingRoute53RecordsData);
resolve(groupBy_1.default(recordData, 'region'));
});