Fix brace node extraction from class initializer
This commit is contained in:
@@ -44,10 +44,9 @@ public class JetClassInitializer extends JetDeclarationImpl implements JetStatem
|
||||
return body;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
@Nullable
|
||||
public PsiElement getOpenBraceNode() {
|
||||
ASTNode openBraceNode = getNode().findChildByType(JetTokens.LBRACE);
|
||||
assert openBraceNode != null;
|
||||
return openBraceNode.getPsi();
|
||||
JetExpression body = getBody();
|
||||
return (body instanceof JetBlockExpression) ? ((JetBlockExpression) body).getLBrace() : null;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -35,7 +35,10 @@ public class JetDeclarationMover extends AbstractJetUpDownMover {
|
||||
new JetVisitorVoid() {
|
||||
@Override
|
||||
public void visitAnonymousInitializer(JetClassInitializer initializer) {
|
||||
memberSuspects.add(initializer.getOpenBraceNode());
|
||||
PsiElement brace = initializer.getOpenBraceNode();
|
||||
if (brace != null) {
|
||||
memberSuspects.add(brace);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
|
||||
Reference in New Issue
Block a user