74 lines
2.4 KiB
Java
74 lines
2.4 KiB
Java
package net.filebot.cli;
|
|
|
|
import java.util.Map;
|
|
import java.util.ResourceBundle;
|
|
|
|
import javax.script.Bindings;
|
|
import javax.script.ScriptContext;
|
|
import javax.script.ScriptEngine;
|
|
import javax.script.ScriptException;
|
|
|
|
import org.codehaus.groovy.control.CompilerConfiguration;
|
|
import org.codehaus.groovy.control.customizers.ImportCustomizer;
|
|
import org.codehaus.groovy.jsr223.GroovyScriptEngineImpl;
|
|
import org.codehaus.groovy.runtime.StackTraceUtils;
|
|
|
|
import groovy.lang.GroovyClassLoader;
|
|
|
|
public class ScriptShell {
|
|
|
|
public static final String ARGV_BINDING_NAME = "args";
|
|
public static final String SHELL_BINDING_NAME = "__shell";
|
|
public static final String SHELL_ARGV_BINDING_NAME = "__args";
|
|
|
|
private final ScriptEngine engine;
|
|
private final ScriptProvider scriptProvider;
|
|
|
|
public ScriptShell(ScriptProvider scriptProvider, Map<String, ?> globals) throws ScriptException {
|
|
this.engine = createScriptEngine();
|
|
this.scriptProvider = scriptProvider;
|
|
|
|
// setup bindings
|
|
Bindings bindings = engine.createBindings();
|
|
bindings.putAll(globals);
|
|
|
|
// bind API objects
|
|
bindings.put(SHELL_BINDING_NAME, this);
|
|
|
|
// setup script context
|
|
engine.getContext().setBindings(bindings, ScriptContext.GLOBAL_SCOPE);
|
|
}
|
|
|
|
public ScriptEngine createScriptEngine() {
|
|
ResourceBundle bundle = ResourceBundle.getBundle(ScriptShell.class.getName());
|
|
|
|
CompilerConfiguration config = new CompilerConfiguration();
|
|
config.setScriptBaseClass(bundle.getString("scriptBaseClass"));
|
|
|
|
// default imports
|
|
ImportCustomizer imports = new ImportCustomizer();
|
|
imports.addStarImports(bundle.getString("starImport").split(", "));
|
|
imports.addStaticStars(bundle.getString("starStaticImport").split(", "));
|
|
config.addCompilationCustomizers(imports);
|
|
|
|
GroovyClassLoader classLoader = new GroovyClassLoader(Thread.currentThread().getContextClassLoader(), config);
|
|
return new GroovyScriptEngineImpl(classLoader);
|
|
}
|
|
|
|
public Object evaluate(final String script, final Bindings bindings) throws Throwable {
|
|
try {
|
|
return engine.eval(script, bindings);
|
|
} catch (Throwable e) {
|
|
while (e.getClass() == ScriptException.class && e.getCause() != null) {
|
|
e = e.getCause();
|
|
}
|
|
throw StackTraceUtils.deepSanitize(e); // make Groovy stacktrace human-readable
|
|
}
|
|
}
|
|
|
|
public Object runScript(String name, Bindings bindings) throws Throwable {
|
|
return evaluate(scriptProvider.getScript(name), bindings);
|
|
}
|
|
|
|
}
|