@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
111 lines (110 loc) • 3.95 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 ec2_1 = __importDefault(require("aws-sdk/clients/ec2"));
const groupBy_1 = __importDefault(require("lodash/groupBy"));
const isEmpty_1 = __importDefault(require("lodash/isEmpty"));
const sdk_1 = __importDefault(require("@cloudgraph/sdk"));
const utils_1 = require("../../utils");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const format_1 = require("../../utils/format");
const logger_1 = __importDefault(require("../../properties/logger"));
/**
* EBS Snapshot
*/
const lt = { ...logger_1.default };
const { logger } = sdk_1.default;
const serviceName = 'EBS';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
const listEbsSnapshotAttribute = async ({ ec2, snapshotId, }) => new Promise(resolve => ec2.describeSnapshotAttribute({
Attribute: 'createVolumePermission',
SnapshotId: snapshotId,
}, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ec2:describeSnapshotAttribute',
err,
});
}
if (isEmpty_1.default(data)) {
return resolve([]);
}
return resolve(data?.CreateVolumePermissions ?? []);
}));
const listEbsSnapshots = async ({ ec2, region, nextToken: NextToken = '', ebsSnapshotData, resolveRegion, }) => {
let args = {
OwnerIds: ['self']
};
if (NextToken) {
args = { ...args, NextToken };
}
ec2.describeSnapshots(args, async (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'ec2:describeSnapshots',
err,
});
}
/**
* No EBS snapshot data for this region
*/
if (isEmpty_1.default(data)) {
return resolveRegion();
}
const { NextToken: nextToken, Snapshots: snapshots } = data || {};
logger.debug(lt.fetchedEbsSnapshots(snapshots.length));
/**
* No EBS Snapshot Found
*/
if (isEmpty_1.default(snapshots)) {
return resolveRegion();
}
/**
* Check to see if there are more
*/
if (nextToken) {
listEbsSnapshots({ region, nextToken, ec2, ebsSnapshotData, resolveRegion });
}
const ebsVolumes = [];
for (const { Tags, SnapshotId, ...snapshot } of snapshots) {
let snapshotAttributes = [];
if (SnapshotId) {
snapshotAttributes = await listEbsSnapshotAttribute({
ec2,
snapshotId: SnapshotId,
});
}
ebsVolumes.push({
...snapshot,
region,
SnapshotId,
Permissions: snapshotAttributes,
Tags: format_1.convertAwsTagsToTagMap(Tags),
});
}
ebsSnapshotData.push(...ebsVolumes);
/**
* If this is the last page of data then return the instances
*/
if (!nextToken) {
resolveRegion();
}
});
};
exports.default = async ({ regions, config, }) => new Promise(async (resolve) => {
const ebsSnapshotData = [];
const volumePromises = [];
// Get all the EBS data for each region with its snapshots
for (const region of regions.split(',')) {
const ec2 = new ec2_1.default({ ...config, region, endpoint });
volumePromises.push(new Promise(resolveRegion => listEbsSnapshots({ ec2, region, ebsSnapshotData, resolveRegion })));
}
logger.debug(lt.fetchingEbsSnapshotData);
// Fetch EBS volumes
await Promise.all(volumePromises);
errorLog.reset();
resolve(groupBy_1.default(ebsSnapshotData, 'region'));
});