package org.apache.maven.plugin.assembly.archive.phase;
   
   /*
    * 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 java.io.File;
  import java.util.Iterator;
  import java.util.List;
  
  import org.apache.maven.plugin.assembly.AssemblerConfigurationSource;
  import org.apache.maven.plugin.assembly.AssemblyContext;
  import org.apache.maven.plugin.assembly.InvalidAssemblerConfigurationException;
  import org.apache.maven.plugin.assembly.archive.ArchiveCreationException;
  import org.apache.maven.plugin.assembly.archive.phase.wrappers.RepoBuilderConfigSourceWrapper;
  import org.apache.maven.plugin.assembly.archive.phase.wrappers.RepoInfoWrapper;
  import org.apache.maven.plugin.assembly.archive.task.AddDirectoryTask;
  import org.apache.maven.plugin.assembly.format.AssemblyFormattingException;
  import org.apache.maven.plugin.assembly.model.Assembly;
  import org.apache.maven.plugin.assembly.model.Repository;
  import org.apache.maven.plugin.assembly.utils.AssemblyFormatUtils;
  import org.apache.maven.plugin.assembly.utils.TypeConversionUtils;
  import org.apache.maven.shared.repository.RepositoryAssembler;
  import org.apache.maven.shared.repository.RepositoryAssemblyException;
  import org.apache.maven.shared.repository.RepositoryBuilderConfigSource;
  import org.apache.maven.shared.repository.model.RepositoryInfo;
  import org.codehaus.plexus.archiver.Archiver;
  import org.codehaus.plexus.component.annotations.Component;
  import org.codehaus.plexus.component.annotations.Requirement;
  import org.codehaus.plexus.logging.AbstractLogEnabled;
  
  /**
   * @version $Id: RepositoryAssemblyPhase.java 1163853 2011-08-31 22:42:32Z jdcasey $
   */
  @Component( role = AssemblyArchiverPhase.class, hint = "repositories" )
  public class RepositoryAssemblyPhase
      extends AbstractLogEnabled
      implements AssemblyArchiverPhase
  {
  
      @Requirement
      private RepositoryAssembler repositoryAssembler;
  
      public RepositoryAssemblyPhase()
      {
          // used for plexus.
      }
  
      // introduced for testing.
      public RepositoryAssemblyPhase( final RepositoryAssembler repositoryAssembler )
      {
          this.repositoryAssembler = repositoryAssembler;
      }
  
      /**
       * {@inheritDoc}
       */
      public void execute( final Assembly assembly, final Archiver archiver,
                           final AssemblerConfigurationSource configSource, final AssemblyContext context )
          throws ArchiveCreationException, AssemblyFormattingException, InvalidAssemblerConfigurationException
      {
          final List<Repository> repositoriesList = assembly.getRepositories();
  
          final File tempRoot = configSource.getTemporaryRootDirectory();
  
          for ( final Iterator<Repository> i = repositoriesList.iterator(); i.hasNext(); )
          {
              final Repository repository = i.next();
  
              final String outputDirectory =
                  AssemblyFormatUtils.getOutputDirectory( repository.getOutputDirectory(), configSource.getProject(),
                                                          null, configSource.getFinalName(), configSource );
  
              final File repositoryDirectory = new File( tempRoot, outputDirectory );
  
              if ( !repositoryDirectory.exists() )
              {
                  repositoryDirectory.mkdirs();
              }
  
              try
              {
                  getLogger().debug( "Assembling repository to: " + repositoryDirectory );
                  repositoryAssembler.buildRemoteRepository( repositoryDirectory, wrap( repository ), wrap( configSource ) );
                 getLogger().debug( "Finished assembling repository to: " + repositoryDirectory );
             }
             catch ( final RepositoryAssemblyException e )
             {
                 throw new ArchiveCreationException( "Failed to assemble repository: " + e.getMessage(), e );
             }
 
             final AddDirectoryTask task = new AddDirectoryTask( repositoryDirectory );
 
             final int dirMode = TypeConversionUtils.modeToInt( repository.getDirectoryMode(), getLogger() );
             if ( dirMode != -1 )
             {
                 task.setDirectoryMode( dirMode );
             }
 
             final int fileMode = TypeConversionUtils.modeToInt( repository.getFileMode(), getLogger() );
             if ( fileMode != -1 )
             {
                 task.setFileMode( fileMode );
             }
 
             task.setOutputDirectory( outputDirectory );
 
             task.execute( archiver, configSource );
         }
     }
 
     private RepositoryBuilderConfigSource wrap( final AssemblerConfigurationSource configSource )
     {
         return new RepoBuilderConfigSourceWrapper( configSource );
     }
 
     private RepositoryInfo wrap( final Repository repository )
     {
         return new RepoInfoWrapper( repository );
     }
 
 }