Put removing builder inference restrictions under the compiler flag

compiler/frontend/src/org/jetbrains/kotlin/resolve/calls/inference/BuilderInferenceSession.kt

@@ -131,6 +131,22 @@ class BuilderInferenceSession(
         if (skipCall(callInfo.callResolutionResult)) return
 
         commonCalls.add(callInfo)
+
+        val resultingDescriptor = callInfo.resolvedCall.resultingDescriptor
+
+        // This check is similar to one for old inference, see getCoroutineInferenceData() function
+        val checkCall = resultingDescriptor is LocalVariableDescriptor || anyReceiverContainStubType(resultingDescriptor)
+
+        if (!checkCall) return
+
+        val isApplicableCall = callComponents.statelessCallbacks.isApplicableCallForBuilderInference(
+            resultingDescriptor,
+            callComponents.languageVersionSettings
+        )
+
+        if (!isApplicableCall) {
+            hasInapplicableCall = true
+        }
     }
 
     private fun anyReceiverContainStubType(descriptor: CallableDescriptor): Boolean {

compiler/frontend/src/org/jetbrains/kotlin/resolve/calls/inference/CoroutineInferenceUtil.kt

@@ -299,6 +299,8 @@ class CoroutineInferenceSupport(
 private fun KotlinType.containsTypeTemplate() = contains { it is TypeTemplate || it is StubTypeForBuilderInference }
 
 fun isApplicableCallForBuilderInference(descriptor: CallableDescriptor, languageVersionSettings: LanguageVersionSettings): Boolean {
+    if (languageVersionSettings.supportsFeature(LanguageFeature.StableBuilderInference)) return true
+
     if (!languageVersionSettings.supportsFeature(LanguageFeature.ExperimentalBuilderInference)) {
         return isGoodCallForOldCoroutines(descriptor)
     }

compiler/frontend/src/org/jetbrains/kotlin/resolve/calls/tower/KotlinResolutionStatelessCallbacksImpl.kt

@@ -95,6 +95,13 @@ class KotlinResolutionStatelessCallbacksImpl(
     override fun isCoroutineCall(argument: KotlinCallArgument, parameter: ValueParameterDescriptor): Boolean =
         isCoroutineCallWithAdditionalInference(parameter, argument.psiCallArgument.valueArgument, languageVersionSettings)
 
+    override fun isApplicableCallForBuilderInference(
+        descriptor: CallableDescriptor,
+        languageVersionSettings: LanguageVersionSettings,
+    ): Boolean {
+        return org.jetbrains.kotlin.resolve.calls.inference.isApplicableCallForBuilderInference(descriptor, languageVersionSettings)
+    }
+
     override fun isOldIntersectionIsEmpty(types: Collection<KotlinType>): Boolean {
         return TypeIntersector.intersectTypes(types) == null
     }