@spotinst/spinnaker-deck/app/scripts/modules/amazon/src/securityGroup/securityGroup.transformer.js

Spinnaker-Deck service, forked with support to Spotinst

'use strict';

import { module } from 'angular';
import { groupBy } from 'lodash';

import { VpcReader } from '../vpc/VpcReader';

export const AMAZON_SECURITYGROUP_SECURITYGROUP_TRANSFORMER = 'spinnaker.amazon.securityGroup.transformer';
export const name = AMAZON_SECURITYGROUP_SECURITYGROUP_TRANSFORMER; // for backwards compatibility
module(AMAZON_SECURITYGROUP_SECURITYGROUP_TRANSFORMER, []).factory('awsSecurityGroupTransformer', function () {
  function normalizeSecurityGroup(securityGroup) {
    return VpcReader.listVpcs().then(addVpcNameToSecurityGroup(securityGroup));
  }

  function addVpcNameToSecurityGroup(securityGroup) {
    return function (vpcs) {
      const matches = vpcs.filter(function (test) {
        return test.id === securityGroup.vpcId;
      });
      securityGroup.vpcName = matches.length ? matches[0].name : '';
    };
  }

  function compress(securityGroups) {
    const grouped = groupBy(securityGroups, 'vpcId');
    Object.keys(grouped).forEach((vpcId) => {
      grouped[vpcId] = grouped[vpcId].map((g) => [g.name, g.id]);
    });
    return grouped;
  }

  function decompress(groupedGroups) {
    const flattened = [];
    Object.keys(groupedGroups).forEach((vpcId) => {
      groupedGroups[vpcId].forEach((g) => {
        flattened.push({ name: g[0], id: g[1], vpcId });
      });
    });
    return flattened;
  }

  return {
    normalizeSecurityGroup,
    compress,
    decompress,
    supportsCompression: true,
  };
});