package org.apache.maven.plugin.surefire;
   
   /*
    * 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 javax.annotation.Nonnull;
  import javax.annotation.Nullable;
  
  import java.util.Collection;
  import java.util.Iterator;
  import java.util.LinkedHashMap;
  import java.util.LinkedHashSet;
  import java.util.List;
  import java.util.Map;
  import java.util.Set;
  
  import org.apache.maven.artifact.Artifact;
  import org.apache.maven.artifact.repository.ArtifactRepository;
  import org.apache.maven.artifact.resolver.ArtifactResolutionException;
  import org.apache.maven.artifact.resolver.ArtifactResolutionRequest;
  import org.apache.maven.artifact.resolver.ArtifactResolutionResult;
  import org.apache.maven.artifact.resolver.ResolutionErrorHandler;
  import org.apache.maven.artifact.resolver.filter.ArtifactFilter;
  import org.apache.maven.artifact.versioning.DefaultArtifactVersion;
  import org.apache.maven.artifact.versioning.InvalidVersionSpecificationException;
  import org.apache.maven.artifact.versioning.OverConstrainedVersionException;
  import org.apache.maven.artifact.versioning.VersionRange;
  import org.apache.maven.model.Dependency;
  import org.apache.maven.model.Plugin;
  import org.apache.maven.plugin.MojoExecutionException;
  import org.apache.maven.plugin.surefire.log.api.ConsoleLogger;
  import org.apache.maven.repository.RepositorySystem;
  
  import static java.util.Arrays.asList;
  import static org.apache.maven.artifact.Artifact.SCOPE_COMPILE;
  import static org.apache.maven.artifact.Artifact.SCOPE_COMPILE_PLUS_RUNTIME;
  import static org.apache.maven.artifact.Artifact.SCOPE_RUNTIME;
  import static org.apache.maven.artifact.ArtifactUtils.artifactMapByVersionlessId;
  import static org.apache.maven.artifact.versioning.VersionRange.createFromVersionSpec;
  
  /**
   * Does dependency resolution and artifact handling for the surefire plugin.
   *
   * @author Stephen Connolly
   * @author Kristian Rosenvold
   */
  final class SurefireDependencyResolver
  {
      static final String PROVIDER_GROUP_ID = "org.apache.maven.surefire";
  
      private static final String[] PROVIDER_CLASSPATH_ORDER = {
          "surefire-junit3",
          "surefire-junit4",
          "surefire-junit47",
          "surefire-testng",
          "surefire-junit-platform",
          "surefire-api",
          "surefire-logger-api",
          "surefire-shared-utils",
          "common-java5",
          "common-junit3",
          "common-junit4",
          "common-junit48",
          "common-testng-utils"
      };
  
      private final RepositorySystem repositorySystem;
  
      private final ConsoleLogger log;
  
      private final ArtifactRepository localRepository;
  
      private final List<ArtifactRepository> pluginRemoteRepositories;
  
      private final List<ArtifactRepository> projectRemoteRepositories;
  
      private final ResolutionErrorHandler resolutionErrorHandler;
  
      private final String pluginName;
  
      private final boolean offline;
  
      SurefireDependencyResolver( RepositorySystem repositorySystem, ConsoleLogger log,
                                 ArtifactRepository localRepository,
                                 List<ArtifactRepository> pluginRemoteRepositories,
                                 List<ArtifactRepository> projectRemoteRepositories,
                                 ResolutionErrorHandler resolutionErrorHandler,
                                 String pluginName, boolean offline )
     {
         this.repositorySystem = repositorySystem;
         this.log = log;
         this.localRepository = localRepository;
         this.pluginRemoteRepositories = pluginRemoteRepositories;
         this.projectRemoteRepositories = projectRemoteRepositories;
         this.resolutionErrorHandler = resolutionErrorHandler;
         this.pluginName = pluginName;
         this.offline = offline;
     }
 
     static boolean isWithinVersionSpec( @Nullable Artifact artifact, @Nonnull String versionSpec )
     {
         if ( artifact == null )
         {
             return false;
         }
         try
         {
             VersionRange range = createFromVersionSpec( versionSpec );
             try
             {
                 return range.containsVersion( artifact.getSelectedVersion() );
             }
             catch ( NullPointerException e )
             {
                 return range.containsVersion( new DefaultArtifactVersion( artifact.getBaseVersion() ) );
             }
         }
         catch ( InvalidVersionSpecificationException | OverConstrainedVersionException e )
         {
             throw new RuntimeException( "Bug in plugin. Please report with stacktrace" );
         }
     }
 
     Map<String, Artifact> resolvePluginDependencies( Plugin plugin, Map<String, Artifact> pluginResolvedDependencies )
         throws MojoExecutionException
     {
         Map<String, Artifact> resolved = new LinkedHashMap<>();
         Collection<Dependency> pluginDependencies = plugin.getDependencies();
 
         for ( Dependency dependency : pluginDependencies )
         {
             Artifact dependencyArtifact = repositorySystem.createDependencyArtifact( dependency );
             ArtifactResolutionResult artifactResolutionResult = resolvePluginArtifact( dependencyArtifact );
             for ( Artifact artifact : artifactResolutionResult.getArtifacts() )
             {
                 String key = artifact.getGroupId() + ":" + artifact.getArtifactId();
                 Artifact resolvedPluginDependency = pluginResolvedDependencies.get( key );
                 if ( resolvedPluginDependency != null )
                 {
                     resolved.put( key, artifact );
                 }
             }
         }
         return resolved;
     }
 
     ArtifactResolutionResult resolvePluginArtifact( Artifact artifact ) throws MojoExecutionException
     {
         return resolvePluginArtifact( artifact, new RuntimeArtifactFilter() );
     }
 
     ArtifactResolutionResult resolveProjectArtifact( Artifact artifact ) throws MojoExecutionException
     {
         return resolveProjectArtifact( artifact, new RuntimeArtifactFilter() );
     }
 
     private ArtifactResolutionResult resolvePluginArtifact( Artifact artifact, ArtifactFilter filter )
         throws MojoExecutionException
     {
         return resolveArtifact( artifact, pluginRemoteRepositories, filter );
     }
 
     private ArtifactResolutionResult resolveProjectArtifact( Artifact artifact, ArtifactFilter filter )
         throws MojoExecutionException
     {
         return resolveArtifact( artifact, projectRemoteRepositories, filter );
     }
 
     private ArtifactResolutionResult resolveArtifact( Artifact artifact, List<ArtifactRepository> repositories,
                                                       ArtifactFilter filter ) throws MojoExecutionException
     {
         ArtifactResolutionRequest request = new ArtifactResolutionRequest()
             .setOffline( offline )
             .setArtifact( artifact )
             .setLocalRepository( localRepository )
             .setResolveTransitively( true )
             .setCollectionFilter( filter )
             .setRemoteRepositories( repositories );
 
         ArtifactResolutionResult result = repositorySystem.resolve( request );
         try
         {
             resolutionErrorHandler.throwErrors( request, result );
         }
         catch ( ArtifactResolutionException e )
         {
             throw new MojoExecutionException( e.getMessage(), e );
         }
 
         return result;
     }
 
     @Nonnull
     Set<Artifact> getProviderClasspath( String providerArtifactId, String providerVersion )
         throws MojoExecutionException
     {
         Dependency provider = toProviderDependency( providerArtifactId, providerVersion );
 
         Artifact providerArtifact = repositorySystem.createDependencyArtifact( provider );
 
         ArtifactResolutionResult result = resolvePluginArtifact( providerArtifact );
 
         if ( log.isDebugEnabled() )
         {
             for ( Artifact artifact : result.getArtifacts() )
             {
                 String artifactPath = artifact.getFile().getAbsolutePath();
                 String scope = artifact.getScope();
                 log.debug( "Adding to " + pluginName + " test classpath: " + artifactPath + " Scope: " + scope );
             }
         }
 
         return orderProviderArtifacts( result.getArtifacts() );
     }
 
     @Nonnull
     Map<String, Artifact> getProviderClasspathAsMap( String providerArtifactId, String providerVersion )
         throws MojoExecutionException
     {
         return artifactMapByVersionlessId( getProviderClasspath( providerArtifactId, providerVersion ) );
     }
 
     // FIXME
     // method argument should be unchanged
     // what if providerArtifacts will be unmodifiable
     private static Set<Artifact> orderProviderArtifacts( Set<Artifact> providerArtifacts )
     {
         Set<Artifact> orderedProviderArtifacts = new LinkedHashSet<>();
         for ( String order : PROVIDER_CLASSPATH_ORDER )
         {
             Iterator<Artifact> providerArtifactsIt = providerArtifacts.iterator();
             while ( providerArtifactsIt.hasNext() )
             {
                 Artifact providerArtifact = providerArtifactsIt.next();
                 if ( providerArtifact.getArtifactId().equals( order ) )
                 {
                     orderedProviderArtifacts.add( providerArtifact );
                     providerArtifactsIt.remove();
                 }
             }
         }
         orderedProviderArtifacts.addAll( providerArtifacts );
         return orderedProviderArtifacts;
     }
 
     private static Dependency toProviderDependency( String providerArtifactId, String providerVersion )
     {
         Dependency dependency = new Dependency();
         dependency.setGroupId( PROVIDER_GROUP_ID );
         dependency.setArtifactId( providerArtifactId );
         dependency.setVersion( providerVersion );
         dependency.setType( "jar" );
         return dependency;
     }
 
     static class RuntimeArtifactFilter implements ArtifactFilter
     {
         private static final Collection<String> SCOPES =
             asList( SCOPE_COMPILE, SCOPE_COMPILE_PLUS_RUNTIME, SCOPE_RUNTIME );
 
         private final Artifact filter;
 
         RuntimeArtifactFilter()
         {
             this( null );
         }
 
         RuntimeArtifactFilter( Artifact filter )
         {
             this.filter = filter;
         }
 
         @Override
         public boolean include( Artifact artifact )
         {
             String scope = artifact.getScope();
             return ( filter == null || artifact.equals( filter ) )
                 && !artifact.isOptional() && ( scope == null || SCOPES.contains( scope ) );
         }
     }
 }