split declaration: fix comments in context

This commit is contained in:
Anna Kozlova
2018-06-04 15:24:54 +03:00
parent 446c6110f1
commit 1ef91b8eb4
2 changed files with 5 additions and 7 deletions

View File

@@ -27,7 +27,6 @@ import com.intellij.psi.javadoc.PsiDocComment;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.refactoring.util.RefactoringUtil;
import com.intellij.util.IncorrectOperationException;
import com.siyeh.ig.psiutils.CommentTracker;
import org.jetbrains.annotations.NotNull;
/**
@@ -148,11 +147,10 @@ public class SplitDeclarationAction extends PsiElementBaseIntentionAction {
.createStatementFromText(var.getName() + "=xxx;", decl);
statement = (PsiExpressionStatement)CodeStyleManager.getInstance(project).reformat(statement);
PsiAssignmentExpression assignment = (PsiAssignmentExpression)statement.getExpression();
CommentTracker commentTracker = new CommentTracker();
PsiExpression initializer = var.getInitializer();
PsiExpression rExpression = RefactoringUtil.convertInitializerToNormalExpression(initializer, var.getType());
commentTracker.replace(assignment.getRExpression(), rExpression);
assignment.getRExpression().replace(rExpression);
PsiElement block = decl.getParent();
if (block instanceof PsiForStatement) {
@@ -188,7 +186,7 @@ public class SplitDeclarationAction extends PsiElementBaseIntentionAction {
return (PsiAssignmentExpression)((PsiExpressionStatement)block.addAfter(statement, decl)).getExpression();
}
finally {
commentTracker.deleteAndRestoreComments(initializer);
initializer.delete();
}
}
}

View File

@@ -2,8 +2,8 @@
import java.util.*;
class Test {
{
List<String> l //end line comment
;
l = new ArrayList<>();
List<String> l;
l = new ArrayList//end line comment
<>();
}
}