From a72fcfa3adf5be93bcf62e8ed26c08d10939b364 Mon Sep 17 00:00:00 2001 From: lresende Date: Thu, 11 Sep 2008 04:10:13 +0000 Subject: Branch before trunk cleanup git-svn-id: http://svn.us.apache.org/repos/asf/tuscany@694106 13f79535-47bb-0310-9956-ffa450edef68 --- .../script/ScriptImplementation.java | 99 +++ .../script/ScriptImplementationActivator.java | 49 ++ .../sca/implementation/script/ScriptInvoker.java | 84 +++ .../script/ScriptInvokerFactory.java | 173 +++++ .../script/engines/TuscanyJRubyScriptEngine.java | 697 +++++++++++++++++++++ 5 files changed, 1102 insertions(+) create mode 100644 branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptImplementation.java create mode 100644 branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptImplementationActivator.java create mode 100644 branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptInvoker.java create mode 100644 branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptInvokerFactory.java create mode 100644 branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/engines/TuscanyJRubyScriptEngine.java (limited to 'branches/trunk-20080910/modules/implementation-script/src/main/java') diff --git a/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptImplementation.java b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptImplementation.java new file mode 100644 index 0000000000..87820c18ed --- /dev/null +++ b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptImplementation.java @@ -0,0 +1,99 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ +package org.apache.tuscany.sca.implementation.script; + +import java.net.MalformedURLException; +import java.net.URL; + +import org.apache.tuscany.sca.contribution.Artifact; +import org.apache.tuscany.sca.contribution.ContributionFactory; +import org.apache.tuscany.sca.contribution.DefaultContributionFactory; +import org.apache.tuscany.sca.contribution.resolver.ModelResolver; +import org.apache.tuscany.sca.extension.helper.utils.ResourceHelper; + +/** + * Represents a Script implementation. + * + * @version $Rev$ $Date$ + */ +public class ScriptImplementation { + + protected String scriptName; + protected URL scriptURL; + protected String scriptSrc; + protected String scriptLanguage; + + public String getScript() { + return scriptName; + } + + public void setScript(String scriptName) { + this.scriptName = scriptName; + } + + public void setLanguage(String language) { + this.scriptLanguage = language; + } + + public void setElementText(String elementText) { + scriptSrc = elementText; + } + + public String getScriptLanguage() { + if (scriptLanguage == null || scriptLanguage.length() < 1) { + int i = scriptName.lastIndexOf('.'); + if (i > 0) { + scriptLanguage = scriptName.substring(i + 1); + } + } + return scriptLanguage; + } + + public String getScriptSrc() { + if (scriptSrc == null) { + if (scriptName == null) { + throw new IllegalArgumentException("script name is null and no inline source used"); + } + if (scriptURL == null) { + throw new RuntimeException("No script: " + scriptName); + } + + scriptSrc = ResourceHelper.readResource(scriptURL); + } + return scriptSrc; + } + + public void resolve(ModelResolver resolver) { + + if (scriptName != null) { + //FIXME The contribution factory should be injected + ContributionFactory contributionFactory = new DefaultContributionFactory(); + Artifact artifact = contributionFactory.createArtifact(); + artifact.setURI(scriptName); + artifact = resolver.resolveModel(Artifact.class, artifact); + if (artifact.getLocation() != null) { + try { + scriptURL = new URL(artifact.getLocation()); + } catch (MalformedURLException e) { + throw new RuntimeException(e); + } + } + } + } +} diff --git a/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptImplementationActivator.java b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptImplementationActivator.java new file mode 100644 index 0000000000..a7c4b9074f --- /dev/null +++ b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptImplementationActivator.java @@ -0,0 +1,49 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +package org.apache.tuscany.sca.implementation.script; + +import org.apache.tuscany.sca.assembly.ComponentType; +import org.apache.tuscany.sca.extension.helper.ImplementationActivator; +import org.apache.tuscany.sca.extension.helper.InvokerFactory; +import org.apache.tuscany.sca.extension.helper.utils.PropertyValueObjectFactory; +import org.apache.tuscany.sca.runtime.RuntimeComponent; + +/** + * + * @version $Rev$ $Date$ + */ +public class ScriptImplementationActivator implements ImplementationActivator { + + // TODO: seems wrong to need PropertyValueObjectFactory, could it be on Property somehow? + protected PropertyValueObjectFactory propertyFactory; + + public ScriptImplementationActivator(PropertyValueObjectFactory propertyFactory) { + this.propertyFactory = propertyFactory; + } + + public Class getImplementationClass() { + return ScriptImplementation.class; + } + + public InvokerFactory createInvokerFactory(RuntimeComponent rc, ComponentType ct, ScriptImplementation implementation) { + return new ScriptInvokerFactory(rc, ct, implementation, propertyFactory); + } + +} diff --git a/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptInvoker.java b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptInvoker.java new file mode 100644 index 0000000000..2ccb414be8 --- /dev/null +++ b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptInvoker.java @@ -0,0 +1,84 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +package org.apache.tuscany.sca.implementation.script; + +import javax.script.Invocable; +import javax.script.ScriptEngine; +import javax.script.ScriptException; + +import org.apache.axiom.om.OMElement; +import org.apache.bsf.xml.XMLHelper; +import org.apache.tuscany.sca.interfacedef.Operation; +import org.apache.tuscany.sca.invocation.Invoker; +import org.apache.tuscany.sca.invocation.Message; + +/** + * Perform the actual script invocation + * + * @version $Rev$ $Date$ + */ +public class ScriptInvoker implements Invoker { + + protected ScriptEngine scriptEngine; + protected XMLHelper xmlHelper; + protected Operation operation; + + public ScriptInvoker(ScriptEngine scriptEngine, XMLHelper xmlHelper, Operation operation) { + this.scriptEngine = scriptEngine; + this.xmlHelper = xmlHelper; + this.operation = operation; + } + + protected Object doInvoke(Object[] objects, Operation op) throws ScriptException { + if (xmlHelper != null) { + objects[0] = xmlHelper.toScriptXML((OMElement)objects[0]); + } + + Operation oper = operation; // static setting + if (oper.getName() == null) { // if no static setting + oper = op; // use dynamic setting + } + Object response; + try { + response = ((Invocable)scriptEngine).invokeFunction(oper.getName(), objects); + } catch (ScriptException e) { + throw e; + } catch (Exception e) { + throw new ScriptException(e); + } + + if (xmlHelper != null) { + response = xmlHelper.toOMElement(response); + } + + return response; + } + + public Message invoke(Message msg) { + try { + Object resp = doInvoke((Object[])msg.getBody(), msg.getOperation()); + msg.setBody(resp); + } catch (ScriptException e) { + msg.setFaultBody(e.getCause()); + } + return msg; + } + +} diff --git a/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptInvokerFactory.java b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptInvokerFactory.java new file mode 100644 index 0000000000..9359590424 --- /dev/null +++ b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/ScriptInvokerFactory.java @@ -0,0 +1,173 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, + * software distributed under the License is distributed on an + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + * KIND, either express or implied. See the License for the + * specific language governing permissions and limitations + * under the License. + */ + +package org.apache.tuscany.sca.implementation.script; + +import java.io.StringReader; +import java.security.AccessController; +import java.security.PrivilegedAction; + +import javax.script.Invocable; +import javax.script.ScriptEngine; +import javax.script.ScriptEngineManager; +import javax.script.ScriptException; + +import org.apache.axiom.om.OMElement; +import org.apache.bsf.xml.XMLHelper; +import org.apache.tuscany.sca.assembly.ComponentReference; +import org.apache.tuscany.sca.assembly.ComponentType; +import org.apache.tuscany.sca.assembly.Property; +import org.apache.tuscany.sca.assembly.Reference; +import org.apache.tuscany.sca.assembly.Service; +import org.apache.tuscany.sca.core.factory.ObjectCreationException; +import org.apache.tuscany.sca.core.factory.ObjectFactory; +import org.apache.tuscany.sca.extension.helper.InvokerFactory; +import org.apache.tuscany.sca.extension.helper.utils.PropertyValueObjectFactory; +import org.apache.tuscany.sca.implementation.script.engines.TuscanyJRubyScriptEngine; +import org.apache.tuscany.sca.interfacedef.InterfaceContract; +import org.apache.tuscany.sca.interfacedef.Operation; +import org.apache.tuscany.sca.interfacedef.java.JavaInterface; +import org.apache.tuscany.sca.interfacedef.wsdl.WSDLInterfaceContract; +import org.apache.tuscany.sca.invocation.Invoker; +import org.apache.tuscany.sca.runtime.RuntimeComponent; + +/** + * + * @version $Rev$ $Date$ + */ +public class ScriptInvokerFactory implements InvokerFactory { + + protected ScriptEngine scriptEngine; + protected XMLHelper xmlHelper; + + protected RuntimeComponent rc; + protected ComponentType ct; + protected ScriptImplementation implementation; + protected PropertyValueObjectFactory propertyFactory; + + + /** + * @param rc + * @param ct + * @param implementation + * @param propertyFactory + */ + public ScriptInvokerFactory(RuntimeComponent rc, + ComponentType ct, + ScriptImplementation implementation, + PropertyValueObjectFactory propertyFactory) { + super(); + this.rc = rc; + this.ct = ct; + this.implementation = implementation; + this.propertyFactory = propertyFactory; + } + + public Invoker createInvoker(Operation operation) { + init(rc, ct, implementation, propertyFactory); + return new ScriptInvoker(scriptEngine, xmlHelper, operation); + } + + protected synchronized void init(RuntimeComponent rc, ComponentType ct, ScriptImplementation implementation, PropertyValueObjectFactory propertyFactory) { + if(scriptEngine!=null) { + return; + } + try { + scriptEngine = getScriptEngineByExtension(implementation.getScriptLanguage()); + if (scriptEngine == null) { + throw new ObjectCreationException("no script engine found for language: " + implementation.getScriptLanguage()); + } + if (!(scriptEngine instanceof Invocable)) { + throw new ObjectCreationException("script engine does not support Invocable: " + scriptEngine); + } + + for (Reference reference : ct.getReferences()) { + scriptEngine.put(reference.getName(), createReferenceProxy(reference.getName(), rc)); + } + + for (Property property : ct.getProperties()) { + ObjectFactory propertyValueFactory = propertyFactory.createValueFactory(property); + if ( propertyValueFactory != null) { + scriptEngine.put(property.getName(), propertyValueFactory.getInstance()); + } + } + + scriptEngine.eval(new StringReader(implementation.getScriptSrc())); + + } catch (ScriptException e) { + throw new ObjectCreationException(e); + } + + // set the databinding and XMLHelper for WSDL interfaces + for (Service service : rc.getServices()) { + InterfaceContract ic = service.getInterfaceContract(); + if (ic instanceof WSDLInterfaceContract) { + // Set to use the Axiom data binding + ic.getInterface().resetDataBinding(OMElement.class.getName()); + xmlHelper = XMLHelper.getArgHelper(scriptEngine); + } + } + } + + /** + * TODO: RuntimeComponent should provide a method like this + */ + @SuppressWarnings("unchecked") + protected Object createReferenceProxy(String name, RuntimeComponent component) { + for (ComponentReference reference : component.getReferences()) { + if (reference.getName().equals(name)) { + Class iface = ((JavaInterface)reference.getInterfaceContract().getInterface()).getJavaClass(); + return component.getComponentContext().getService(iface, name); + } + } + throw new IllegalArgumentException("reference " + name + " not found on component: " + component); + } + + /** + * Hack for now to work around a problem with the JRuby script engine + */ + protected ScriptEngine getScriptEngineByExtension(String scriptExtn) { + if ("rb".equals(scriptExtn)) { + return new TuscanyJRubyScriptEngine(); + } else { + if ("py".equals(scriptExtn)) { + pythonCachedir(); + } + // Allow privileged access to run access classes. Requires RuntimePermission + // for accessClassInPackage.sun.misc. + ScriptEngineManager scriptEngineManager = + AccessController.doPrivileged(new PrivilegedAction() { + public ScriptEngineManager run() { + return new ScriptEngineManager(); + } + }); + return scriptEngineManager.getEngineByExtension(scriptExtn); + } + } + + /** + * If the Python home isn't set then let Tuscany suppress messages other than errors + * See TUSCANY-1950 + */ + protected void pythonCachedir() { + if (System.getProperty("python.home") == null) { + System.setProperty("python.verbose", "error"); + } + } +} diff --git a/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/engines/TuscanyJRubyScriptEngine.java b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/engines/TuscanyJRubyScriptEngine.java new file mode 100644 index 0000000000..324246c32b --- /dev/null +++ b/branches/trunk-20080910/modules/implementation-script/src/main/java/org/apache/tuscany/sca/implementation/script/engines/TuscanyJRubyScriptEngine.java @@ -0,0 +1,697 @@ +/* +* Copyright (c) 2006, Sun Microsystems, Inc. +* All rights reserved. +* +* Redistribution and use in source and binary forms, with or without +* modification, are permitted provided that the following conditions are met: +* +* - Redistributions of source code must retain the above copyright notice, this +* list of conditions and the following disclaimer. +* +* - Redistributions in binary form must reproduce the above copyright notice, +* this list of conditions and the following disclaimer in the documentation +* and/or other materials provided with the distribution. +* +* - Neither the name of the Sun Microsystems, Inc. nor the names of +* contributors may be used to endorse or promote products derived from this +* software without specific prior written permission. +* +* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND +* CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED +* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED +* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A +* PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE +* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY +* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, +* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF +* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER +* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN +* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE +* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS +* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH +* DAMAGE. +*/ +package org.apache.tuscany.sca.implementation.script.engines; + +import java.io.File; +import java.io.FileInputStream; +import java.io.FileNotFoundException; +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; +import java.io.PrintStream; +import java.io.Reader; +import java.io.StringWriter; +import java.io.UnsupportedEncodingException; +import java.io.Writer; +import java.lang.reflect.InvocationHandler; +import java.lang.reflect.Method; +import java.lang.reflect.Proxy; +import java.nio.ByteBuffer; +import java.nio.CharBuffer; +import java.nio.charset.Charset; +import java.nio.charset.CharsetDecoder; +import java.nio.charset.CoderResult; +import java.nio.charset.CodingErrorAction; +import java.security.AccessController; +import java.security.PrivilegedAction; +import java.util.Arrays; +import java.util.Collections; +import java.util.HashSet; +import java.util.Iterator; +import java.util.List; +import java.util.Set; + +import javax.script.AbstractScriptEngine; +import javax.script.Bindings; +import javax.script.Compilable; +import javax.script.CompiledScript; +import javax.script.Invocable; +import javax.script.ScriptContext; +import javax.script.ScriptEngine; +import javax.script.ScriptEngineFactory; +import javax.script.ScriptException; +import javax.script.SimpleBindings; + +import org.jruby.Ruby; +import org.jruby.RubyException; +import org.jruby.RubyIO; +import org.jruby.RubyObject; +import org.jruby.ast.Node; +import org.jruby.exceptions.RaiseException; +import org.jruby.internal.runtime.GlobalVariable; +import org.jruby.internal.runtime.GlobalVariables; +import org.jruby.internal.runtime.ReadonlyAccessor; +import org.jruby.internal.runtime.ValueAccessor; +import org.jruby.javasupport.Java; +import org.jruby.javasupport.JavaEmbedUtils; +import org.jruby.javasupport.JavaObject; +import org.jruby.javasupport.JavaUtil; +import org.jruby.runtime.Block; +import org.jruby.runtime.IAccessor; +import org.jruby.runtime.builtin.IRubyObject; +import org.jruby.util.KCode; + +import com.sun.script.jruby.JRubyScriptEngineFactory; + +/** + * This class is a copy of the class com.sun.script.ruby.JRubyScriptEngine with some minor modifications + * to work around problems with Tuscany setting SCA properties and references as global variable in JRuby + * Should only need it temporarily till a new BSF release fixes it. + * + * @version $Rev$ $Date$ + */ +@SuppressWarnings("unchecked") +public class TuscanyJRubyScriptEngine extends AbstractScriptEngine + implements Compilable, Invocable { + + // my factory, may be null + private ScriptEngineFactory factory; + private Ruby runtime; + + public TuscanyJRubyScriptEngine() { + // Allow privileged access to ready properties. Requires PropertyPermission in security + // policy. + String rubyPath = AccessController.doPrivileged(new PrivilegedAction() { + public String run() { + return System.getProperty("com.sun.script.jruby.loadpath"); + } + }); + init(rubyPath); + } + + public TuscanyJRubyScriptEngine(String loadPath) { + init(loadPath); + } + + // my implementation for CompiledScript + private class JRubyCompiledScript extends CompiledScript { + // my compiled code + private Node node; + + JRubyCompiledScript (Node node) { + this.node = node; + } + + public ScriptEngine getEngine() { + return TuscanyJRubyScriptEngine.this; + } + + public Object eval(ScriptContext ctx) throws ScriptException { + return evalNode(node, ctx); + } + } + + // Compilable methods + public CompiledScript compile(String script) + throws ScriptException { + Node node = compileScript(script, context); + return new JRubyCompiledScript(node); + } + + public CompiledScript compile (Reader reader) + throws ScriptException { + Node node = compileScript(reader, context); + return new JRubyCompiledScript(node); + } + + // Invocable methods + public Object invokeFunction(String name, Object... args) + throws ScriptException, NoSuchMethodException { + return invokeImpl(null, name, args, Object.class); + } + + public Object invokeMethod(Object obj, String name, Object... args) + throws ScriptException, NoSuchMethodException { + if (obj == null) { + throw new IllegalArgumentException("script object is null"); + } + return invokeImpl(obj, name, args, Object.class); + } + + public Object getInterface(Object obj, Class clazz) { + if (obj == null) { + throw new IllegalArgumentException("script object is null"); + } + return makeInterface(obj, clazz); + } + + public Object getInterface(Class clazz) { + return makeInterface(null, clazz); + } + + private T makeInterface(Object obj, Class clazz) { + if (clazz == null || !clazz.isInterface()) { + throw new IllegalArgumentException("interface Class expected"); + } + final Object thiz = obj; + return (T) Proxy.newProxyInstance( + clazz.getClassLoader(), + new Class[] { clazz }, + new InvocationHandler() { + public Object invoke(Object proxy, Method m, Object[] args) + throws Throwable { + return invokeImpl(thiz, m.getName(), + args, m.getReturnType()); + } + }); + } + + // ScriptEngine methods + public synchronized Object eval(String str, ScriptContext ctx) + throws ScriptException { + Node node = compileScript(str, ctx); + return evalNode(node, ctx); + } + + public synchronized Object eval(Reader reader, ScriptContext ctx) + throws ScriptException { + Node node = compileScript(reader, ctx); + return evalNode(node, ctx); + } + + public ScriptEngineFactory getFactory() { + synchronized (this) { + if (factory == null) { + factory = new JRubyScriptEngineFactory(); + } + } + return factory; + } + + public Bindings createBindings() { + return new SimpleBindings(); + } + + // package-private methods + void setFactory(ScriptEngineFactory factory) { + this.factory = factory; + } + + // internals only below this point + + private Object rubyToJava(IRubyObject value) { + return rubyToJava(value, Object.class); + } + + private Object rubyToJava(IRubyObject value, Class type) { + return JavaUtil.convertArgument( + runtime, + Java.ruby_to_java(value, value, Block.NULL_BLOCK), + type); + } + + private IRubyObject javaToRuby(Object value) { + if (value instanceof IRubyObject) { + return (IRubyObject) value; + } + IRubyObject result = JavaUtil.convertJavaToRuby(runtime, value); + if (result instanceof JavaObject) { + return runtime.getModule("JavaUtilities").callMethod(runtime.getCurrentContext(), "wrap", result); + } + return result; + } + + private synchronized Node compileScript(String script, ScriptContext ctx) + throws ScriptException { + GlobalVariables oldGlobals = runtime.getGlobalVariables(); + try { + setErrorWriter(ctx.getErrorWriter()); + setGlobalVariables(ctx); + String filename = (String) ctx.getAttribute(ScriptEngine.FILENAME); + if (filename == null) { + filename = ""; + } + return runtime.parseEval(script, filename, null, 0); + } catch (RaiseException e) { + RubyException re = e.getException(); + runtime.printError(re); + throw new ScriptException(e); + } catch (Exception e) { + throw new ScriptException(e); + } finally { + if (oldGlobals != null) { + setGlobalVariables(oldGlobals); + } + } + } + + private synchronized Node compileScript(Reader reader, ScriptContext ctx) + throws ScriptException { + GlobalVariables oldGlobals = runtime.getGlobalVariables(); + try { + setErrorWriter(ctx.getErrorWriter()); + setGlobalVariables(ctx); + String filename = (String) ctx.getAttribute(ScriptEngine.FILENAME); + if (filename == null) { + filename = ""; + String script = getRubyScript(reader); + return runtime.parseEval(script, filename, null, 0); + } + InputStream inputStream = getRubyReader(filename); + return runtime.parseFile(inputStream, filename, null); + } catch (RaiseException e) { + RubyException re = e.getException(); + runtime.printError(re); + throw new ScriptException(e); + } catch (Exception exp) { + throw new ScriptException(exp); + } finally { + if (oldGlobals != null) { + setGlobalVariables(oldGlobals); + } + } + } + + private String getRubyScript(Reader reader) throws IOException { + StringBuffer sb = new StringBuffer(); + char[] cbuf; + while (true) { + cbuf = new char[8*1024]; + int chars = reader.read(cbuf, 0, cbuf.length); + if (chars < 0) { + break; + } + sb.append(cbuf, 0, chars); + } + cbuf = null; + return (new String(sb)).trim(); + } + + private InputStream getRubyReader(String filename) throws FileNotFoundException { + File file = new File(filename); + return new FileInputStream(file); + } + + private void setGlobalVariables(final ScriptContext ctx) { + ctx.setAttribute("context", ctx, ScriptContext.ENGINE_SCOPE); + setGlobalVariables(new GlobalVariables(runtime) { + GlobalVariables parent = runtime.getGlobalVariables(); + + @Override + public void define(String name, IAccessor accessor) { + assert name != null; + assert accessor != null; + assert name.startsWith("$"); + synchronized (ctx) { + Bindings engineScope = ctx.getBindings(ScriptContext.ENGINE_SCOPE); + engineScope.put(name, new GlobalVariable(accessor)); + } + } + + @Override + public void defineReadonly(String name, IAccessor accessor) { + assert name != null; + assert accessor != null; + assert name.startsWith("$"); + synchronized (ctx) { + Bindings engineScope = ctx.getBindings(ScriptContext.ENGINE_SCOPE); + engineScope.put(name, new GlobalVariable(new + ReadonlyAccessor(name, accessor))); + } + } + + @Override + public boolean isDefined(String name) { + assert name != null; + assert name.startsWith("$"); + synchronized (ctx) { + String modifiedName = name.substring(1); + boolean defined = ctx.getAttributesScope(modifiedName) != -1; + return defined ? true : parent.isDefined(name); + } + } + + @Override + public void alias(String name, String oldName) { + assert name != null; + assert oldName != null; + assert name.startsWith("$"); + assert oldName.startsWith("$"); + + if (runtime.getSafeLevel() >= 4) { + throw runtime.newSecurityError("Insecure: can't alias global variable"); + } + + synchronized (ctx) { + int scope = ctx.getAttributesScope(name); + if (scope == -1) { + scope = ScriptContext.ENGINE_SCOPE; + } + + IRubyObject value = get(oldName); + ctx.setAttribute(name, rubyToJava(value), scope); + } + } + + @Override + public IRubyObject get(String name) { + assert name != null; + assert name.startsWith("$"); + + synchronized (ctx) { + // skip '$' and try + String modifiedName = name.substring(1); + int scope = ctx.getAttributesScope(modifiedName); + if (scope == -1) { + return parent.get(name); + } + + Object obj = ctx.getAttribute(modifiedName, scope); + if (obj instanceof IAccessor) { + return ((IAccessor)obj).getValue(); + } else { + return javaToRuby(obj); + } + } + } + + @Override + public IRubyObject set(String name, IRubyObject value) { + assert name != null; + assert name.startsWith("$"); + + if (runtime.getSafeLevel() >= 4) { + throw runtime.newSecurityError("Insecure: can't change global variable value"); + } + + synchronized (ctx) { + // skip '$' and try + String modifiedName = name.substring(1); + int scope = ctx.getAttributesScope(modifiedName); + if (scope == -1) { + scope = ScriptContext.ENGINE_SCOPE; + } + IRubyObject oldValue = get(name); + Object obj = ctx.getAttribute(modifiedName, scope); + if (obj instanceof IAccessor) { + ((IAccessor)obj).setValue(value); + } else { + ctx.setAttribute(modifiedName, rubyToJava(value), scope); + if ("KCODE".equals(modifiedName)) { + setKCode((String)rubyToJava(value)); + } else if ("stdout".equals(modifiedName)) { + equalOutputs((RubyObject)value); + } + } + return oldValue; + } + } + + @Override + public Set getNames() { + HashSet set = new HashSet(); + synchronized (ctx) { + for (Object scope : ctx.getScopes()) { + Bindings b = ctx.getBindings((Integer)scope); + if (b != null) { + for (Object key: b.keySet()) { + set.add(key); + } + } + } + } + for (Iterator names = parent.getNames().iterator(); names.hasNext();) { + set.add(names.next()); + } + return Collections.unmodifiableSet(set); + } + + @Override + public IRubyObject getDefaultSeparator() { + return parent.getDefaultSeparator(); + } + }); + } + + private void setGlobalVariables(GlobalVariables globals) { + runtime.setGlobalVariables(globals); + } + + private synchronized Object evalNode(Node node, ScriptContext ctx) + throws ScriptException { + GlobalVariables oldGlobals = runtime.getGlobalVariables(); + try { + setWriterOutputStream(ctx.getWriter()); + setErrorWriter(ctx.getErrorWriter()); + setGlobalVariables(ctx); + return rubyToJava(runtime.runNormally(node, false)); + } catch (Exception exp) { + throw new ScriptException(exp); + } finally { + try { + JavaEmbedUtils.terminate(runtime); + } catch (RaiseException e) { + RubyException re = e.getException(); + runtime.printError(re); + if (!runtime.fastGetClass("SystemExit").isInstance(re)) { + throw new ScriptException(e); + } + } finally { + if (oldGlobals != null) { + setGlobalVariables(oldGlobals); + } + } + } + } + + private void init(final String loadPath) { + AccessController.doPrivileged(new PrivilegedAction() { + public Object run() { + runtime = Ruby.newInstance(); + IAccessor d = new ValueAccessor(runtime.newString("