Implement "move class-level declaration up/down" action handler
This commit is contained in:
@@ -17,7 +17,10 @@
|
||||
package org.jetbrains.jet.lang.psi;
|
||||
|
||||
import com.intellij.lang.ASTNode;
|
||||
import com.intellij.psi.PsiElement;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.annotations.Nullable;
|
||||
import org.jetbrains.jet.lexer.JetTokens;
|
||||
|
||||
public class JetClassInitializer extends JetDeclarationImpl implements JetStatementExpression {
|
||||
public JetClassInitializer(@NotNull ASTNode node) {
|
||||
@@ -40,4 +43,11 @@ public class JetClassInitializer extends JetDeclarationImpl implements JetStatem
|
||||
assert body != null;
|
||||
return body;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public PsiElement getOpenBraceNode() {
|
||||
ASTNode openBraceNode = getNode().findChildByType(JetTokens.LBRACE);
|
||||
assert openBraceNode != null;
|
||||
return openBraceNode.getPsi();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -17,9 +17,11 @@
|
||||
package org.jetbrains.jet.lang.psi;
|
||||
|
||||
import com.intellij.lang.ASTNode;
|
||||
import com.intellij.psi.PsiElement;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.annotations.Nullable;
|
||||
import org.jetbrains.jet.JetNodeTypes;
|
||||
import org.jetbrains.jet.lexer.JetTokens;
|
||||
|
||||
public class JetClassObject extends JetDeclarationImpl implements JetStatementExpression {
|
||||
public JetClassObject(@NotNull ASTNode node) {
|
||||
@@ -41,4 +43,9 @@ public class JetClassObject extends JetDeclarationImpl implements JetStatementEx
|
||||
return (JetObjectDeclaration) findChildByType(JetNodeTypes.OBJECT_DECLARATION);
|
||||
}
|
||||
|
||||
@Nullable @IfNotParsed
|
||||
public PsiElement getClassKeywordNode() {
|
||||
ASTNode keywordNode = getNode().findChildByType(JetTokens.CLASS_KEYWORD);
|
||||
return keywordNode != null ? keywordNode.getPsi() : null;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user