| /* |
| * Copyright 2003-2010 Dave Griffith, Bas Leijdekkers |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package com.siyeh.ig.errorhandling; |
| |
| import com.intellij.codeInspection.ProblemDescriptor; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.psi.*; |
| import com.intellij.psi.tree.IElementType; |
| import com.intellij.psi.util.PsiTreeUtil; |
| import com.intellij.util.IncorrectOperationException; |
| import com.siyeh.InspectionGadgetsBundle; |
| import com.siyeh.ig.BaseInspection; |
| import com.siyeh.ig.BaseInspectionVisitor; |
| import com.siyeh.ig.InspectionGadgetsFix; |
| import com.intellij.psi.util.FileTypeUtils; |
| import org.jetbrains.annotations.NotNull; |
| |
| public class EmptyFinallyBlockInspection extends BaseInspection { |
| @Override |
| @NotNull |
| public String getDisplayName() { |
| return InspectionGadgetsBundle.message("empty.finally.block.display.name"); |
| } |
| |
| @Override |
| public boolean isEnabledByDefault() { |
| return true; |
| } |
| |
| @Override |
| @NotNull |
| protected String buildErrorString(Object... infos) { |
| return InspectionGadgetsBundle.message("empty.finally.block.problem.descriptor"); |
| } |
| |
| @Override |
| protected InspectionGadgetsFix buildFix(Object... infos) { |
| final Integer count = (Integer)infos[0]; |
| if (count.intValue() == 0) { |
| return new RemoveTryFinallyBlockFix(); |
| } |
| else { |
| return new RemoveFinallyBlockFix(); |
| } |
| } |
| |
| private static class RemoveTryFinallyBlockFix extends InspectionGadgetsFix { |
| @Override |
| @NotNull |
| public String getFamilyName() { |
| return getName(); |
| } |
| @Override |
| @NotNull |
| public String getName() { |
| return InspectionGadgetsBundle.message("remove.try.finally.block.quickfix"); |
| } |
| |
| @Override |
| protected void doFix(Project project, ProblemDescriptor descriptor) throws IncorrectOperationException { |
| final PsiElement element = descriptor.getPsiElement(); |
| final PsiTryStatement tryStatement = PsiTreeUtil.getParentOfType(element, PsiTryStatement.class); |
| if (tryStatement == null) { |
| return; |
| } |
| final PsiCodeBlock tryBlock = tryStatement.getTryBlock(); |
| if (tryBlock == null) { |
| return; |
| } |
| final PsiElement parent = tryStatement.getParent(); |
| if (parent == null) { |
| return; |
| } |
| |
| final PsiResourceList resources = tryStatement.getResourceList(); |
| if (resources != null) { |
| final PsiElementFactory factory = JavaPsiFacade.getElementFactory(project); |
| for (PsiResourceVariable resource : resources.getResourceVariables()) { |
| final PsiStatement statement = factory.createStatementFromText(resource.getText() + ";", parent); |
| parent.addBefore(statement, tryStatement); |
| } |
| } |
| |
| final PsiElement first = tryBlock.getFirstBodyElement(); |
| final PsiElement last = tryBlock.getLastBodyElement(); |
| if (first != null && last != null) { |
| parent.addRangeAfter(first, last, tryStatement); |
| } |
| |
| tryStatement.delete(); |
| } |
| } |
| |
| private static class RemoveFinallyBlockFix extends InspectionGadgetsFix { |
| @Override |
| @NotNull |
| public String getFamilyName() { |
| return getName(); |
| } |
| @Override |
| @NotNull |
| public String getName() { |
| return InspectionGadgetsBundle.message("remove.finally.block.quickfix"); |
| } |
| |
| @Override |
| protected void doFix(Project project, ProblemDescriptor descriptor) |
| throws IncorrectOperationException { |
| final PsiElement element = descriptor.getPsiElement(); |
| final PsiTryStatement tryStatement = |
| PsiTreeUtil.getParentOfType(element, PsiTryStatement.class); |
| if (tryStatement == null) { |
| return; |
| } |
| final PsiCodeBlock finallyBlock = tryStatement.getFinallyBlock(); |
| if (finallyBlock == null) { |
| return; |
| } |
| deleteUntilFinally(finallyBlock); |
| } |
| |
| private static void deleteUntilFinally(PsiElement element) { |
| if (element instanceof PsiJavaToken) { |
| final PsiJavaToken keyword = (PsiJavaToken)element; |
| final IElementType tokenType = keyword.getTokenType(); |
| if (tokenType.equals(JavaTokenType.FINALLY_KEYWORD)) { |
| keyword.delete(); |
| return; |
| } |
| } |
| deleteUntilFinally(element.getPrevSibling()); |
| if (!(element instanceof PsiWhiteSpace)) { |
| element.delete(); |
| } |
| } |
| } |
| |
| @Override |
| public BaseInspectionVisitor buildVisitor() { |
| return new EmptyFinallyBlockVisitor(); |
| } |
| |
| private static class EmptyFinallyBlockVisitor |
| extends BaseInspectionVisitor { |
| |
| @Override |
| public void visitTryStatement( |
| @NotNull PsiTryStatement statement) { |
| super.visitTryStatement(statement); |
| if (FileTypeUtils.isInServerPageFile(statement.getContainingFile())) { |
| return; |
| } |
| final PsiCodeBlock finallyBlock = statement.getFinallyBlock(); |
| if (finallyBlock == null) { |
| return; |
| } |
| if (finallyBlock.getStatements().length != 0) { |
| return; |
| } |
| final PsiCodeBlock[] catchBlocks = statement.getCatchBlocks(); |
| final PsiElement[] children = statement.getChildren(); |
| for (final PsiElement child : children) { |
| final String childText = child.getText(); |
| if (PsiKeyword.FINALLY.equals(childText)) { |
| registerError(child, Integer.valueOf(catchBlocks.length)); |
| return; |
| } |
| } |
| } |
| } |
| } |