@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
149 lines (148 loc) • 5.93 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 elasticache_1 = __importDefault(require("aws-sdk/clients/elasticache"));
const groupBy_1 = __importDefault(require("lodash/groupBy"));
const isEmpty_1 = __importDefault(require("lodash/isEmpty"));
const logger_1 = __importDefault(require("../../properties/logger"));
const format_1 = require("../../utils/format");
const utils_1 = require("../../utils");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const lt = { ...logger_1.default };
const { logger } = sdk_1.default;
const serviceName = 'ElastiCache cluster';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
const getElasticacheClusters = async (elastiCache) => new Promise(resolve => {
const clusterList = [];
const descClustersOpts = {};
const listAllClusters = (token) => {
if (token) {
descClustersOpts.Marker = token;
}
try {
elastiCache.describeCacheClusters(descClustersOpts, (err, data) => {
const { Marker, CacheClusters = [] } = data || {};
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'elastiCache:describeCacheClusters',
err,
});
}
clusterList.push(...CacheClusters);
if (Marker) {
listAllClusters(Marker);
}
else {
resolve(clusterList);
}
});
}
catch (error) {
resolve([]);
}
};
listAllClusters();
});
const getResourceTags = async (elastiCache, arn) => new Promise(resolve => {
try {
elastiCache.listTagsForResource({ ResourceName: arn }, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'elastiCache:listTagsForResource',
err,
});
return resolve({});
}
const { TagList: tags = [] } = data || {};
resolve(format_1.convertAwsTagsToTagMap(tags));
});
}
catch (error) {
resolve({});
}
});
const getCacheSubnetGroup = async ({ elastiCache, cacheSubnetGroupName, }) => new Promise(resolve => {
const args = {
CacheSubnetGroupName: cacheSubnetGroupName,
};
try {
elastiCache.describeCacheSubnetGroups(args, (err, data) => {
if (err) {
errorLog.generateAwsErrorLog({
functionName: 'elastiCache:describeCacheSubnetGroups',
err,
});
}
/**
* No cache subnet group for this region
*/
if (isEmpty_1.default(data)) {
return resolve({});
}
const { CacheSubnetGroups: cacheSubnetGroups } = data || {};
const result = cacheSubnetGroups && cacheSubnetGroups?.length > 0
? cacheSubnetGroups[0]
: {};
resolve(result);
});
}
catch (error) {
resolve({});
}
});
exports.default = async ({ regions, config, }) => new Promise(async (resolve) => {
const elastiCacheData = [];
const regionPromises = [];
const tagsPromises = [];
// Get all the clusters for the region
regions.split(',').map(region => {
const regionPromise = new Promise(async (resolveRegion) => {
const elastiCache = new elasticache_1.default({ ...config, region, endpoint });
let clusters = await getElasticacheClusters(elastiCache);
if (!isEmpty_1.default(clusters)) {
// Get cache subnet groups for each cluster
const cacheSubnetGroupNames = clusters.map(cluster => cluster.CacheSubnetGroupName);
if (!isEmpty_1.default(cacheSubnetGroupNames)) {
const cacheSubnetGroups = await Promise.all(cacheSubnetGroupNames.map(cacheSubnetGroupName => getCacheSubnetGroup({
elastiCache,
cacheSubnetGroupName,
})));
if (!isEmpty_1.default(cacheSubnetGroups)) {
clusters = clusters.map(cluster => {
const cacheSubnetGroup = cacheSubnetGroups.find((group) => group.CacheSubnetGroupName ===
cluster.CacheSubnetGroupName);
return {
...cluster,
CacheSubnetGroup: cacheSubnetGroup || {},
};
});
}
}
elastiCacheData.push(...clusters.map(cluster => ({
...cluster,
region,
})));
}
resolveRegion();
});
regionPromises.push(regionPromise);
});
await Promise.all(regionPromises);
logger.debug(lt.fetchedElasticacheClusters(elastiCacheData.length));
// get all tags for each cluster
elastiCacheData.map(({ ARN: arn, region }, idx) => {
const elastiCache = new elasticache_1.default({ ...config, region, endpoint });
const tagsPromise = new Promise(async (resolveTags) => {
elastiCacheData[idx].Tags = await getResourceTags(elastiCache, arn);
resolveTags();
});
tagsPromises.push(tagsPromise);
});
await Promise.all(tagsPromises);
errorLog.reset();
resolve(groupBy_1.default(elastiCacheData, 'region'));
});