Class ClasspathScanner

etc/future/scanner/ClasspathScanner.java:13–104

import java.util.jar.JarInputStream;

public class ClasspathScanner {

    private String pkg;
    private boolean recur = false;

    protected String getPackage() {
        return pkg;
    }

    public ClasspathScanner(String pkg, boolean subpackages) {
        recur = subpackages;
        sanitizePackage(pkg);
    }

    public ClasspathScanner(String pkg) {
        sanitizePackage(pkg);
    }

    private void sanitizePackage(String pkgName) {
        if ((pkgName == null) || (pkgName.trim().length() == 0)) throw new IllegalArgumentException("Base package cannot be null");
        pkg = pkgName.replace('.', '/');
        if (pkg.endsWith("*")) pkg = pkg.substring(0, pkg.length() - 1);
        if (pkg.endsWith("/")) pkg = pkg.substring(0, pkg.length() - 1);
    }

    protected ClassLoader getClassLoader() {
        return Thread.currentThread().getContextClassLoader();
    }

    protected boolean isJARPath(String path) {
        return (path.indexOf("!") > 0) & (path.indexOf(".jar") > 0);
    }

    protected void add(Set<String> classes, String className) {
        if ((className.startsWith(pkg)) && (className.endsWith(".class"))) {
            boolean add = recur ? true : className.substring(pkg.length() + 1).indexOf("/") < 0;
            if (add) classes.add(className.substring(0, className.length() - 6).replace('/', '.'));
        }
    }

    protected Set<String> getClassesFromJAR(String path) throws IOException {
        Set<String> classes = new HashSet<String>();
        String jarPath = path.substring(0, path.indexOf("!")).substring(path.indexOf(":") + 1);
        JarInputStream jarFile = new JarInputStream(new FileInputStream(jarPath));
        JarEntry jarEntry;
        do {
            jarEntry = jarFile.getNextJarEntry();
            if (jarEntry != null) add(classes, jarEntry.getName());
        } while (jarEntry != null);
        return classes;
    }

    protected Set<String> getClassesFromDirectory(String path) {
        Set<String> classes = new HashSet<String>();
        File directory = new File(path);
        if (directory.exists()) {
            for (String file : directory.list()) {
                File f = new File(directory, file);