forked from External/mage
106 lines
4.1 KiB
Java
106 lines
4.1 KiB
Java
package mage.util;
|
|
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.IOException;
|
|
import java.net.URI;
|
|
import java.net.URISyntaxException;
|
|
import java.net.URL;
|
|
import java.util.*;
|
|
import java.util.jar.JarEntry;
|
|
import java.util.jar.JarInputStream;
|
|
|
|
/**
|
|
* @author North
|
|
*/
|
|
public final class ClassScanner {
|
|
|
|
private static void checkClassForInclusion(List<Class> cards, Class type, String name, ClassLoader cl) {
|
|
try {
|
|
Class clazz = Class.forName(name, true, cl);
|
|
if (clazz.getEnclosingClass() == null && type.isAssignableFrom(clazz)) {
|
|
cards.add(clazz);
|
|
}
|
|
} catch (ClassNotFoundException ex) {
|
|
// ignored
|
|
}
|
|
}
|
|
|
|
public static List<Class> findClasses(ClassLoader classLoader, List<String> packages, Class<?> type) {
|
|
List<Class> cards = new ArrayList<>();
|
|
try {
|
|
if (classLoader == null) classLoader = Thread.currentThread().getContextClassLoader();
|
|
assert classLoader != null;
|
|
|
|
Map<String, String> dirs = new HashMap<>();
|
|
Set<String> jars = new TreeSet<>();
|
|
for (String packageName : packages) {
|
|
String path = packageName.replace('.', '/');
|
|
Enumeration<URL> resources = classLoader.getResources(path);
|
|
|
|
while (resources.hasMoreElements()) {
|
|
URL resource = resources.nextElement();
|
|
String filePath = resource.getFile();
|
|
if (filePath.startsWith("file:")) {
|
|
filePath = filePath.substring("file:".length(), filePath.lastIndexOf('!'));
|
|
jars.add(filePath);
|
|
} else {
|
|
try {
|
|
filePath = resource.toURI().getPath();
|
|
} catch (URISyntaxException e) {
|
|
throw new RuntimeException(e);
|
|
}
|
|
dirs.put(filePath, packageName);
|
|
}
|
|
}
|
|
}
|
|
|
|
for (Map.Entry<String, String> dir : dirs.entrySet()) {
|
|
cards.addAll(findClasses(classLoader, new File(dir.getKey()), dir.getValue(), type));
|
|
}
|
|
|
|
for (String filePath : jars) {
|
|
File file = new File(CardUtil.urlDecode(filePath));
|
|
cards.addAll(findClassesInJar(classLoader, file, packages, type));
|
|
}
|
|
} catch (IOException ex) {
|
|
}
|
|
return cards;
|
|
}
|
|
|
|
private static List<Class> findClasses(ClassLoader classLoader, File directory, String packageName, Class<?> type) {
|
|
List<Class> cards = new ArrayList<>();
|
|
if (directory == null || !directory.exists()) return cards;
|
|
|
|
for (File file : directory.listFiles()) {
|
|
if (file.getName().endsWith(".class")) {
|
|
String name = packageName + '.' + file.getName().substring(0, file.getName().length() - 6);
|
|
checkClassForInclusion(cards, type, name, classLoader);
|
|
}
|
|
}
|
|
return cards;
|
|
}
|
|
|
|
private static List<Class> findClassesInJar(ClassLoader classLoader, File file, List<String> packages, Class<?> type) {
|
|
List<Class> cards = new ArrayList<>();
|
|
if (!file.exists()) return cards;
|
|
|
|
|
|
try (JarInputStream jarFile = new JarInputStream(new FileInputStream(file))) {
|
|
while (true) {
|
|
JarEntry jarEntry = jarFile.getNextJarEntry();
|
|
if (jarEntry == null) {
|
|
break;
|
|
}
|
|
if (jarEntry.getName().endsWith(".class")) {
|
|
String className = jarEntry.getName().replace(".class", "").replace('/', '.');
|
|
int packageNameEnd = className.lastIndexOf('.');
|
|
String packageName = packageNameEnd != -1 ? className.substring(0, packageNameEnd) : "";
|
|
if (packages.contains(packageName)) checkClassForInclusion(cards, type, className, classLoader);
|
|
}
|
|
}
|
|
} catch (IOException ex) {
|
|
}
|
|
return cards;
|
|
}
|
|
}
|