com.erigir.maven.plugin.AbstractSeedyMojo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of seedy-maven-plugin Show documentation
Seedy (CD) is a maven plugin for automating continuous deployment to Amazon Web Services
There is a newer version: 0.71
package com.erigir.maven.plugin;

import com.amazonaws.auth.AWSCredentials;
import com.amazonaws.auth.BasicSessionCredentials;
import com.amazonaws.auth.DefaultAWSCredentialsProviderChain;
import com.amazonaws.services.s3.AmazonS3;
import com.amazonaws.services.s3.AmazonS3Client;
import com.amazonaws.services.securitytoken.AWSSecurityTokenService;
import com.amazonaws.services.securitytoken.AWSSecurityTokenServiceClient;
import com.amazonaws.services.securitytoken.model.AssumeRoleRequest;
import com.amazonaws.services.securitytoken.model.AssumeRoleResult;
import org.apache.maven.plugin.Mojo;
import org.apache.maven.plugin.MojoExecutionException;

import java.text.SimpleDateFormat;
import java.util.Date;
/**
 Copyright 2014 Christopher Weiss

 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.
 **/


/**
 * @author cweiss
 * @since .01
 */
public abstract class AbstractSeedyMojo extends org.apache.maven.plugin.AbstractMojo {
    private AWSCredentials credentials;

    protected synchronized final AWSCredentials credentials()
            throws MojoExecutionException {
        if (credentials == null) {
            getLog().info("Fetching AWS Credentials");
            credentials = new DefaultAWSCredentialsProviderChain().getCredentials();
            if (credentials == null) {
                throw new MojoExecutionException("Couldn't fetch credentials - either set aws.accessKeyId and aws.secretKey or (better) grant this machine an IAM role");
            }

            String assumedRoleARN = getAssumedRoleArn();
            if (assumedRoleARN!=null && assumedRoleARN.trim().length()>0)
            {
                getLog().info("Assumed Role Arn supplied - attempting to assume that role");
                assumedRoleARN = assumedRoleARN.trim();
                String assumedRoleExternalId = getAssumedRoleExternalId();
                if (assumedRoleExternalId==null || assumedRoleExternalId.trim().length()==0)
                {
                    throw new MojoExecutionException("If you set an assumed role arn you must also set assumed role external id");
                }
                assumedRoleExternalId=assumedRoleExternalId.trim();
                String sessionName = "seedy-"+System.currentTimeMillis();
                AssumeRoleRequest arr = new AssumeRoleRequest()
                        .withRoleSessionName(sessionName)
                        .withRoleArn(assumedRoleARN)
                        .withExternalId(assumedRoleExternalId);

                AWSSecurityTokenService sts = new AWSSecurityTokenServiceClient(credentials);
                AssumeRoleResult result = sts.assumeRole(arr);
                getLog().info("Successfully assumed role - using new credentials");
                credentials = new BasicSessionCredentials(result.getCredentials().getAccessKeyId(), result.getCredentials().getSecretAccessKey(), result.getCredentials().getSessionToken());
            }


            getLog().info("Using credentials : " + obscure(credentials.getAWSAccessKeyId(), 2) + " / " + obscure(credentials.getAWSSecretKey(), 2));
        }
        return credentials;
    }

    public abstract String getAssumedRoleArn();
    public abstract String getAssumedRoleExternalId();

    protected AmazonS3 s3()
            throws MojoExecutionException {
        return new AmazonS3Client(credentials());
    }

    protected final int buildNumber(Integer def)
            throws MojoExecutionException {
        Integer rval = null;
        String env = propertyOrEnvVariable("BUILD_NUMBER");
        if (env == null) {
            if (def == null) {
                throw new MojoExecutionException("No environment variable 'BUILD_NUMBER' found and no default set");
            } else {
                getLog().info("No environment variable 'BUILD_NUMBER' found, using default (Jenkins would set this)");
                rval = def;
            }
        } else {
            rval = new Integer(env);
        }
        getLog().info("Using build number : "+rval);
        return rval;
    }

    protected final String buildId() {
        String rval = propertyOrEnvVariable("BUILD_ID");
        if (rval == null) {
            getLog().info("No environment variable 'BUILD_ID' found, defaulting to timestamp (Jenkins would set this)");
            rval = new SimpleDateFormat("yyyy-MM-dd_hh-mm-ss").format(new Date());
        }
        getLog().info("Using build id : "+rval);
        return rval;
    }

    /**
     * Looks first at java properties, then environmental variables
     * @param varName String containing the name of the variable to look up
     * @return String containing the variable, or null if none found
     */
    protected final String propertyOrEnvVariable(String varName)
    {
        String rval = System.getProperty(varName);
        if (rval==null)
        {
            rval = System.getenv(varName);
        }
        return rval;
    }

    protected final void safeSleep(long ms) {
        try {
            Thread.sleep(ms);
        } catch (InterruptedException ie) {
            getLog().info("Sleep interrupted");
        }
    }

    protected final String obscure(String input, int save) {
        String rval = input;
        if (input != null && input.length() > save * 2) {
            StringBuilder b = new StringBuilder();
            b.append(input.substring(0, save));
            while (b.length() < input.length() - save) {
                b.append("*");
            }
            b.append(input.substring(input.length() - save));
            rval = b.toString();
        }
        return rval;
    }

}