Skip to content

Commit 91ea659

Browse files
Arthurm1Arthurm1
committed
convert to public API
1 parent b3d6ae2 commit 91ea659

64 files changed

Lines changed: 1990 additions & 1756 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎semanticdb-javac/src/main/java/com/sourcegraph/semanticdb_javac/CompilerRange.java‎

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,12 @@ public enum CompilerRange {
1818
/** Map the compiler start/end positions to SemanticDB start/end positions. */
1919
FROM_START_TO_END,
2020

21+
/**
22+
* Map the compiler point position to SemanticDB end and use (point - symbol name length) for the
23+
* SemanticDB start position.
24+
*/
25+
FROM_END_TO_SYMBOL_NAME,
26+
2127
/**
2228
* Map the compiler point position to SemanticDB start and use (point + symbol name length) for
2329
* the SemanticDB end position.
@@ -59,6 +65,16 @@ public boolean isFromPoint() {
5965
}
6066
}
6167

68+
public boolean isFromEndPoint() {
69+
switch (this) {
70+
case FROM_END_TO_SYMBOL_NAME:
71+
case FROM_END_WITH_TEXT_SEARCH:
72+
return true;
73+
default:
74+
return false;
75+
}
76+
}
77+
6278
public boolean isFromTextSearch() {
6379
switch (this) {
6480
case FROM_TEXT_SEARCH:
@@ -70,6 +86,15 @@ public boolean isFromTextSearch() {
7086
}
7187
}
7288

89+
public boolean isPlusOne() {
90+
switch (this) {
91+
case FROM_POINT_TO_SYMBOL_NAME_PLUS_ONE:
92+
return true;
93+
default:
94+
return false;
95+
}
96+
}
97+
7398
public boolean isFromEnd() {
7499
return this == CompilerRange.FROM_END_WITH_TEXT_SEARCH;
75100
}

‎semanticdb-javac/src/main/java/com/sourcegraph/semanticdb_javac/EmptyEndPosTable.java‎

Lines changed: 0 additions & 22 deletions
This file was deleted.

‎semanticdb-javac/src/main/java/com/sourcegraph/semanticdb_javac/GlobalSymbolsCache.java‎

Lines changed: 51 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,29 @@
11
package com.sourcegraph.semanticdb_javac;
22

3-
import com.sun.tools.javac.code.Symbol;
4-
53
import javax.lang.model.element.Element;
64
import javax.lang.model.element.ElementKind;
75
import javax.lang.model.element.ExecutableElement;
8-
import java.util.*;
6+
import javax.lang.model.element.TypeElement;
7+
import javax.lang.model.element.PackageElement;
8+
import javax.lang.model.element.TypeParameterElement;
9+
import javax.lang.model.element.VariableElement;
10+
import javax.lang.model.element.QualifiedNameable;
11+
import java.util.IdentityHashMap;
12+
import java.util.ArrayList;
913

1014
import static com.sourcegraph.semanticdb_javac.Debugging.pprint;
1115

1216
/** Cache of SemanticDB symbols that can be referenced between files. */
1317
public final class GlobalSymbolsCache {
1418

15-
private final IdentityHashMap<Symbol, String> globals = new IdentityHashMap<>();
19+
private final IdentityHashMap<Element, String> globals = new IdentityHashMap<>();
1620
private final SemanticdbJavacOptions options;
1721

1822
public GlobalSymbolsCache(SemanticdbJavacOptions options) {
1923
this.options = options;
2024
}
2125

22-
public String semanticdbSymbol(Symbol sym, LocalSymbolsCache locals) {
26+
public String semanticdbSymbol(Element sym, LocalSymbolsCache locals) {
2327
String result = globals.get(sym);
2428
if (result != null) return result;
2529
String localResult = locals.get(sym);
@@ -31,33 +35,48 @@ public String semanticdbSymbol(Symbol sym, LocalSymbolsCache locals) {
3135
return result;
3236
}
3337

34-
public String semanticdbSymbol(Element element, LocalSymbolsCache locals) {
35-
return semanticdbSymbol((Symbol) element, locals);
36-
}
37-
38-
public boolean isNone(Symbol sym) {
38+
public boolean isNone(Element sym) {
3939
return sym == null;
4040
}
4141

42-
private String uncachedSemanticdbSymbol(Symbol sym, LocalSymbolsCache locals) {
43-
if (isNone(sym)) return SemanticdbSymbols.NONE;
44-
String owner = semanticdbSymbol(sym.owner, locals);
45-
if (owner.equals(SemanticdbSymbols.NONE)) {
42+
private String uncachedSemanticdbSymbol(Element sym, LocalSymbolsCache locals) {
43+
44+
if (isNone(sym)) return SemanticdbSymbols.ROOT_PACKAGE;
45+
if (sym instanceof PackageElement && ((PackageElement) sym).isUnnamed())
4646
return SemanticdbSymbols.ROOT_PACKAGE;
47-
} else if (SemanticdbSymbols.isLocal(owner) || isAnonymousClass(sym) || isLocalVariable(sym)) {
47+
String owner = semanticdbSymbol(sym.getEnclosingElement(), locals);
48+
49+
if (SemanticdbSymbols.isLocal(owner) || isAnonymousClass(sym) || isLocalVariable(sym)) {
4850
return locals.put(sym);
4951
}
52+
53+
if (sym instanceof PackageElement) {
54+
StringBuilder sb = new StringBuilder();
55+
String qualifiedName = ((PackageElement) sym).getQualifiedName().toString();
56+
String[] names = qualifiedName.split("\\.");
57+
for (String name : names) {
58+
if (name.length() > 0) {
59+
SemanticdbSymbols.Descriptor desc =
60+
new SemanticdbSymbols.Descriptor(SemanticdbSymbols.Descriptor.Kind.Package, name);
61+
sb.append(desc.encode());
62+
}
63+
}
64+
65+
return sb.toString();
66+
}
5067
SemanticdbSymbols.Descriptor desc = semanticdbDescriptor(sym);
5168
if (options.verboseEnabled && desc.kind == SemanticdbSymbols.Descriptor.Kind.None) {
52-
pprint(sym.getQualifiedName().toString());
69+
if (sym instanceof QualifiedNameable)
70+
pprint(((QualifiedNameable) sym).getQualifiedName().toString());
71+
else pprint(sym.getSimpleName().toString());
5372
pprint(
5473
String.format(
5574
"sym: %s (%s - superclass %s)", sym, sym.getClass(), sym.getClass().getSuperclass()));
5675
}
5776
return SemanticdbSymbols.global(owner, desc);
5877
}
5978

60-
private boolean isLocalVariable(Symbol sym) {
79+
private boolean isLocalVariable(Element sym) {
6180
switch (sym.getKind()) {
6281
case PARAMETER:
6382
case EXCEPTION_PARAMETER:
@@ -68,28 +87,25 @@ private boolean isLocalVariable(Symbol sym) {
6887
}
6988
}
7089

71-
private boolean isAnonymousClass(Symbol sym) {
72-
return sym instanceof Symbol.ClassSymbol && sym.name.isEmpty();
90+
private boolean isAnonymousClass(Element sym) {
91+
return sym instanceof TypeElement && sym.getSimpleName().length() == 0;
7392
}
7493

75-
private SemanticdbSymbols.Descriptor semanticdbDescriptor(Symbol sym) {
76-
if (sym instanceof Symbol.ClassSymbol) {
94+
private SemanticdbSymbols.Descriptor semanticdbDescriptor(Element sym) {
95+
if (sym instanceof TypeElement) {
7796
return new SemanticdbSymbols.Descriptor(
78-
SemanticdbSymbols.Descriptor.Kind.Type, sym.name.toString());
79-
} else if (sym instanceof Symbol.MethodSymbol) {
97+
SemanticdbSymbols.Descriptor.Kind.Type, sym.getSimpleName().toString());
98+
} else if (sym instanceof ExecutableElement) {
8099
return new SemanticdbSymbols.Descriptor(
81100
SemanticdbSymbols.Descriptor.Kind.Method,
82-
sym.name.toString(),
83-
methodDisambiguator((Symbol.MethodSymbol) sym));
84-
} else if (sym instanceof Symbol.PackageSymbol) {
85-
return new SemanticdbSymbols.Descriptor(
86-
SemanticdbSymbols.Descriptor.Kind.Package, sym.name.toString());
87-
} else if (sym instanceof Symbol.TypeVariableSymbol) {
101+
sym.getSimpleName().toString(),
102+
methodDisambiguator((ExecutableElement) sym));
103+
} else if (sym instanceof TypeParameterElement) {
88104
return new SemanticdbSymbols.Descriptor(
89-
SemanticdbSymbols.Descriptor.Kind.TypeParameter, sym.name.toString());
90-
} else if (sym instanceof Symbol.VarSymbol) {
105+
SemanticdbSymbols.Descriptor.Kind.TypeParameter, sym.getSimpleName().toString());
106+
} else if (sym instanceof VariableElement) {
91107
return new SemanticdbSymbols.Descriptor(
92-
SemanticdbSymbols.Descriptor.Kind.Term, sym.name.toString());
108+
SemanticdbSymbols.Descriptor.Kind.Term, sym.getSimpleName().toString());
93109
} else {
94110
return SemanticdbSymbols.Descriptor.NONE;
95111
}
@@ -113,11 +129,11 @@ private SemanticdbSymbols.Descriptor semanticdbDescriptor(Symbol sym) {
113129
* <p><a href="https://scalameta.org/docs/semanticdb/specification.html#symbol-2">Link to
114130
* SemanticDB spec</a>.
115131
*/
116-
private String methodDisambiguator(Symbol.MethodSymbol sym) {
117-
Iterable<? extends Element> elements = sym.owner.getEnclosedElements();
132+
private String methodDisambiguator(ExecutableElement sym) {
133+
Iterable<? extends Element> elements = sym.getEnclosingElement().getEnclosedElements();
118134
ArrayList<ExecutableElement> methods = new ArrayList<>();
119135
for (Element e : elements) {
120-
if (e instanceof ExecutableElement && e.getSimpleName() == sym.name) {
136+
if (e instanceof ExecutableElement && e.getSimpleName() == sym.getSimpleName()) {
121137
methods.add((ExecutableElement) e);
122138
}
123139
}

‎semanticdb-javac/src/main/java/com/sourcegraph/semanticdb_javac/LocalSymbolsCache.java‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,20 @@
11
package com.sourcegraph.semanticdb_javac;
22

3-
import com.sun.tools.javac.code.Symbol;
3+
import javax.lang.model.element.Element;
44

55
import java.util.IdentityHashMap;
66

77
/** Cache of SemanticDB symbols that are local to a single file. */
88
public final class LocalSymbolsCache {
99

10-
private final IdentityHashMap<Symbol, String> symbols = new IdentityHashMap<>();
10+
private final IdentityHashMap<Element, String> symbols = new IdentityHashMap<>();
1111
private int localsCounter = -1;
1212

13-
public String get(Symbol sym) {
13+
public String get(Element sym) {
1414
return symbols.get(sym);
1515
}
1616

17-
public String put(Symbol sym) {
17+
public String put(Element sym) {
1818
localsCounter++;
1919
String result = SemanticdbSymbols.local(localsCounter);
2020
symbols.put(sym, result);

‎semanticdb-javac/src/main/java/com/sourcegraph/semanticdb_javac/RangeFinder.java‎

Lines changed: 71 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -1,60 +1,93 @@
11
package com.sourcegraph.semanticdb_javac;
22

3-
import com.sun.source.tree.CompilationUnitTree;
4-
import com.sun.source.tree.LineMap;
5-
import com.sun.source.util.TreePath;
6-
import com.sun.source.util.Trees;
3+
import javax.tools.Diagnostic;
74

85
import javax.lang.model.element.Element;
9-
import javax.lang.model.element.Name;
10-
import java.io.IOException;
11-
import java.util.HashMap;
126
import java.util.Optional;
13-
import java.util.regex.Matcher;
14-
import java.util.regex.Pattern;
15-
16-
import static com.sourcegraph.semanticdb_javac.Debugging.pprint;
177

188
public class RangeFinder {
19-
public static Optional<Semanticdb.Range> findRange(
20-
TreePath path,
21-
Trees trees,
22-
CompilationUnitTree root,
9+
public static class StartEndRange {
10+
public int start;
11+
public int end;
12+
13+
StartEndRange(int start, int end) {
14+
this.start = start;
15+
this.end = end;
16+
}
17+
}
18+
19+
public static Optional<StartEndRange> findRange(
2320
Element element,
24-
int startPos,
21+
String name,
22+
int originalStartPos,
23+
int originalEndPos,
2524
String source,
2625
boolean fromEnd) {
27-
LineMap lineMap = root.getLineMap();
28-
Name name = element.getSimpleName();
29-
if (name.contentEquals("<init>")) name = element.getEnclosingElement().getSimpleName();
30-
31-
int endPos = (int) trees.getSourcePositions().getEndPosition(root, path.getLeaf());
32-
// false for anonymous classes
33-
if (name.length() != 0) {
34-
startPos = findNameIn(name, fromEnd ? endPos : startPos, source, fromEnd);
35-
endPos = startPos + name.length();
36-
}
26+
int startPos = findNameIn(name, originalStartPos, originalEndPos, element, source, fromEnd);
27+
int endPos = startPos + name.length();
3728

3829
if (endPos == -1 || startPos == -1) {
3930
return Optional.empty();
4031
}
4132

42-
Semanticdb.Range range =
43-
Semanticdb.Range.newBuilder()
44-
.setStartLine((int) lineMap.getLineNumber(startPos) - 1)
45-
.setStartCharacter((int) lineMap.getColumnNumber(startPos) - 1)
46-
.setEndLine((int) lineMap.getLineNumber(endPos) - 1)
47-
.setEndCharacter((int) lineMap.getColumnNumber(endPos) - 1)
48-
.build();
49-
return Optional.of(range);
33+
return Optional.of(new StartEndRange(startPos, endPos));
34+
}
35+
36+
private static int findNameFromEnd(
37+
String name,
38+
int originalStartPos,
39+
int originalEndPos,
40+
int end,
41+
Element element,
42+
String source) {
43+
if (end < 0) return -1;
44+
int offset = source.lastIndexOf(name, end);
45+
if (offset == -1 && originalStartPos != Diagnostic.NOPOS && originalEndPos != Diagnostic.NOPOS)
46+
return originalStartPos;
47+
if (offset == -1) {
48+
return -1;
49+
}
50+
int endOfWord = offset + name.length();
51+
// found name in wrong word? e.g. finding `"A"` in `A("A")`
52+
if (offset > 0 && Character.isJavaIdentifierPart(source.charAt(offset - 1)))
53+
return findNameFromEnd(name, originalStartPos, originalEndPos, offset - 1, element, source);
54+
if (endOfWord < source.length() && Character.isJavaIdentifierPart(source.charAt(endOfWord)))
55+
return findNameFromEnd(name, originalStartPos, originalEndPos, offset - 1, element, source);
56+
57+
return offset;
58+
}
59+
60+
private static int findNameFromStart(
61+
String name,
62+
int start,
63+
int originalStartPos,
64+
int originalEndPos,
65+
Element element,
66+
String source) {
67+
if (start >= source.length()) return -1;
68+
int offset = source.indexOf(name, start);
69+
if (offset == -1 && originalStartPos != Diagnostic.NOPOS && originalEndPos != Diagnostic.NOPOS)
70+
return originalStartPos;
71+
if (offset == -1) {
72+
return -1;
73+
}
74+
int end = offset + name.length();
75+
// found name in wrong word? e.g. finding `"A"` in `A("A")`
76+
if (offset > 0 && Character.isJavaIdentifierPart(source.charAt(offset - 1)))
77+
return findNameFromStart(name, end + 1, originalStartPos, originalEndPos, element, source);
78+
if (end < source.length() && Character.isJavaIdentifierPart(source.charAt(end)))
79+
return findNameFromStart(name, end + 1, originalStartPos, originalEndPos, element, source);
80+
81+
return offset;
5082
}
5183

52-
private static int findNameIn(CharSequence name, int start, String source, boolean fromEnd) {
53-
if (source.equals("")) return -1;
84+
private static int findNameIn(
85+
String name, int start, int end, Element element, String source, boolean fromEnd) {
86+
if (source.length() == 0) return -1;
5487

5588
int offset;
56-
if (fromEnd) offset = source.lastIndexOf(name.toString(), start);
57-
else offset = source.indexOf(name.toString(), start);
89+
if (fromEnd) offset = findNameFromEnd(name, start, start, end, element, source);
90+
else offset = findNameFromStart(name, start, end, end, element, source);
5891
if (offset > -1) {
5992
return offset;
6093
}

0 commit comments

Comments
 (0)