177 lines
5.2 KiB
Java
177 lines
5.2 KiB
Java
package pw.yumc.YumCore.engine;
|
|
|
|
import java.io.File;
|
|
import java.io.Reader;
|
|
import java.lang.reflect.InvocationTargetException;
|
|
import java.lang.reflect.Method;
|
|
import java.net.MalformedURLException;
|
|
import java.net.URL;
|
|
import java.net.URLClassLoader;
|
|
import java.util.HashMap;
|
|
|
|
import javax.script.Bindings;
|
|
import javax.script.Invocable;
|
|
import javax.script.ScriptContext;
|
|
import javax.script.ScriptEngine;
|
|
import javax.script.ScriptEngineFactory;
|
|
import javax.script.ScriptEngineManager;
|
|
import javax.script.ScriptException;
|
|
import javax.script.SimpleBindings;
|
|
|
|
/**
|
|
* 喵式脚本引擎
|
|
*
|
|
* @author 喵♂呜
|
|
* @since 2016年8月29日 上午7:51:43
|
|
*/
|
|
public class MiaoScriptEngine implements ScriptEngine, Invocable {
|
|
private static MiaoScriptEngine DEFAULT;
|
|
private static ScriptEngineManager manager;
|
|
private ScriptEngine engine;
|
|
|
|
static {
|
|
manager = new ScriptEngineManager(ClassLoader.getSystemClassLoader());
|
|
}
|
|
|
|
public static void setBindings(Bindings bindings) {
|
|
manager.setBindings(bindings);
|
|
}
|
|
|
|
public static Bindings getBindings() {
|
|
return manager.getBindings();
|
|
}
|
|
|
|
public MiaoScriptEngine() {
|
|
this("js");
|
|
}
|
|
|
|
public MiaoScriptEngine(final String engineType) {
|
|
this(manager, engineType);
|
|
}
|
|
|
|
public MiaoScriptEngine(ScriptEngineManager engineManager) {
|
|
this(engineManager, "js");
|
|
}
|
|
|
|
public MiaoScriptEngine(ScriptEngineManager engineManager, final String engineType) {
|
|
try {
|
|
engine = engineManager.getEngineByName(engineType);
|
|
} catch (final NullPointerException ignored) {
|
|
}
|
|
if (engine == null) {
|
|
File nashorn = new File(System.getProperty("java.ext.dirs").split(File.pathSeparator)[0], "nashorn.jar");
|
|
if (nashorn.exists()) {
|
|
try {
|
|
Method method = URLClassLoader.class.getDeclaredMethod("addURL", URL.class);
|
|
// 设置方法的访问权限
|
|
method.setAccessible(true);
|
|
// 获取系统类加载器
|
|
URL url = nashorn.toURI().toURL();
|
|
method.invoke(Thread.currentThread().getContextClassLoader(), url);
|
|
engineManager = new ScriptEngineManager();
|
|
engine = engineManager.getEngineByName(engineType);
|
|
} catch (NoSuchMethodException | MalformedURLException | IllegalAccessException | InvocationTargetException ignored) {
|
|
ignored.printStackTrace();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public static MiaoScriptEngine getDefault() {
|
|
if (DEFAULT == null) {
|
|
DEFAULT = new MiaoScriptEngine();
|
|
}
|
|
return DEFAULT;
|
|
}
|
|
|
|
@Override
|
|
public Bindings createBindings() {
|
|
return new SimpleBindings(new HashMap<>(engine.getBindings(ScriptContext.GLOBAL_SCOPE)));
|
|
}
|
|
|
|
@Override
|
|
public Object eval(final Reader reader) throws ScriptException {
|
|
return engine.eval(reader);
|
|
}
|
|
|
|
@Override
|
|
public Object eval(final Reader reader, final Bindings n) throws ScriptException {
|
|
return engine.eval(reader, n);
|
|
}
|
|
|
|
@Override
|
|
public Object eval(final Reader reader, final ScriptContext context) throws ScriptException {
|
|
return engine.eval(reader, context);
|
|
}
|
|
|
|
@Override
|
|
public Object eval(final String script) throws ScriptException {
|
|
return engine.eval(script);
|
|
}
|
|
|
|
@Override
|
|
public Object eval(final String script, final Bindings n) throws ScriptException {
|
|
return engine.eval(script, n);
|
|
}
|
|
|
|
@Override
|
|
public Object eval(final String script, final ScriptContext context) throws ScriptException {
|
|
return engine.eval(script, context);
|
|
}
|
|
|
|
@Override
|
|
public Object get(final String key) {
|
|
return engine.get(key);
|
|
}
|
|
|
|
@Override
|
|
public Bindings getBindings(final int scope) {
|
|
return engine.getBindings(scope);
|
|
}
|
|
|
|
@Override
|
|
public ScriptContext getContext() {
|
|
return engine.getContext();
|
|
}
|
|
|
|
@Override
|
|
public ScriptEngineFactory getFactory() {
|
|
return engine.getFactory();
|
|
}
|
|
|
|
@Override
|
|
public <T> T getInterface(final Class<T> clasz) {
|
|
return ((Invocable) engine).getInterface(clasz);
|
|
}
|
|
|
|
@Override
|
|
public <T> T getInterface(final Object thiz, final Class<T> clasz) {
|
|
return ((Invocable) engine).getInterface(thiz, clasz);
|
|
}
|
|
|
|
@Override
|
|
public Object invokeFunction(final String name, final Object... args) throws ScriptException, NoSuchMethodException {
|
|
return ((Invocable) engine).invokeFunction(name, args);
|
|
}
|
|
|
|
@Override
|
|
public Object invokeMethod(final Object thiz, final String name, final Object... args) throws ScriptException, NoSuchMethodException {
|
|
return ((Invocable) engine).invokeMethod(thiz, name, args);
|
|
}
|
|
|
|
@Override
|
|
public void put(final String key, final Object value) {
|
|
engine.put(key, value);
|
|
}
|
|
|
|
@Override
|
|
public void setBindings(final Bindings bindings, final int scope) {
|
|
engine.setBindings(bindings, scope);
|
|
}
|
|
|
|
@Override
|
|
public void setContext(final ScriptContext context) {
|
|
engine.setContext(context);
|
|
}
|
|
}
|