br.com.ingenieux.mojo.cloudfront.UpdateDistributionMojo Maven / Gradle / Ivy

Go to download
package br.com.ingenieux.mojo.cloudfront;

import static org.apache.commons.lang.StringUtils.isBlank;
import static org.apache.commons.lang.StringUtils.strip;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import org.apache.commons.codec.digest.DigestUtils;
import org.apache.http.HttpResponse;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpHead;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.maven.plugins.annotations.Mojo;

import com.amazonaws.services.cloudfront.model.CreateInvalidationRequest;
import com.amazonaws.services.cloudfront.model.InvalidationBatch;
import com.amazonaws.services.cloudfront.model.Paths;
import com.amazonaws.services.s3.model.PutObjectRequest;

@Mojo(name="update-distribution", requiresProject=true)
public class UpdateDistributionMojo extends SeedS3DistributionMojo {
	private final HttpClient httpClient = new DefaultHttpClient();

	@Override
	protected void fireDistributionExecuted(Distribution d) {
		if (results.isEmpty())
			return;

		List paths = new ArrayList();

		InvalidationBatch batch = new InvalidationBatch("invalidate-" + d.id);
		CreateInvalidationRequest invalidationRequest = new CreateInvalidationRequest()
				.withDistributionId(d.id).withInvalidationBatch(batch);

		for (Map.Entry entry : results.entrySet()) {
			String key = entry.getKey();
			PutObjectRequest request = entry.getValue();

			if (null != request && request.getBucketName().equals(d.s3Bucket)) {
				paths.add("/" + request.getKey());
			} else {
				paths.add("/" + key);
			}
		}

		if (!paths.isEmpty()) {
			batch.setPaths(new Paths().withItems(paths));

			invalidationRequest.withInvalidationBatch(batch);

			getService().createInvalidation(invalidationRequest);
		} else {
			getLog().info("No need to invalidate. Skipping");
		}
	}

	@Override
	protected Collection getPutObjectRequest(Distribution d,
			String path, File file, String contentType) throws IOException {
		String md5sum = DigestUtils.md5Hex(new FileInputStream(file));

		String cacheETag = getETag(d, path);

		if (md5sum.equals(cacheETag)) {
			getLog().info("Ignoring resource " + path + " due to same eTag");
			return Collections.emptyList();
		}

		return super.getPutObjectRequest(d, path, file, contentType);
	}

	protected String getETag(Distribution d, String path) throws IOException,
			ClientProtocolException {
		if (!isBlank(d.s3Bucket))
			return s3Client.getObjectMetadata(d.s3Bucket, path).getETag();

		String assetUrl = String.format("http://%s/%s", d.domainName, path);

		getLog().info("Checking for eTag on " + assetUrl);

		HttpResponse result = httpClient.execute(new HttpHead(assetUrl));

		getLog().info("Status Code:" + result.getStatusLine().getStatusCode());

		if (result.containsHeader("ETag")) {
			String eTag = strip(result.getFirstHeader("ETag").getValue(), "\"");
			getLog().info("ETag is " + eTag);

			return eTag;
		}

		return "";
	}

	@Override
	protected void executeOnCustomDistribution(Distribution d)
			throws IOException {
		getLog().warn(
				"Distribution id "
						+ d.id
						+ " does not have an s3Bucket assigned (perhaps its a custom origin?). Scanning webappDirectory for matches");
		List pathList = fetchLocalDistributionFiles(d);

		for (String path : pathList) {
			File file = new File(webappDirectory, path);

			String md5Sum = DigestUtils.md5Hex(new FileInputStream(file));
			String eTag = getETag(d, path);

			if (isBlank(eTag)) {
				getLog().info(
						"Resource "
								+ path
								+ " does not exist. No need to invalidate on a Custom Origin");
			} else if (!md5Sum.equals(eTag)) {
				getLog().info(
						"Resource "
								+ path
								+ " needs to be invalidated. Adding to Invalidation Request Queue");

				results.put(path, null);
			} else {
				getLog().info(
						"Resource "
								+ path
								+ " matches existing eTag. Not adding to invalidation queue");
			}
		}

	}
}