Skip to content

Instantly share code, notes, and snippets.

@raphw

raphw/MavenPomResolver.java

Last active December 8, 2024 12:29
Show Gist options
  • Save raphw/78b0d5264636c3d2a3af339c3259c17c to your computer and use it in GitHub Desktop.
Save raphw/78b0d5264636c3d2a3af339c3259c17c to your computer and use it in GitHub Desktop.
Download ZIP
Maven POM resolution (standalone)
Raw
MavenPomResolver.java
package codes.rafael.mavenpom;
import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Path;
import java.util.*;
import java.util.function.Supplier;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.IntStream;
import java.util.stream.Stream;
public class MavenPomResolver {
private static final String NAMESPACE_4_0_0 = "http://maven.apache.org/POM/4.0.0";
private static final Set<String> IMPLICITS = Set.of("groupId", "artifactId", "version", "packaging");
private static final Pattern PROPERTY = Pattern.compile("(\\$\\{([\\w.]+)})");
private final MavenRepository repository;
private final DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
public MavenPomResolver(MavenRepository repository) {
this.repository = repository;
factory.setNamespaceAware(true);
}
public List<MavenDependency> dependencies(String groupId,
String artifactId,
String version,
MavenDependencyScope scope) throws IOException {
SequencedMap<DependencyKey, DependencyInclusion> dependencies = new LinkedHashMap<>();
Map<DependencyKey, MavenDependencyScope> overrides = new HashMap<>();
Map<DependencyCoordinates, UnresolvedPom> poms = new HashMap<>();
Queue<ContextualPom> queue = new ArrayDeque<>(Set.of(new ContextualPom(
resolve(assembleOrCached(groupId, artifactId, version, new HashSet<>(), poms), poms, null),
scope,
Set.of(),
Map.of())));
do {
ContextualPom current = queue.remove();
Map<DependencyKey, DependencyValue> managedDependencies = new HashMap<>(current.pom().managedDependencies());
managedDependencies.putAll(current.managedDependencies());
for (Map.Entry<DependencyKey, DependencyValue> entry : current.pom().dependencies().entrySet()) {
if (!current.exclusions().contains(new DependencyExclusion(
entry.getKey().groupId(),
entry.getKey().artifactId()))) {
DependencyValue primary = current.managedDependencies().get(entry.getKey()), value = primary == null
? entry.getValue().with(current.pom().managedDependencies().get(entry.getKey()))
: primary.with(entry.getValue());
boolean optional = switch (value.optional()) {
case "true" -> true;
case "false" -> false;
case null -> false;
default -> throw new IllegalStateException("Unexpected value: " + value);
};
if (optional && current.pom().origin() != null) {
continue;
}
MavenDependencyScope actual = toScope(value.scope()), derived = switch (current.scope()) {
case null -> actual == MavenDependencyScope.IMPORT ? null : actual;
case COMPILE -> switch (actual) {
case COMPILE, RUNTIME -> actual;
default -> null;
};
case PROVIDED, RUNTIME, TEST -> switch (actual) {
case COMPILE, RUNTIME -> current.scope();
default -> null;
};
case SYSTEM, IMPORT -> null;
};
if (derived == null) {
continue;
}
DependencyInclusion previous = dependencies.get(entry.getKey());
if (previous != null) {
if (previous.scope().ordinal() > overrides.getOrDefault(entry.getKey(), derived).ordinal()) {
overrides.put(entry.getKey(), derived);
}
continue;
}
dependencies.put(entry.getKey(), new DependencyInclusion(value.version(),
optional,
derived,
value.systemPath() == null ? null : Path.of(value.systemPath()),
new HashSet<>()));
if (current.pom().origin() != null) {
dependencies.get(current.pom().origin()).transitives().add(entry.getKey());
}
Set<DependencyExclusion> exclusions;
if (value.exclusions() == null || value.exclusions().isEmpty()) {
exclusions = current.exclusions();
} else {
exclusions = new HashSet<>(current.exclusions());
exclusions.addAll(value.exclusions());
}
queue.add(new ContextualPom(resolve(assembleOrCached(entry.getKey().groupId(),
entry.getKey().artifactId(),
value.version(),
new HashSet<>(),
poms), poms, entry.getKey()), derived, exclusions, managedDependencies));
}
}
} while (!queue.isEmpty());
Queue<DependencyKey> keys = new ArrayDeque<>(overrides.keySet());
while (!keys.isEmpty()) {
DependencyKey key = keys.remove();
Set<DependencyKey> transitives = dependencies.get(key).transitives();
transitives.forEach(transitive -> {
MavenDependencyScope current = overrides.get(transitive), candidate = overrides.get(key);
if (current == null || candidate.ordinal() < current.ordinal()) {
overrides.put(transitive, candidate);
}
});
keys.addAll(transitives);
}
return dependencies.entrySet().stream().map(entry -> new MavenDependency(entry.getKey().groupId(),
entry.getKey().artifactId(),
entry.getValue().version(),
entry.getKey().type(),
entry.getKey().classifier(),
overrides.getOrDefault(entry.getKey(), entry.getValue().scope()),
entry.getValue().path(),
entry.getValue().optional())).toList();
}
private UnresolvedPom assemble(InputStream inputStream,
Set<DependencyCoordinates> children,
Map<DependencyCoordinates, UnresolvedPom> poms) throws IOException,
SAXException,
ParserConfigurationException {
Document document;
try (inputStream) {
document = factory.newDocumentBuilder().parse(inputStream);
}
return switch (document.getDocumentElement().getNamespaceURI()) {
case NAMESPACE_4_0_0 -> {
DependencyCoordinates parent = toChildren400(document.getDocumentElement(), "parent")
.findFirst()
.map(node -> new DependencyCoordinates(
toTextChild400(node, "groupId").orElseThrow(missing("parent.groupId")),
toTextChild400(node, "artifactId").orElseThrow(missing("parent.artifactId")),
toTextChild400(node, "version").orElseThrow(missing("parent.version"))))
.orElse(null);
Map<String, String> properties = new HashMap<>();
Map<DependencyKey, DependencyValue> managedDependencies = new HashMap<>();
SequencedMap<DependencyKey, DependencyValue> dependencies = new LinkedHashMap<>();
if (parent != null) {
if (!children.add(new DependencyCoordinates(parent.groupId(), parent.artifactId(), parent.version()))) {
throw new IllegalStateException("Circular dependency to "
+ parent.groupId() + ":" + parent.artifactId() + ":" + parent.version());
}
UnresolvedPom resolution = assembleOrCached(parent.groupId(),
parent.artifactId(),
parent.version(),
children,
poms);
properties.putAll(resolution.properties());
IMPLICITS.forEach(property -> {
String value = resolution.properties().get(property);
if (value != null) {
properties.put("parent." + property, value);
properties.put("project.parent." + property, value);
}
});
managedDependencies.putAll(resolution.managedDependencies());
dependencies.putAll(resolution.dependencies());
}
IMPLICITS.forEach(property -> toChildren400(document.getDocumentElement(), property)
.findFirst()
.ifPresent(node -> {
properties.put(property, node.getTextContent());
properties.put("project." + property, node.getTextContent());
}));
toChildren400(document.getDocumentElement(), "properties")
.limit(1)
.flatMap(MavenPomResolver::toChildren)
.filter(node -> node.getNodeType() == Node.ELEMENT_NODE)
.forEach(node -> properties.put(node.getLocalName(), node.getTextContent()));
toChildren400(document.getDocumentElement(), "dependencyManagement")
.limit(1)
.flatMap(node -> toChildren400(node, "dependencies"))
.limit(1)
.flatMap(node -> toChildren400(node, "dependency"))
.map(MavenPomResolver::toDependency400)
.forEach(entry -> managedDependencies.put(entry.getKey(), entry.getValue()));
toChildren400(document.getDocumentElement(), "dependencies")
.limit(1)
.flatMap(node -> toChildren400(node, "dependency"))
.map(MavenPomResolver::toDependency400)
.forEach(entry -> dependencies.putLast(entry.getKey(), entry.getValue()));
yield new UnresolvedPom(properties, managedDependencies, dependencies);
}
case null, default -> throw new IllegalArgumentException(
"Unknown namespace: " + document.getDocumentElement().getNamespaceURI());
};
}
private UnresolvedPom assembleOrCached(String groupId,
String artifactId,
String version,
Set<DependencyCoordinates> children,
Map<DependencyCoordinates, UnresolvedPom> poms) throws IOException {
if (version == null) {
throw new IllegalArgumentException("No version specified for " + groupId + ":" + artifactId);
}
DependencyCoordinates coordinates = new DependencyCoordinates(groupId, artifactId, version);
UnresolvedPom pom = poms.get(coordinates);
if (pom == null) {
try {
pom = assemble(repository.fetch(groupId,
artifactId,
version,
"pom",
null,
null).toInputStream(), children, poms);
} catch (RuntimeException | SAXException | ParserConfigurationException e) {
throw new IllegalStateException("Failed to resolve " + groupId + ":" + artifactId + ":" + version, e);
}
poms.put(coordinates, pom);
}
return pom;
}
private ResolvedPom resolve(UnresolvedPom pom,
Map<DependencyCoordinates, UnresolvedPom> poms,
DependencyKey origin) throws IOException {
Map<DependencyKey, DependencyValue> managedDependencies = new HashMap<>();
for (Map.Entry<DependencyKey, DependencyValue> entry : pom.managedDependencies().entrySet()) {
DependencyKey key = entry.getKey().resolve(pom.properties());
DependencyValue value = entry.getValue().resolve(pom.properties());
if (Objects.equals(value.scope(), "import") && Objects.equals(key.type(), "pom")) {
UnresolvedPom imported = assembleOrCached(key.groupId(),
key.artifactId(),
value.version(),
new HashSet<>(),
poms);
imported.managedDependencies().forEach((importedKey, importedValue) -> managedDependencies.putIfAbsent(
importedKey.resolve(imported.properties()),
importedValue.resolve(imported.properties())));
} else {
managedDependencies.put(key, value);
}
}
SequencedMap<DependencyKey, DependencyValue> dependencies = new LinkedHashMap<>();
pom.dependencies().forEach((key, value) -> dependencies.putLast(
key.resolve(pom.properties()),
value.resolve(pom.properties())));
return new ResolvedPom(managedDependencies, dependencies, origin);
}
private static Stream<Node> toChildren(Node node) {
NodeList children = node.getChildNodes();
return IntStream.iterate(0,
index -> index < children.getLength(),
index -> index + 1).mapToObj(children::item);
}
private static Stream<Node> toChildren400(Node node, String localName) {
return toChildren(node).filter(child -> Objects.equals(child.getLocalName(), localName)
&& Objects.equals(child.getNamespaceURI(), NAMESPACE_4_0_0));
}
private static Optional<String> toTextChild400(Node node, String localName) {
return toChildren400(node, localName).map(Node::getTextContent).findFirst();
}
private static Map.Entry<DependencyKey, DependencyValue> toDependency400(Node node) {
return Map.entry(
new DependencyKey(
toTextChild400(node, "groupId").orElseThrow(missing("groupId")),
toTextChild400(node, "artifactId").orElseThrow(missing("artifactId")),
toTextChild400(node, "type").orElse("jar"),
toTextChild400(node, "classifier").orElse(null)),
new DependencyValue(
toTextChild400(node, "version").orElse(null),
toTextChild400(node, "scope").orElse(null),
toTextChild400(node, "systemPath").orElse(null),
toChildren400(node, "exclusions")
.findFirst()
.map(exclusions -> toChildren400(exclusions, "exclusion")
.map(child -> new DependencyExclusion(
toTextChild400(child, "groupId").orElseThrow(missing("exclusion.groupId")),
toTextChild400(child, "artifactId").orElseThrow(missing("exclusion.artifactId"))))
.toList())
.orElse(null),
toTextChild400(node, "optional").orElse(null)));
}
private static String property(String text, Map<String, String> properties) {
return property(text, properties, Set.of());
}
private static String property(String text, Map<String, String> properties, Set<String> previous) {
if (text != null && text.contains("$")) {
Matcher matcher = PROPERTY.matcher(text);
StringBuilder sb = new StringBuilder();
while (matcher.find()) {
String property = matcher.group(2);
String replacement = properties.get(property);
if (replacement == null) {
replacement = System.getProperty(property);
}
if (replacement == null) {
throw new IllegalStateException("Property not defined: " + property);
} else {
HashSet<String> duplicates = new HashSet<>(previous);
if (!duplicates.add(property)) {
throw new IllegalStateException("Circular property definition of: " + property);
}
matcher.appendReplacement(sb, property(replacement, properties, duplicates));
}
}
return matcher.appendTail(sb).toString();
} else {
return text;
}
}
private static MavenDependencyScope toScope(String scope) {
return switch (scope) {
case "compile" -> MavenDependencyScope.COMPILE;
case "provided" -> MavenDependencyScope.PROVIDED;
case "runtime" -> MavenDependencyScope.RUNTIME;
case "test" -> MavenDependencyScope.TEST;
case "system" -> MavenDependencyScope.SYSTEM;
case "import" -> MavenDependencyScope.IMPORT;
case null -> MavenDependencyScope.COMPILE;
default -> throw new IllegalArgumentException("");
};
}
private static Supplier<IllegalStateException> missing(String property) {
return () -> new IllegalStateException("Property not defined: " + property);
}
private record DependencyKey(String groupId,
String artifactId,
String type,
String classifier) {
private DependencyKey resolve(Map<String, String> properties) {
return new DependencyKey(property(groupId, properties),
property(artifactId, properties),
property(type, properties),
property(classifier, properties));
}
}
private record DependencyValue(String version,
String scope,
String systemPath,
List<DependencyExclusion> exclusions,
String optional) {
private DependencyValue resolve(Map<String, String> properties) {
return new DependencyValue(property(version, properties),
property(scope, properties),
property(systemPath, properties),
exclusions == null ? null : exclusions.stream().map(exclusion -> new DependencyExclusion(
property(exclusion.groupId(), properties),
property(exclusion.artifactId(), properties))).toList(),
property(optional, properties)
);
}
private DependencyValue with(DependencyValue supplement) {
if (supplement == null) {
return this;
}
return new DependencyValue(version == null ? supplement.version() : version,
scope == null ? supplement.scope() : scope,
systemPath == null ? supplement.systemPath() : systemPath,
exclusions == null ? supplement.exclusions() : exclusions,
optional == null ? supplement.optional() : optional);
}
}
private record DependencyInclusion(String version,
boolean optional,
MavenDependencyScope scope,
Path path,
Set<DependencyKey> transitives) {
}
private record DependencyExclusion(String groupId, String artifactId) {
}
private record DependencyCoordinates(String groupId, String artifactId, String version) {
}
private record UnresolvedPom(Map<String, String> properties,
Map<DependencyKey, DependencyValue> managedDependencies,
SequencedMap<DependencyKey, DependencyValue> dependencies) {
}
private record ResolvedPom(Map<DependencyKey, DependencyValue> managedDependencies,
SequencedMap<DependencyKey, DependencyValue> dependencies,
DependencyKey origin) {
}
private record ContextualPom(ResolvedPom pom,
MavenDependencyScope scope,
Set<DependencyExclusion> exclusions,
Map<DependencyKey, DependencyValue> managedDependencies) {
}
}
Raw
MavenRepository.java
package codes.rafael.mavenpom;
import java.io.FilterInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URI;
import java.nio.channels.FileChannel;
import java.nio.file.Files;
import java.nio.file.Path;
import java.security.DigestInputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;
import java.util.Base64;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
public class MavenRepository implements Repository {
private final URI repository;
private final Path local;
private final Map<String, URI> validations;
public MavenRepository() {
String environment = System.getenv("MAVEN_REPOSITORY_URI");
if (environment != null && !environment.endsWith("/")) {
environment += "/";
}
repository = URI.create(environment == null ? "https://repo1.maven.org/maven2/" : environment);
Path local = Path.of(System.getProperty("user.home"), ".m2", "repository");
this.local = Files.isDirectory(local) ? local : null;
validations = Map.of("SHA1", repository);
}
public MavenRepository(URI repository, Path local, Map<String, URI> validations) {
this.repository = repository;
this.local = local;
this.validations = validations;
}
@Override
public InputStreamSource fetch(String coordinate) throws IOException {
String[] elements = coordinate.split(":");
return switch (elements.length) {
case 4 -> fetch(elements[0], elements[1], elements[2], "jar", null, null);
case 5 -> fetch(elements[0], elements[1], elements[2], elements[3], null, null);
case 6 -> fetch(elements[0], elements[1], elements[2], elements[4], elements[3], null);
default -> throw new IllegalArgumentException("Insufficient Maven coordinate: " + coordinate);
};
}
public InputStreamSource fetch(String groupId,
String artifactId,
String version,
String type,
String classifier,
String checksum) throws IOException {
return fetch(repository, groupId, artifactId, version, type, classifier, checksum).materialize();
}
private LazyInputStreamSource fetch(URI repository,
String groupId,
String artifactId,
String version,
String type,
String classifier,
String checksum) throws IOException {
String path = groupId.replace('.', '/')
+ "/" + artifactId
+ "/" + version
+ "/" + artifactId + "-" + version + (classifier == null ? "" : "-" + classifier)
+ "." + type + (checksum == null ? "" : ("." + checksum));
Path cached = local == null ? null : local.resolve(path);
if (cached != null) {
if (Files.exists(cached)) {
boolean valid = true;
if (checksum == null) {
Map<LazyInputStreamSource, byte[]> results = new HashMap<>();
for (Map.Entry<String, URI> entry : validations.entrySet()) {
LazyInputStreamSource source = fetch(entry.getValue(),
groupId,
artifactId,
version,
type,
classifier,
entry.getKey().toLowerCase());
if (valid) {
MessageDigest digest;
try {
digest = MessageDigest.getInstance(entry.getKey());
} catch (NoSuchAlgorithmException e) {
throw new IllegalStateException(e);
}
try (FileChannel channel = FileChannel.open(cached)) {
digest.update(channel.map(FileChannel.MapMode.READ_ONLY, 0, channel.size()));
}
byte[] expected;
try (InputStream inputStream = source.toInputStream()) {
expected = inputStream.readAllBytes();
}
results.put(source, expected);
valid = Arrays.equals(Base64.getDecoder().decode(expected), digest.digest());
} else {
results.put(source, null);
}
}
if (valid) {
for (Map.Entry<LazyInputStreamSource, byte[]> entry : results.entrySet()) {
entry.getKey().storeIfNotPresent(entry.getValue());
}
} else {
Files.delete(cached);
for (LazyInputStreamSource source : results.keySet()) {
source.deleteIfPresent();
}
}
}
if (valid) {
return new StoredInputStreamSource(cached);
}
} else {
Files.createDirectories(cached.getParent());
}
}
Map<LazyInputStreamSource, MessageDigest> digests = new HashMap<>();
if (checksum == null) {
for (Map.Entry<String, URI> entry : validations.entrySet()) {
LazyInputStreamSource source = fetch(entry.getValue(),
groupId,
artifactId,
version,
type,
classifier,
entry.getKey().toLowerCase());
MessageDigest digest;
try {
digest = MessageDigest.getInstance(entry.getKey());
} catch (NoSuchAlgorithmException e) {
throw new IllegalStateException(e);
}
digests.put(source, digest);
}
}
URI uri = repository.resolve(path);
if (cached == null) {
return () -> ValidatingInputStream.of(uri.toURL().openStream(), digests);
} else {
return new LatentInputStreamSource(cached,
uri,
digests,
artifactId + "-" + version + (classifier == null ? "" : "-" + classifier),
type + (checksum == null ? "" : ("." + checksum)));
}
}
@FunctionalInterface
private interface LazyInputStreamSource extends InputStreamSource {
default void deleteIfPresent() throws IOException {
}
default void storeIfNotPresent(byte[] bytes) throws IOException {
}
default InputStreamSource materialize() throws IOException {
return this;
}
}
record StoredInputStreamSource(Path path) implements LazyInputStreamSource {
@Override
public void deleteIfPresent() throws IOException {
Files.delete(path);
}
@Override
public InputStream toInputStream() throws IOException {
return Files.newInputStream(path);
}
@Override
public Optional<Path> getPath() {
return Optional.of(path);
}
}
record LatentInputStreamSource(Path path,
URI uri,
Map<LazyInputStreamSource, MessageDigest> digests,
String prefix,
String suffix) implements LazyInputStreamSource {
@Override
public InputStream toInputStream() throws IOException {
return ValidatingInputStream.of(uri.toURL().openStream(), digests);
}
@Override
public void storeIfNotPresent(byte[] bytes) throws IOException {
Path temporary = Files.createTempFile(prefix, suffix);
try (OutputStream outputStream = Files.newOutputStream(temporary)) {
outputStream.write(bytes);
} catch (Throwable t) {
Files.delete(temporary);
throw t;
}
Files.move(temporary, path);
}
@Override
public InputStreamSource materialize() throws IOException {
Path temporary = Files.createTempFile(prefix, suffix);
try (InputStream inputStream = toInputStream();
OutputStream outputStream = Files.newOutputStream(temporary)) {
inputStream.transferTo(outputStream);
} catch (Throwable t) {
Files.delete(temporary);
throw t;
}
return new StoredInputStreamSource(Files.move(temporary, path));
}
}
private static class ValidatingInputStream extends FilterInputStream {
private final Map<LazyInputStreamSource, MessageDigest> digests;
private ValidatingInputStream(InputStream inputStream, Map<LazyInputStreamSource, MessageDigest> digests) {
super(inputStream);
this.digests = digests;
}
private static InputStream of(InputStream inputStream, Map<LazyInputStreamSource, MessageDigest> digests) {
if (digests.isEmpty()) {
return inputStream;
}
for (MessageDigest digest : digests.values()) {
inputStream = new DigestInputStream(inputStream, digest);
}
return new ValidatingInputStream(inputStream, digests);
}
@Override
public void close() throws IOException {
super.close();
boolean valid = true;
Map<LazyInputStreamSource, byte[]> results = new HashMap<>();
for (Map.Entry<LazyInputStreamSource, MessageDigest> entry : digests.entrySet()) {
byte[] expected;
try (InputStream inputStream = entry.getKey().toInputStream()) {
expected = inputStream.readAllBytes();
}
results.put(entry.getKey(), expected);
if (!(valid = Arrays.equals(Base64.getDecoder().decode(expected), entry.getValue().digest()))) {
break;
}
}
if (valid) {
for (Map.Entry<LazyInputStreamSource, byte[]> entry : results.entrySet()) {
entry.getKey().storeIfNotPresent(entry.getValue());
}
} else {
for (LazyInputStreamSource source : digests.keySet()) {
source.deleteIfPresent();
}
throw new IOException("Failed checksum validation");
}
}
}
}
@Geolykt
Copy link

Geolykt commented Dec 8, 2024

Talking about the test scope and unintuitive behaviour concerning it: The test dependencies of a test dependency are transitively included, which is probably the most irritating behaviour I've seen so far.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment