scala_maven.MavenArtifactResolver Maven / Gradle / Ivy

Go to download
/*
 * This is free and unencumbered software released into the public domain.
 * See UNLICENSE.
 */
package scala_maven;

import java.util.NoSuchElementException;
import java.util.Set;
import org.apache.maven.artifact.Artifact;
import org.apache.maven.artifact.resolver.ArtifactResolutionRequest;
import org.apache.maven.execution.MavenSession;
import org.apache.maven.repository.RepositorySystem;

public class MavenArtifactResolver {

  /** Constant {@link String} for "pom". Used to specify the Maven POM artifact type. */
  public static final String POM = "pom";
  /** Constant {@link String} for "jar". Used to specify the Maven JAR artifact type. */
  public static final String JAR = "jar";

  private final RepositorySystem repositorySystem;
  private final MavenSession session;

  public MavenArtifactResolver(RepositorySystem repositorySystem, MavenSession session) {
    this.repositorySystem = repositorySystem;
    this.session = session;
  }

  public Artifact getJar(String groupId, String artifactId, String version, String classifier) {
    Artifact artifact = createJarArtifact(groupId, artifactId, version, classifier);
    Set resolvedArtifacts = resolve(artifact, false);
    if (resolvedArtifacts.isEmpty()) {
      throw new NoSuchElementException(
          String.format(
              "Could not resolve artifact %s:%s:%s:%s", groupId, artifactId, version, classifier));
    }
    return resolvedArtifacts.iterator().next();
  }

  public Set getJarAndDependencies(
      String groupId, String artifactId, String version, String classifier) {
    Artifact artifact = createJarArtifact(groupId, artifactId, version, classifier);
    return resolve(artifact, true);
  }

  private Artifact createJarArtifact(
      String groupId, String artifactId, String version, String classifier) {
    return classifier == null
        ? repositorySystem.createArtifact(groupId, artifactId, version, JAR)
        : repositorySystem.createArtifactWithClassifier(
            groupId, artifactId, version, JAR, classifier);
  }

  private Set resolve(Artifact artifact, boolean transitively) {
    ArtifactResolutionRequest request =
        new ArtifactResolutionRequest()
            .setArtifact(artifact)
            .setResolveRoot(true)
            .setResolveTransitively(transitively)
            .setServers(session.getRequest().getServers())
            .setMirrors(session.getRequest().getMirrors())
            .setProxies(session.getRequest().getProxies())
            .setLocalRepository(session.getLocalRepository())
            .setRemoteRepositories(session.getCurrentProject().getRemoteArtifactRepositories());
    return repositorySystem.resolve(request).getArtifacts();
  }
}