/*
    * DaCapo Benchmark Maven Plugin
    * Copyright (C) 2010 Technische Universität Darmstadt
    * info@scalabench.org
    *
    * 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 org.scalabench.plugins.dacapo;
  
  import java.io.File;
  import java.io.IOException;
  import java.util.List;
  
  import org.apache.maven.plugin.MojoExecutionException;
  import org.apache.maven.plugin.MojoFailureException;
  import org.codehaus.plexus.archiver.ArchiverException;
  import org.codehaus.plexus.archiver.FileSet;
  import org.codehaus.plexus.archiver.util.DefaultFileSet;
  import org.codehaus.plexus.archiver.zip.ZipArchiver;
  
  /**
   * Package the benchmark's input data.
   * 
   * @since 0.1.0
   * 
   * @threadSafe
   * @goal package-data
   * @phase prepare-package
   */
  public class PackageDataMojo extends AbstractDacapoBenchmarkMojo {
      
      /**
       * Main directory for input data to include as a ZIP file in the JAR.
       * 
       * @required
       * @parameter default-value="${basedir}/src/main/data"
       */
      private File dataSourceDirectory;
      
      /**
       * Additional directories for input data to include.
       * 
       * @parameter
       */
      private List<File> additionalDataSourceDirectories;
      
      /**
       * The data output directory.
       * 
       * @required
       * @readonly
       * @parameter expression="${project.build.outputDirectory}/dat"
       */
      private File dataOutputDirectory;
      
      /**
       * Should the input data be compressed?
       * 
       * @required
       * @parameter default-value=true
       */
      private boolean compressData;
      
      /**
       * @component role="org.codehaus.plexus.archiver.Archiver" roleHint="zip"
       */
      private ZipArchiver zipArchiver;
      
      @Override
      public void execute() throws MojoExecutionException, MojoFailureException {
          final File zipDataFile = new File(dataOutputDirectory, getBenchmarkName() + ".zip");
          
          try {
              addDirectory(dataSourceDirectory);
              
              if (additionalDataSourceDirectories != null)
                  for (File additionalDataSourceDirectory : additionalDataSourceDirectories)
                      addDirectory(additionalDataSourceDirectory);
              
              createDataPackage(zipDataFile);
          } catch (ArchiverException e) {
              throw new MojoExecutionException("Could not package input data", e);
          } catch (IOException e) {
              throw new MojoExecutionException("Could not package input data", e);
          }
      }
      
      private void createDataPackage(final File zipDataFile) throws ArchiverException, IOException {
          if (!zipArchiver.getResources().hasNext()) {
             logWarn("No input data to package");
         } else {
             zipArchiver.setDestFile(zipDataFile);
             zipArchiver.setCompress(compressData);
             zipArchiver.createArchive();
         }
     }
     
     private void addDirectory(File directory) throws ArchiverException {
         if (!directory.exists()) {
             logInfo("Skipping non-existing data source directory %s", directory);
         } else {
             logInfo("Packaging data source directory %s", directory);
             final FileSet fs = createDirectoryFileSet(directory);
             zipArchiver.addFileSet(fs);
         }
     }
     
     private FileSet createDirectoryFileSet(File directory) {
         final DefaultFileSet fs = new DefaultFileSet();
         fs.setDirectory(directory);
         fs.setPrefix(getBenchmarkName() + File.separatorChar);
         return fs;
     }
 }