Migrate to FirEnumEntry

This commit is contained in:
simon.ogorodnik
2019-12-30 13:49:31 +03:00
committed by Mikhail Glukhikh
parent 5b5a5292b7
commit 65a44e2e20
27 changed files with 182 additions and 317 deletions
@@ -68,14 +68,7 @@ val FirClassSymbol<*>.superConeTypes
val FirClass<*>.superConeTypes get() = superTypeRefs.mapNotNull { it.coneTypeSafe<ConeClassLikeType>() }
fun FirRegularClass.collectEnumEntries(): Collection<FirPropertySymbol> {
fun FirRegularClass.collectEnumEntries(): Collection<FirEnumEntry> {
assert(classKind == ClassKind.ENUM_CLASS)
return declarations
.mapNotNull {
if (it !is FirProperty) return@mapNotNull null
val initializer = it.initializer
if (initializer == null || initializer !is FirAnonymousObject || initializer.classKind != ClassKind.ENUM_ENTRY)
return@mapNotNull null
return@mapNotNull it.symbol
}
return declarations.filterIsInstance<FirEnumEntry>()
}