@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
144 lines (143 loc) • 5.41 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.getTags = exports.getRestApisForRegion = void 0;
const sdk_1 = __importDefault(require("@cloudgraph/sdk"));
const apigateway_1 = __importDefault(require("aws-sdk/clients/apigateway"));
const apigatewayv2_1 = __importDefault(require("aws-sdk/clients/apigatewayv2"));
const isEmpty_1 = __importDefault(require("lodash/isEmpty"));
const groupBy_1 = __importDefault(require("lodash/groupBy"));
const generateArns_1 = require("../../utils/generateArns");
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 services_1 = __importDefault(require("../../enums/services"));
const data_1 = require("../apiGatewayDomainName/data");
const lt = { ...logger_1.default };
const { logger } = sdk_1.default;
const MAX_REST_API = 500;
const serviceName = 'API Gateway Rest API';
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 getRestApisForRegion = async (apiGw) => new Promise(resolve => {
const restApiList = [];
const getRestApisOpts = {};
const listAllRestApis = (token) => {
getRestApisOpts.limit = MAX_REST_API;
if (token) {
getRestApisOpts.position = token;
}
try {
apiGw.getRestApis(getRestApisOpts, (err, data) => {
const { position, items = [] } = data || {};
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'apiGw:getRestApis',
err,
});
}
restApiList.push(...items);
if (position) {
listAllRestApis(position);
}
else {
resolve(restApiList);
}
});
}
catch (error) {
resolve([]);
}
};
listAllRestApis();
});
exports.getRestApisForRegion = getRestApisForRegion;
const getTags = async ({ apiGw, arn, }) => new Promise(resolve => {
try {
apiGw.getTags({ resourceArn: arn }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'apiGw:getTags',
err,
});
resolve({});
}
const { tags = {} } = data || {};
resolve(tags);
});
}
catch (error) {
resolve({});
}
});
exports.getTags = getTags;
exports.default = async ({ regions, config, rawData, }) => new Promise(async (resolve) => {
const apiGatewayData = [];
let domainNamesData = [];
const regionPromises = [];
const tagsPromises = [];
const existingData = rawData.find(({ name }) => name === services_1.default.apiGatewayDomainName)
?.data || {};
regions.split(',').map(region => {
const apiGw = new apigateway_1.default({
...config,
region,
endpoint,
...customRetrySettings,
});
const regionPromise = new Promise(async (resolveRegion) => {
if (isEmpty_1.default(existingData)) {
const apiGwV2 = new apigatewayv2_1.default({
...config,
region,
endpoint,
...customRetrySettings,
});
domainNamesData = await data_1.getDomainNamesForRegion(apiGwV2);
}
else {
domainNamesData = existingData[region] || [];
}
const restApiList = await exports.getRestApisForRegion(apiGw);
if (!isEmpty_1.default(restApiList)) {
apiGatewayData.push(...restApiList.map(restApi => ({
...restApi,
domainNames: domainNamesData?.map(domain => domain.DomainName) || [],
region,
})));
}
resolveRegion();
});
regionPromises.push(regionPromise);
});
await Promise.all(regionPromises);
logger.debug(lt.fetchedApiGatewayRestApis(apiGatewayData.length));
// get all tags for each rest api
apiGatewayData.map(({ id, region }, idx) => {
const apiGw = new apigateway_1.default({
...config,
region,
endpoint,
...customRetrySettings,
});
const tagsPromise = new Promise(async (resolveTags) => {
const arn = generateArns_1.apiGatewayRestApiArn({
restApiArn: generateArns_1.apiGatewayArn({ region }),
id,
});
apiGatewayData[idx].Tags = await exports.getTags({ apiGw, arn });
resolveTags();
});
tagsPromises.push(tagsPromise);
});
logger.debug(lt.gettingApiGatewayTags);
await Promise.all(tagsPromises);
errorLog.reset();
resolve(groupBy_1.default(apiGatewayData, 'region'));
});