com.tvarit.plugin.AutoScalingMojo.java Source code

Introduction

Here is the source code for com.tvarit.plugin.AutoScalingMojo.java

Source

/*
 * Tvarit is an AWS DevOps Automation Tool for JEE applications.
 * See http://www.tvarit.io
 *     Copyright (C) 2016. Sachin Dole.
 *
 *     This program is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     This program is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU General Public License for more details.
 *
 *     You should have received a copy of the GNU General Public License
 *     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package com.tvarit.plugin;

import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.services.cloudformation.AmazonCloudFormationClient;
import com.amazonaws.services.cloudformation.model.Capability;
import com.amazonaws.services.cloudformation.model.CreateStackRequest;
import com.amazonaws.services.cloudformation.model.Stack;
import com.amazonaws.services.ec2.AmazonEC2Client;
import com.amazonaws.services.s3.AmazonS3Client;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.Parameter;
import org.apache.maven.project.MavenProject;

import java.net.MalformedURLException;
import java.util.List;

@Mojo(name = "setup-autoscaling")
public class AutoScalingMojo extends AbstractMojo {

    @Parameter(required = true, readonly = true, property = "mySecretKey")
    private String secretKey;
    @Parameter(required = true, readonly = true, property = "myAccessKey")
    private String accessKey;
    @Parameter
    private String templateUrl;
    @Parameter(required = true)
    private String projectName;
    @Parameter(required = true)
    private String bucketName;
    @Parameter
    private String lambdaCodeS3Key;

    @Override
    public void execute() throws MojoExecutionException, MojoFailureException {
        getLog().debug("Starting " + this.getClass().getSimpleName() + " execution ");
        final BasicAWSCredentials awsCredentials = new BasicAWSCredentials(accessKey, secretKey);
        AmazonS3Client amazonS3Client = new AmazonS3Client(awsCredentials);
        final MavenProject project = (MavenProject) this.getPluginContext().getOrDefault("project", null);
        String lambdaCodeS3Bucket = this.bucketName;
        if (lambdaCodeS3Key == null) {
            lambdaCodeS3Key = new LambdaS3BucketKeyMaker().makeKey(project);
            lambdaCodeS3Bucket = "tvarit";
        }
        AmazonCloudFormationClient amazonCloudFormationClient = new AmazonCloudFormationClient(awsCredentials);
        AmazonEC2Client amazonEC2Client = new AmazonEC2Client(awsCredentials);
        List<com.amazonaws.services.cloudformation.model.Parameter> allParams = new AsgParameterMaker().make(
                amazonEC2Client, amazonCloudFormationClient, project, projectName, lambdaCodeS3Key,
                lambdaCodeS3Bucket);
        final String stackName = projectName + "-asg";
        if (templateUrl == null)
            try {
                templateUrl = new TemplateUrlMaker().makeUrl(project, "autoscaling.template").toString();
            } catch (MalformedURLException e) {
                throw new MojoExecutionException(
                        "Could not create default url for templates. Please open an issue on github.", e);
            }
        final CreateStackRequest createStackRequest = new CreateStackRequest()
                .withCapabilities(Capability.CAPABILITY_IAM).withStackName(stackName).withParameters(allParams)
                .withTemplateURL(templateUrl);
        final Stack stack = new StackMaker().makeStack(createStackRequest, amazonCloudFormationClient, getLog());
        new S3WarUploadEventToInvokeLambdaMaker().make(amazonS3Client, bucketName, stack);
        getLog().info("Finished completing stack");

    }
}