azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[FIR] Use stubs to get context receiver list and name of label

Browse Source 
Thirteenth step for ^KT-52615

Merge-request: KT-MR-8911
Merged-by: Egor Kulikov <Egor.Kulikov@jetbrains.com>
This commit is contained in:
Egor Kulikov
2023-02-21 10:44:02 +00:00
committed by Space Team
parent 1dd9751f99
commit 694ef8e52d
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/psi/src/org/jetbrains/kotlin/psi/KtContextReceiver.kt
+4 -1
View File
@@ -24,7 +24,10 @@ class KtContextReceiver : KtElementImplStub<KotlinContextReceiverStub> {
return targetLabel()?.getReferencedName() return targetLabel()?.getReferencedName()
} }
fun labelNameAsName(): Name? = targetLabel()?.getReferencedNameAsName() fun labelNameAsName(): Name? {
stub?.let { stub -> return stub.getLabel()?.let { Name.identifier(it) } }
return targetLabel()?.getReferencedNameAsName()
}
fun typeReference(): KtTypeReference? = getStubOrPsiChild(KtStubElementTypes.TYPE_REFERENCE) fun typeReference(): KtTypeReference? = getStubOrPsiChild(KtStubElementTypes.TYPE_REFERENCE)
compiler/psi/src/org/jetbrains/kotlin/psi/KtContextReceiverList.kt
+1 -2
View File
@@ -6,7 +6,6 @@
package org.jetbrains.kotlin.psi package org.jetbrains.kotlin.psi
import com.intellij.lang.ASTNode import com.intellij.lang.ASTNode
import org.jetbrains.kotlin.KtNodeTypes
import org.jetbrains.kotlin.psi.stubs.KotlinPlaceHolderStub import org.jetbrains.kotlin.psi.stubs.KotlinPlaceHolderStub
import org.jetbrains.kotlin.psi.stubs.elements.KtStubElementTypes import org.jetbrains.kotlin.psi.stubs.elements.KtStubElementTypes
@@ -18,7 +17,7 @@ class KtContextReceiverList : KtElementImplStub<KotlinPlaceHolderStub<KtContextR
return visitor.visitContextReceiverList(this, data) return visitor.visitContextReceiverList(this, data)
} }
fun contextReceivers(): List<KtContextReceiver> = findChildrenByType(KtNodeTypes.CONTEXT_RECEIVER) fun contextReceivers(): List<KtContextReceiver> = getStubOrPsiChildrenAsList(KtStubElementTypes.CONTEXT_RECEIVER)
fun typeReferences(): List<KtTypeReference> = contextReceivers().mapNotNull { it.typeReference() } fun typeReferences(): List<KtTypeReference> = contextReceivers().mapNotNull { it.typeReference() }