net.badbird5907.blib.util.ReflectionUtils Maven / Gradle / Ivy

Go to download
package net.badbird5907.blib.util;

import com.google.common.collect.ImmutableSet;
import org.bukkit.plugin.Plugin;
import org.reflections.util.ClasspathHelper;
import org.reflections.vfs.Vfs;

import java.net.URL;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

public class ReflectionUtils {
    public static Collection> getClassesInPackage(Plugin plugin, String packageName) {
        Set> classes = new HashSet<>();
        for (URL url : ClasspathHelper.forClassLoader(ClasspathHelper.contextClassLoader(),
                ClasspathHelper.staticClassLoader(), plugin
                        .getClass().getClassLoader())) {
            Vfs.Dir dir = Vfs.fromURL(url);
            try {
                for (Vfs.File file : dir.getFiles()) {
                    String name = file.getRelativePath().replace("/", ".").replace(".class", "");
                    if (name.startsWith(packageName))
                        classes.add(Class.forName(name));
                }
            } catch (Exception ex) {
                ex.printStackTrace();
            } finally {
                dir.close();
            }
        }
        return ImmutableSet.copyOf(classes);
    }
}