1 /** 2 * Portions of this software was developed by employees of the National Institute 3 * of Standards and Technology (NIST), an agency of the Federal Government and is 4 * being made available as a public service. Pursuant to title 17 United States 5 * Code Section 105, works of NIST employees are not subject to copyright 6 * protection in the United States. This software may be subject to foreign 7 * copyright. Permission in the United States and in foreign countries, to the 8 * extent that NIST may hold copyright, to use, copy, modify, create derivative 9 * works, and distribute this software and its documentation without fee is hereby 10 * granted on a non-exclusive basis, provided that this notice and disclaimer 11 * of warranty appears in all copies. 12 * 13 * THE SOFTWARE IS PROVIDED 'AS IS' WITHOUT ANY WARRANTY OF ANY KIND, EITHER 14 * EXPRESSED, IMPLIED, OR STATUTORY, INCLUDING, BUT NOT LIMITED TO, ANY WARRANTY 15 * THAT THE SOFTWARE WILL CONFORM TO SPECIFICATIONS, ANY IMPLIED WARRANTIES OF 16 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND FREEDOM FROM 17 * INFRINGEMENT, AND ANY WARRANTY THAT THE DOCUMENTATION WILL CONFORM TO THE 18 * SOFTWARE, OR ANY WARRANTY THAT THE SOFTWARE WILL BE ERROR FREE. IN NO EVENT 19 * SHALL NIST BE LIABLE FOR ANY DAMAGES, INCLUDING, BUT NOT LIMITED TO, DIRECT, 20 * INDIRECT, SPECIAL OR CONSEQUENTIAL DAMAGES, ARISING OUT OF, RESULTING FROM, 21 * OR IN ANY WAY CONNECTED WITH THIS SOFTWARE, WHETHER OR NOT BASED UPON WARRANTY, 22 * CONTRACT, TORT, OR OTHERWISE, WHETHER OR NOT INJURY WAS SUSTAINED BY PERSONS OR 23 * PROPERTY OR OTHERWISE, AND WHETHER OR NOT LOSS WAS SUSTAINED FROM, OR AROSE OUT 24 * OF THE RESULTS OF, OR USE OF, THE SOFTWARE OR SERVICES PROVIDED HEREUNDER. 25 */ 26 27 package gov.nist.secauto.swid.plugin.entry.archive; 28 29 import org.apache.maven.artifact.Artifact; 30 import org.apache.maven.project.MavenProject; 31 import org.codehaus.plexus.archiver.ArchiveEntry; 32 33 import java.nio.file.Path; 34 import java.nio.file.Paths; 35 import java.util.HashMap; 36 import java.util.Map; 37 import java.util.function.Predicate; 38 39 public class ArtifactMap { 40 41 protected static Map<String, Artifact> initialize(MavenProject project) { 42 Map<String, Artifact> retval = new HashMap<>(); 43 Artifact target = project.getArtifact(); 44 retval.put(target.getFile().getAbsolutePath(), target); 45 46 project.getArtifacts().stream().filter(includeOnlyCompileAndRuntime()) 47 .forEach(a -> retval.put(a.getFile().getAbsolutePath(), a)); 48 return retval; 49 } 50 51 protected static Predicate<Artifact> includeOnlyCompileAndRuntime() { 52 return new Predicate<Artifact>() { 53 54 @Override 55 public boolean test(Artifact artifact) { 56 String scope = artifact.getScope(); 57 return Artifact.SCOPE_COMPILE.equals(scope) || Artifact.SCOPE_RUNTIME.equals(scope); 58 } 59 }; 60 } 61 62 private final Map<String, Artifact> artifactFileToArtifactMap; 63 64 protected ArtifactMap(Map<String, Artifact> map) { 65 artifactFileToArtifactMap = map; 66 } 67 68 public ArtifactMap(MavenProject project) { 69 this(initialize(project)); 70 } 71 72 /** 73 * Retrieve the artifact for an archiver entry. 74 * 75 * @param entry 76 * the archive entry 77 * @return the matching artifact or {@code null} if a matching artifact was not found 78 */ 79 public Artifact lookupArtifactForArchiveEntry(ArchiveEntry entry) { 80 Path entryPath = Paths.get(entry.getResource().getName()); 81 String path = entryPath.toString(); 82 return artifactFileToArtifactMap.get(path); 83 } 84 }