Register reference contributor for collection literals

#KT-18551 Fixed
This commit is contained in:
Mikhail Zarechenskiy
2017-06-20 23:57:59 +03:00
parent 78b238a05b
commit 837df66c19
7 changed files with 70 additions and 1 deletions
@@ -17,13 +17,21 @@
package org.jetbrains.kotlin.psi
import com.intellij.lang.ASTNode
import com.intellij.psi.PsiElement
import com.intellij.psi.util.PsiTreeUtil
import org.jetbrains.kotlin.lexer.KtTokens
class KtCollectionLiteralExpression(node: ASTNode) : KtExpressionImpl(node), KtReferenceExpression {
override fun <R, D> accept(visitor: KtVisitor<R, D>, data: D): R {
return visitor.visitCollectionLiteralExpression(this, data)
}
val leftBracket: PsiElement?
get() = findChildByType(KtTokens.LBRACKET)
val rightBracket: PsiElement?
get() = findChildByType(KtTokens.LBRACKET)
fun getInnerExpressions(): List<KtExpression> {
return PsiTreeUtil.getChildrenOfTypeAsList(this, KtExpression::class.java)
}