From c505f989295d0a2426ae2bab7ae1aead67d164ea Mon Sep 17 00:00:00 2001 From: anna Date: Mon, 21 Oct 2013 18:08:49 +0200 Subject: [PATCH] IG: allow to proceed with replaced statement --- .../src/com/siyeh/ig/InspectionGadgetsFix.java | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/plugins/InspectionGadgets/InspectionGadgetsAnalysis/src/com/siyeh/ig/InspectionGadgetsFix.java b/plugins/InspectionGadgets/InspectionGadgetsAnalysis/src/com/siyeh/ig/InspectionGadgetsFix.java index d0feec6766b3..ab3e0a10ff22 100644 --- a/plugins/InspectionGadgets/InspectionGadgetsAnalysis/src/com/siyeh/ig/InspectionGadgetsFix.java +++ b/plugins/InspectionGadgets/InspectionGadgetsAnalysis/src/com/siyeh/ig/InspectionGadgetsFix.java @@ -92,7 +92,7 @@ public abstract class InspectionGadgetsFix implements LocalQuickFix { styleManager.shortenClassReferences(element); } - protected static void replaceExpressionAndShorten(@NotNull PsiExpression expression, @NotNull @NonNls String newExpressionText) { + protected static PsiElement replaceExpressionAndShorten(@NotNull PsiExpression expression, @NotNull @NonNls String newExpressionText) { final Project project = expression.getProject(); final JavaPsiFacade psiFacade = JavaPsiFacade.getInstance(project); final PsiElementFactory factory = psiFacade.getElementFactory(); @@ -101,17 +101,17 @@ public abstract class InspectionGadgetsFix implements LocalQuickFix { final JavaCodeStyleManager javaCodeStyleManager = JavaCodeStyleManager.getInstance(project); javaCodeStyleManager.shortenClassReferences(replacementExp); final CodeStyleManager styleManager = CodeStyleManager.getInstance(project); - styleManager.reformat(replacementExp); + return styleManager.reformat(replacementExp); } - protected static void replaceStatement(@NotNull PsiStatement statement, @NotNull @NonNls String newStatementText) { + protected static PsiElement replaceStatement(@NotNull PsiStatement statement, @NotNull @NonNls String newStatementText) { final Project project = statement.getProject(); final JavaPsiFacade psiFacade = JavaPsiFacade.getInstance(project); final PsiElementFactory factory = psiFacade.getElementFactory(); final PsiStatement newStatement = factory.createStatementFromText(newStatementText, statement); final PsiElement replacementExp = statement.replace(newStatement); final CodeStyleManager styleManager = CodeStyleManager.getInstance(project); - styleManager.reformat(replacementExp); + return styleManager.reformat(replacementExp); } protected static void replaceStatementAndShortenClassNames(@NotNull PsiStatement statement, @NotNull @NonNls String newStatementText) {