jruby · Jun 18, 2015 · Jun 18, 2015 · Jun 18, 2015 · Jun 18, 2015
Showing with 105 additions and 0 deletions.

+105 −0 truffle/src/main/java/org/jruby/truffle/Main.java
diff --git a/truffle/src/main/java/org/jruby/truffle/Main.java b/truffle/src/main/java/org/jruby/truffle/Main.java
@@ -0,0 +1,105 @@
+/*
+ * Copyright (c) 2015 Oracle and/or its affiliates. All rights reserved. This
+ * code is released under a tri EPL/GPL/LGPL license. You can use it,
+ * redistribute it and/or modify it under the terms of the:
+ *
+ * Eclipse Public License version 1.0
+ * GNU General Public License version 2
+ * GNU Lesser General Public License version 2.1
+ */
+package org.jruby.truffle;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Iterator;
+import java.util.Map;
+import java.util.concurrent.atomic.AtomicBoolean;
+
+import org.jruby.Ruby;
+import org.jruby.RubyInstanceConfig;
+import org.jruby.ast.RootNode;
+import org.jruby.internal.runtime.GlobalVariable;
+import org.jruby.internal.runtime.ValueAccessor;
+import org.jruby.runtime.IAccessor;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+
+public class Main {
+
+    public static void main(String[] args) {
+        final RubyInstanceConfig config = new RubyInstanceConfig();
+        config.setHardExit(true);
+        config.processArguments(args);
+
+        InputStream in = config.getScriptSource();
+        String filename = config.displayedFileName();
+
+        final Ruby runtime = Ruby.newInstance(config);
+        final AtomicBoolean didTeardown = new AtomicBoolean();
+
+        config.setCompileMode(RubyInstanceConfig.CompileMode.TRUFFLE);
+
+        if (config.isHardExit()) {
+            // We're the command-line JRuby, and should set a shutdown hook for
+            // teardown.
+            Runtime.getRuntime().addShutdownHook(new Thread() {
+                public void run() {
+                    if (didTeardown.compareAndSet(false, true)) {
+                        runtime.tearDown();
+                    }
+                }
+            });
+        }
+
+        if (in == null) {
+            return;
+        } else {
+
+            // Global variables
+            IAccessor d = new ValueAccessor(runtime.newString(filename));
+            runtime.getGlobalVariables().define("$PROGRAM_NAME", d,
+                    GlobalVariable.Scope.GLOBAL);
+            runtime.getGlobalVariables().define("$0", d,
+                    GlobalVariable.Scope.GLOBAL);
+
+            for (Iterator i = config.getOptionGlobals().entrySet().iterator(); i.hasNext();) {
+                Map.Entry entry = (Map.Entry) i.next();
+                Object value = entry.getValue();
+                IRubyObject varvalue;
+                if (value != null) {
+                    varvalue = runtime.newString(value.toString());
+                } else {
+                    varvalue = runtime.getTrue();
+                }
+                runtime.getGlobalVariables().set(
+                        "$" + entry.getKey().toString(), varvalue);
+            }
+
+            RootNode scriptNode = (RootNode) runtime
+                    .parseFromMain(filename, in);
+
+            // If no DATA, we're done with the stream, shut it down
+			if (runtime.fetchGlobalConstant("DATA") == null) {
+				try {
+					in.close();
+				} catch (IOException ioe) {
+				}
+			}
+			ThreadContext context = runtime.getCurrentContext();
+
+			String oldFile = context.getFile();
+			int oldLine = context.getLine();
+
+			try {
+				context.setFileAndLine(scriptNode.getPosition());
+				runtime.getTruffleContext().execute(scriptNode);
+			} finally {
+				context.setFileAndLine(oldFile, oldLine);
+				runtime.shutdownTruffleContextIfRunning();
+			}
+
+		}
+
+	}
+
+}