Examples with CaseTree com.sun.source.tree.CaseTree used on opensource projects

Example 1 with CaseTree

use of com.sun.source.tree.CaseTree in project error-prone by google.

the class ControlFlowVisitor method visitSwitch.

@Override
public Result visitSwitch(SwitchTree node, BreakContext cxt) {
    Result result = null;
    boolean seenDefault = false;
    cxt.loopDepth++;
    try {
        for (CaseTree caseTree : node.getCases()) {
            if (caseTree.getExpression() == null) {
                seenDefault = true;
            }
            if (result == null) {
                result = caseTree.accept(this, cxt);
            } else {
                result = result.or(caseTree.accept(this, cxt));
            }
        }
        if (!seenDefault) {
            result = result.or(NEVER_EXITS);
        }
        return result;
    } finally {
        cxt.loopDepth--;
    }
}

Example 2 with CaseTree

use of com.sun.source.tree.CaseTree in project error-prone by google.

the class WildcardImport method qualifiedNameFix.

/**
 * Add an import for {@code owner}, and qualify all on demand imported references to members of
 * owner by owner's simple name.
 */
private static void qualifiedNameFix(final SuggestedFix.Builder fix, final Symbol owner, VisitorState state) {
    fix.addImport(owner.getQualifiedName().toString());
    final JCCompilationUnit unit = (JCCompilationUnit) state.getPath().getCompilationUnit();
    new TreePathScanner<Void, Void>() {

        @Override
        public Void visitIdentifier(IdentifierTree tree, Void unused) {
            Symbol sym = ASTHelpers.getSymbol(tree);
            if (sym == null) {
                return null;
            }
            Tree parent = getCurrentPath().getParentPath().getLeaf();
            if (parent.getKind() == Tree.Kind.CASE && ((CaseTree) parent).getExpression().equals(tree) && sym.owner.getKind() == ElementKind.ENUM) {
                // switch cases can refer to enum constants by simple name without importing them
                return null;
            }
            if (sym.owner.equals(owner) && unit.starImportScope.includes(sym)) {
                fix.prefixWith(tree, owner.getSimpleName() + ".");
            }
            return null;
        }
    }.scan(unit, null);
}

Example 3 with CaseTree

use of com.sun.source.tree.CaseTree in project error-prone by google.

the class MissingDefault method matchSwitch.

@Override
public Description matchSwitch(SwitchTree tree, VisitorState state) {
    Type switchType = ASTHelpers.getType(tree.getExpression());
    if (switchType.asElement().getKind() == ElementKind.ENUM) {
        // by MissingCasesInEnumSwitch
        return NO_MATCH;
    }
    Optional<? extends CaseTree> maybeDefault = tree.getCases().stream().filter(c -> c.getExpression() == null).findFirst();
    if (!maybeDefault.isPresent()) {
        Description.Builder description = buildDescription(tree);
        if (!tree.getCases().isEmpty()) {
            // Inserting the default after the last case is easier than finding the closing brace
            // for the switch statement. Hopefully we don't often see switches with zero cases.
            CaseTree lastCase = getLast(tree.getCases());
            String replacement;
            if (lastCase.getStatements().isEmpty() || Reachability.canCompleteNormally(Iterables.getLast(lastCase.getStatements()))) {
                replacement = "\nbreak;\ndefault: // fall out\n";
            } else {
                replacement = "\ndefault: // fall out\n";
            }
            description.addFix(SuggestedFix.postfixWith(getLast(tree.getCases()), replacement));
        }
        return description.build();
    }
    CaseTree defaultCase = maybeDefault.get();
    if (!defaultCase.getStatements().isEmpty()) {
        return NO_MATCH;
    }
    // If `default` case is empty, and last in switch, add `// fall out` comment
    // TODO(epmjohnston): Maybe move comment logic to go/bugpattern/FallThrough
    int idx = tree.getCases().indexOf(defaultCase);
    if (idx != tree.getCases().size() - 1) {
        return NO_MATCH;
    }
    int end = state.getEndPosition(tree);
    if (ErrorProneTokens.getTokens(state.getSourceCode().subSequence(state.getEndPosition(defaultCase), end).toString(), state.context).stream().anyMatch(t -> !t.comments().isEmpty())) {
        return NO_MATCH;
    }
    return buildDescription(defaultCase).setMessage("Default case should be documented with a comment").addFix(SuggestedFix.postfixWith(defaultCase, " // fall out")).build();
}

Example 4 with CaseTree

use of com.sun.source.tree.CaseTree in project error-prone by google.

the class SwitchDefault method matchSwitch.

@Override
public Description matchSwitch(SwitchTree tree, VisitorState state) {
    Optional<? extends CaseTree> maybeDefault = tree.getCases().stream().filter(c -> c.getExpression() == null).findAny();
    if (!maybeDefault.isPresent()) {
        return NO_MATCH;
    }
    // Collect all case trees in the statement group containing the default
    List<CaseTree> defaultStatementGroup = new ArrayList<>();
    Iterator<? extends CaseTree> it = tree.getCases().iterator();
    while (it.hasNext()) {
        CaseTree caseTree = it.next();
        defaultStatementGroup.add(caseTree);
        if (caseTree.getExpression() == null) {
            while (it.hasNext() && caseTree.getStatements().isEmpty()) {
                caseTree = it.next();
                defaultStatementGroup.add(caseTree);
            }
            break;
        }
        if (!caseTree.getStatements().isEmpty()) {
            defaultStatementGroup.clear();
        }
    }
    // Find the position of the default case within the statement group
    int idx = defaultStatementGroup.indexOf(maybeDefault.get());
    SuggestedFix.Builder fix = SuggestedFix.builder();
    CaseTree defaultTree = defaultStatementGroup.get(idx);
    if (it.hasNext()) {
        // Only emit a fix if the default doesn't fall through.
        if (!Reachability.canCompleteNormally(getLast(defaultStatementGroup))) {
            int start = ((JCTree) defaultStatementGroup.get(0)).getStartPosition();
            int end = state.getEndPosition(getLast(defaultStatementGroup));
            String replacement;
            String source = state.getSourceCode().toString();
            // If the default case isn't the last case in its statement group, move it to the end.
            if (idx != defaultStatementGroup.size() - 1) {
                int caseEnd = ((JCTree) getLast(defaultStatementGroup).getStatements().get(0)).getStartPosition();
                int cutStart = ((JCTree) defaultTree).getStartPosition();
                int cutEnd = state.getEndPosition(defaultTree);
                replacement = source.substring(start, cutStart) + source.substring(cutEnd, caseEnd) + "\n" + source.substring(cutStart, cutEnd) + source.substring(caseEnd, end);
            } else {
                replacement = source.substring(start, end);
            }
            fix.replace(start, end, "").postfixWith(getLast(tree.getCases()), replacement);
        }
    } else if (idx != defaultStatementGroup.size() - 1) {
        // If the default case isn't the last case in its statement group, move it to the end.
        fix.delete(defaultTree).prefixWith(getLast(defaultStatementGroup).getStatements().get(0), state.getSourceForNode(defaultTree));
    } else {
        return NO_MATCH;
    }
    Description.Builder description = buildDescription(defaultStatementGroup.get(0));
    if (!fix.isEmpty()) {
        description.addFix(fix.build());
    }
    return description.build();
}

Example 5 with CaseTree

use of com.sun.source.tree.CaseTree in project checker-framework by typetools.

the class FenumVisitor method visitSwitch.

@Override
public Void visitSwitch(SwitchTree node, Void p) {
    ExpressionTree expr = node.getExpression();
    AnnotatedTypeMirror exprType = atypeFactory.getAnnotatedType(expr);
    for (CaseTree caseExpr : node.getCases()) {
        List<? extends ExpressionTree> realCaseExprs = TreeUtils.caseTreeGetExpressions(caseExpr);
        // Check all the case options against the switch expression type:
        for (ExpressionTree realCaseExpr : realCaseExprs) {
            AnnotatedTypeMirror caseType = atypeFactory.getAnnotatedType(realCaseExpr);
            // There is currently no "switch" message key, so it is treated
            // identically to "type.incompatible".
            this.commonAssignmentCheck(exprType, caseType, caseExpr, "type.incompatible");
        }
    }
    return super.visitSwitch(node, p);
}