Adjust PSI usages to new labels syntax

This commit is contained in:
Denis Zharkov
2015-04-27 12:58:04 +03:00
parent 241860d510
commit 89c72d9400
6 changed files with 6 additions and 12 deletions
@@ -25,11 +25,7 @@ open public class JetExpressionWithLabel(node: ASTNode) : JetExpressionImpl(node
findChildByType<JetContainerNode>(JetNodeTypes.LABEL_QUALIFIER)?.
findChildByType(JetNodeTypes.LABEL) as? JetSimpleNameExpression
public fun getLabelName(): String? {
val labelElement = getTargetLabel()
assert(labelElement == null || labelElement.getText().startsWith("@"))
return labelElement?.getText()?.substring(1)
}
public fun getLabelName(): String? = getTargetLabel()?.getIdentifier()?.getText()
override fun <R, D> accept(visitor: JetVisitor<R, D>, data: D) = visitor.visitExpressionWithLabel(this, data)
}
@@ -21,5 +21,5 @@ import org.jetbrains.kotlin.lexer.JetTokens
import com.intellij.psi.PsiElement
public class JetLabelReferenceExpression(node: ASTNode) : JetSimpleNameExpressionImpl(node) {
public override fun getReferencedNameElement() = (findChildByType(JetTokens.LABEL_IDENTIFIER): PsiElement?) ?: this
public override fun getReferencedNameElement() = getIdentifier() ?: this
}
@@ -839,8 +839,7 @@ public class JetPsiUtil {
}
public static boolean isLabelIdentifierExpression(PsiElement element) {
return element instanceof JetSimpleNameExpression &&
((JetSimpleNameExpression) element).getReferencedNameElementType() == JetTokens.LABEL_IDENTIFIER;
return element instanceof JetLabelReferenceExpression;
}
@Nullable