refactor the graaljs integration to allow Java code to be called; updated the simulation extension and scripting to due to a change in the FlightConfiguration

This commit is contained in:
thzero 2022-07-02 13:55:09 -05:00
parent 0337ed4c78
commit 1b1eecaa4b
5 changed files with 290 additions and 291 deletions

View File

@ -270,5 +270,32 @@
<SOURCES /> <SOURCES />
</library> </library>
</orderEntry> </orderEntry>
<orderEntry type="module-library">
<library>
<CLASSES>
<root url="jar://$MODULE_DIR$/lib/js-scriptengine-22.1.0.1.jar!/" />
</CLASSES>
<JAVADOC />
<SOURCES />
</library>
</orderEntry>
<orderEntry type="module-library">
<library>
<CLASSES>
<root url="jar://$MODULE_DIR$/lib/js-22.1.0.1.jar!/" />
</CLASSES>
<JAVADOC />
<SOURCES />
</library>
</orderEntry>
<orderEntry type="module-library">
<library>
<CLASSES>
<root url="jar://$MODULE_DIR$/lib/graal-sdk-22.1.0.1.jar!/" />
</CLASSES>
<JAVADOC />
<SOURCES />
</library>
</orderEntry>
</component> </component>
</module> </module>

View File

@ -0,0 +1,99 @@
package net.sf.openrocket.scripting;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineFactory;
import java.util.*;
import com.oracle.truffle.js.scriptengine.GraalJSScriptEngine;
import org.graalvm.polyglot.Context;
import org.graalvm.polyglot.HostAccess;
public class GraalJSScriptEngineFactory implements ScriptEngineFactory {
private static final String ENGINE_NAME = "Graal.js";
private static final String LANGUAGE = "ECMAScript";
private static final String LANGUAGE_VERSION = "ECMAScript 262 Edition 11";
private static final String[] NAMES = new String[]{"js", "JS", "JavaScript", "javascript", LANGUAGE, LANGUAGE.toLowerCase(), ENGINE_NAME, ENGINE_NAME.toLowerCase(), "Graal-js", "graal-js", "Graal.JS", "Graal-JS", "GraalJS", "GraalJSPolyglot"};
private static final String[] MIME_TYPES = new String[]{"application/javascript", "application/ecmascript", "text/javascript", "text/ecmascript"};
private static final String[] EXTENSIONS = new String[]{"js", "mjs"};
private static final List names;
private static final List mimeTypes;
private static final List extensions;
public GraalJSScriptEngineFactory() {
}
public ScriptEngine getScriptEngine() {
return GraalJSScriptEngine.create(null,
Context.newBuilder("js")
.allowHostAccess(HostAccess.ALL)
.allowHostClassLookup(s -> true)
.option("js.ecmascript-version", "2021"));
}
public String getEngineName() {
return ENGINE_NAME;
}
public String getEngineVersion() {
return ((GraalJSScriptEngine)getScriptEngine()).getPolyglotEngine().getVersion();
}
public List<String> getExtensions() {
return extensions;
}
public String getLanguageVersion() {
return LANGUAGE_VERSION;
}
public String getLanguageName() {
return LANGUAGE;
}
public List<String> getMimeTypes() {
return mimeTypes;
}
public List<String> getNames() {
return names;
}
public String getMethodCallSyntax(final String obj, final String method, final String... args) {
return null;
}
public String getOutputStatement(final String toDisplay) {
return "print(" + toDisplay + ")";
}
public Object getParameter(String key) {
switch (key) {
case "javax.script.name":
return "javascript";
case "javax.script.engine":
return this.getEngineName();
case "javax.script.engine_version":
return this.getEngineVersion();
case "javax.script.language":
return this.getLanguageName();
case "javax.script.language_version":
return this.getLanguageVersion();
default:
return null;
}
}
public String getProgram(final String... statements) {
return null;
}
static {
List<String> nameList = Arrays.asList(NAMES);
List<String> mimeTypeList = Arrays.asList(MIME_TYPES);
List<String> extensionList = Arrays.asList(EXTENSIONS);
names = Collections.unmodifiableList(nameList);
mimeTypes = Collections.unmodifiableList(mimeTypeList);
extensions = Collections.unmodifiableList(extensionList);
}
}

View File

@ -66,8 +66,7 @@ public class ScriptEngineManagerRedux {
* @see java.lang.Thread#getContextClassLoader * @see java.lang.Thread#getContextClassLoader
*/ */
public ScriptEngineManagerRedux() { public ScriptEngineManagerRedux() {
ClassLoader ctxtLoader = Thread.currentThread().getContextClassLoader(); init(Thread.currentThread().getContextClassLoader());
init(ctxtLoader);
} }
/** /**
@ -83,96 +82,13 @@ public class ScriptEngineManagerRedux {
init(loader); init(loader);
} }
private void init(final ClassLoader loader) {
globalScope = new SimpleBindings();
engineSpis = new TreeSet<ScriptEngineFactory>(Comparator.comparing(
ScriptEngineFactory::getEngineName,
Comparator.nullsLast(Comparator.naturalOrder()))
);
nameAssociations = new HashMap<String, ScriptEngineFactory>();
extensionAssociations = new HashMap<String, ScriptEngineFactory>();
mimeTypeAssociations = new HashMap<String, ScriptEngineFactory>();
initEngines(loader);
}
private ServiceLoader<ScriptEngineFactory> getServiceLoader(final ClassLoader loader) {
if (loader != null) {
return ServiceLoader.load(ScriptEngineFactory.class, loader);
} else {
return ServiceLoader.loadInstalled(ScriptEngineFactory.class);
}
}
private void initEngines(final ClassLoader loader) {
Iterator<ScriptEngineFactory> itr = null;
try {
ServiceLoader<ScriptEngineFactory> sl = AccessController.doPrivileged(
new PrivilegedAction<ServiceLoader<ScriptEngineFactory>>() {
@Override
public ServiceLoader<ScriptEngineFactory> run() {
return getServiceLoader(loader);
}
});
itr = sl.iterator();
} catch (ServiceConfigurationError err) {
// } catch (Exception err) {
System.err.println("Can't find ScriptEngineFactory providers: " +
err.getMessage());
if (DEBUG) {
err.printStackTrace();
}
// do not throw any exception here. user may want to
// manage his/her own factories using this manager
// by explicit registratation (by registerXXX) methods.
return;
}
try {
while (itr.hasNext()) {
try {
ScriptEngineFactory fact = itr.next();
engineSpis.add(fact);
} catch (ServiceConfigurationError err) {
// } catch (Exception err) {
System.err.println("ScriptEngineManager providers.next(): "
+ err.getMessage());
if (DEBUG) {
err.printStackTrace();
}
// one factory failed, but check other factories...
continue;
}
}
} catch (ServiceConfigurationError err) {
// } catch (Exception err) {
System.err.println("ScriptEngineManager providers.hasNext(): "
+ err.getMessage());
if (DEBUG) {
err.printStackTrace();
}
// do not throw any exception here. user may want to
// manage his/her own factories using this manager
// by explicit registratation (by registerXXX) methods.
return;
}
}
/** /**
* <code>setBindings</code> stores the specified <code>Bindings</code> * Gets the value for the specified key in the Global Scope
* in the <code>globalScope</code> field. ScriptEngineManager sets this * @param key The key whose value is to be returned.
* <code>Bindings</code> as global bindings for <code>ScriptEngine</code> * @return The value for the specified key.
* objects created by it.
*
* @param bindings The specified <code>Bindings</code>
* @throws IllegalArgumentException if bindings is null.
*/ */
public void setBindings(Bindings bindings) { public Object get(String key) {
if (bindings == null) { return _globalScope.get(key);
throw new IllegalArgumentException("Global scope cannot be null.");
}
globalScope = bindings;
} }
/** /**
@ -183,27 +99,7 @@ public class ScriptEngineManagerRedux {
* @return The globalScope field. * @return The globalScope field.
*/ */
public Bindings getBindings() { public Bindings getBindings() {
return globalScope; return _globalScope;
}
/**
* Sets the specified key/value pair in the Global Scope.
* @param key Key to set
* @param value Value to set.
* @throws NullPointerException if key is null.
* @throws IllegalArgumentException if key is empty string.
*/
public void put(String key, Object value) {
globalScope.put(key, value);
}
/**
* Gets the value for the specified key in the Global Scope
* @param key The key whose value is to be returned.
* @return The value for the specified key.
*/
public Object get(String key) {
return globalScope.get(key);
} }
/** /**
@ -222,140 +118,46 @@ public class ScriptEngineManagerRedux {
* created <code>ScriptEngine</code>. * created <code>ScriptEngine</code>.
* @throws NullPointerException if shortName is null. * @throws NullPointerException if shortName is null.
*/ */
public ScriptEngine getEngineByName(String shortName) { private Map<String, ScriptEngineFactory> _factoriesByName = new HashMap<>();
if (shortName == null) throw new NullPointerException(); public synchronized ScriptEngine getEngineByName(String shortName) {
//look for registered name first if (shortName == null) {
Object obj; throw new NullPointerException();
if (null != (obj = nameAssociations.get(shortName))) {
ScriptEngineFactory spi = (ScriptEngineFactory)obj;
try {
ScriptEngine engine = spi.getScriptEngine();
engine.setBindings(getBindings(), ScriptContext.GLOBAL_SCOPE);
return engine;
} catch (Exception exp) {
if (DEBUG) exp.printStackTrace();
}
} }
for (ScriptEngineFactory spi : engineSpis) { String key = shortName.toLowerCase();
List<String> names = null; if (_factoriesByName.containsKey(key)) {
try { return getEngineByFactory(_factoriesByName.get(key));
names = spi.getNames(); }
} catch (Exception exp) {
if (DEBUG) exp.printStackTrace();
}
if (names != null) { // Look for registered name first
for (String name : names) { ScriptEngineFactory factoryNamed;
if (shortName.equals(name)) { if (null != (factoryNamed = _nameAssociations.get(key))) {
try { try {
ScriptEngine engine = spi.getScriptEngine(); _factoriesByName.put(key, factoryNamed);
engine.setBindings(getBindings(), ScriptContext.GLOBAL_SCOPE); return getEngineByFactory(factoryNamed);
return engine; } catch (Exception exp) {
} catch (Exception exp) { if (DEBUG) {
if (DEBUG) exp.printStackTrace(); exp.printStackTrace();
}
}
} }
} }
} }
return null; Optional<String> factoryName;
} List<String> names;
for (ScriptEngineFactory factory : _scriptEngineFactories) {
/**
* Look up and create a <code>ScriptEngine</code> for a given extension. The algorithm
* used by <code>getEngineByName</code> is used except that the search starts
* by looking for a <code>ScriptEngineFactory</code> registered to handle the
* given extension using <code>registerEngineExtension</code>.
* @param extension The given extension
* @return The engine to handle scripts with this extension. Returns <code>null</code>
* if not found.
* @throws NullPointerException if extension is null.
*/
public ScriptEngine getEngineByExtension(String extension) {
if (extension == null) throw new NullPointerException();
//look for registered extension first
Object obj;
if (null != (obj = extensionAssociations.get(extension))) {
ScriptEngineFactory spi = (ScriptEngineFactory)obj;
try { try {
ScriptEngine engine = spi.getScriptEngine(); factoryName = factory.getNames().stream().filter(l -> l.equalsIgnoreCase(shortName)).findFirst();
engine.setBindings(getBindings(), ScriptContext.GLOBAL_SCOPE); if (factoryName.isPresent()) {
return engine; _factoriesByName.put(key, factory);
return getEngineByFactory(factory);
}
} catch (Exception exp) { } catch (Exception exp) {
if (DEBUG) exp.printStackTrace(); if (DEBUG) {
} exp.printStackTrace();
}
for (ScriptEngineFactory spi : engineSpis) {
List<String> exts = null;
try {
exts = spi.getExtensions();
} catch (Exception exp) {
if (DEBUG) exp.printStackTrace();
}
if (exts == null) continue;
for (String ext : exts) {
if (extension.equals(ext)) {
try {
ScriptEngine engine = spi.getScriptEngine();
engine.setBindings(getBindings(), ScriptContext.GLOBAL_SCOPE);
return engine;
} catch (Exception exp) {
if (DEBUG) exp.printStackTrace();
}
} }
} }
} }
return null;
}
/**
* Look up and create a <code>ScriptEngine</code> for a given mime type. The algorithm
* used by <code>getEngineByName</code> is used except that the search starts
* by looking for a <code>ScriptEngineFactory</code> registered to handle the
* given mime type using <code>registerEngineMimeType</code>.
* @param mimeType The given mime type
* @return The engine to handle scripts with this mime type. Returns <code>null</code>
* if not found.
* @throws NullPointerException if mimeType is null.
*/
public ScriptEngine getEngineByMimeType(String mimeType) {
if (mimeType == null) throw new NullPointerException();
//look for registered types first
Object obj;
if (null != (obj = mimeTypeAssociations.get(mimeType))) {
ScriptEngineFactory spi = (ScriptEngineFactory)obj;
try {
ScriptEngine engine = spi.getScriptEngine();
engine.setBindings(getBindings(), ScriptContext.GLOBAL_SCOPE);
return engine;
} catch (Exception exp) {
if (DEBUG) exp.printStackTrace();
}
}
for (ScriptEngineFactory spi : engineSpis) {
List<String> types = null;
try {
types = spi.getMimeTypes();
} catch (Exception exp) {
if (DEBUG) exp.printStackTrace();
}
if (types == null) continue;
for (String type : types) {
if (mimeType.equals(type)) {
try {
ScriptEngine engine = spi.getScriptEngine();
engine.setBindings(getBindings(), ScriptContext.GLOBAL_SCOPE);
return engine;
} catch (Exception exp) {
if (DEBUG) exp.printStackTrace();
}
}
}
}
return null; return null;
} }
@ -364,12 +166,19 @@ public class ScriptEngineManagerRedux {
* found by the discovery mechanism. * found by the discovery mechanism.
* @return List of all discovered <code>ScriptEngineFactory</code>s. * @return List of all discovered <code>ScriptEngineFactory</code>s.
*/ */
public List<ScriptEngineFactory> getEngineFactories() { public synchronized List<ScriptEngineFactory> getEngineFactories() {
List<ScriptEngineFactory> res = new ArrayList<ScriptEngineFactory>(engineSpis.size()); return List.copyOf(_scriptEngineFactories);
for (ScriptEngineFactory spi : engineSpis) { }
res.add(spi);
} /**
return Collections.unmodifiableList(res); * Sets the specified key/value pair in the Global Scope.
* @param key Key to set
* @param value Value to set.
* @throws NullPointerException if key is null.
* @throws IllegalArgumentException if key is empty string.
*/
public void put(String key, Object value) {
_globalScope.put(key, value);
} }
/** /**
@ -380,51 +189,112 @@ public class ScriptEngineManagerRedux {
* @throws NullPointerException if any of the parameters is null. * @throws NullPointerException if any of the parameters is null.
*/ */
public void registerEngineName(String name, ScriptEngineFactory factory) { public void registerEngineName(String name, ScriptEngineFactory factory) {
if (name == null || factory == null) throw new NullPointerException(); if (name == null || factory == null) {
nameAssociations.put(name, factory); throw new NullPointerException();
}
_nameAssociations.put(name.toLowerCase(), factory);
} }
/** /**
* Registers a <code>ScriptEngineFactory</code> to handle a mime type. * <code>setBindings</code> stores the specified <code>Bindings</code>
* Overrides any such association found using the Discovery mechanism. * in the <code>globalScope</code> field. ScriptEngineManager sets this
* <code>Bindings</code> as global bindings for <code>ScriptEngine</code>
* objects created by it.
* *
* @param type The mime type to be associated with the * @param bindings The specified <code>Bindings</code>
* <code>ScriptEngineFactory</code>. * @throws IllegalArgumentException if bindings is null.
*
* @param factory The class to associate with the given mime type.
* @throws NullPointerException if any of the parameters is null.
*/ */
public void registerEngineMimeType(String type, ScriptEngineFactory factory) { public void setBindings(Bindings bindings) {
if (type == null || factory == null) throw new NullPointerException(); if (bindings == null) {
mimeTypeAssociations.put(type, factory); throw new IllegalArgumentException("Global scope cannot be null.");
}
_globalScope = bindings;
} }
/** private ScriptEngine getEngineByFactory(ScriptEngineFactory factory) {
* Registers a <code>ScriptEngineFactory</code> to handle an extension. if (factory == null) {
* Overrides any such association found using the Discovery mechanism. return null;
* }
* @param extension The extension type to be associated with the
* <code>ScriptEngineFactory</code>. ScriptEngine engine = factory.getScriptEngine();
* @param factory The class to associate with the given extension. engine.setBindings(getBindings(), ScriptContext.GLOBAL_SCOPE);
* @throws NullPointerException if any of the parameters is null. return engine;
*/ }
public void registerEngineExtension(String extension, ScriptEngineFactory factory) {
if (extension == null || factory == null) throw new NullPointerException(); private ServiceLoader<ScriptEngineFactory> getServiceLoader(final ClassLoader loader) {
extensionAssociations.put(extension, factory); if (loader != null) {
return ServiceLoader.load(ScriptEngineFactory.class, loader);
} else {
return ServiceLoader.loadInstalled(ScriptEngineFactory.class);
}
}
private void init(final ClassLoader loader) {
_scriptEngineFactories = new TreeSet<>(Comparator.comparing(
ScriptEngineFactory::getEngineName,
Comparator.nullsLast(Comparator.naturalOrder()))
);
initEngines(loader);
}
private void initEngines(final ClassLoader loader) {
Iterator<ScriptEngineFactory> itr;
try {
ServiceLoader<ScriptEngineFactory> loaders = AccessController.doPrivileged(
new PrivilegedAction<ServiceLoader<ScriptEngineFactory>>() {
@Override
public ServiceLoader<ScriptEngineFactory> run() {
return getServiceLoader(loader);
}
});
itr = loaders.iterator();
} catch (ServiceConfigurationError err) {
// } catch (Exception err) {
System.err.println("Can't find ScriptEngineFactory providers: " + err.getMessage());
if (DEBUG) {
err.printStackTrace();
}
// do not throw any exception here. user may want to
// manage his/her own factories using this manager
// by explicit registration (by registerXXX) methods.
return;
}
try {
while (itr.hasNext()) {
try {
ScriptEngineFactory factory = itr.next();
_scriptEngineFactories.add(factory);
} catch (ServiceConfigurationError err) {
// } catch (Exception err) {
System.err.println("ScriptEngineManager providers.next(): " + err.getMessage());
if (DEBUG) {
err.printStackTrace();
}
// one factory failed, but check other factories...
}
}
} catch (ServiceConfigurationError err) {
// } catch (Exception err) {
System.err.println("ScriptEngineManager providers.hasNext(): " + err.getMessage());
if (DEBUG) {
err.printStackTrace();
}
// do not throw any exception here. user may want to
// manage his/her own factories using this manager
// by explicit registratation (by registerXXX) methods.
}
} }
/** Set of script engine factories discovered. */ /** Set of script engine factories discovered. */
private TreeSet<ScriptEngineFactory> engineSpis; private TreeSet<ScriptEngineFactory> _scriptEngineFactories;
/** Map of engine name to script engine factory. */ /** Map of engine name to script engine factory. */
private HashMap<String, ScriptEngineFactory> nameAssociations; private HashMap<String, ScriptEngineFactory> _nameAssociations = new HashMap<>();
/** Map of script file extension to script engine factory. */
private HashMap<String, ScriptEngineFactory> extensionAssociations;
/** Map of script MIME type to script engine factory. */
private HashMap<String, ScriptEngineFactory> mimeTypeAssociations;
/** Global bindings associated with script engines created by this manager. */ /** Global bindings associated with script engines created by this manager. */
private Bindings globalScope; private Bindings _globalScope = new SimpleBindings();
} }

View File

@ -4,7 +4,6 @@ import java.math.BigInteger;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
import java.security.MessageDigest; import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException; import java.security.NoSuchAlgorithmException;
import java.util.Arrays;
import java.util.List; import java.util.List;
import java.util.prefs.BackingStoreException; import java.util.prefs.BackingStoreException;
@ -12,6 +11,7 @@ import javax.script.ScriptEngine;
import javax.script.ScriptEngineFactory; import javax.script.ScriptEngineFactory;
import net.sf.openrocket.scripting.ScriptEngineManagerRedux; import net.sf.openrocket.scripting.ScriptEngineManagerRedux;
import net.sf.openrocket.scripting.GraalJSScriptEngineFactory;
import net.sf.openrocket.startup.Preferences; import net.sf.openrocket.startup.Preferences;
import net.sf.openrocket.util.ArrayList; import net.sf.openrocket.util.ArrayList;
import net.sf.openrocket.util.BugException; import net.sf.openrocket.util.BugException;
@ -31,19 +31,25 @@ public class ScriptingUtil {
/** The name to be chosen from a list of alternatives. If not found, will use the default name. */ /** The name to be chosen from a list of alternatives. If not found, will use the default name. */
private static final List<String> PREFERRED_LANGUAGE_NAMES = List.of("JavaScript"); private static final List<String> PREFERRED_LANGUAGE_NAMES = List.of("JavaScript");
private static ScriptEngineManagerRedux manager;
@Inject @Inject
Preferences prefs; Preferences prefs;
private static ScriptEngineManagerRedux manager;
public ScriptingUtil() { public ScriptingUtil() {
if (manager == null) { if (manager == null) {
// using the ScriptEngineManger from javax.script package pulls in the sun.misc.ServiceConfigurationError // using the ScriptEngineManger from javax.script package pulls in the sun.misc.ServiceConfigurationError
// which is removed in Java 9+ which causes a ClassNotFoundException to be thrown. // which is removed in Java 9+ which causes a ClassNotFoundException to be thrown.
manager = new ScriptEngineManagerRedux(); manager = new ScriptEngineManagerRedux();
manager.registerEngineName("Javascript", new GraalJSScriptEngineFactory());
} }
} }
public ScriptEngine getEngineByName(String shortName) {
return manager.getEngineByName(shortName);
}
/** /**
* Return the preferred internal language name based on a script language name. * Return the preferred internal language name based on a script language name.
@ -59,29 +65,16 @@ public class ScriptingUtil {
if (engine == null) { if (engine == null) {
return null; return null;
} }
return getLanguage(engine.getFactory());
}
public ScriptEngine getEngineByName(String shortName) { return getLanguageByFactory(engine.getFactory());
return manager.getEngineByName(shortName);
} }
public List<String> getLanguages() { public List<String> getLanguages() {
List<String> langs = new ArrayList<>(); List<String> languages = new ArrayList<>();
for (ScriptEngineFactory factory : manager.getEngineFactories()) { for (ScriptEngineFactory factory : manager.getEngineFactories()) {
langs.add(getLanguage(factory)); languages.add(getLanguageByFactory(factory));
} }
return langs; return languages;
}
private String getLanguage(ScriptEngineFactory factory) {
for (String name : factory.getNames()) {
if (PREFERRED_LANGUAGE_NAMES.contains(name)) {
return name;
}
}
return factory.getLanguageName();
} }
/** /**
@ -126,6 +119,16 @@ public class ScriptingUtil {
throw new BugException(e); throw new BugException(e);
} }
} }
private String getLanguageByFactory(ScriptEngineFactory factory) {
for (String name : factory.getNames()) {
if (PREFERRED_LANGUAGE_NAMES.contains(name)) {
return name;
}
}
return factory.getLanguageName();
}
static String normalize(String script) { static String normalize(String script) {
return script.replaceAll("\r", "").trim(); return script.replaceAll("\r", "").trim();