package org.kuali.maven.mojo.s3;

 import java.io.IOException;

 import java.io.InputStream;

 import org.apache.maven.plugin.MojoExecutionException;

 import org.apache.maven.settings.Server;

 import org.springframework.core.io.DefaultResourceLoader;

 import org.springframework.core.io.Resource;

 import org.springframework.core.io.ResourceLoader;

 import com.amazonaws.auth.AWSCredentials;

 import com.amazonaws.auth.BasicAWSCredentials;

 import com.amazonaws.services.s3.internal.Mimetypes;

 import com.amazonaws.services.s3.model.CannedAccessControlList;

 import com.amazonaws.services.s3.model.ObjectMetadata;

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

 /**

  * Abstract mojo containing S3 mojo essentials.

  */

 public abstract class S3Mojo extends BaseMojo {

     /**

      * Helper class for determining the mime types of objects.

      */

     Mimetypes mimeTypes = Mimetypes.getInstance();

     /**

      * This is the id of the server from settings.xml containing Amazon AWS

      * credentials.

      *

      * @parameter expression="${serverId}"

      */

     private String serverId;

     /**

      * Only update the bucket hierarchy underneath this prefix.

      *

      * @parameter expression="${prefix}"

      */

     private String prefix;

     /**

      * The delimiter used to organize keys into a hierarchy.

      *

      * @parameter expression="${delimiter}" default-value="/"

      */

     private String delimiter;

     /**

      * Maximum number of keys to return per query.

      *

      * @parameter expression="${maxKeys}"

      */

     private Integer maxKeys;

     /**

      * Amazon AWS Access Key Id. See also <code>serverId</code>.

      *

      * @parameter expression="${accessKeyId}"

      */

     private String accessKeyId;

     /**

      * Amazon AWS Secret Access Key. See also <code>serverId</code>.

      *

      * @parameter expression="${secretAccessKey}"

      */

     private String secretAccessKey;

     /**

      * The name of the bucket to update.

      *

      * @parameter expression="${bucket}"

      * @required

      */

     private String bucket;

     protected ObjectMetadata getObjectMetadata(final String location,

             final Resource resource) throws IOException {

         ObjectMetadata om = new ObjectMetadata();

         String contentType = mimeTypes.getMimetype(location);

         om.setContentType(contentType);

         om.setContentLength(resource.contentLength());

         return om;

     }

     protected PutObjectRequest getPutObjectRequest(final String location, final String key)

             throws IOException {

         ResourceLoader loader = new DefaultResourceLoader();

         Resource resource = loader.getResource(location);

         InputStream in = resource.getInputStream();

         ObjectMetadata objectMetadata = getObjectMetadata(location, resource);

         PutObjectRequest request = new PutObjectRequest(getBucket(), key, in,

                 objectMetadata);

         request.setCannedAcl(CannedAccessControlList.PublicRead);

         return request;

     }

     protected PutObjectRequest getPutObjectRequest(final String location)

             throws IOException {

         String key = location.substring(1);

         return getPutObjectRequest(location, key);

     }

     protected String getAuthenticationErrorMessage() {

         StringBuffer sb = new StringBuffer();

         sb.append("\n\nError: accessKeyId and secretAccessKey must be provided.\n");

         sb.append("Provide them in the plugin configuration or specify them on the command line:\n\n");

         sb.append("-DaccessKeyId=XXXX\n");

         sb.append("-DsecretAccessKey=XXXX\n");

         sb.append("\n");

         sb.append("You can also provide them in settings.xml as username and password:\n\n");

         sb.append("<server>\n");

         sb.append("  <id>[serverId]</id>\n");

         sb.append("  <username>[AWS Access Key ID]</username>\n");

         sb.append("  <password>[AWS Secret Access Key]</password>\n");

         sb.append("</server>\n\n.\n");

         return sb.toString();

     }

     protected void updateCredentials() {

         if (getServerId() == null) {

             return;

         }

         Server server = getSettings().getServer(getServerId());

         if (getAccessKeyId() == null) {

             setAccessKeyId(server.getUsername());

         }

         if (getSecretAccessKey() == null) {

             setSecretAccessKey(server.getPassword());

         }

     }

     protected void validateCredentials() throws MojoExecutionException {

         if (getAccessKeyId() == null || getSecretAccessKey() == null) {

             throw new MojoExecutionException(getAuthenticationErrorMessage());

         }

     }

     protected AWSCredentials getCredentials() throws MojoExecutionException {

         return new BasicAWSCredentials(getAccessKeyId(), getSecretAccessKey());

     }

     public String getAccessKeyId() {

         return accessKeyId;

     }

     public void setAccessKeyId(final String accessKeyId) {

         this.accessKeyId = accessKeyId;

     }

     public String getSecretAccessKey() {

         return secretAccessKey;

     }

     public void setSecretAccessKey(final String secretAccessKey) {

         this.secretAccessKey = secretAccessKey;

     }

     public String getBucket() {

         return bucket;

     }

     public void setBucket(final String bucket) {

         this.bucket = bucket;

     }

     public String getPrefix() {

         return prefix;

     }

     public void setPrefix(final String prefix) {

         this.prefix = prefix;

     }

     public String getDelimiter() {

         return delimiter;

     }

     public void setDelimiter(final String delimiter) {

         this.delimiter = delimiter;

     }

     public Integer getMaxKeys() {

         return maxKeys;

     }

     public void setMaxKeys(final Integer maxKeys) {

         this.maxKeys = maxKeys;

     }

     public String getServerId() {

         return serverId;

     }

     public void setServerId(final String serverId) {

         this.serverId = serverId;

     }

 }