mirror of https://github.com/DanilaFe/abacus
81 lines
3.1 KiB
Java
81 lines
3.1 KiB
Java
package org.nwapw.abacus.plugin;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.net.URL;
|
|
import java.net.URLClassLoader;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.util.ArrayList;
|
|
import java.util.Enumeration;
|
|
import java.util.List;
|
|
import java.util.jar.JarEntry;
|
|
import java.util.jar.JarFile;
|
|
import java.util.stream.Collectors;
|
|
|
|
/**
|
|
* Class that loads plugin classes from their jars.
|
|
*/
|
|
public class ClassFinder {
|
|
|
|
/**
|
|
* Loads all the plugin classes from the given plugin folder.
|
|
*
|
|
* @param filePath the path for the plugin folder.
|
|
* @return the list of all loaded classes.
|
|
* @throws IOException thrown if an error occurred scanning the plugin folder.
|
|
* @throws ClassNotFoundException thrown if the class listed in the file doesn't get loaded.
|
|
*/
|
|
public static List<Class<?>> loadJars(String filePath) throws IOException, ClassNotFoundException {
|
|
return loadJars(new File(filePath));
|
|
}
|
|
|
|
/**
|
|
* Loads all the plugin classes from the given plugin folder.
|
|
*
|
|
* @param pluginFolderPath the folder in which to look for plugins.
|
|
* @return the list of all loaded classes.
|
|
* @throws IOException thrown if an error occurred scanning the plugin folder.
|
|
* @throws ClassNotFoundException thrown if the class listed in the file doesn't get loaded.
|
|
*/
|
|
public static List<Class<?>> loadJars(File pluginFolderPath) throws IOException, ClassNotFoundException {
|
|
ArrayList<Class<?>> toReturn = new ArrayList<>();
|
|
if (!pluginFolderPath.exists()) return toReturn;
|
|
ArrayList<File> files = Files.walk(pluginFolderPath.toPath())
|
|
.map(Path::toFile)
|
|
.filter(f -> f.getName().endsWith(".jar"))
|
|
.collect(Collectors.toCollection(ArrayList::new));
|
|
for (File file : files) {
|
|
toReturn.addAll(loadJar(file));
|
|
}
|
|
return toReturn;
|
|
}
|
|
|
|
/**
|
|
* Loads the classes from a single path, given by the file.
|
|
*
|
|
* @param jarLocation the location of the jar to load.
|
|
* @return the list of loaded classes loaded from the jar.
|
|
* @throws IOException thrown if there was an error reading the file
|
|
* @throws ClassNotFoundException thrown if the class could not be loaded.
|
|
*/
|
|
public static List<Class<?>> loadJar(File jarLocation) throws IOException, ClassNotFoundException {
|
|
ArrayList<Class<?>> loadedClasses = new ArrayList<>();
|
|
String path = jarLocation.getPath();
|
|
URL[] urls = new URL[]{new URL("jar:file:" + path + "!/")};
|
|
|
|
URLClassLoader classLoader = URLClassLoader.newInstance(urls);
|
|
JarFile jarFolder = new JarFile(jarLocation);
|
|
Enumeration jarEntityList = jarFolder.entries();
|
|
|
|
while (jarEntityList.hasMoreElements()) {
|
|
JarEntry jarEntity = (JarEntry) jarEntityList.nextElement();
|
|
if (jarEntity.getName().endsWith(".class")) {
|
|
loadedClasses.add(classLoader.loadClass(jarEntity.getName().replace('/', '.').substring(0, jarEntity.getName().length() - 6)));
|
|
}
|
|
}
|
|
return loadedClasses;
|
|
}
|
|
|
|
}
|