Spring/spring-tools
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

create nicer symbols for request mapping annotations

Browse Source
This commit is contained in:
Martin Lippert
2017-08-31 13:31:26 +02:00
parent 3d201b8367
commit a9e325c32e
1 changed files with 77 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
headless-services/boot-java-language-server/src/main/java/org/springframework/ide/vscode/boot/java/requestmapping/RequestMappingSymbolProvider.java
View File

@@ -10,7 +10,16 @@
*******************************************************************************/
package org.springframework.ide.vscode.boot.java.requestmapping;
import java.util.Iterator;
import java.util.List;
import org.eclipse.jdt.core.dom.Annotation;
import org.eclipse.jdt.core.dom.Expression;
import org.eclipse.jdt.core.dom.MemberValuePair;
import org.eclipse.jdt.core.dom.NormalAnnotation;
import org.eclipse.jdt.core.dom.QualifiedName;
import org.eclipse.jdt.core.dom.SingleMemberAnnotation;
import org.eclipse.jdt.core.dom.StringLiteral;
import org.eclipse.lsp4j.Location;
import org.eclipse.lsp4j.SymbolInformation;
import org.eclipse.lsp4j.SymbolKind;
@@ -25,7 +34,16 @@ public class RequestMappingSymbolProvider implements SymbolProvider {
@Override
public SymbolInformation getSymbol(Annotation node, TextDocument doc) {
try {
SymbolInformation symbol = new SymbolInformation(node.toString(), SymbolKind.Interface,
String path = getPath(node);
String method = getMethod(node);
if (path != null && !path.startsWith("/")) {
path = "/" + path;
}
String symbolLabel = "@" + path + " -- " + method;
SymbolInformation symbol = new SymbolInformation(symbolLabel, SymbolKind.Interface,
new Location(doc.getUri(), doc.toRange(node.getStartPosition(), node.getLength())));
return symbol;
}
@@ -36,5 +54,63 @@ public class RequestMappingSymbolProvider implements SymbolProvider {
return null;
}
private String getMethod(Annotation node) {
if (node.isNormalAnnotation()) {
NormalAnnotation normNode = (NormalAnnotation) node;
List<?> values = normNode.values();
for (Iterator<?> iterator = values.iterator(); iterator.hasNext();) {
Object object = iterator.next();
if (object instanceof MemberValuePair) {
MemberValuePair pair = (MemberValuePair) object;
String valueName = pair.getName().getIdentifier();
if (valueName != null && valueName.equals("method")) {
Expression expression = pair.getValue();
if (expression instanceof QualifiedName) {
QualifiedName qualifiedName = (QualifiedName) expression;
return qualifiedName.getName().toString();
}
}
}
}
}
String type = node.getTypeName().toString();
if (type != null && type.endsWith("Mapping") && !type.startsWith("Request")) {
String method = type.substring(0, type.length() - 7).toUpperCase();
return method;
}
return "(no method defined)";
}
private String getPath(Annotation node) {
if (node.isNormalAnnotation()) {
NormalAnnotation normNode = (NormalAnnotation) node;
List<?> values = normNode.values();
for (Iterator<?> iterator = values.iterator(); iterator.hasNext();) {
Object object = iterator.next();
if (object instanceof MemberValuePair) {
MemberValuePair pair = (MemberValuePair) object;
String valueName = pair.getName().getIdentifier();
if (valueName != null && (valueName.equals("value") || valueName.equals("path"))) {
Expression expression = pair.getValue();
if (expression instanceof StringLiteral) {
StringLiteral literal = (StringLiteral) expression;
return literal.getLiteralValue();
}
}
}
}
} else if (node.isSingleMemberAnnotation()) {
SingleMemberAnnotation singleNode = (SingleMemberAnnotation) node;
Expression expression = singleNode.getValue();
if (expression instanceof StringLiteral) {
StringLiteral literal = (StringLiteral) expression;
return literal.getLiteralValue();
}
}
return null;
}
}