SecurityGroupProvider.java

package org.zalando.stups.fullstop.plugin;

import com.amazonaws.AmazonServiceException;
import com.amazonaws.regions.Region;
import com.amazonaws.services.ec2.AmazonEC2Client;
import com.amazonaws.services.ec2.model.DescribeSecurityGroupsRequest;
import com.amazonaws.services.ec2.model.DescribeSecurityGroupsResult;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.zalando.stups.fullstop.aws.ClientProvider;

import java.util.List;

/**
 * @author jbellmann
 */
public class SecurityGroupProvider {

    public static final String INVALID_GROUP_NOT_FOUND = "InvalidGroup.NotFound";

    private final Logger log = LoggerFactory.getLogger(SecurityGroupProvider.class);

    private final ClientProvider clientProvider;

    public SecurityGroupProvider(final ClientProvider clientProvider) {
        this.clientProvider = clientProvider;
    }

    public String getSecurityGroup(final List<String> securityGroupIds, final Region region, final String accountId) {

        DescribeSecurityGroupsResult result = null;
        final ObjectMapper objectMapper = new ObjectMapper();
        String securityGroups = null;

        final AmazonEC2Client amazonEC2Client = clientProvider.getClient(AmazonEC2Client.class, accountId, region);

        if (amazonEC2Client == null) {
            throw new RuntimeException(
                    String.format(
                            "Somehow we could not create an Client with accountId: %s and region: %s", accountId,
                            region.toString()));
        } else {

            try {
                final DescribeSecurityGroupsRequest request = new DescribeSecurityGroupsRequest();
                request.setGroupIds(securityGroupIds);
                result = amazonEC2Client.describeSecurityGroups(request);
            } catch (final AmazonServiceException e) {
                if (e.getErrorCode().equals(INVALID_GROUP_NOT_FOUND)) {
                    log.warn(e.getMessage());
                } else {
                    log.error(e.getMessage());
                }
                return null;
            }

            try {
                securityGroups = objectMapper.writeValueAsString(result);
            } catch (final JsonProcessingException e) {
                log.error(e.getMessage());
                return null;
            }

            return securityGroups;
        }
    }

}