[FIR IDE] Fix structure element non analyzable elements recording for analyzable properties

This commit is contained in:
Igor Yakovlev
2021-01-21 17:19:11 +03:00
parent 5480faf5c5
commit 65b6a95f64
@@ -181,6 +181,13 @@ internal class NonReanalyzableDeclarationStructureElement(
companion object {
private val recorder = object : FirElementsRecorder() {
override fun visitProperty(property: FirProperty, data: MutableMap<KtElement, FirElement>) {
val psi = property.psi as? KtProperty ?: return super.visitProperty(property, data)
if (!FileElementFactory.isReanalyzableContainer(psi) || KtPsiUtil.isLocal(psi)) {
super.visitProperty(property, data)
}
}
override fun visitSimpleFunction(simpleFunction: FirSimpleFunction, data: MutableMap<KtElement, FirElement>) {
val psi = simpleFunction.psi as? KtNamedFunction ?: return super.visitSimpleFunction(simpleFunction, data)
if (!FileElementFactory.isReanalyzableContainer(psi) || KtPsiUtil.isLocal(psi)) {