mirror of
https://github.com/jneug/zeichenmaschine.git
synced 2026-04-14 06:33:34 +02:00
284 lines
10 KiB
Java
284 lines
10 KiB
Java
|
|
|
|
import java.io.File;
|
|
import java.io.FilenameFilter;
|
|
import java.io.Serializable;
|
|
import java.net.MalformedURLException;
|
|
import java.net.URL;
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
import java.util.Enumeration;
|
|
import java.util.List;
|
|
import java.util.jar.JarEntry;
|
|
import java.util.jar.JarFile;
|
|
|
|
/**
|
|
* Find classes in the classpath (reads JARs and classpath folders).
|
|
*
|
|
* @author Pål Brattberg, brattberg@gmail.com
|
|
* @see http://gist.github.com/pal
|
|
*/
|
|
@SuppressWarnings("unchecked")
|
|
public class ClasspathInspector {
|
|
static boolean DEBUG = false;
|
|
|
|
|
|
public static List<String> getAllKnownClassNames() {
|
|
List<String> classNames = new ArrayList<String>();
|
|
List<File> classLocations = getClassLocationsForCurrentClasspath();
|
|
for (File file : classLocations) {
|
|
classNames.addAll(getClassNamesFromPath(file));
|
|
}
|
|
return classNames;
|
|
}
|
|
|
|
public static List<Class> getAllKnownClasses() {
|
|
List<Class> classFiles = new ArrayList<Class>();
|
|
List<File> classLocations = getClassLocationsForCurrentClasspath();
|
|
for (File file : classLocations) {
|
|
classFiles.addAll(getClassesFromPath(file));
|
|
}
|
|
return classFiles;
|
|
}
|
|
|
|
public static List<Class> getMatchingClasses(Class interfaceOrSuperclass) {
|
|
List<Class> matchingClasses = new ArrayList<Class>();
|
|
List<Class> classes = getAllKnownClasses();
|
|
log("checking %s classes", classes.size());
|
|
for (Class clazz : classes) {
|
|
if (interfaceOrSuperclass.isAssignableFrom(clazz)) {
|
|
matchingClasses.add(clazz);
|
|
log("class %s is assignable from %s", interfaceOrSuperclass, clazz);
|
|
}
|
|
}
|
|
return matchingClasses;
|
|
}
|
|
|
|
public static List<Class> getMatchingClasses(String validPackagePrefix, Class interfaceOrSuperclass) {
|
|
throw new IllegalStateException("Not yet implemented!");
|
|
}
|
|
|
|
public static List<Class> getMatchingClasses(String validPackagePrefix) {
|
|
throw new IllegalStateException("Not yet implemented!");
|
|
}
|
|
|
|
private static Collection<? extends Class> getClassesFromPath(File path) {
|
|
if (path.isDirectory()) {
|
|
return getClassesFromDirectory(path);
|
|
} else {
|
|
return getClassesFromJarFile(path);
|
|
}
|
|
}
|
|
|
|
private static Collection<String> getClassNamesFromPath(File path) {
|
|
if (path.isDirectory()) {
|
|
return getClassNamesFromDirectory(path);
|
|
} else {
|
|
return getClassNamesFromJarFile(path);
|
|
}
|
|
}
|
|
|
|
private static String fromFileToClassName(final String fileName) {
|
|
return fileName.substring(0, fileName.length() - 6).replaceAll("/|\\\\", "\\.");
|
|
}
|
|
|
|
private static List<Class> getClassesFromJarFile(File path) {
|
|
List<Class> classes = new ArrayList<Class>();
|
|
log("getClassesFromJarFile: Getting classes for " + path);
|
|
|
|
try {
|
|
if (path.canRead()) {
|
|
JarFile jar = new JarFile(path);
|
|
Enumeration<JarEntry> en = jar.entries();
|
|
while (en.hasMoreElements()) {
|
|
JarEntry entry = en.nextElement();
|
|
if (entry.getName().endsWith("class")) {
|
|
String className = fromFileToClassName(entry.getName());
|
|
log("\tgetClassesFromJarFile: found " + className);
|
|
Class claz = Class.forName(className);
|
|
classes.add(claz);
|
|
}
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
throw new RuntimeException("Failed to read classes from jar file: " + path, e);
|
|
}
|
|
|
|
return classes;
|
|
}
|
|
|
|
private static List<String> getClassNamesFromJarFile(File path) {
|
|
List<String> classes = new ArrayList<>();
|
|
log("getClassNamesFromJarFile: Getting classes for " + path);
|
|
|
|
try {
|
|
if (path.canRead()) {
|
|
JarFile jar = new JarFile(path);
|
|
Enumeration<JarEntry> en = jar.entries();
|
|
while (en.hasMoreElements()) {
|
|
JarEntry entry = en.nextElement();
|
|
if (entry.getName().endsWith("class")) {
|
|
String className = fromFileToClassName(entry.getName());
|
|
classes.add(className);
|
|
log("\tgetClassesFromJarFile: found " + className);
|
|
}
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
throw new RuntimeException("Failed to read classnames from jar file: " + path, e);
|
|
}
|
|
|
|
return classes;
|
|
}
|
|
|
|
private static List<Class> getClassesFromDirectory(File path) {
|
|
List<Class> classes = new ArrayList<Class>();
|
|
log("getClassesFromDirectory: Getting classes for " + path);
|
|
|
|
// get jar files from top-level directory
|
|
List<File> jarFiles = listFiles(path, new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File dir, String name) {
|
|
return name.endsWith(".jar");
|
|
}
|
|
}, false);
|
|
for (File file : jarFiles) {
|
|
classes.addAll(getClassesFromJarFile(file));
|
|
}
|
|
|
|
// get all class-files
|
|
List<File> classFiles = listFiles(path, new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File dir, String name) {
|
|
return name.endsWith(".class");
|
|
}
|
|
}, true);
|
|
|
|
// List<URL> urlList = new ArrayList<URL>();
|
|
// List<String> classNameList = new ArrayList<String>();
|
|
int substringBeginIndex = path.getAbsolutePath().length() + 1;
|
|
for (File classfile : classFiles) {
|
|
String className = classfile.getAbsolutePath().substring(substringBeginIndex);
|
|
className = fromFileToClassName(className);
|
|
log("Found class %s in path %s: ", className, path);
|
|
try {
|
|
classes.add(Class.forName(className));
|
|
} catch (Throwable e) {
|
|
log("Couldn't create class %s. %s: ", className, e);
|
|
}
|
|
|
|
}
|
|
|
|
return classes;
|
|
}
|
|
|
|
private static List<String> getClassNamesFromDirectory(File path) {
|
|
List<String> classes = new ArrayList<>();
|
|
log("getClassNamesFromDirectory: Getting classes for " + path);
|
|
|
|
// get jar files from top-level directory
|
|
List<File> jarFiles = listFiles(path, new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File dir, String name) {
|
|
return name.endsWith(".jar");
|
|
}
|
|
}, false);
|
|
for (File file : jarFiles) {
|
|
classes.addAll(getClassNamesFromJarFile(file));
|
|
}
|
|
|
|
// get all class-files
|
|
List<File> classFiles = listFiles(path, new FilenameFilter() {
|
|
@Override
|
|
public boolean accept(File dir, String name) {
|
|
return name.endsWith(".class");
|
|
}
|
|
}, true);
|
|
|
|
// List<URL> urlList = new ArrayList<URL>();
|
|
// List<String> classNameList = new ArrayList<String>();
|
|
int substringBeginIndex = path.getAbsolutePath().length() + 1;
|
|
for (File classfile : classFiles) {
|
|
String className = classfile.getAbsolutePath().substring(substringBeginIndex);
|
|
className = fromFileToClassName(className);
|
|
log("Found class %s in path %s: ", className, path);
|
|
classes.add(className);
|
|
}
|
|
|
|
return classes;
|
|
}
|
|
|
|
private static List<File> listFiles(File directory, FilenameFilter filter, boolean recurse) {
|
|
List<File> files = new ArrayList<File>();
|
|
File[] entries = directory.listFiles();
|
|
|
|
// Go over entries
|
|
for (File entry : entries) {
|
|
// If there is no filter or the filter accepts the
|
|
// file / directory, add it to the list
|
|
if (filter == null || filter.accept(directory, entry.getName())) {
|
|
files.add(entry);
|
|
}
|
|
|
|
// If the file is a directory and the recurse flag
|
|
// is set, recurse into the directory
|
|
if (recurse && entry.isDirectory()) {
|
|
files.addAll(listFiles(entry, filter, recurse));
|
|
}
|
|
}
|
|
|
|
// Return collection of files
|
|
return files;
|
|
}
|
|
|
|
public static List<File> getClassLocationsForCurrentClasspath() {
|
|
List<File> urls = new ArrayList<File>();
|
|
String javaClassPath = System.getProperty("java.class.path");
|
|
if (javaClassPath != null) {
|
|
for (String path : javaClassPath.split(File.pathSeparator)) {
|
|
urls.add(new File(path));
|
|
}
|
|
}
|
|
return urls;
|
|
}
|
|
|
|
// todo: this is only partial, probably
|
|
public static URL normalize(URL url) throws MalformedURLException {
|
|
String spec = url.getFile();
|
|
|
|
// get url base - remove everything after ".jar!/??" , if exists
|
|
final int i = spec.indexOf("!/");
|
|
if (i != -1) {
|
|
spec = spec.substring(0, spec.indexOf("!/"));
|
|
}
|
|
|
|
// uppercase windows drive
|
|
url = new URL(url, spec);
|
|
final String file = url.getFile();
|
|
final int i1 = file.indexOf(':');
|
|
if (i1 != -1) {
|
|
String drive = file.substring(i1 - 1, 2).toUpperCase();
|
|
url = new URL(url, file.substring(0, i1 - 1) + drive + file.substring(i1));
|
|
}
|
|
|
|
return url;
|
|
}
|
|
|
|
private static void log(String pattern, final Object... args) {
|
|
if (DEBUG)
|
|
System.out.printf(pattern + "\n", args);
|
|
}
|
|
|
|
public static void main(String[] args) {
|
|
// find all classes in classpath
|
|
List<Class> allClasses = ClasspathInspector.getAllKnownClasses();
|
|
System.out.printf("There are %s classes available in the classpath\n", allClasses.size());
|
|
|
|
// find all classes that implement/subclass an interface/superclass
|
|
List<Class> serializableClasses = ClasspathInspector.getMatchingClasses(Serializable.class);
|
|
for (Class clazz : serializableClasses) {
|
|
System.out.printf("%s is Serializable\n", clazz);
|
|
}
|
|
}
|
|
}
|