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
|
/*
* Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
import javax.tools.*;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URI;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
class Compiler {
final private Map<String,String> input;
private List<String> options;
Compiler(Map<String,String> input) {
this.input = input;
}
Compiler setRelease(int release) {
// Setting the -release option does not work for some reason
// so do it the old fashioned way
// options = Arrays.asList("-release", String.valueOf(release));
String target = String.valueOf(release);
options = Arrays.asList("-source", target, "-target", target, "-classpath", "");
return this;
}
Map<String,byte[]> compile() {
List<SourceFileObject> cunits = createCompilationUnits();
Map<String,ClassFileObject> cfos = createClassFileObjects();
JavaCompiler jc = ToolProvider.getSystemJavaCompiler();
JavaFileManager jfm = new CustomFileManager(jc.getStandardFileManager(null, null, null), cfos);
if(!jc.getTask(null, jfm, null, options, null, cunits).call()) {
throw new RuntimeException("Compilation failed");
}
return createOutput(cfos);
}
private List<SourceFileObject> createCompilationUnits() {
return input.entrySet().stream()
.map(e -> new SourceFileObject(e.getKey(), e.getValue())).collect(Collectors.toList());
}
private Map<String,ClassFileObject> createClassFileObjects() {
return input.keySet().stream()
.collect(Collectors.toMap(k -> k, k -> new ClassFileObject(k)));
}
private Map<String,byte[]> createOutput(Map<String,ClassFileObject> cfos) {
return cfos.keySet().stream().collect(Collectors.toMap(k -> k, k -> cfos.get(k).getBytes()));
}
private static class SourceFileObject extends SimpleJavaFileObject {
private final String source;
SourceFileObject(String name, String source) {
super(URI.create("string:///" + name.replace('.', '/') + Kind.SOURCE.extension), Kind.SOURCE);
this.source = source;
}
@Override
public CharSequence getCharContent(boolean ignoreEncodingErrors) {
return source;
}
}
private static class ClassFileObject extends SimpleJavaFileObject {
private final ByteArrayOutputStream baos = new ByteArrayOutputStream();
ClassFileObject(String className) {
super(URI.create(className), Kind.CLASS);
}
@Override
public OutputStream openOutputStream() throws IOException {
return baos;
}
public byte[] getBytes() {
return baos.toByteArray();
}
}
private static class CustomFileManager extends ForwardingJavaFileManager<JavaFileManager> {
private final Map<String,ClassFileObject> cfos;
CustomFileManager(JavaFileManager jfm, Map<String,ClassFileObject> cfos) {
super(jfm);
this.cfos = cfos;
}
@Override
public JavaFileObject getJavaFileForOutput(JavaFileManager.Location loc, String name,
JavaFileObject.Kind kind, FileObject sibling) throws IOException {
ClassFileObject cfo = cfos.get(name);
return cfo;
}
}
}
|