com.simpligility.maven.provisioner.MavenRepositoryHelper.java Source code

Introduction

Here is the source code for com.simpligility.maven.provisioner.MavenRepositoryHelper.java

Source

/** 
 * Copyright simpligility technologies inc. http://www.simpligility.com
 * Licensed under Eclipse Public License - v 1.0 http://www.eclipse.org/legal/epl-v10.html
 */
package com.simpligility.maven.provisioner;

import java.io.File;
import java.util.ArrayList;
import java.util.Collection;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.apache.commons.io.filefilter.AndFileFilter;
import org.apache.commons.io.filefilter.DirectoryFileFilter;
import org.apache.commons.io.filefilter.IOFileFilter;
import org.apache.commons.io.filefilter.NotFileFilter;
import org.apache.commons.io.filefilter.SuffixFileFilter;
import org.apache.commons.io.filefilter.TrueFileFilter;
import org.apache.commons.io.filefilter.WildcardFileFilter;
import org.eclipse.aether.DefaultRepositorySystemSession;
import org.eclipse.aether.RepositorySystem;
import org.eclipse.aether.artifact.Artifact;
import org.eclipse.aether.artifact.DefaultArtifact;
import org.eclipse.aether.deployment.DeployRequest;
import org.eclipse.aether.repository.Authentication;
import org.eclipse.aether.repository.RemoteRepository;
import org.eclipse.aether.util.repository.AuthenticationBuilder;
import org.slf4j.LoggerFactory;
import org.slf4j.Logger;

public class MavenRepositoryHelper {
    private static Logger logger = LoggerFactory.getLogger("MavenRepositoryHelper");

    private File repositoryPath;

    private RepositorySystem system;

    private DefaultRepositorySystemSession session;

    public MavenRepositoryHelper(File repositoryPath) {
        this.repositoryPath = repositoryPath;
        initialize();
    }

    private void initialize() {
        system = RepositoryHandler.getRepositorySystem();
        session = RepositoryHandler.getRepositorySystemSession(system, repositoryPath);
    }

    public void deployToRemote(String targetUrl, String username, String password) {
        // Using commons-io, if performance or so is a problem it might be worth looking at the Java 8 streams API
        // e.g. http://blog.jooq.org/2014/01/24/java-8-friday-goodies-the-new-new-io-apis/
        // not yet though..
        Collection<File> subDirectories = FileUtils.listFilesAndDirs(repositoryPath,
                (IOFileFilter) DirectoryFileFilter.DIRECTORY, TrueFileFilter.INSTANCE);
        Collection<File> leafDirectories = new ArrayList<File>();
        for (File subDirectory : subDirectories) {
            if (isLeafVersionDirectory(subDirectory)) {

                leafDirectories.add(subDirectory);
            }
        }
        for (File leafDirectory : leafDirectories) {
            String leafAbsolutePath = leafDirectory.getAbsoluteFile().toString();
            int repoAbsolutePathLength = repositoryPath.getAbsoluteFile().toString().length();
            String leafRepoPath = leafAbsolutePath.substring(repoAbsolutePathLength + 1, leafAbsolutePath.length());

            Gav gav = GavUtil.getGavFromRepositoryPath(leafRepoPath);

            // only interested in files using the artifactId-version* pattern
            // don't bother with .sha1 files
            IOFileFilter fileFilter = new AndFileFilter(
                    new WildcardFileFilter(gav.getArtifactId() + "-" + gav.getVersion() + "*"),
                    new NotFileFilter(new SuffixFileFilter("sha1")));
            Collection<File> artifacts = FileUtils.listFiles(leafDirectory, fileFilter, null);

            Authentication auth = new AuthenticationBuilder().addUsername(username).addPassword(password).build();

            RemoteRepository distRepo = new RemoteRepository.Builder("repositoryIdentifier", "default", targetUrl)
                    .setAuthentication(auth).build();

            DeployRequest deployRequest = new DeployRequest();
            deployRequest.setRepository(distRepo);
            for (File file : artifacts) {
                String extension;
                if (file.getName().endsWith("tar.gz")) {
                    extension = "tar.gz";
                } else {
                    extension = FilenameUtils.getExtension(file.getName());
                }

                String baseFileName = gav.getFilenameStart() + "." + extension;
                String fileName = file.getName();
                String g = gav.getGroupdId();
                String a = gav.getArtifactId();
                String v = gav.getVersion();

                Artifact artifact = null;
                if (gav.getPomFilename().equals(fileName)) {
                    artifact = new DefaultArtifact(g, a, "pom", v);
                } else if (gav.getJarFilename().equals(fileName)) {
                    artifact = new DefaultArtifact(g, a, "jar", v);
                } else if (gav.getSourceFilename().equals(fileName)) {
                    artifact = new DefaultArtifact(g, a, "sources", "jar", v);
                } else if (gav.getJavadocFilename().equals(fileName)) {
                    artifact = new DefaultArtifact(g, a, "javadoc", "jar", v);
                } else if (baseFileName.equals(fileName)) {
                    artifact = new DefaultArtifact(g, a, extension, v);
                } else {
                    String classifier = file.getName().substring(gav.getFilenameStart().length() + 1,
                            file.getName().length() - ("." + extension).length());
                    artifact = new DefaultArtifact(g, a, classifier, extension, v);
                }

                if (artifact != null) {
                    artifact = artifact.setFile(file);
                    deployRequest.addArtifact(artifact);
                }

            }

            try {
                system.deploy(session, deployRequest);
            } catch (Exception e) {
                logger.info("Deployment failed with " + e.getMessage() + ", artifact might be deployed already.");
            }
        }
    }

    /**
     * Determine if it is a leaf directory with artifacts in it. Criteria used is that there is no subdirectory.
     * 
     * @param subDirectory
     * @return
     */
    private boolean isLeafVersionDirectory(File subDirectory) {
        boolean isLeafVersionDirectory;
        Collection<File> subDirectories = FileUtils.listFilesAndDirs(subDirectory,
                (IOFileFilter) DirectoryFileFilter.DIRECTORY, TrueFileFilter.INSTANCE);
        // it finds at least itself so have to check for > 1
        isLeafVersionDirectory = subDirectories.size() > 1 ? false : true;
        return isLeafVersionDirectory;
    }

}