package org.apache.maven.plugin.assembly.archive.task;

 /*

  * 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.plugin.assembly.AssemblerConfigurationSource;

 import org.apache.maven.plugin.assembly.archive.ArchiveCreationException;

 import org.apache.maven.plugin.assembly.format.AssemblyFormattingException;

 import org.apache.maven.plugin.assembly.format.FileSetFormatter;

 import org.apache.maven.plugin.assembly.model.FileSet;

 import org.apache.maven.plugin.assembly.utils.AssemblyFormatUtils;

 import org.apache.maven.plugin.assembly.utils.TypeConversionUtils;

 import org.apache.maven.project.MavenProject;

 import org.codehaus.plexus.archiver.Archiver;

 import org.codehaus.plexus.logging.Logger;

 import org.codehaus.plexus.logging.console.ConsoleLogger;

 import java.io.File;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.Arrays;

 import java.util.Iterator;

 import java.util.List;

 /**

  * @version $Id: AddFileSetsTask.java 999624 2010-09-21 20:40:03Z jdcasey $

  */

 public class AddFileSetsTask

     implements ArchiverTask

 {

     private final List<FileSet> fileSets;

     private Logger logger;

     private MavenProject project;

     private MavenProject moduleProject;

     public AddFileSetsTask( final List<FileSet> fileSets )

     {

         this.fileSets = fileSets;

     }

     public AddFileSetsTask( final FileSet... fileSets )

     {

         this.fileSets = new ArrayList<FileSet>( Arrays.asList( fileSets ) );

     }

     public void execute( final Archiver archiver, final AssemblerConfigurationSource configSource )

         throws ArchiveCreationException, AssemblyFormattingException

     {

         // don't need this check here. it's more efficient here, but the logger is not actually

         // used until addFileSet(..)...and the check should be there in case someone extends the

         // class.

         // checkLogger();

         final File archiveBaseDir = configSource.getArchiveBaseDirectory();

         if ( archiveBaseDir != null )

         {

             if ( !archiveBaseDir.exists() )

             {

                 throw new ArchiveCreationException( "The archive base directory '" + archiveBaseDir.getAbsolutePath()

                                 + "' does not exist" );

             }

             else if ( !archiveBaseDir.isDirectory() )

             {

                 throw new ArchiveCreationException( "The archive base directory '" + archiveBaseDir.getAbsolutePath()

                                 + "' exists, but it is not a directory" );

             }

         }

         for ( final Iterator<FileSet> i = fileSets.iterator(); i.hasNext(); )

         {

             final FileSet fileSet = i.next();

             addFileSet( fileSet, archiver, configSource, archiveBaseDir );

         }

     }

     protected void addFileSet( final FileSet fileSet, final Archiver archiver,

                                final AssemblerConfigurationSource configSource, final File archiveBaseDir )

         throws AssemblyFormattingException, ArchiveCreationException

     {

         // throw this check in just in case someone extends this class...

         checkLogger();

         final FileSetFormatter fileSetFormatter = new FileSetFormatter( configSource, logger );

         if ( project == null )

         {

             project = configSource.getProject();

         }

         final File basedir = project.getBasedir();

         String destDirectory = fileSet.getOutputDirectory();

         if ( destDirectory == null )

         {

             destDirectory = fileSet.getDirectory();

         }

         destDirectory =

             AssemblyFormatUtils.getOutputDirectory( destDirectory, configSource.getProject(), moduleProject, project,

                                                     configSource.getFinalName(), configSource );

         if ( logger.isDebugEnabled() )

         {

             logger.debug( "FileSet[" + destDirectory + "]" + " dir perms: "

                             + Integer.toString( archiver.getOverrideDirectoryMode(), 8 ) + " file perms: "

                             + Integer.toString( archiver.getOverrideFileMode(), 8 )

                             + ( fileSet.getLineEnding() == null ? "" : " lineEndings: " + fileSet.getLineEnding() ) );

         }

         logger.debug( "The archive base directory is '" + archiveBaseDir + "'" );

         File fileSetDir = getFileSetDirectory( fileSet, basedir, archiveBaseDir );

         if ( fileSetDir.exists() )

         {

             try

             {

                 fileSetDir = fileSetFormatter.formatFileSetForAssembly( fileSetDir, fileSet );

             }

             catch ( final IOException e )

             {

                 throw new ArchiveCreationException( "Error fixing file-set line endings for assembly: "

                                 + e.getMessage(), e );

             }

             logger.debug( "Adding file-set from directory: '" + fileSetDir.getAbsolutePath()

                             + "'\nassembly output directory is: \'" + destDirectory + "\'" );

             final AddDirectoryTask task = new AddDirectoryTask( fileSetDir );

             final int dirMode = TypeConversionUtils.modeToInt( fileSet.getDirectoryMode(), logger );

             if ( dirMode != -1 )

             {

                 task.setDirectoryMode( dirMode );

             }

             final int fileMode = TypeConversionUtils.modeToInt( fileSet.getFileMode(), logger );

             if ( fileMode != -1 )

             {

                 task.setFileMode( fileMode );

             }

             task.setUseDefaultExcludes( fileSet.isUseDefaultExcludes() );

             final List<String> excludes = fileSet.getExcludes();

             excludes.add( "**/*.filtered" );

             excludes.add( "**/*.formatted" );

             task.setExcludes( excludes );

             task.setIncludes( fileSet.getIncludes() );

             task.setOutputDirectory( destDirectory );

             task.execute( archiver, configSource );

         }

     }

     protected File getFileSetDirectory( final FileSet fileSet, final File basedir, final File archiveBaseDir )

         throws ArchiveCreationException, AssemblyFormattingException

     {

         String sourceDirectory = fileSet.getDirectory();

         if ( sourceDirectory == null || sourceDirectory.trim()

                                                        .length() < 1 )

         {

             sourceDirectory = basedir.getAbsolutePath();

         }

         File fileSetDir = null;

         if ( archiveBaseDir == null )

         {

             fileSetDir = new File( sourceDirectory );

             if ( !fileSetDir.isAbsolute() )

             {

                 fileSetDir = new File( basedir, sourceDirectory );

             }

         }

         else

         {

             fileSetDir = new File( archiveBaseDir, sourceDirectory );

         }

         return fileSetDir;

     }

     private void checkLogger()

     {

         if ( logger == null )

         {

             logger = new ConsoleLogger( Logger.LEVEL_INFO, "AddFileSetsTask-internal" );

         }

     }

     public void setLogger( final Logger logger )

     {

         this.logger = logger;

     }

     public void setProject( final MavenProject project )

     {

         this.project = project;

     }

     public MavenProject getModuleProject()

     {

         return moduleProject;

     }

     public void setModuleProject( final MavenProject moduleProject )

     {

         this.moduleProject = moduleProject;

     }

 }