org.xmlsh.aws.gradle.cloudformation.AmazonCloudFormationCreateStackTask.java Source code

Introduction

Here is the source code for org.xmlsh.aws.gradle.cloudformation.AmazonCloudFormationCreateStackTask.java

Source

/*
 * Copyright 2015-2016 David A. Lee. <dlee@calldei.com>
 *
 * This is a derived work from Classmethods, Inc.
 *
 * Current License Terms - Dual licensed under the following :
 *
 * "Simplified BSD License" included in license.txt
 *
 * The original work (derived from) license terms:  Apache License, Version 2.0
 * included in copyright.origin/LICENSE.TXT
 */
/*
 * TOTALLY BOGUS
 * Stupid
 */
package org.xmlsh.aws.gradle.cloudformation;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import lombok.Getter;
import lombok.Setter;

import org.gradle.api.GradleException;
import org.gradle.api.internal.ConventionTask;
import org.gradle.api.tasks.TaskAction;

import com.amazonaws.AmazonServiceException;
import com.amazonaws.services.cloudformation.AmazonCloudFormation;
import com.amazonaws.services.cloudformation.model.Capability;
import com.amazonaws.services.cloudformation.model.CreateStackRequest;
import com.amazonaws.services.cloudformation.model.CreateStackResult;
import com.amazonaws.services.cloudformation.model.DeleteStackRequest;
import com.amazonaws.services.cloudformation.model.DescribeStacksRequest;
import com.amazonaws.services.cloudformation.model.DescribeStacksResult;
import com.amazonaws.services.cloudformation.model.OnFailure;
import com.amazonaws.services.cloudformation.model.Parameter;
import com.amazonaws.services.cloudformation.model.Stack;
import com.amazonaws.services.cloudformation.model.Tag;
import com.amazonaws.services.cloudformation.model.UpdateStackRequest;
import com.amazonaws.services.cloudformation.model.UpdateStackResult;

public class AmazonCloudFormationCreateStackTask extends ConventionTask {

    @Getter
    @Setter
    private String stackName;

    @Getter
    @Setter
    private List<Parameter> cfnStackParams = new ArrayList<>();

    @Getter
    @Setter
    private List<Tag> tags = new ArrayList<>();

    @Getter
    @Setter
    private boolean disableRollback;

    @Getter
    @Setter
    private String onFailure;

    @Getter
    @Setter
    private boolean capabilityIam;

    @Getter
    @Setter
    private String templateBody;

    @Getter
    @Setter
    private boolean usePreviousTemplate = false;

    @Getter
    @Setter
    private List<String> stableStatuses = Arrays.asList("CREATE_COMPLETE", "ROLLBACK_COMPLETE", "UPDATE_COMPLETE",
            "UPDATE_ROLLBACK_COMPLETE");

    public AmazonCloudFormationCreateStackTask() {
        setDescription("Create / Migrate cfn stack.");
        setGroup("AWS");
    }

    @TaskAction
    public void createStack() throws InterruptedException {
        // to enable conventionMappings feature
        String stackName = getStackName();
        if (stackName == null)
            throw new GradleException("stackName is not specified");

        AmazonCloudFormationPluginExtension ext = getProject().getExtensions()
                .getByType(AmazonCloudFormationPluginExtension.class);
        AmazonCloudFormation cfn = ext.getClient();

        try {
            DescribeStacksResult describeStackResult = cfn
                    .describeStacks(new DescribeStacksRequest().withStackName(stackName));
            Stack stack = describeStackResult.getStacks().get(0);
            throw new GradleException("Stack exists. invalid status for create: " + stack.getStackStatus());

        } catch (AmazonServiceException e) {
            if (e.getMessage().contains("does not exist")) {
                getLogger().info("stack {} not found", stackName);
                createStack(cfn);
            } else if (e.getMessage().contains("No updates are to be performed.")) {
                // ignore
            } else {
                throw e;
            }
        }
    }

    private void createStack(AmazonCloudFormation cfn) {
        // to enable conventionMappings feature
        String stackName = getStackName();
        List<Parameter> cfnStackParams = getCfnStackParams();

        getLogger().info("create stack: {}", stackName);

        CreateStackRequest req = new CreateStackRequest().withStackName(stackName).withParameters(cfnStackParams)
                .withTemplateBody(getTemplateBody()).withDisableRollback(isDisableRollback())
                .withOnFailure(getOnFailure()).withTags(getTags());
        if (isCapabilityIam()) {
            req.setCapabilities(Arrays.asList(Capability.CAPABILITY_IAM.toString()));
        }
        CreateStackResult createStackResult = cfn.createStack(req);
        getLogger().info("create requested: {}", createStackResult.getStackId());
    }
}