com.automationrockstars.base.JarUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bmo Show documentation
The newest version!
/*
 * 
 */

package com.automationrockstars.base;

import com.google.common.base.Preconditions;
import com.google.common.base.Predicate;
import com.google.common.base.Splitter;
import com.google.common.base.Throwables;
import com.google.common.collect.FluentIterable;
import com.google.common.collect.Iterables;
import com.google.common.io.ByteStreams;
import com.google.common.reflect.ClassPath;
import com.google.common.reflect.ClassPath.ResourceInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Path;
import java.nio.file.Paths;

public class JarUtils {

    private static final ClassLoader cl = ClassLoader.getSystemClassLoader();
    private static final Logger LOG = LoggerFactory.getLogger(JarUtils.class);
    private static ClassPath cp;
    private static FluentIterable resources = null;

    private static ClassLoader classLoader() {
        return cl;
    }

    public static void unzipToFile(Path outsideDirectory, String... fileOnClassPath) {
        unzipResource(outsideDirectory, findResources(fileOnClassPath).get(0));
    }

    public static void unzipResource(Path outsideDirectory, ResourceInfo resource) {
        outsideDirectory.toFile().mkdirs();
        try (InputStream internalFile = resource.url().openStream(); FileOutputStream out = new FileOutputStream(outsideDirectory.toFile() + "/" + fileNameFromClasspath(resource.getResourceName()))) {
            ByteStreams.copy(internalFile, out);
        } catch (IOException e) {
            Throwables.propagate(e);
        }
    }

    public static String fileNameFromClasspath(String classPathFile) {
        return Iterables.getLast(Splitter.on("/").splitToList(classPathFile));
    }

    public static void unzipDirectory(Path outsideDirectory, String... dirOnClassPath) {
        for (ResourceInfo resource : findResources(dirOnClassPath)) {
            unzipResource(outsideDirectory, resource);
        }
    }

    private static ClassPath cp() {
        try {
            cp = ClassPath.from(classLoader());
        } catch (IOException e) {
        }
        return cp;
    }

    private static FluentIterable resources() {
        if (resources == null) {
            Preconditions.checkNotNull(cp(), "ClassPath scanner cannot be initialized");
            resources = FluentIterable.from(cp().getResources());
        }
        return resources;
    }

    static Predicate matchesName(final String... filterParts) {
        return new Predicate() {

            @Override
            public boolean apply(String resourceName) {
                boolean matches = true;
                for (String filterPart : filterParts) {
                    matches = matches && resourceName.matches(".*" + filterPart + ".*");
                }
                return matches;
            }
        };

    }

    public static FluentIterable findResources(final String... filterParts) {
        String currentPath = Paths.get("").toAbsolutePath().toString();
        if (currentPath.contains(" ")) {
            LOG.error("Current path contains SPACE characters that usually make loading classpath resources failing. Current path is {}", currentPath);
        }
        return resources().filter(new Predicate() {
            @Override
            public boolean apply(ResourceInfo input) {
                return matchesName(filterParts).apply(input.getResourceName());
            }
        });
    }

    public static void reload() {
        resources = null;
        cp = null;
        resources();
    }

}