@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
132 lines (131 loc) • 4.89 kB
JavaScript
;
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 apigateway_1 = __importDefault(require("aws-sdk/clients/apigateway"));
const isEmpty_1 = __importDefault(require("lodash/isEmpty"));
const groupBy_1 = __importDefault(require("lodash/groupBy"));
const logger_1 = __importDefault(require("../../properties/logger"));
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const utils_1 = require("../../utils");
const constants_1 = require("../../config/constants");
const data_1 = require("../apiGatewayRestApi/data");
const services_1 = __importDefault(require("../../enums/services"));
const lt = { ...logger_1.default };
const { logger } = sdk_1.default;
const MAX_RESOURCES = 500;
const serviceName = 'API gateway Resource';
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 getResources = async ({ apiGw, restApiId }) => new Promise(resolve => {
const resourceList = [];
const getResourcesOpts = {
restApiId,
};
const listAllResources = (token) => {
getResourcesOpts.limit = MAX_RESOURCES;
getResourcesOpts.embed = ['methods'];
if (token) {
getResourcesOpts.position = token;
}
try {
apiGw.getResources(getResourcesOpts, (err, data) => {
const { position, items = [] } = data || {};
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'apiGw:getResources',
err,
});
}
/**
* No rest APIs for this region
*/
if (!isEmpty_1.default(data)) {
resourceList.push(...items);
}
if (position) {
listAllResources(position);
}
else {
resolve(resourceList);
}
});
}
catch (error) {
resolve([]);
}
};
listAllResources();
});
exports.default = async ({ regions, config, rawData, }) => new Promise(async (resolve) => {
const apiGatewayData = [];
const apiGatewayResources = [];
const regionPromises = [];
const additionalPromises = [];
const existingData = rawData.find(({ name }) => name === services_1.default.apiGatewayRestApi)?.data ||
{};
if (isEmpty_1.default(existingData)) {
// Refresh data
regions.split(',').map(region => {
const apiGw = new apigateway_1.default({
...config,
region,
endpoint,
...customRetrySettings,
});
const regionPromise = new Promise(async (resolveRegion) => {
const restApiList = await data_1.getRestApisForRegion(apiGw);
if (!isEmpty_1.default(restApiList)) {
apiGatewayData.push(...restApiList.map(restApi => ({
restApiId: restApi.id,
region,
})));
}
resolveRegion();
});
regionPromises.push(regionPromise);
return null;
});
await Promise.all(regionPromises);
}
else {
// Uses existing data
regions.split(',').map(region => {
if (!isEmpty_1.default(existingData[region])) {
apiGatewayData.push(...existingData[region].map(restApi => ({
restApiId: restApi.id,
region,
})));
}
return null;
});
}
apiGatewayData.map(({ restApiId, region }) => {
const apiGw = new apigateway_1.default({
...config,
region,
endpoint,
...customRetrySettings,
});
const additionalPromise = new Promise(async (resolveAdditional) => {
const resources = await getResources({ apiGw, restApiId });
apiGatewayResources.push(...resources.map(resource => ({
...resource,
restApiId,
region,
})));
resolveAdditional();
});
additionalPromises.push(additionalPromise);
return null;
});
await Promise.all(additionalPromises);
logger.debug(lt.fetchedApiGatewayResources(apiGatewayResources.length));
errorLog.reset();
resolve(groupBy_1.default(apiGatewayResources, 'region'));
});