forked from jruby/warbler
-
Notifications
You must be signed in to change notification settings - Fork 1
/
JarMain.java
148 lines (130 loc) · 5.09 KB
/
JarMain.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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
/**
* Copyright (c) 2010-2011 Engine Yard, Inc.
* Copyright (c) 2007-2009 Sun Microsystems, Inc.
* This source code is available under the MIT license.
* See the file LICENSE.txt for details.
*/
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Enumeration;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
public class JarMain implements Runnable {
public static final String MAIN = "/" + JarMain.class.getName().replace('.', '/') + ".class";
private String[] args;
private String path, jarfile;
private boolean debug;
private File extractRoot;
public JarMain(String[] args) throws Exception {
this.args = args;
URL mainClass = getClass().getResource(MAIN);
this.path = mainClass.toURI().getSchemeSpecificPart();
this.jarfile = mainClass.getFile().replace("!" + MAIN, "").replace("file:", "");
this.debug = isDebug();
this.extractRoot = File.createTempFile("jruby", "extract");
this.extractRoot.delete();
this.extractRoot.mkdirs();
Runtime.getRuntime().addShutdownHook(new Thread(this));
}
private URL[] extractJRuby() throws Exception {
JarFile jf = new JarFile(this.jarfile);
List<String> jarNames = new ArrayList<String>();
for (Enumeration<JarEntry> eje = jf.entries(); eje.hasMoreElements(); ) {
String name = eje.nextElement().getName();
if (name.startsWith("META-INF/lib") && name.endsWith(".jar")) {
jarNames.add("/" + name);
}
}
List<URL> urls = new ArrayList<URL>();
for (String name : jarNames) {
urls.add(extractJar(name));
}
return (URL[]) urls.toArray(new URL[urls.size()]);
}
private URL extractJar(String jarpath) throws Exception {
InputStream jarStream = new URL("jar:" + path.replace(MAIN, jarpath)).openStream();
String jarname = jarpath.substring(jarpath.lastIndexOf("/") + 1, jarpath.lastIndexOf("."));
File jarFile = new File(extractRoot, jarname + ".jar");
jarFile.deleteOnExit();
FileOutputStream outStream = new FileOutputStream(jarFile);
try {
byte[] buf = new byte[65536];
int bytesRead = 0;
while ((bytesRead = jarStream.read(buf)) != -1) {
outStream.write(buf, 0, bytesRead);
}
} finally {
jarStream.close();
outStream.close();
}
debug(jarname + ".jar extracted to " + jarFile.getPath());
return jarFile.toURI().toURL();
}
private int launchJRuby(URL[] jars) throws Exception {
System.setProperty("org.jruby.embed.class.path", "");
URLClassLoader loader = new URLClassLoader(jars);
Class scriptingContainerClass = Class.forName("org.jruby.embed.ScriptingContainer", true, loader);
Object scriptingContainer = scriptingContainerClass.newInstance();
Method argv = scriptingContainerClass.getDeclaredMethod("setArgv", new Class[] {String[].class});
argv.invoke(scriptingContainer, new Object[] {args});
Method setClassLoader = scriptingContainerClass.getDeclaredMethod("setClassLoader", new Class[] {ClassLoader.class});
setClassLoader.invoke(scriptingContainer, new Object[] {loader});
debug("invoking " + jarfile + " with: " + Arrays.deepToString(args));
Method runScriptlet = scriptingContainerClass.getDeclaredMethod("runScriptlet", new Class[] {String.class});
return ((Number) runScriptlet.invoke(scriptingContainer, new Object[] {
"begin\n" +
"require 'META-INF/init.rb'\n" +
"require 'META-INF/main.rb'\n" +
"0\n" +
"rescue SystemExit => e\n" +
"e.status\n" +
"end"
})).intValue();
}
private int start() throws Exception {
URL[] u = extractJRuby();
return launchJRuby(u);
}
private void debug(String msg) {
if (debug) {
System.out.println(msg);
}
}
private void delete(File f) {
if (f.isDirectory()) {
File[] children = f.listFiles();
for (int i = 0; i < children.length; i++) {
delete(children[i]);
}
}
f.delete();
}
public void run() {
delete(extractRoot);
}
public static void main(String[] args) {
try {
int exit = new JarMain(args).start();
System.exit(exit);
} catch (Exception e) {
Throwable t = e;
while (t.getCause() != null && t.getCause() != t) {
t = t.getCause();
}
if (isDebug()) {
t.printStackTrace();
}
System.exit(1);
}
}
private static boolean isDebug() {
return System.getProperty("warbler.debug") != null;
}
}