io.github.algomaster99.terminator.commons.maven.MavenModule Maven / Gradle / Ivy

Go to download
package io.github.algomaster99.terminator.commons.maven;

import java.io.FileReader;
import java.io.IOException;
import java.nio.file.Path;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Queue;
import java.util.Set;
import java.util.stream.Collectors;
import org.apache.maven.model.Dependency;
import org.apache.maven.model.Model;
import org.apache.maven.model.io.xpp3.MavenXpp3Reader;
import org.codehaus.plexus.util.xml.pull.XmlPullParserException;

public class MavenModule {
    private final MavenModule parent;

    private final Model self;

    private final Path fileSystemPath;

    private final List submodules = new ArrayList<>();

    private MavenModule(Model self, Path fileSystemPath, MavenModule parent) {
        this.self = self;
        this.fileSystemPath = fileSystemPath;
        this.parent = parent;
    }

    public void addSubmodule(MavenModule child) {
        submodules.add(child);
    }

    public Model getSelf() {
        return self;
    }

    public List getSubmodules() {
        return submodules;
    }

    public Path getFileSystemPath() {
        return fileSystemPath;
    }

    public MavenModule findSubmodule(String artifactIdOfModule) {
        Queue queue = new ArrayDeque<>();
        queue.add(topLevelParent());
        while (!queue.isEmpty()) {
            MavenModule module = queue.poll();
            if (module.getSelf().getArtifactId().equals(artifactIdOfModule)) {
                return module;
            }
            queue.addAll(module.getSubmodules());
        }
        return null;
    }

    public MavenModule topLevelParent() {
        if (parent == null) {
            return this;
        }
        return parent.topLevelParent();
    }

    public Set getSubmodulesThatAreDependencies() {
        Set subModulesThatAreDependencies = new HashSet<>();
        List dependencies = self.getDependencies();
        for (Dependency dependency : dependencies) {
            String artifactId = dependency.getArtifactId();
            MavenModule submodule = findSubmodule(artifactId);
            if (submodule == null) {
                continue;
            }
            subModulesThatAreDependencies.add(submodule);
            // look for transitive dependencies that are submodules
            subModulesThatAreDependencies.addAll(submodule.getSubmodulesThatAreDependencies());
        }
        return subModulesThatAreDependencies;
    }

    /**
     * This is delegated to {@link Model#equals(Object)}.
     */
    @Override
    public boolean equals(Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj == null || getClass() != obj.getClass()) {
            return false;
        }
        MavenModule that = (MavenModule) obj;
        return self.equals(that.self);
    }

    @Override
    public int hashCode() {
        return self.hashCode();
    }

    public static MavenModule createMavenModuleGraph(Path projectRoot) throws IOException, XmlPullParserException {
        Path rootPom = projectRoot.resolve("pom.xml");
        MavenXpp3Reader reader = new MavenXpp3Reader();
        Model rootModel = reader.read(new FileReader(rootPom.toFile()));

        MavenModule root = new MavenModule(rootModel, projectRoot.toAbsolutePath(), null);

        List submodules = rootModel.getModules();

        for (String module : submodules) {
            Path modulePath = projectRoot.resolve(module);
            MavenXpp3Reader moduleReader = new MavenXpp3Reader();
            Model moduleModel = moduleReader.read(
                    new FileReader(modulePath.resolve("pom.xml").toFile()));
            MavenModule mavenModule = new MavenModule(moduleModel, modulePath, root);
            if (moduleModel.getModules() != null) {
                List childModules = moduleModel.getModules();
                List children = childModules.stream()
                        .map(childModule -> {
                            try {
                                return createMavenModuleGraph(modulePath.resolve(childModule));
                            } catch (IOException | XmlPullParserException e) {
                                throw new RuntimeException(e);
                            }
                        })
                        .collect(Collectors.toList());
                children.forEach(mavenModule::addSubmodule);
            }
            root.addSubmodule(mavenModule);
        }
        return root;
    }
}