jp.classmethod.aws.gradle.elasticbeanstalk.AWSElasticBeanstalkCreateEnvironmentTask.java Source code

Introduction

Here is the source code for jp.classmethod.aws.gradle.elasticbeanstalk.AWSElasticBeanstalkCreateEnvironmentTask.java

Source

/*
 * Copyright 2015-2016 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package jp.classmethod.aws.gradle.elasticbeanstalk;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import lombok.Getter;
import lombok.Setter;

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

import com.amazonaws.services.elasticbeanstalk.AWSElasticBeanstalk;
import com.amazonaws.services.elasticbeanstalk.model.CreateEnvironmentRequest;
import com.amazonaws.services.elasticbeanstalk.model.CreateEnvironmentResult;
import com.amazonaws.services.elasticbeanstalk.model.DescribeEnvironmentsRequest;
import com.amazonaws.services.elasticbeanstalk.model.DescribeEnvironmentsResult;
import com.amazonaws.services.elasticbeanstalk.model.Tag;
import com.amazonaws.services.elasticbeanstalk.model.UpdateEnvironmentRequest;

public class AWSElasticBeanstalkCreateEnvironmentTask extends ConventionTask { // NOPMD

    @Getter
    @Setter
    private String appName;

    @Getter
    @Setter
    private String envName;

    @Getter
    @Setter
    private String envDesc = "";

    @Getter
    @Setter
    private String cnamePrefix = java.util.UUID.randomUUID().toString();

    @Getter
    @Setter
    private String templateName;

    @Getter
    @Setter
    private String versionLabel;

    @Getter
    @Setter
    private Tier tier = Tier.WebServer;

    @Getter
    @Setter
    private Map<String, String> tags = new HashMap<String, String>();

    public AWSElasticBeanstalkCreateEnvironmentTask() {
        setDescription("Create/Migrate ElasticBeanstalk Environment.");
        setGroup("AWS");
    }

    @TaskAction
    public void createEnvironment() { // NOPMD
        // to enable conventionMappings feature
        String appName = getAppName();
        String envName = getEnvName();
        String envDesc = getEnvDesc();
        String cnamePrefix = getCnamePrefix();
        String templateName = getTemplateName();
        String versionLabel = getVersionLabel();
        Tier tier = getTier();
        Map<String, String> tags = getTags();

        AwsBeanstalkPluginExtension ext = getProject().getExtensions().getByType(AwsBeanstalkPluginExtension.class);
        AWSElasticBeanstalk eb = ext.getClient();

        DescribeEnvironmentsResult der = eb.describeEnvironments(new DescribeEnvironmentsRequest()
                .withApplicationName(appName).withEnvironmentNames(envName).withIncludeDeleted(false));

        List<Tag> ebTags = tags.entrySet().stream().map(entry -> {
            Tag t = new Tag();
            t.setKey(entry.getKey());
            t.setValue(entry.getValue());
            return t;
        }).collect(Collectors.toList());

        if (der.getEnvironments() == null || der.getEnvironments().isEmpty()) {
            CreateEnvironmentRequest req = new CreateEnvironmentRequest().withApplicationName(appName)
                    .withEnvironmentName(envName).withDescription(envDesc).withTemplateName(templateName)
                    .withVersionLabel(versionLabel);

            if (tier != null) {
                req.withTier(tier.toEnvironmentTier());
                if (tier == Tier.WebServer) {
                    req.withCNAMEPrefix(cnamePrefix);
                }
            }

            if (ebTags != null && !ebTags.isEmpty()) {
                req.withTags(ebTags);
            }

            CreateEnvironmentResult result = eb.createEnvironment(req);
            getLogger().info("environment {} @ {} ({}) created", envName, appName, result.getEnvironmentId());
        } else {
            String environmentId = der.getEnvironments().get(0).getEnvironmentId();

            // Only these two values are required to deploy the a application
            UpdateEnvironmentRequest req = new UpdateEnvironmentRequest().withEnvironmentId(environmentId)
                    .withVersionLabel(versionLabel);

            // All other variables are optional and refer to the environment
            if (isNotBlank(envName)) {
                req.withEnvironmentName(envName);
            }
            if (isNotBlank(envDesc)) {
                req.withDescription(envDesc);
            }
            if (isNotBlank(templateName)) {
                req.withTemplateName(templateName);
            }

            eb.updateEnvironment(req);

            getLogger().info("environment {} @ {} ({}) updated", envName, appName, environmentId);
        }
    }

    // simple helper method to not include apache commons lang's StringUtils only for this
    private boolean isNotBlank(String str) {
        return str != null && !str.trim().isEmpty();
    }
}