UploadSourceBundleMojo.java

/*
 * Copyright (c) 2016 ingenieux Labs
 *
 * 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 br.com.ingenieux.mojo.beanstalk.bundle;

import com.amazonaws.services.s3.model.PutObjectRequest;
import com.amazonaws.services.s3.model.PutObjectResult;

import org.apache.commons.lang.StringUtils;
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.io.File;

import br.com.ingenieux.mojo.aws.util.BeanstalkerS3Client;
import br.com.ingenieux.mojo.beanstalk.AbstractBeanstalkMojo;

import static java.lang.String.format;

/**
 * Uploads a packed war file to Amazon S3 for further Deployment.
 *
 * @since 0.1.0
 */
@Mojo(name = "upload-source-bundle")
public class UploadSourceBundleMojo extends AbstractBeanstalkMojo {

  /**
   * S3 Bucket
   */
  @Parameter(property = "beanstalk.s3Bucket")
  String s3Bucket;

  /**
   * S3 Key
   */
  @Parameter(
    property = "beanstalk.s3Key",
    defaultValue = "${project.artifactId}/${project.build.finalName}-${beanstalk.versionLabel}.${project.packaging}",
    required = true
  )
  String s3Key;

  /**
   * <p> Should we do a multipart upload? Defaults to true </p> <p> Disable when you want to be
   * charged slightly less :) </p>
   */
  @Parameter(property = "beanstalk.multipartUpload", defaultValue = "true")
  boolean multipartUpload = false;

  /**
   * Artifact to Deploy
   */
  @Parameter(property = "beanstalk.artifactFile", defaultValue = "${project.build.directory}/${project.build.finalName}.${project.packaging}", required = true)
  File artifactFile;

  /**
   * Silent Upload?
   */
  @Parameter(property = "beanstalk.silentUpload", defaultValue = "false")
  boolean silentUpload = false;

  /**
   * Version Label to use. Defaults to Project Version
   */
  @Parameter(property = "beanstalk.versionLabel", required = true)
  String versionLabel;

  @Parameter(defaultValue = "${project}")
  MavenProject project;

  protected Object executeInternal() throws Exception {
    String path = artifactFile.getPath();

    if (!(path.endsWith(".war") || path.endsWith(".jar") || path.endsWith(".zip"))) {
      getLog().warn("Not a war/jar/zip file. Skipping");

      return null;
    }

    if (!artifactFile.exists()) {
      throw new MojoFailureException("Artifact File does not exist! (file=" + path + ")");
    }

    BeanstalkerS3Client client = new BeanstalkerS3Client(getAWSCredentials(), getClientConfiguration(), getRegion());

    client.setMultipartUpload(multipartUpload);
    client.setSilentUpload(silentUpload);

    if (StringUtils.isBlank(s3Bucket)) {
      getLog().info("S3 Bucket not defined.");
      s3Bucket = getService().createStorageLocation().getS3Bucket();

      getLog().info("Using defaults, like: " + s3Bucket);

      project.getProperties().put("beanstalk.s3Bucket", s3Bucket);
    }

    getLog().info("Target Path: s3://" + s3Bucket + "/" + s3Key);
    getLog().info("Uploading artifact file: " + path);

    PutObjectResult result = client.putObject(new PutObjectRequest(s3Bucket, s3Key, artifactFile));

    getLog().info("Artifact Uploaded");

    project.getProperties().put("beanstalk.s3Key", s3Key);

    project.getProperties().put("beanstalk.lastUploadedS3Object", format("s3://%s/%s", s3Bucket, s3Key));

    return result;
  }
}