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.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.stream.Collectors;

/**
 * Class that loads plugin classes from their jars.
 */
public class ClassFinder {

    public static ArrayList<Class<?>> loadJars(String filePath) throws IOException, ClassNotFoundException {
        return loadJars(new File(filePath));
    }

    public static ArrayList<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;
    }

    public static ArrayList<Class<?>> loadJar(String path) throws IOException, ClassNotFoundException {
        return loadJar(new File(path));
    }

    public static ArrayList<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;
    }

}

@@ -436,164 +436,909 @@
     
-public static ArrayList%3CClass%3C?%3E%3E loadJars(String filePath) throws IOException, ClassNotFoundException %7B%0A        return loadJars(new File(filePath));%0A    %7D%0A
+/**%0A     * Loads all the plugin classes from the given plugin folder.%0A     * @param filePath the path for the plugin folder.%0A     * @return the list of all loaded classes.%0A     * @throws IOException thrown if an error occurred scanning the plugin folder.%0A     * @throws ClassNotFoundException thrown if the class listed in the file doesn't get loaded.%0A     */%0A    public static ArrayList%3CClass%3C?%3E%3E loadJars(String filePath) throws IOException, ClassNotFoundException %7B%0A        return loadJars(new File(filePath));%0A    %7D%0A%0A    /**%0A     * Loads all the plugin classes from the given plugin folder.%0A     * @param pluginFolderPath the folder in which to look for plugins.%0A     * @return the list of all loaded classes.%0A     * @throws IOException thrown if an error occurred scanning the plugin folder.%0A     * @throws ClassNotFoundException thrown if the class listed in the file doesn't get loaded.%0A     */
 %0A   
@@ -1921,154 +1921,357 @@
     
-public static ArrayList%3CClass%3C?%3E%3E loadJar(String path) throws IOException, ClassNotFoundException %7B%0A        return loadJar(new File(path));%0A
+/**%0A     * Loads the classes from a single path, given by the file.%0A     * @param jarLocation the location of the jar to load.%0A     * @return the list of loaded classes loaded from the jar.%0A     * @throws IOException thrown if there was an error reading the file%0A     * @throws ClassNotFoundException thrown if the class could not be loaded.%0A 
     
-%7D%0A
+*/
 %0A   

Comment class finder.