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
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
|
package org.javacs;
import com.sun.source.tree.*;
import com.sun.source.util.*;
import java.io.IOException;
import java.net.URI;
import java.util.*;
import java.util.logging.Logger;
import javax.lang.model.element.*;
import javax.lang.model.type.TypeKind;
import javax.lang.model.util.*;
import javax.tools.*;
import org.javacs.lsp.Range;
public class CompileBatch implements AutoCloseable {
private final JavaCompilerService parent;
private final ReportProgress progress;
private final TaskPool.Borrow borrow;
private final Trees trees;
private final Elements elements;
private final Types types;
private final List<CompilationUnitTree> roots;
CompileBatch(JavaCompilerService parent, Collection<? extends JavaFileObject> files, ReportProgress progress) {
this.parent = parent;
this.progress = progress;
this.borrow = batchTask(parent, files);
this.trees = Trees.instance(borrow.task);
this.elements = borrow.task.getElements();
this.types = borrow.task.getTypes();
this.roots = new ArrayList<CompilationUnitTree>();
// Print timing information for optimization
var profiler = new Profiler();
borrow.task.addTaskListener(profiler);
// Show progress message through the UI
class CountFiles implements TaskListener {
Set<URI> parse = new HashSet<>(), enter = new HashSet<>(), analyze = new HashSet<>();
void inc(String message) {
var n = parse.size() + enter.size() + analyze.size();
var total = files.size() * 3;
progress.progress(message, n, total);
}
@Override
public void started(TaskEvent e) {
var uri = e.getSourceFile().toUri();
switch (e.getKind()) {
case PARSE:
if (parse.add(uri)) inc("Parse sources");
break;
case ENTER:
if (enter.add(uri)) inc("Enter symbols");
break;
case ANALYZE:
var name = Parser.fileName(uri);
if (analyze.add(uri)) inc("Analyze " + name);
break;
}
}
}
borrow.task.addTaskListener(new CountFiles());
// Compile all roots
try {
for (var t : borrow.task.parse()) roots.add(t);
// The results of borrow.task.analyze() are unreliable when errors are present
// You can get at `Element` values using `Trees`
borrow.task.analyze();
} catch (IOException e) {
throw new RuntimeException(e);
}
profiler.print();
}
@Override
public void close() {
borrow.close();
}
static TaskPool.Borrow batchTask(JavaCompilerService parent, Collection<? extends JavaFileObject> sources) {
parent.diags.clear();
return parent.compiler.getTask(
null,
parent.fileManager,
parent.diags::add,
JavaCompilerService.options(parent.classPath),
Collections.emptyList(),
sources);
}
public Optional<Element> element(URI uri, int line, int character) {
for (var root : roots) {
if (root.getSourceFile().toUri().equals(uri)) {
var path = CompileFocus.findPath(borrow.task, root, line, character);
var el = trees.getElement(path);
return Optional.ofNullable(el);
}
}
// Somehow, uri was not in batch
var names = new StringJoiner(", ");
for (var r : roots) {
names.add(Parser.fileName(r.getSourceFile().toUri()));
}
throw new RuntimeException("File " + uri + " isn't in batch " + names);
}
public Optional<List<TreePath>> definitions(Element el) {
LOG.info(String.format("Search for definitions of `%s` in %d files...", el, roots.size()));
if (el.asType().getKind() == TypeKind.ERROR) {
LOG.info(String.format("...`%s` is an error type, giving up", el.asType()));
return Optional.empty();
}
var refs = new ArrayList<TreePath>();
class FindDefinitions extends TreePathScanner<Void, Void> {
boolean sameSymbol(Element found) {
return el.equals(found);
}
boolean isSubMethod(Element found) {
if (!(el instanceof ExecutableElement)) return false;
if (!(found instanceof ExecutableElement)) return false;
var superMethod = (ExecutableElement) el;
var subMethod = (ExecutableElement) found;
var subType = (TypeElement) subMethod.getEnclosingElement();
// TODO need to check if class is compatible as well
if (elements.overrides(subMethod, superMethod, subType)) {
// LOG.info(String.format("...`%s.%s` overrides `%s`", subType, subMethod, superMethod));
return true;
}
return false;
}
void check(TreePath from) {
var found = trees.getElement(from);
var match = sameSymbol(found) || isSubMethod(found);
if (match) refs.add(from);
}
@Override
public Void visitClass(ClassTree t, Void __) {
check(getCurrentPath());
return super.visitClass(t, null);
}
@Override
public Void visitMethod(MethodTree t, Void __) {
check(getCurrentPath());
return super.visitMethod(t, null);
}
@Override
public Void visitVariable(VariableTree t, Void __) {
check(getCurrentPath());
return super.visitVariable(t, null);
}
}
var finder = new FindDefinitions();
for (var r : roots) {
finder.scan(r, null);
}
return Optional.of(refs);
}
public Optional<List<TreePath>> references(Element to) {
LOG.info(String.format("Search for references to %s...", to));
// If to is an error, we won't be able to find anything
if (to.asType().getKind() == TypeKind.ERROR) {
LOG.info(String.format("...`%s` is an error type, giving up", to.asType()));
return Optional.empty();
}
// Otherwise, scan roots for references
List<TreePath> list = new ArrayList<TreePath>();
var map = Map.of(to, list);
var finder = new FindReferences(borrow.task);
for (var r : roots) {
finder.scan(r, map);
}
LOG.info(String.format("...found %d references", list.size()));
return Optional.of(list);
}
/**
* Find all elements in `file` that get turned into code-lenses. This needs to match the result of
* `ParseFile#declarations`
*/
public List<Element> declarations(URI file) {
for (var r : roots) {
if (!r.getSourceFile().toUri().equals(file)) continue;
var paths = ParseFile.declarations(r);
var els = new ArrayList<Element>();
for (var p : paths) {
var e = trees.getElement(p);
assert e != null;
els.add(e);
}
return els;
}
// Couldn't find file! Throw an error.
var message = new StringJoiner(", ");
for (var r : roots) {
message.add(Parser.fileName(r.getSourceFile().toUri()));
}
throw new RuntimeException(file + " is not in " + message);
}
public Index index(URI from, List<Element> declarations) {
for (var r : roots) {
if (r.getSourceFile().toUri().equals(from)) {
return new Index(borrow.task, r, parent.diags, declarations);
}
}
throw new RuntimeException(from + " is not in compiled batch");
}
public Optional<Range> range(TreePath path) {
var uri = path.getCompilationUnit().getSourceFile().toUri();
var contents = FileStore.contents(uri);
return ParseFile.range(borrow.task, contents, path);
}
private static final Logger LOG = Logger.getLogger("main");
}
|
