mirror of
https://github.com/ViaVersion/ViaProxy.git
synced 2024-09-07 01:08:06 +02:00
Added transformer support for plugins
This commit is contained in:
parent
bda7244b90
commit
d9b01a3b63
@ -17,15 +17,21 @@
|
|||||||
*/
|
*/
|
||||||
package net.raphimc.viaproxy.plugins;
|
package net.raphimc.viaproxy.plugins;
|
||||||
|
|
||||||
|
import com.vdurmont.semver4j.Semver;
|
||||||
|
import net.lenni0451.classtransform.TransformerManager;
|
||||||
|
import net.lenni0451.classtransform.additionalclassprovider.GuavaClassPathProvider;
|
||||||
|
import net.lenni0451.classtransform.utils.loader.InjectionClassLoader;
|
||||||
|
import net.lenni0451.classtransform.utils.tree.IClassProvider;
|
||||||
import net.lenni0451.lambdaevents.LambdaManager;
|
import net.lenni0451.lambdaevents.LambdaManager;
|
||||||
import net.lenni0451.lambdaevents.generator.LambdaMetaFactoryGenerator;
|
import net.lenni0451.lambdaevents.generator.LambdaMetaFactoryGenerator;
|
||||||
|
import net.raphimc.viaproxy.ViaProxy;
|
||||||
|
import net.raphimc.viaproxy.util.URLClassProvider;
|
||||||
import net.raphimc.viaproxy.util.logging.Logger;
|
import net.raphimc.viaproxy.util.logging.Logger;
|
||||||
import org.yaml.snakeyaml.Yaml;
|
import org.yaml.snakeyaml.Yaml;
|
||||||
|
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
import java.net.URL;
|
import java.net.URL;
|
||||||
import java.net.URLClassLoader;
|
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
@ -37,6 +43,7 @@ public class PluginManager {
|
|||||||
public static final File PLUGINS_DIR = new File("plugins");
|
public static final File PLUGINS_DIR = new File("plugins");
|
||||||
|
|
||||||
private static final Yaml YAML = new Yaml();
|
private static final Yaml YAML = new Yaml();
|
||||||
|
private static final IClassProvider ROOT_CLASS_PROVIDER = new GuavaClassPathProvider();
|
||||||
private static final List<ViaProxyPlugin> PLUGINS = new ArrayList<>();
|
private static final List<ViaProxyPlugin> PLUGINS = new ArrayList<>();
|
||||||
|
|
||||||
public static List<ViaProxyPlugin> getPlugins() {
|
public static List<ViaProxyPlugin> getPlugins() {
|
||||||
@ -64,7 +71,10 @@ public class PluginManager {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private static void loadAndScanJar(final File file) throws Throwable {
|
private static void loadAndScanJar(final File file) throws Throwable {
|
||||||
URLClassLoader loader = new URLClassLoader(new URL[]{file.toURI().toURL()}, PluginManager.class.getClassLoader());
|
URL url = file.toURI().toURL();
|
||||||
|
URLClassProvider classProvider = new URLClassProvider(ROOT_CLASS_PROVIDER, url);
|
||||||
|
TransformerManager transformerManager = new TransformerManager(classProvider);
|
||||||
|
InjectionClassLoader loader = new InjectionClassLoader(transformerManager, PluginManager.class.getClassLoader(), url);
|
||||||
InputStream viaproxyYml = loader.getResourceAsStream("viaproxy.yml");
|
InputStream viaproxyYml = loader.getResourceAsStream("viaproxy.yml");
|
||||||
if (viaproxyYml == null) throw new IllegalStateException("Plugin '" + file.getName() + "' does not have a viaproxy.yml");
|
if (viaproxyYml == null) throw new IllegalStateException("Plugin '" + file.getName() + "' does not have a viaproxy.yml");
|
||||||
Map<String, Object> yaml = YAML.load(viaproxyYml);
|
Map<String, Object> yaml = YAML.load(viaproxyYml);
|
||||||
@ -72,6 +82,10 @@ public class PluginManager {
|
|||||||
if (!yaml.containsKey("author")) throw new IllegalStateException("Plugin '" + file.getName() + "' does not have a author attribute in the viaproxy.yml");
|
if (!yaml.containsKey("author")) throw new IllegalStateException("Plugin '" + file.getName() + "' does not have a author attribute in the viaproxy.yml");
|
||||||
if (!yaml.containsKey("version")) throw new IllegalStateException("Plugin '" + file.getName() + "' does not have a version attribute in the viaproxy.yml");
|
if (!yaml.containsKey("version")) throw new IllegalStateException("Plugin '" + file.getName() + "' does not have a version attribute in the viaproxy.yml");
|
||||||
if (!yaml.containsKey("main")) throw new IllegalStateException("Plugin '" + file.getName() + "' does not have a main attribute in the viaproxy.yml");
|
if (!yaml.containsKey("main")) throw new IllegalStateException("Plugin '" + file.getName() + "' does not have a main attribute in the viaproxy.yml");
|
||||||
|
Semver minVersion = new Semver(yaml.getOrDefault("minVersion", "0.0.0").toString());
|
||||||
|
if (!ViaProxy.VERSION.equals("${version}") && minVersion.isLowerThan(new Semver(ViaProxy.VERSION))) {
|
||||||
|
throw new IllegalStateException("Plugin '" + file.getName() + "' requires a newer version of ViaProxy (v" + minVersion + ")");
|
||||||
|
}
|
||||||
|
|
||||||
String main = (String) yaml.get("main");
|
String main = (String) yaml.get("main");
|
||||||
|
|
||||||
@ -83,6 +97,7 @@ public class PluginManager {
|
|||||||
ViaProxyPlugin plugin = (ViaProxyPlugin) instance;
|
ViaProxyPlugin plugin = (ViaProxyPlugin) instance;
|
||||||
PLUGINS.add(plugin);
|
PLUGINS.add(plugin);
|
||||||
|
|
||||||
|
plugin.registerTransformers(transformerManager, classProvider);
|
||||||
plugin.onEnable();
|
plugin.onEnable();
|
||||||
Logger.LOGGER.info("Successfully loaded plugin '" + yaml.get("name") + "' by " + yaml.get("author") + " (v" + yaml.get("version") + ")");
|
Logger.LOGGER.info("Successfully loaded plugin '" + yaml.get("name") + "' by " + yaml.get("author") + " (v" + yaml.get("version") + ")");
|
||||||
}
|
}
|
||||||
|
@ -17,8 +17,14 @@
|
|||||||
*/
|
*/
|
||||||
package net.raphimc.viaproxy.plugins;
|
package net.raphimc.viaproxy.plugins;
|
||||||
|
|
||||||
|
import net.lenni0451.classtransform.TransformerManager;
|
||||||
|
import net.lenni0451.classtransform.utils.tree.IClassProvider;
|
||||||
|
|
||||||
public abstract class ViaProxyPlugin {
|
public abstract class ViaProxyPlugin {
|
||||||
|
|
||||||
public abstract void onEnable();
|
public abstract void onEnable();
|
||||||
|
|
||||||
|
public void registerTransformers(final TransformerManager transformerManager, final IClassProvider classProvider) {
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,57 @@
|
|||||||
|
/*
|
||||||
|
* This file is part of ViaProxy - https://github.com/RaphiMC/ViaProxy
|
||||||
|
* Copyright (C) 2023 RK_01/RaphiMC and contributors
|
||||||
|
*
|
||||||
|
* This program is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* 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 General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
package net.raphimc.viaproxy.util;
|
||||||
|
|
||||||
|
import net.lenni0451.classtransform.utils.ASMUtils;
|
||||||
|
import net.lenni0451.classtransform.utils.tree.IClassProvider;
|
||||||
|
import org.apache.commons.io.IOUtils;
|
||||||
|
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.function.Supplier;
|
||||||
|
|
||||||
|
public class URLClassProvider implements IClassProvider {
|
||||||
|
|
||||||
|
private final IClassProvider parent;
|
||||||
|
private final List<URL> urls;
|
||||||
|
|
||||||
|
public URLClassProvider(final IClassProvider parent, final URL... urls) {
|
||||||
|
this.parent = parent;
|
||||||
|
this.urls = Arrays.asList(urls);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public byte[] getClass(String name) {
|
||||||
|
for (URL url : this.urls) {
|
||||||
|
try (InputStream is = new URL("jar:" + url + "!/" + ASMUtils.slash(name) + ".class").openStream()) {
|
||||||
|
IOUtils.toByteArray(is);
|
||||||
|
} catch (Throwable ignored) {
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return this.parent.getClass(name);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Map<String, Supplier<byte[]>> getAllClasses() {
|
||||||
|
return this.parent.getAllClasses();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user