Analysis API FIR: fix isDefaultEnumEntryMember

This commit is contained in:
Mikhail Glukhikh
2022-08-04 14:55:30 +02:00
committed by Space
parent 41b73abc49
commit 10ffdab438
@@ -5,6 +5,7 @@
package org.jetbrains.kotlin.analysis.api.fir.renderer
import org.jetbrains.kotlin.KtFakeSourceElementKind
import org.jetbrains.kotlin.analysis.api.components.KtDeclarationRendererOptions
import org.jetbrains.kotlin.analysis.utils.printer.PrettyPrinter
import org.jetbrains.kotlin.analysis.utils.printer.prettyPrint
@@ -356,14 +357,7 @@ internal class FirIdeRenderer private constructor(
private fun FirDeclaration.isDefaultEnumEntryMember(firClass: FirClass): Boolean {
if (firClass.classKind != ClassKind.ENUM_CLASS) return false
if (this is FirConstructor) return isPrimary && valueParameters.isEmpty()
if (this !is FirSimpleFunction) return false
if (name == StandardNames.ENUM_VALUES && valueParameters.isEmpty()) return true
if (name == StandardNames.ENUM_VALUE_OF) {
return valueParameters.singleOrNull()?.returnTypeRef?.isString == true
}
return false
return source?.kind == KtFakeSourceElementKind.EnumGeneratedDeclaration
}
private fun FirDeclaration.isDefaultPrimaryConstructor() =