me.remigio07.chatplugin.bootstrap.JARLibraryLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api Show documentation
A complete yet lightweight plugin which handles just too many features!
There is a newer version: 1.9.10
/*
 * 	ChatPlugin - A complete yet lightweight plugin which handles just too many features!
 * 	Copyright 2024  Remigio07
 * 	
 * 	This program is distributed in the hope that it will be useful,
 * 	but WITHOUT ANY WARRANTY; without even the implied warranty of
 * 	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * 	GNU Affero General Public License for more details.
 * 	
 * 	You should have received a copy of the GNU Affero General Public License
 * 	along with this program.  If not, see .
 * 	
 * 	
 */

package me.remigio07.chatplugin.bootstrap;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.InvocationTargetException;
import java.net.URL;
import java.net.URLClassLoader;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * Class used to load ChatPlugin's libraries from JAR files.
 */
public class JARLibraryLoader extends URLClassLoader {
	
	private static JARLibraryLoader instance = new JARLibraryLoader();
	
	static {
		registerAsParallelCapable();
	}
	
	private JARLibraryLoader() {
		super(new URL[0], JARLibraryLoader.class.getClassLoader());
	}
	
	/**
	 * Initializes ChatPlugin.
	 * 
	 * Uses {@link Environment#getCurrent()}'s
	 * {@link Environment#getEnableMethodArgsTypes()}.
	 * 
	 * @param args Enable method's args
	 */
	public void initialize(Object... args) {
		try {
			Environment environment = Environment.getCurrent();
			List jars = Stream.of(extract(false), extract(true)).filter(Objects::nonNull).collect(Collectors.toList());
			
			if (instance.getURLs().length == 0)
				for (URL jar : jars)
					instance.addURL(jar);
			Class mainClass = Class.forName("me.remigio07.chatplugin.ChatPlugin" + (jars.size() == 2 ? "Premium" : "") + "Impl", true, instance);
			
			if (environment == Environment.VELOCITY)
				mainClass.getMethod("load", Object.class, Object.class, Object.class).invoke(null, args);
			else mainClass.getMethod("load", Object.class, Object.class).invoke(null, args);
		} catch (IOException | ClassNotFoundException | IllegalAccessException | InvocationTargetException | NoSuchMethodException  e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * Loads a library contained in the specified JAR file.
	 * 
	 * @param target Library to load
	 * @throws IOException If target is not a valid JAR file or an I/O error occurrs
	 * @throws SecurityException If the {@link SecurityManager} denies the access to target
	 */
	public void load(File target) throws IOException {
		addURL(target.toURI().toURL());
	}
	
	private URL extract(boolean premium) throws IOException {
		URL url = JARLibraryLoader.class.getClassLoader().getResource("ChatPlugin" + (premium ? "-PREMIUM" : "") + ".jar");
		
		if (url == null)
			return null;
		try (InputStream input = url.openStream()) {
			Path path = Files.createTempFile("ChatPlugin" + (premium ? "-PREMIUM" : ""), ".jar.tmp");
			
			path.toFile().deleteOnExit();
			Files.copy(input, path, StandardCopyOption.REPLACE_EXISTING);
			return path.toUri().toURL();
		}
	}
	
	/**
	 * Gets the loader's current instance.
	 * 
	 * @return Loader's instance
	 */
	public static JARLibraryLoader getInstance() {
		return instance;
	}
	
}