From 81d829ce3bfd11410ad246c5d4dab051367693e4 Mon Sep 17 00:00:00 2001 From: Yan Zhulanow Date: Fri, 4 Aug 2023 19:04:24 +0900 Subject: [PATCH] [LL API] Collect context for enum entries --- .../analysis/low/level/api/fir/util/ContextCollector.kt | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt index 0f1c190ba83..5573cf56645 100644 --- a/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt +++ b/analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/ContextCollector.kt @@ -274,10 +274,16 @@ private class ContextCollectorVisitor( } override fun visitEnumEntry(enumEntry: FirEnumEntry) { + dumpContext(enumEntry.psi, ContextKind.SELF) + enumEntry.lazyResolveToPhase(FirResolvePhase.BODY_RESOLVE) context.forEnumEntry { - super.visitEnumEntry(enumEntry) + dumpContext(enumEntry.psi, ContextKind.BODY) + + onActive { + super.visitEnumEntry(enumEntry) + } } }