-
-
Notifications
You must be signed in to change notification settings - Fork 56
/
SimpleScript.java
97 lines (87 loc) · 3.48 KB
/
SimpleScript.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package net.citizensnpcs.api.scripting;
import javax.script.Bindings;
import javax.script.CompiledScript;
import javax.script.Invocable;
import javax.script.ScriptContext;
import javax.script.ScriptEngine;
import javax.script.ScriptException;
import com.google.common.base.Throwables;
public class SimpleScript implements Script {
private final Bindings bindings;
private final ScriptEngine engine;
private final Invocable invocable;
@SuppressWarnings("unused")
private final Object root;
public SimpleScript(CompiledScript src, ContextProvider[] providers) throws ScriptException {
this.engine = src.getEngine();
this.invocable = (Invocable) engine;
this.bindings = engine.createBindings();
for (ContextProvider provider : providers)
provider.provide(this);
this.root = src.eval(bindings);
}
@Override
public <T> T convertToInterface(Object obj, Class<T> expected) {
if (obj == null || expected == null)
throw new IllegalArgumentException("arguments should not be null");
if (expected.isAssignableFrom(obj.getClass()))
return expected.cast(obj);
synchronized (engine) {
Bindings old = engine.getBindings(ScriptContext.ENGINE_SCOPE);
engine.setBindings(bindings, ScriptContext.ENGINE_SCOPE);
T t = invocable.getInterface(expected);
engine.setBindings(old, ScriptContext.ENGINE_SCOPE);
return t;
}
}
@Override
public Object getAttribute(String name) {
if (name == null)
throw new IllegalArgumentException("name should not be null");
return bindings.get(name);
}
@Override
public Object invoke(Object instance, String name, Object... args) {
if (instance == null || name == null)
throw new IllegalArgumentException("instance and method name should not be null");
try {
synchronized (engine) {
Bindings old = engine.getBindings(ScriptContext.ENGINE_SCOPE);
engine.setBindings(bindings, ScriptContext.ENGINE_SCOPE);
Object ret = invocable.invokeMethod(instance, name, args);
engine.setBindings(old, ScriptContext.ENGINE_SCOPE);
return ret;
}
} catch (ScriptException e) {
Throwables.getRootCause(e).printStackTrace();
} catch (NoSuchMethodException e) {
throw new RuntimeException(e);
}
return null;
}
@Override
public Object invoke(String name, Object... args) {
if (name == null)
throw new IllegalArgumentException("name should not be null");
try {
synchronized (engine) {
Bindings old = engine.getBindings(ScriptContext.ENGINE_SCOPE);
engine.setBindings(bindings, ScriptContext.ENGINE_SCOPE);
Object ret = invocable.invokeFunction(name, args);
engine.setBindings(old, ScriptContext.ENGINE_SCOPE);
return ret;
}
} catch (ScriptException e) {
Throwables.getRootCause(e).printStackTrace();
} catch (NoSuchMethodException e) {
throw new RuntimeException(e);
}
return null;
}
@Override
public void setAttribute(String name, Object value) {
if (name == null || value == null)
throw new IllegalArgumentException("arguments should not be null");
bindings.put(name, value);
}
}