From 790f2d13ae4b5135a55b810ae25c700def812635 Mon Sep 17 00:00:00 2001
From: Jinseong Jeon <jsjeon@google.com>
Date: Sat, 12 Feb 2022 00:13:05 -0800
Subject: [PATCH] RAW FIR: record non-null value parameter of Enum#valueOf
 upfront

---
 .../src/org/jetbrains/kotlin/fir/EnumClassUtils.kt     | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/EnumClassUtils.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/EnumClassUtils.kt
index 14efa90779e..7e128f1a9e8 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/EnumClassUtils.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/EnumClassUtils.kt
@@ -26,7 +26,6 @@ import org.jetbrains.kotlin.fir.symbols.impl.FirNamedFunctionSymbol
 import org.jetbrains.kotlin.fir.symbols.impl.FirValueParameterSymbol
 import org.jetbrains.kotlin.fir.types.builder.buildResolvedTypeRef
 import org.jetbrains.kotlin.fir.types.impl.ConeClassLikeTypeImpl
-import org.jetbrains.kotlin.fir.types.impl.FirImplicitStringTypeRef
 import org.jetbrains.kotlin.name.CallableId
 import org.jetbrains.kotlin.name.FqName
 import org.jetbrains.kotlin.name.StandardClassIds
@@ -91,7 +90,14 @@ fun FirRegularClassBuilder.generateValueOfFunction(
             source = sourceElement
             origin = FirDeclarationOrigin.Source
             this.moduleData = moduleData
-            returnTypeRef = FirImplicitStringTypeRef(source)
+            returnTypeRef = buildResolvedTypeRef {
+                source = sourceElement
+                type = ConeClassLikeTypeImpl(
+                    ConeClassLikeLookupTagImpl(StandardClassIds.String),
+                    emptyArray(),
+                    isNullable = false
+                )
+            }
             name = DEFAULT_VALUE_PARAMETER
             this@vp.symbol = FirValueParameterSymbol(DEFAULT_VALUE_PARAMETER)
             isCrossinline = false