Initial implementation of loop-to-call-chain intention

This commit is contained in:
Valentin Kipyatkov
2016-03-08 18:12:23 +01:00
parent c50cf13611
commit 12b1a99a6a
80 changed files with 3462 additions and 17 deletions
@@ -17,6 +17,7 @@
package org.jetbrains.kotlin.psi;
import com.intellij.lang.ASTNode;
import com.intellij.psi.PsiElement;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.kotlin.KtNodeTypes;
@@ -48,7 +49,12 @@ public class KtForExpression extends KtLoopExpression {
}
@Nullable @IfNotParsed
public ASTNode getInKeywordNode() {
return getNode().findChildByType(KtTokens.IN_KEYWORD);
public PsiElement getInKeyword() {
return findChildByType(KtTokens.IN_KEYWORD);
}
@NotNull
public PsiElement getForKeyword() {
return findChildByType(KtTokens.FOR_KEYWORD);
}
}