package org.apache.maven.plugin.ear;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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.
   */
  
  import org.apache.maven.archiver.MavenArchiveConfiguration;
  import org.apache.maven.archiver.MavenArchiver;
  import org.apache.maven.execution.MavenSession;
  import org.apache.maven.plugin.MojoExecutionException;
  import org.apache.maven.plugin.MojoFailureException;
  import org.apache.maven.plugin.ear.util.EarMavenArchiver;
  import org.apache.maven.plugin.ear.util.JavaEEVersion;
  import org.apache.maven.plugins.annotations.Component;
  import org.apache.maven.plugins.annotations.LifecyclePhase;
  import org.apache.maven.plugins.annotations.Mojo;
  import org.apache.maven.plugins.annotations.Parameter;
  import org.apache.maven.plugins.annotations.ResolutionScope;
  import org.apache.maven.project.MavenProjectHelper;
  import org.apache.maven.shared.filtering.MavenFileFilter;
  import org.apache.maven.shared.filtering.MavenFilteringException;
  import org.apache.maven.shared.filtering.MavenResourcesExecution;
  import org.apache.maven.shared.filtering.MavenResourcesFiltering;
  import org.codehaus.plexus.archiver.Archiver;
  import org.codehaus.plexus.archiver.ArchiverException;
  import org.codehaus.plexus.archiver.UnArchiver;
  import org.codehaus.plexus.archiver.jar.JarArchiver;
  import org.codehaus.plexus.archiver.jar.Manifest;
  import org.codehaus.plexus.archiver.jar.Manifest.Attribute;
  import org.codehaus.plexus.archiver.jar.ManifestException;
  import org.codehaus.plexus.archiver.manager.ArchiverManager;
  import org.codehaus.plexus.archiver.manager.NoSuchArchiverException;
  import org.codehaus.plexus.archiver.zip.ZipArchiver;
  import org.codehaus.plexus.archiver.zip.ZipUnArchiver;
  import org.codehaus.plexus.util.DirectoryScanner;
  import org.codehaus.plexus.util.FileUtils;
  import org.codehaus.plexus.util.StringUtils;
  
  import java.io.File;
  import java.io.FileReader;
  import java.io.IOException;
  import java.io.PrintWriter;
  import java.util.ArrayList;
  import java.util.Arrays;
  import java.util.List;
  import java.util.zip.ZipException;
  
  /**
   * Builds J2EE Enterprise Archive (EAR) files.
   *
   * @author <a href="snicoll@apache.org">Stephane Nicoll</a>
   * @version $Id: EarMojo.java 1359710 2012-07-10 14:53:53Z tchemit $
   */
  @Mojo( name = "ear", defaultPhase = LifecyclePhase.PACKAGE, threadSafe = true,
         requiresDependencyResolution = ResolutionScope.TEST )
  public class EarMojo
      extends AbstractEarMojo
  {
      private static final String[] EMPTY_STRING_ARRAY = { };
  
  
      /**
       * Single directory for extra files to include in the EAR.
       */
      @Parameter( defaultValue = "${basedir}/src/main/application", required = true )
      private File earSourceDirectory;
  
      /**
       * The comma separated list of tokens to include in the EAR.
       */
      @Parameter( alias = "includes", defaultValue = "**" )
      private String earSourceIncludes;
  
      /**
       * The comma separated list of tokens to exclude from the EAR.
       */
      @Parameter( alias = "excludes" )
      private String earSourceExcludes;
  
      /**
       * Specify that the EAR sources should be filtered.
       *
       * @since 2.3.2
       */
     @Parameter( defaultValue = "false" )
     private boolean filtering;
 
     /**
      * Filters (property files) to include during the interpolation of the pom.xml.
      *
      * @since 2.3.2
      */
     @Parameter
     private List filters;
 
     /**
      * A list of file extensions that should not be filtered if
      * filtering is enabled.
      *
      * @since 2.3.2
      */
     @Parameter
     private List nonFilteredFileExtensions;
 
     /**
      * To escape interpolated value with Windows path
      * c:\foo\bar will be replaced with c:\\foo\\bar.
      *
      * @since 2.3.2
      */
     @Parameter( property = "maven.ear.escapedBackslashesInFilePath", defaultValue = "false" )
     private boolean escapedBackslashesInFilePath;
 
     /**
      * Expression preceded with this String won't be interpolated
      * \${foo} will be replaced with ${foo}.
      *
      * @since 2.3.2
      */
     @Parameter( property = "maven.ear.escapeString" )
     protected String escapeString;
 
     /**
      * The location of the manifest file to be used within the EAR file. If
      * no value if specified, the default location in the workDirectory is
      * taken. If the file does not exist, a manifest will be generated
      * automatically.
      */
     @Parameter
     private File manifestFile;
 
     /**
      * The location of a custom application.xml file to be used
      * within the EAR file.
      */
     @Parameter
     private String applicationXml;
 
     /**
      * The directory for the generated EAR.
      */
     @Parameter( defaultValue = "${project.build.directory}", required = true )
     private String outputDirectory;
 
     /**
      * The name of the EAR file to generate.
      */
     @Parameter( alias = "earName", defaultValue = "${project.build.finalName}", required = true )
     private String finalName;
 
     /**
      * The comma separated list of artifact's type(s) to unpack
      * by default.
      */
     @Parameter
     private String unpackTypes;
 
     /**
      * Classifier to add to the artifact generated. If given, the artifact will
      * be an attachment instead.
      */
     @Parameter
     private String classifier;
 
     /**
      * A comma separated list of tokens to exclude when packaging the EAR.
      * By default nothing is excluded. Note that you can use the Java Regular
      * Expressions engine to include and exclude specific pattern using the
      * expression %regex[].
      * Hint: read the about (?!Pattern).
      *
      * @since 2.7
      */
     @Parameter
     private String packagingExcludes;
 
     /**
      * A comma separated list of tokens to include when packaging the EAR.
      * By default everything is included. Note that you can use the Java Regular
      * Expressions engine to include and exclude specific pattern using the
      * expression %regex[].
      *
      * @since 2.7
      */
     @Parameter
     private String packagingIncludes;
 
     /**
      * Whether to create skinny WARs or not. A skinny WAR is a WAR that does not
      * have all of its dependencies in WEB-INF/lib. Instead those dependencies
      * are shared between the WARs through the EAR.
      *
      * @since 2.7
      */
     @Parameter( property = "maven.ear.skinnyWars", defaultValue = "false" )
     private boolean skinnyWars;
 
     /**
      * The Jar archiver.
      */
     @Component( role = Archiver.class, hint = "jar" )
     private JarArchiver jarArchiver;
 
     /**
      * The Zip archiver.
      */
     @Component( role = Archiver.class, hint = "zip" )
     private ZipArchiver zipArchiver;
 
     /**
      * The Zip Un archiver.
      */
     @Component( role = UnArchiver.class, hint = "zip" )
     private ZipUnArchiver zipUnArchiver;
 
     /**
      * The archive configuration to use.
      * See <a href="http://maven.apache.org/shared/maven-archiver/index.html">Maven Archiver Reference</a>.
      */
     @Parameter
     private MavenArchiveConfiguration archive = new MavenArchiveConfiguration();
 
     /**
      */
     @Component
     private MavenProjectHelper projectHelper;
 
     /**
      * The archive manager.
      */
     @Component
     private ArchiverManager archiverManager;
 
     /**
      */
     @Component( role = MavenFileFilter.class, hint = "default" )
     private MavenFileFilter mavenFileFilter;
 
     /**
      */
     @Component( role = MavenResourcesFiltering.class, hint = "default" )
     private MavenResourcesFiltering mavenResourcesFiltering;
 
     /**
      * @since 2.3.2
      */
     @Component
     private MavenSession session;
 
 
     private List filterWrappers;
 
 
     public void execute()
         throws MojoExecutionException, MojoFailureException
     {
         // Initializes ear modules
         super.execute();
 
         final JavaEEVersion javaEEVersion = JavaEEVersion.getJavaEEVersion( version );
 
         // Initializes unpack types
         List<String> unpackTypesList = new ArrayList<String>();
         if ( unpackTypes != null )
         {
             unpackTypesList = Arrays.asList( unpackTypes.split( "," ) );
             for ( String type : unpackTypesList )
             {
                 if ( !EarModuleFactory.standardArtifactTypes.contains( type ) )
                 {
                     throw new MojoExecutionException(
                         "Invalid type [" + type + "] supported types are " + EarModuleFactory.standardArtifactTypes );
                 }
             }
             getLog().debug( "Initialized unpack types " + unpackTypesList );
         }
 
         // Copy modules
         try
         {
             for ( EarModule module : getModules() )
             {
                 if ( module instanceof JavaModule )
                 {
                     getLog().warn( "JavaModule is deprecated (" + module + "), please use JarModule instead." );
                 }
                 if ( module instanceof Ejb3Module )
                 {
                     getLog().warn( "Ejb3Module is deprecated (" + module + "), please use EjbModule instead." );
                 }
                 final File sourceFile = module.getArtifact().getFile();
                 final File destinationFile = buildDestinationFile( getWorkDirectory(), module.getUri() );
                 if ( !sourceFile.isFile() )
                 {
                     throw new MojoExecutionException(
                         "Cannot copy a directory: " + sourceFile.getAbsolutePath() + "; Did you package/install " +
                             module.getArtifact() + "?" );
                 }
 
                 if ( destinationFile.getCanonicalPath().equals( sourceFile.getCanonicalPath() ) )
                 {
                     getLog().info(
                         "Skipping artifact [" + module + "], as it already exists at [" + module.getUri() + "]" );
                     continue;
                 }
 
                 // If the module is within the unpack list, make sure that no unpack wasn't forced (null or true)
                 // If the module is not in the unpack list, it should be true
                 if ( ( unpackTypesList.contains( module.getType() ) &&
                     ( module.shouldUnpack() == null || module.shouldUnpack().booleanValue() ) ) ||
                     ( module.shouldUnpack() != null && module.shouldUnpack().booleanValue() ) )
                 {
                     getLog().info( "Copying artifact [" + module + "] to [" + module.getUri() + "] (unpacked)" );
                     // Make sure that the destination is a directory to avoid plexus nasty stuff :)
                     destinationFile.mkdirs();
                     unpack( sourceFile, destinationFile );
 
                     if ( skinnyWars && module.changeManifestClasspath() )
                     {
                         changeManifestClasspath( module, destinationFile );
                     }
                 }
                 else
                 {
                     if ( sourceFile.lastModified() > destinationFile.lastModified() )
                     {
                         getLog().info( "Copying artifact [" + module + "] to [" + module.getUri() + "]" );
                         FileUtils.copyFile( sourceFile, destinationFile );
 
                         if ( skinnyWars && module.changeManifestClasspath() )
                         {
                             changeManifestClasspath( module, destinationFile );
                         }
                     }
                     else
                     {
                         getLog().debug(
                             "Skipping artifact [" + module + "], as it is already up to date at [" + module.getUri() +
                                 "]" );
                     }
                 }
             }
         }
         catch ( IOException e )
         {
             throw new MojoExecutionException( "Error copying EAR modules", e );
         }
         catch ( ArchiverException e )
         {
             throw new MojoExecutionException( "Error unpacking EAR modules", e );
         }
         catch ( NoSuchArchiverException e )
         {
             throw new MojoExecutionException( "No Archiver found for EAR modules", e );
         }
 
         // Copy source files
         try
         {
             File earSourceDir = earSourceDirectory;
             if ( earSourceDir.exists() )
             {
                 getLog().info( "Copy ear sources to " + getWorkDirectory().getAbsolutePath() );
                 String[] fileNames = getEarFiles( earSourceDir );
                 for ( int i = 0; i < fileNames.length; i++ )
                 {
                     copyFile( new File( earSourceDir, fileNames[i] ), new File( getWorkDirectory(), fileNames[i] ) );
                 }
             }
 
             if ( applicationXml != null && !"".equals( applicationXml ) )
             {
                 //rename to application.xml
                 getLog().info( "Including custom application.xml[" + applicationXml + "]" );
                 File metaInfDir = new File( getWorkDirectory(), META_INF );
                 copyFile( new File( applicationXml ), new File( metaInfDir, "/application.xml" ) );
             }
 
         }
         catch ( IOException e )
         {
             throw new MojoExecutionException( "Error copying EAR sources", e );
         }
         catch ( MavenFilteringException e )
         {
             throw new MojoExecutionException( "Error filtering EAR sources", e );
         }
 
         // Check if deployment descriptor is there
         File ddFile = new File( getWorkDirectory(), APPLICATION_XML_URI );
         if ( !ddFile.exists() && ( javaEEVersion.lt( JavaEEVersion.Five ) ) )
         {
             throw new MojoExecutionException(
                 "Deployment descriptor: " + ddFile.getAbsolutePath() + " does not exist." );
         }
 
         try
         {
             File earFile = getEarFile( outputDirectory, finalName, classifier );
             final MavenArchiver archiver = new EarMavenArchiver( getModules() );
             final JarArchiver jarArchiver = getJarArchiver();
             getLog().debug( "Jar archiver implementation [" + jarArchiver.getClass().getName() + "]" );
             archiver.setArchiver( jarArchiver );
             archiver.setOutputFile( earFile );
 
             // Include custom manifest if necessary
             includeCustomManifestFile();
 
             getLog().debug( "Excluding " + Arrays.asList( getPackagingExcludes() ) + " from the generated EAR." );
             getLog().debug( "Including " + Arrays.asList( getPackagingIncludes() ) + " in the generated EAR." );
 
             archiver.getArchiver().addDirectory( getWorkDirectory(), getPackagingIncludes(), getPackagingExcludes() );
             archiver.createArchive( session, getProject(), archive );
 
             if ( classifier != null )
             {
                 projectHelper.attachArtifact( getProject(), "ear", classifier, earFile );
             }
             else
             {
                 getProject().getArtifact().setFile( earFile );
             }
         }
         catch ( Exception e )
         {
             throw new MojoExecutionException( "Error assembling EAR", e );
         }
     }
 
     public String getApplicationXml()
     {
         return applicationXml;
     }
 
     public void setApplicationXml( String applicationXml )
     {
         this.applicationXml = applicationXml;
     }
 
     /**
      * Returns a string array of the excludes to be used
      * when assembling/copying the ear.
      *
      * @return an array of tokens to exclude
      */
     protected String[] getExcludes()
     {
         @SuppressWarnings( "unchecked" ) List<String> excludeList =
             new ArrayList<String>( FileUtils.getDefaultExcludesAsList() );
         if ( earSourceExcludes != null && !"".equals( earSourceExcludes ) )
         {
             excludeList.addAll( Arrays.asList( StringUtils.split( earSourceExcludes, "," ) ) );
         }
 
         // if applicationXml is specified, omit the one in the source directory
         if ( getApplicationXml() != null && !"".equals( getApplicationXml() ) )
         {
             excludeList.add( "**/" + META_INF + "/application.xml" );
         }
 
         return (String[]) excludeList.toArray( EMPTY_STRING_ARRAY );
     }
 
     /**
      * Returns a string array of the includes to be used
      * when assembling/copying the ear.
      *
      * @return an array of tokens to include
      */
     protected String[] getIncludes()
     {
         return StringUtils.split( StringUtils.defaultString( earSourceIncludes ), "," );
     }
 
     public String[] getPackagingExcludes()
     {
         if ( StringUtils.isEmpty( packagingExcludes ) )
         {
             return new String[0];
         }
         else
         {
             return StringUtils.split( packagingExcludes, "," );
         }
     }
 
     public void setPackagingExcludes( String packagingExcludes )
     {
         this.packagingExcludes = packagingExcludes;
     }
 
     public String[] getPackagingIncludes()
     {
         if ( StringUtils.isEmpty( packagingIncludes ) )
         {
             return new String[]{ "**" };
         }
         else
         {
             return StringUtils.split( packagingIncludes, "," );
         }
     }
 
     public void setPackagingIncludes( String packagingIncludes )
     {
         this.packagingIncludes = packagingIncludes;
     }
 
     private static File buildDestinationFile( File buildDir, String uri )
     {
         return new File( buildDir, uri );
     }
 
     private void includeCustomManifestFile()
     {
         if ( manifestFile == null )
         {
             manifestFile = new File( getWorkDirectory(), "META-INF/MANIFEST.MF" );
         }
 
         if ( !manifestFile.exists() )
         {
             getLog().info( "Could not find manifest file: " + manifestFile + " - Generating one" );
         }
         else
         {
             getLog().info( "Including custom manifest file [" + manifestFile + "]" );
             archive.setManifestFile( manifestFile );
         }
     }
 
     /**
      * Returns the EAR file to generate, based on an optional classifier.
      *
      * @param basedir    the output directory
      * @param finalName  the name of the ear file
      * @param classifier an optional classifier
      * @return the EAR file to generate
      */
     private static File getEarFile( String basedir, String finalName, String classifier )
     {
         if ( classifier == null )
         {
             classifier = "";
         }
         else if ( classifier.trim().length() > 0 && !classifier.startsWith( "-" ) )
         {
             classifier = "-" + classifier;
         }
 
         return new File( basedir, finalName + classifier + ".ear" );
     }
 
     /**
      * Returns a list of filenames that should be copied
      * over to the destination directory.
      *
      * @param sourceDir the directory to be scanned
      * @return the array of filenames, relative to the sourceDir
      */
     private String[] getEarFiles( File sourceDir )
     {
         DirectoryScanner scanner = new DirectoryScanner();
         scanner.setBasedir( sourceDir );
         scanner.setExcludes( getExcludes() );
         scanner.addDefaultExcludes();
 
         scanner.setIncludes( getIncludes() );
 
         scanner.scan();
 
         return scanner.getIncludedFiles();
     }
 
     /**
      * Unpacks the module into the EAR structure.
      *
      * @param source  File to be unpacked.
      * @param destDir Location where to put the unpacked files.
      */
     public void unpack( File source, File destDir )
         throws NoSuchArchiverException, IOException, ArchiverException
     {
         UnArchiver unArchiver = archiverManager.getUnArchiver( "zip" );
         unArchiver.setSourceFile( source );
         unArchiver.setDestDirectory( destDir );
 
         // Extract the module
         unArchiver.extract();
     }
 
     /**
      * Returns the {@link JarArchiver} implementation used
      * to package the EAR file.
      * <p/>
      * By default the archiver is obtained from the Plexus container.
      *
      * @return the archiver
      */
     protected JarArchiver getJarArchiver()
     {
         return jarArchiver;
     }
 
     private void copyFile( File source, File target )
         throws MavenFilteringException, IOException, MojoExecutionException
     {
         if ( filtering && !isNonFilteredExtension( source.getName() ) )
         {
             // Silly that we have to do this ourselves
             if ( target.getParentFile() != null && !target.getParentFile().exists() )
             {
                 target.getParentFile().mkdirs();
             }
             mavenFileFilter.copyFile( source, target, true, getFilterWrappers(), null );
         }
         else
         {
             FileUtils.copyFile( source, target );
         }
     }
 
     public boolean isNonFilteredExtension( String fileName )
     {
         return !mavenResourcesFiltering.filteredFileExtension( fileName, nonFilteredFileExtensions );
     }
 
     private List getFilterWrappers()
         throws MojoExecutionException
     {
         if ( filterWrappers == null )
         {
             try
             {
                 MavenResourcesExecution mavenResourcesExecution = new MavenResourcesExecution();
                 mavenResourcesExecution.setEscapeString( escapeString );
                 filterWrappers =
                     mavenFileFilter.getDefaultFilterWrappers( project, filters, escapedBackslashesInFilePath,
                                                               this.session, mavenResourcesExecution );
             }
             catch ( MavenFilteringException e )
             {
                 getLog().error( "Fail to build filtering wrappers " + e.getMessage() );
                 throw new MojoExecutionException( e.getMessage(), e );
             }
         }
         return filterWrappers;
     }
 
     private void changeManifestClasspath( EarModule module, File original )
         throws MojoFailureException
     {
         try
         {
             File workDirectory;
 
             // Handle the case that the destination might be a directory (project-038)
             if ( original.isFile() )
             {
                 // Create a temporary work directory
                 workDirectory =
                     new File( new File( generatedDescriptorLocation, "temp" ), module.getArtifact().getArtifactId() );
                 workDirectory.mkdirs();
                 getLog().debug( "Created a temporary work directory: " + workDirectory.getAbsolutePath() );
 
                 // Unpack the archive to a temporary work directory
                 zipUnArchiver.setSourceFile( original );
                 zipUnArchiver.setDestDirectory( workDirectory );
                 zipUnArchiver.extract();
             }
             else
             {
                 workDirectory = original;
             }
 
             // Create a META-INF/MANIFEST.MF file if it doesn't exist (project-038)
             File metaInfDirectory = new File( workDirectory, "META-INF" );
             boolean newMetaInfCreated = metaInfDirectory.mkdirs();
             if ( newMetaInfCreated )
             {
                 getLog().debug(
                     "This project did not have a META-INF directory before, so a new directory was created." );
             }
             File manifestFile = new File( metaInfDirectory, "MANIFEST.MF" );
             boolean newManifestCreated = manifestFile.createNewFile();
             if ( newManifestCreated )
             {
                 getLog().debug(
                     "This project did not have a META-INF/MANIFEST.MF file before, so a new file was created." );
             }
 
             // Read the manifest from disk
             Manifest mf = new Manifest( new FileReader( manifestFile ) );
             Attribute classPath = mf.getMainSection().getAttribute( "Class-Path" );
             List<String> classPathElements = new ArrayList<String>();
 
             if ( classPath != null )
             {
                 classPathElements.addAll( Arrays.asList( classPath.getValue().split( " " ) ) );
             }
             else
             {
                 classPath = new Attribute( "Class-Path", "" );
                 mf.getMainSection().addConfiguredAttribute( classPath );
             }
 
             // Modify the classpath entries in the manifest
             for ( EarModule o : getModules() )
             {
                 if ( o instanceof JarModule )
                 {
                     JarModule jm = (JarModule) o;
 
                     if ( module.getLibDir() != null )
                     {
                         File artifact =
                             new File( new File( workDirectory, module.getLibDir() ), jm.getBundleFileName() );
 
                         if ( artifact.exists() )
                         {
                             if ( !artifact.delete() )
                             {
                                 getLog().error( "Could not delete '" + artifact + "'" );
                             }
                         }
                     }
 
                     if ( classPathElements.contains( jm.getBundleFileName() ) )
                     {
                         classPathElements.set( classPathElements.indexOf( jm.getBundleFileName() ), jm.getUri() );
                     }
                     else
                     {
                         classPathElements.add( jm.getUri() );
                     }
                 }
             }
             classPath.setValue( StringUtils.join( classPathElements.iterator(), " " ) );
 
             // Write the manifest to disk
             PrintWriter pw = new PrintWriter( manifestFile );
             mf.write( pw );
             pw.close();
 
             if ( original.isFile() )
             {
                 // Pack up the archive again from the work directory
                 if ( !original.delete() )
                 {
                     getLog().error( "Could not delete original artifact file " + original );
                 }
 
                 getLog().debug( "Zipping module" );
                 zipArchiver.setDestFile( original );
                 zipArchiver.addDirectory( workDirectory );
                 zipArchiver.createArchive();
             }
         }
         catch ( ManifestException e )
         {
             throw new MojoFailureException( e.getMessage() );
         }
         catch ( ZipException e )
         {
             throw new MojoFailureException( e.getMessage() );
         }
         catch ( IOException e )
         {
             throw new MojoFailureException( e.getMessage() );
         }
         catch ( ArchiverException e )
         {
             throw new MojoFailureException( e.getMessage() );
         }
     }
 }