tech.guyi.ipojo.compile.lib.maven.MavenHelper Maven / Gradle / Ivy
The newest version!
package tech.guyi.ipojo.compile.lib.maven;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.reflect.TypeToken;
import lombok.SneakyThrows;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.maven.repository.internal.MavenRepositorySystemUtils;
import org.eclipse.aether.DefaultRepositorySystemSession;
import org.eclipse.aether.RepositorySystem;
import org.eclipse.aether.RepositorySystemSession;
import org.eclipse.aether.artifact.Artifact;
import org.eclipse.aether.artifact.DefaultArtifact;
import org.eclipse.aether.collection.CollectRequest;
import org.eclipse.aether.collection.CollectResult;
import org.eclipse.aether.collection.DependencyCollectionException;
import org.eclipse.aether.graph.Dependency;
import org.eclipse.aether.graph.DependencyNode;
import org.eclipse.aether.graph.DependencyVisitor;
import org.eclipse.aether.repository.LocalRepository;
import org.eclipse.aether.repository.RemoteRepository;
import org.eclipse.aether.resolution.ArtifactRequest;
import org.eclipse.aether.resolution.ArtifactResolutionException;
import org.eclipse.aether.util.repository.AuthenticationBuilder;
import tech.guyi.ipojo.compile.lib.maven.util.Booter;
import tech.guyi.ipojo.compile.lib.configuration.entry.Project;
import tech.guyi.ipojo.compile.lib.configuration.entry.Repository;
import tech.guyi.ipojo.compile.lib.configuration.entry.Server;
import java.io.*;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.*;
import java.util.stream.Collectors;
public class MavenHelper {
private static final Gson gson = new GsonBuilder().setPrettyPrinting().create();
private static final String cacheDir = ".compile.cache";
private static final String dependencyCachePath = cacheDir + "/dependency.cache.json";
private static Map> dependencyCache = new HashMap<>();
static {
if (Files.exists(Paths.get(dependencyCachePath))){
try {
dependencyCache = gson.fromJson(
IOUtils.toString(new FileInputStream(dependencyCachePath), StandardCharsets.UTF_8),
new TypeToken>>(){}.getType()
);
} catch (IOException e) {
e.printStackTrace();
}
}
}
@SneakyThrows
private static void addCache(String name, Set dependencies){
dependencyCache.put(name,dependencies);
String json = gson.toJson(dependencyCache);
if (Files.notExists(Paths.get(cacheDir))){
Files.createDirectory(Paths.get(cacheDir));
}
FileUtils.writeByteArrayToFile(new File(dependencyCachePath),json.getBytes());
}
private static RepositorySystemSession buildSession(tech.guyi.ipojo.compile.lib.configuration.entry.Dependency root, RepositorySystem system, Project project){
return root.get(project)
.filter(path -> Files.exists(Paths.get(path)))
.map(p -> {
DefaultRepositorySystemSession session = MavenRepositorySystemUtils.newSession();
LocalRepository localRepo = new LocalRepository( project.getLocalRepository());
session.setLocalRepositoryManager( system.newLocalRepositoryManager( session, localRepo ));
return session;
})
.orElseGet(() -> Booter.newRepositorySystemSession(system,project.getLocalRepository()));
}
public static Set getDependencies(
Project project,
tech.guyi.ipojo.compile.lib.configuration.entry.Dependency root){
try {
if (dependencyCache.containsKey(root.getName())){
return dependencyCache.get(root.getName());
}
RepositorySystem system = Booter.newRepositorySystem();
RepositorySystemSession session = buildSession(root,system,project);
CollectRequest request = buildRequest(root,project.getRepositories(),project.getServers());
CollectResult result = system.collectDependencies( session, request );
Set dependencies = new HashSet<>();
result.getRoot().accept(new DependencyVisitor() {
@Override
public boolean visitEnter(DependencyNode node) {
dependencies.add(
new tech.guyi.ipojo.compile.lib.configuration.entry.Dependency(
node.getDependency().getArtifact().getGroupId(),
node.getDependency().getArtifact().getArtifactId(),
node.getDependency().getArtifact().getVersion(),
node.getDependency().getScope()
)
);
return true;
}
@Override
public boolean visitLeave(DependencyNode dependencyNode) {
return true;
}
});
addCache(root.getName(),dependencies);
dependencies.add(root);
dependencies
.stream()
.filter(dependency -> dependency.get(project)
.map(path -> Files.notExists(Paths.get(path)))
.orElse(false))
.forEach(dependency -> resolveArtifact(request,system,session,dependency.getName()));
return dependencies;
} catch (DependencyCollectionException e) {
e.printStackTrace();
}
return Collections.singleton(root);
}
public static void resolveArtifact(CollectRequest request,RepositorySystem system,RepositorySystemSession session,String coords){
try {
Artifact tmp = new DefaultArtifact(coords);
ArtifactRequest artifactRequest = new ArtifactRequest();
artifactRequest.setRepositories(request.getRepositories());
artifactRequest.setArtifact(tmp);
system.resolveArtifact(session,artifactRequest);
} catch (ArtifactResolutionException e) {
e.printStackTrace();
}
}
public static void resolveArtifact(CollectRequest request,String localRepository,String coords){
RepositorySystem system = Booter.newRepositorySystem();
RepositorySystemSession session = Booter.newRepositorySystemSession(system,localRepository);
resolveArtifact(request,system,session,coords);
}
public static void resolveArtifact(List repositories,Set servers,String localRepository, String coords){
resolveArtifact(buildRequest(coords,"",repositories,servers),localRepository,coords);
}
private static CollectRequest buildRequest(
String coords,
String scope,
List repositories,
Set servers){
Artifact artifact = new DefaultArtifact(coords);
CollectRequest request = new CollectRequest();
request.setRoot(new Dependency(artifact,scope));
if (repositories == null || repositories.isEmpty()){
request.setRepositories(getDefaultRepository());
}else{
request.setRepositories(
repositories
.stream()
.map(repo -> {
RemoteRepository.Builder builder = new RemoteRepository.Builder(
repo.getId(),
repo.getType(),
repo.getUrl());
servers.stream()
.filter(server -> server.getId().equals(repo.getId()))
.findFirst()
.ifPresent(server -> builder.setAuthentication(
new AuthenticationBuilder()
.addUsername(server.getUsername())
.addPassword(server.getPassword())
.build()
)
);
return builder.build();
}).collect(Collectors.toList())
);
}
return request;
}
private static CollectRequest buildRequest(
tech.guyi.ipojo.compile.lib.configuration.entry.Dependency root,
List repositories,
Set servers){
return buildRequest(root.getName(),root.getScope(),repositories,servers);
}
private static List getDefaultRepository(){
return Arrays.asList(
new RemoteRepository.Builder(
"aliyun",
"default",
"https://maven.aliyun.com/repository/public"
).build(),
new RemoteRepository.Builder(
"guyi",
"default",
"http://nexus.guyi-maple.top/repository/maven-public/"
).build()
);
}
}