org.sonar.java.JavaFilesCache Maven / Gradle / Ivy
/*
* SonarQube Java
* Copyright (C) 2012 SonarSource
* [email protected]
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 3 of the License, or (at your option) any later version.
*
* This program 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02
*/
package org.sonar.java;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.collect.HashMultimap;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Multimap;
import org.apache.commons.lang.StringUtils;
import org.sonar.java.model.InternalSyntaxToken;
import org.sonar.java.model.JavaTree;
import org.sonar.java.signature.MethodSignaturePrinter;
import org.sonar.java.signature.MethodSignatureScanner;
import org.sonar.plugins.java.api.JavaFileScanner;
import org.sonar.plugins.java.api.JavaFileScannerContext;
import org.sonar.plugins.java.api.tree.AnnotationTree;
import org.sonar.plugins.java.api.tree.BaseTreeVisitor;
import org.sonar.plugins.java.api.tree.ClassTree;
import org.sonar.plugins.java.api.tree.ExpressionTree;
import org.sonar.plugins.java.api.tree.IdentifierTree;
import org.sonar.plugins.java.api.tree.LiteralTree;
import org.sonar.plugins.java.api.tree.MemberSelectExpressionTree;
import org.sonar.plugins.java.api.tree.MethodTree;
import org.sonar.plugins.java.api.tree.NewArrayTree;
import org.sonar.plugins.java.api.tree.Tree;
import java.io.File;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
public class JavaFilesCache extends BaseTreeVisitor implements JavaFileScanner {
@VisibleForTesting
Map resourcesCache = Maps.newHashMap();
@VisibleForTesting
Map methodStartLines = Maps.newHashMap();
@VisibleForTesting
Multimap suppressWarningLines = HashMultimap.create();
private File currentFile;
private Deque currentClassKey = new LinkedList();
private Deque parent = new LinkedList();
private Deque anonymousInnerClassCounter = new LinkedList();
private String currentPackage;
public Map getResourcesCache() {
return resourcesCache;
}
public Map getMethodStartLines() {
return methodStartLines;
}
public Multimap getSuppressWarningLines() {
return suppressWarningLines;
}
public boolean hasSuppressWarningLines() {
return !suppressWarningLines.isEmpty();
}
@Override
public void scanFile(JavaFileScannerContext context) {
JavaTree.CompilationUnitTreeImpl tree = (JavaTree.CompilationUnitTreeImpl) context.getTree();
currentPackage = tree.packageNameAsString().replace('.', '/');
currentFile = context.getFile();
currentClassKey.clear();
parent.clear();
anonymousInnerClassCounter.clear();
suppressWarningLines.clear();
scan(tree);
}
@Override
public void visitClass(ClassTree tree) {
String className = "";
if (tree.simpleName() != null) {
className = tree.simpleName().name();
}
String key = getClassKey(className);
currentClassKey.push(key);
parent.push(tree);
anonymousInnerClassCounter.push(0);
resourcesCache.put(key, currentFile);
handleSuppressWarning(tree);
super.visitClass(tree);
currentClassKey.pop();
parent.pop();
anonymousInnerClassCounter.pop();
}
private String getClassKey(String className) {
String key = className;
if (StringUtils.isNotEmpty(currentPackage)) {
key = currentPackage + "/" + className;
}
if ("".equals(className) || (parent.peek() != null && parent.peek().is(Tree.Kind.METHOD))) {
// inner class declared within method
int count = anonymousInnerClassCounter.pop() + 1;
key = currentClassKey.peek() + "$" + count + className;
anonymousInnerClassCounter.push(count);
} else if (currentClassKey.peek() != null) {
key = currentClassKey.peek() + "$" + className;
}
return key;
}
@Override
public void visitMethod(MethodTree tree) {
parent.push(tree);
String methodKey = currentClassKey.peek() + "#" + MethodSignaturePrinter.print(MethodSignatureScanner.scan(tree));
methodStartLines.put(methodKey, ((JavaTree) tree.simpleName()).getLine());
handleSuppressWarning(tree);
super.visitMethod(tree);
parent.pop();
}
private void handleSuppressWarning(ClassTree tree) {
int endLine = ((InternalSyntaxToken) tree.closeBraceToken()).getLine();
handleSuppressWarning(tree.modifiers().annotations(), endLine);
}
private void handleSuppressWarning(MethodTree tree) {
int endLine = ((JavaTree) tree.simpleName()).getLine();
// if we have no block, then we assume method is on one line on the method name line.
if (tree.block() != null) {
endLine = ((InternalSyntaxToken) tree.block().closeBraceToken()).getLine();
}
handleSuppressWarning(tree.modifiers().annotations(), endLine);
}
private void handleSuppressWarning(List annotationTrees, int endLine) {
int startLine = 0;
List warnings = Lists.newArrayList();
for (AnnotationTree annotationTree : annotationTrees) {
if (isSuppressWarningsAnnotation(annotationTree)) {
startLine = ((JavaTree) annotationTree).getLine();
warnings.addAll(getSuppressWarningArgs(annotationTree));
break;
}
}
for (int i = startLine; i <= endLine; i++) {
suppressWarningLines.putAll(i, warnings);
}
}
private boolean isSuppressWarningsAnnotation(AnnotationTree annotationTree) {
boolean suppressWarningsType = false;
Tree type = annotationTree.annotationType();
if (type.is(Tree.Kind.IDENTIFIER)) {
suppressWarningsType = "SuppressWarnings".equals(((IdentifierTree) type).name());
} else if (type.is(Tree.Kind.MEMBER_SELECT)) {
MemberSelectExpressionTree mset = (MemberSelectExpressionTree) type;
suppressWarningsType = "SuppressWarnings".equals(mset.identifier().name()) &&
mset.expression().is(Tree.Kind.MEMBER_SELECT) && "lang".equals(((MemberSelectExpressionTree) mset.expression()).identifier().name());
}
return suppressWarningsType;
}
private List getSuppressWarningArgs(AnnotationTree annotationTree) {
return getValueFromExpression(annotationTree.arguments().get(0));
}
private List getValueFromExpression(ExpressionTree expression) {
List args = Lists.newArrayList();
if (expression.is(Tree.Kind.STRING_LITERAL)) {
args.add(trimQuotes(((LiteralTree) expression).value()));
} else if (expression.is(Tree.Kind.NEW_ARRAY)) {
for (ExpressionTree initializer : ((NewArrayTree) expression).initializers()) {
args.addAll(getValueFromExpression(initializer));
}
}
return args;
}
private String trimQuotes(String value) {
return value.substring(1, value.length() - 1);
}
}