| /** |
| * Licensed 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 |
| * <p> |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * <p> |
| * 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. |
| */ |
| package org.apache.winegrower.extension.build.maven; |
| |
| import org.apache.maven.model.Dependency; |
| import org.apache.maven.plugin.AbstractMojo; |
| import org.apache.maven.plugins.annotations.Component; |
| import org.apache.maven.plugins.annotations.Parameter; |
| import org.apache.maven.project.DefaultDependencyResolutionRequest; |
| import org.apache.maven.project.DependencyResolutionException; |
| import org.apache.maven.project.DependencyResolutionRequest; |
| import org.apache.maven.project.MavenProject; |
| import org.apache.maven.project.ProjectDependenciesResolver; |
| import org.apache.maven.shared.dependency.graph.DependencyGraphBuilder; |
| import org.eclipse.aether.RepositorySystem; |
| import org.eclipse.aether.RepositorySystemSession; |
| import org.eclipse.aether.artifact.DefaultArtifact; |
| import org.eclipse.aether.graph.DependencyNode; |
| import org.eclipse.aether.graph.DependencyVisitor; |
| import org.eclipse.aether.impl.ArtifactResolver; |
| import org.eclipse.aether.repository.LocalRepositoryManager; |
| import org.eclipse.aether.repository.RemoteRepository; |
| import org.eclipse.aether.resolution.ArtifactRequest; |
| import org.eclipse.aether.resolution.ArtifactResolutionException; |
| import org.eclipse.aether.resolution.ArtifactResult; |
| |
| import java.io.File; |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.List; |
| import java.util.stream.Stream; |
| |
| import static java.util.Optional.ofNullable; |
| |
| public abstract class LibsMojo extends AbstractMojo { |
| @Parameter(defaultValue = "${repositorySystemSession}", readonly = true) |
| private RepositorySystemSession session; |
| |
| @Parameter(defaultValue = "${project.remoteProjectRepositories}", readonly = true) |
| private List<RemoteRepository> remoteRepositories; |
| |
| @Component |
| private RepositorySystem repositorySystem; |
| |
| @Component |
| private ArtifactResolver resolver; |
| |
| @Component |
| private ProjectDependenciesResolver dependenciesResolver; |
| |
| @Component |
| private DependencyGraphBuilder graphBuilder; |
| |
| @Parameter(property = "winegrower.libs") |
| private Collection<String> libs; |
| |
| protected Stream<File> collectLibs() { |
| return ofNullable(libs) |
| .map(value -> value.stream().flatMap(l -> { |
| final boolean transitive = l.endsWith("?transitive"); |
| final String coords = transitive ? l.substring(0, l.length() - "?transitive".length()) : l; |
| final String[] c = coords.split(":"); |
| if (c.length < 3 || c.length > 5) { |
| throw new IllegalArgumentException("libs syntax is groupId:artifactId:version[:classifier][:type[?transitive]]"); |
| } |
| if (!transitive) { |
| return Stream.of(resolve(c[0], c[1], c[2], c.length == 4 ? c[3] : "")); |
| } else { |
| return collectTransitiveDependencies(new Dependency() {{ |
| setGroupId(c[0]); |
| setArtifactId(c[1]); |
| setVersion(c[2]); |
| if (c.length == 4 && !"-".equals(c[3])) { |
| setClassifier(c[3]); |
| } |
| if (c.length == 5) { |
| setType(c[4]); |
| } |
| }}); |
| } |
| }).filter(it -> !it.getName().endsWith(".pom"))) |
| .orElseGet(Stream::empty); |
| } |
| |
| protected Stream<File> collectTransitiveDependencies(final Dependency dependency) { |
| final DependencyResolutionRequest request = new DefaultDependencyResolutionRequest(); |
| request.setMavenProject(new MavenProject() {{ |
| getDependencies().add(dependency); |
| }}); |
| request.setRepositorySession(session); |
| try { |
| final Collection<File> files = new ArrayList<>(); |
| dependenciesResolver.resolve(request).getDependencyGraph().accept(new DependencyVisitor() { |
| @Override |
| public boolean visitEnter(final DependencyNode node) { |
| return true; |
| } |
| |
| @Override |
| public boolean visitLeave(final DependencyNode node) { |
| final org.eclipse.aether.artifact.Artifact artifact = node.getArtifact(); |
| files.add(artifact.getFile()); |
| return true; |
| } |
| }); |
| return files.stream(); |
| } catch (final DependencyResolutionException e) { |
| throw new IllegalStateException(e.getMessage(), e); |
| } |
| } |
| |
| protected File resolve(final String group, final String artifact, final String version, final String classifier) { |
| final DefaultArtifact art = new DefaultArtifact(group, artifact, classifier, "jar", version); |
| final ArtifactRequest artifactRequest = new ArtifactRequest().setArtifact(art).setRepositories(remoteRepositories); |
| |
| final LocalRepositoryManager lrm = session.getLocalRepositoryManager(); |
| art.setFile(new File(lrm.getRepository().getBasedir(), lrm.getPathForLocalArtifact(artifactRequest.getArtifact()))); |
| |
| try { |
| final ArtifactResult result = repositorySystem.resolveArtifact(session, artifactRequest); |
| if (result.isMissing()) { |
| throw new IllegalStateException("Can't find commons-cli, please add it to the pom."); |
| } |
| return result.getArtifact().getFile(); |
| } catch (final ArtifactResolutionException e) { |
| throw new IllegalStateException(e.getMessage(), e); |
| } |
| } |
| } |