KTIJ-20826 Approximate builder inference stub types as type parameters

This is a temporary fix to avoid crashes in FIR IDE

Of all stub types, only `ConeStubTypeForChainInference` is handled now,
so we can catch the other possible errors with stub types (in the
delegates' inference, for example)

The fix introduced in this commit should be reworked when KT-50916 is
fixed

^KTIJ-20826 Fixed

analysis/analysis-api-fir/src/org/jetbrains/kotlin/analysis/api/fir/KtSymbolByFirBuilder.kt

@@ -30,10 +30,12 @@ import org.jetbrains.kotlin.fir.declarations.impl.FirOuterClassTypeParameterRef
 import org.jetbrains.kotlin.fir.java.declarations.FirJavaField
 import org.jetbrains.kotlin.fir.resolve.getContainingClass
 import org.jetbrains.kotlin.fir.resolve.getSymbolByLookupTag
+import org.jetbrains.kotlin.fir.resolve.inference.ConeTypeParameterBasedTypeVariable
 import org.jetbrains.kotlin.fir.resolve.originalConstructorIfTypeAlias
 import org.jetbrains.kotlin.fir.resolve.providers.symbolProvider
 import org.jetbrains.kotlin.fir.resolve.substitution.ConeSubstitutor
 import org.jetbrains.kotlin.fir.resolve.substitution.ConeSubstitutorByMap
+import org.jetbrains.kotlin.fir.scopes.impl.toConeType
 import org.jetbrains.kotlin.fir.symbols.ConeClassLikeLookupTag
 import org.jetbrains.kotlin.fir.symbols.ConeTypeParameterLookupTag
 import org.jetbrains.kotlin.fir.symbols.FirBasedSymbol
@@ -438,6 +440,15 @@ internal class KtSymbolByFirBuilder private constructor(
                     is ConeDefinitelyNotNullType -> KtFirDefinitelyNotNullType(coneType, token, this@KtSymbolByFirBuilder)
                     is ConeCapturedType -> KtFirCapturedType(coneType, token, this@KtSymbolByFirBuilder)
                     is ConeIntegerLiteralType -> KtFirIntegerLiteralType(coneType, token, this@KtSymbolByFirBuilder)
+                    is ConeStubTypeForChainInference -> {
+                        // TODO this is a temporary hack to prevent FIR IDE from crashing on builder inference, see KT-50916
+                        val typeVariable = coneType.constructor.variable as? ConeTypeParameterBasedTypeVariable
+                        val typeParameterSymbol = typeVariable?.typeParameterSymbol ?: throwUnexpectedElementError(coneType)
+                        val coneTypeParameterType = typeParameterSymbol.toConeType() as ConeTypeParameterType
+
+                        KtFirTypeParameterType(coneTypeParameterType, token, this@KtSymbolByFirBuilder)
+                    }
+
                     else -> throwUnexpectedElementError(coneType)
                 }
             }

analysis/analysis-api-fir/tests/org/jetbrains/kotlin/analysis/api/fir/fir/FirResolveCallTestGenerated.java

@@ -172,6 +172,12 @@ public class FirResolveCallTestGenerated extends AbstractResolveCallTest {
         runTest("analysis/analysis-api/testData/components/callResolver/resolveCall/arrayOfInAnnotation.kt");
     }
 
+    @Test
+    @TestMetadata("builderInference.kt")
+    public void testBuilderInference() throws Exception {
+        runTest("analysis/analysis-api/testData/components/callResolver/resolveCall/builderInference.kt");
+    }
+
     @Test
     @TestMetadata("calleeExpressionOfImplicitInvoke.kt")
     public void testCalleeExpressionOfImplicitInvoke() throws Exception {