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

Go to download
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.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;
    }

}