/
Main.java
82 lines (66 loc) · 2.74 KB
/
Main.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
/*
* 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 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;
import java.io.IOException;
import java.io.InputStream;
import java.util.Map;
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);
config.setCompileMode(RubyInstanceConfig.CompileMode.TRUFFLE);
if (in == null) {
return;
} else {
// Global variables
IAccessor programName = new ValueAccessor(runtime.newString(filename));
runtime.getGlobalVariables().define("$PROGRAM_NAME", programName, GlobalVariable.Scope.GLOBAL);
runtime.getGlobalVariables().define("$0", programName, GlobalVariable.Scope.GLOBAL);
for (Map.Entry<String, String> entry : config.getOptionGlobals().entrySet()) {
final IRubyObject varvalue;
if (entry.getValue() != null) {
varvalue = runtime.newString(entry.getValue());
} else {
varvalue = runtime.getTrue();
}
runtime.getGlobalVariables().set("$" + entry.getKey(), 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);
}
}
}
}