package org.apache.maven.archetype.downloader;
   
   /*
    * 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.artifact.Artifact;
  import org.apache.maven.artifact.factory.ArtifactFactory;
  import org.apache.maven.artifact.repository.ArtifactRepository;
  import org.apache.maven.artifact.resolver.ArtifactNotFoundException;
  import org.apache.maven.artifact.resolver.ArtifactResolutionException;
  import org.apache.maven.artifact.resolver.ArtifactResolver;
  import org.codehaus.plexus.component.annotations.Component;
  import org.codehaus.plexus.component.annotations.Requirement;
  
  import java.io.File;
  import java.util.ArrayList;
  import java.util.List;
  
  /**
   * @author Jason van Zyl
   */
  @Component( role = Downloader.class )
  public class DefaultDownloader
      implements Downloader
  {
      @Requirement
      private ArtifactResolver artifactResolver;
  
      @Requirement
      private ArtifactFactory artifactFactory;
  
      public File download( String groupId, String artifactId, String version, ArtifactRepository archetypeRepository,
                            ArtifactRepository localRepository, List<ArtifactRepository> remoteRepositories )
          throws DownloadException, DownloadNotFoundException
     {
          Artifact artifact =
              artifactFactory.createArtifact( groupId, artifactId, version, Artifact.SCOPE_RUNTIME, "jar" );
          Artifact artifactPom =
              artifactFactory.createArtifact( groupId, artifactId, version, Artifact.SCOPE_RUNTIME, "pom" );
  
          List<ArtifactRepository> repositories = new ArrayList<ArtifactRepository>( remoteRepositories );
          if ( repositories.isEmpty() && archetypeRepository != null )
          {
              repositories.add( archetypeRepository );
          }
          else if ( repositories.isEmpty() && localRepository != null )
          {
              repositories.add( localRepository );
          }
  
          ArtifactRepository localRepo = localRepository;
          try
          {
              artifactResolver.resolve( artifact, repositories, localRepo );
          }
          catch ( ArtifactResolutionException e )
          {
              throw new DownloadException( "Error downloading " + artifact.getId() + ".", e );
          }
          catch ( ArtifactNotFoundException e )
          {
              throw new DownloadNotFoundException( "Requested " + artifact.getId() + " download does not exist.", e );
          }
          try
          {
              artifactResolver.resolve( artifactPom, repositories, localRepo );
          }
          catch ( ArtifactResolutionException e )
          {
              throw new DownloadException( "Error downloading POM for " + artifact.getId() + ".", e );
          }
          catch ( ArtifactNotFoundException e )
          {
              throw new DownloadNotFoundException( "Requested " + artifact.getId() + " download's POM does not exist.", e );
          }
  
          return artifact.getFile();
      }
  
      public File downloadOld( String groupId, String artifactId, String version, ArtifactRepository archetypeRepository,
                               ArtifactRepository localRepository, List<ArtifactRepository> remoteRepositories )
          throws DownloadException, DownloadNotFoundException
      {
         Artifact artifact =
             artifactFactory.createArtifact( groupId, artifactId, version, Artifact.SCOPE_RUNTIME, "jar" );
         try
         {
             artifactResolver.resolve( artifact, remoteRepositories, localRepository );
         }
         catch ( ArtifactResolutionException e )
         {
             throw new DownloadException( "Error downloading " + artifact.getId() + ".", e );
         }
         catch ( ArtifactNotFoundException e )
         {
             throw new DownloadNotFoundException( "Requested " + artifact.getId() + " download does not exist.", e );
         }
 
         return artifact.getFile();
     }
 }