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

FIR IDE: find source of deserialized enum entry

Browse Source
This commit is contained in:
Jinseong Jeon
2021-10-12 14:21:42 -07:00
committed by Space
parent 8023e7fb02
commit 26673d2b08
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/FirIdeDeserializedDeclarationSourceProvider.kt
+12
View File
@@ -32,6 +32,7 @@ object FirIdeDeserializedDeclarationSourceProvider {
is FirClass -> provideSourceForClass(fir, project) is FirClass -> provideSourceForClass(fir, project)
is FirTypeAlias -> provideSourceForTypeAlias(fir, project) is FirTypeAlias -> provideSourceForTypeAlias(fir, project)
is FirConstructor -> provideSourceForConstructor(fir, project) is FirConstructor -> provideSourceForConstructor(fir, project)
is FirEnumEntry -> provideSourceForEnumEntry(fir, project)
else -> null else -> null
} }
} }
@@ -97,6 +98,17 @@ object FirIdeDeserializedDeclarationSourceProvider {
return null return null
} }
private fun provideSourceForEnumEntry(
enumEntry: FirEnumEntry,
project: Project
): PsiElement? {
val candidates = enumEntry.containingKtClass(project)?.body?.enumEntries
?.filter { it.name == enumEntry.name.asString() }
.orEmpty()
return candidates.firstOrNull(KtElement::isCompiled)
}
private fun FirDeclaration.scope(project: Project): GlobalSearchScope { private fun FirDeclaration.scope(project: Project): GlobalSearchScope {
return GlobalSearchScope.allScope(project) return GlobalSearchScope.allScope(project)
/* TODO: /* TODO: