[FIR] Introduce common super type for context dependent resolution modes

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/ResolutionMode.kt

@@ -12,21 +12,16 @@ import org.jetbrains.kotlin.fir.types.*
 import org.jetbrains.kotlin.fir.types.builder.buildResolvedTypeRef
 
 sealed class ResolutionMode(val forceFullCompletion: Boolean) {
-    object ContextDependent : ResolutionMode(forceFullCompletion = false) {
+    sealed class ContextDependent : ResolutionMode(forceFullCompletion = false) {
-        override fun toString(): String = "ContextDependent"
+        companion object Default : ContextDependent() {
+            override fun toString(): String = "ContextDependent"
+        }
+
+        data object Delegate : ContextDependent()
     }
 
-    object ContextDependentDelegate : ResolutionMode(forceFullCompletion = false) {
+    data object ContextIndependent : ResolutionMode(forceFullCompletion = true)
-        override fun toString(): String = "ContextDependentDelegate"
+    data object ReceiverResolution : ResolutionMode(forceFullCompletion = true)
-    }
-
-    object ContextIndependent : ResolutionMode(forceFullCompletion = true) {
-        override fun toString(): String = "ContextIndependent"
-    }
-
-    object ReceiverResolution : ResolutionMode(forceFullCompletion = true) {
-        override fun toString(): String = "ReceiverResolution"
-    }
 
     class WithExpectedType(
         val expectedTypeRef: FirResolvedTypeRef,

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/inference/FirCallCompleter.kt

@@ -112,7 +112,7 @@ class FirCallCompleter(
                 runCompletionForCall(candidate, completionMode, call, initialType, analyzer)
 
                 // Add top-level delegate call as partially resolved to inference session
-                if (resolutionMode is ResolutionMode.ContextDependentDelegate) {
+                if (resolutionMode is ResolutionMode.ContextDependent.Delegate) {
                     require(inferenceSession is FirDelegatedPropertyInferenceSession)
                     inferenceSession.addPartiallyResolvedCall(call)
                 }

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/body/resolve/BodyResolveContext.kt

@@ -685,7 +685,7 @@ class BodyResolveContext(
         mode: ResolutionMode,
         f: () -> T
     ): T {
-        require(mode !is ResolutionMode.ContextDependent && mode !is ResolutionMode.ContextDependentDelegate)
+        require(mode !is ResolutionMode.ContextDependent)
         if (mode !is ResolutionMode.LambdaResolution) {
             storeContextForAnonymousFunction(anonymousFunction)
         }

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/body/resolve/FirDeclarationsResolveTransformer.kt

@@ -271,10 +271,10 @@ open class FirDeclarationsResolveTransformer(
             dataFlowAnalyzer.enterDelegateExpression()
             // Resolve delegate expression, after that, delegate will contain either expr.provideDelegate or expr
             if (property.isLocal) {
-                property.transformDelegate(transformer, ResolutionMode.ContextDependentDelegate)
+                property.transformDelegate(transformer, ResolutionMode.ContextDependent.Delegate)
             } else {
                 context.forPropertyInitializer {
-                    property.transformDelegate(transformer, ResolutionMode.ContextDependentDelegate)
+                    property.transformDelegate(transformer, ResolutionMode.ContextDependent.Delegate)
                 }
             }
 
@@ -326,8 +326,8 @@ open class FirDeclarationsResolveTransformer(
         data: ResolutionMode,
     ): FirStatement {
         // First, resolve delegate expression in dependent context, and add potentially partially resolved call to inference session
-        // (that is why we use ContextDependentDelegate instead of plain ContextDependent)
+        // (that is why we use ContextDependent.Delegate instead of plain ContextDependent)
-        val delegateExpression = wrappedDelegateExpression.expression.transformSingle(transformer, ResolutionMode.ContextDependentDelegate)
+        val delegateExpression = wrappedDelegateExpression.expression.transformSingle(transformer, ResolutionMode.ContextDependent.Delegate)
             .transformSingle(components.integerLiteralAndOperatorApproximationTransformer, null)
 
         // Second, replace result type of delegate expression with stub type if delegate not yet resolved
@@ -773,7 +773,7 @@ open class FirDeclarationsResolveTransformer(
         }
 
         return when (data) {
-            is ResolutionMode.ContextDependent, is ResolutionMode.ContextDependentDelegate -> {
+            is ResolutionMode.ContextDependent -> {
                 context.storeContextForAnonymousFunction(anonymousFunction)
                 anonymousFunction
             }

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/body/resolve/FirExpressionsResolveTransformer.kt

@@ -498,7 +498,10 @@ open class FirExpressionsResolveTransformer(transformer: FirAbstractBodyResolveT
         )
 
         val approximationIsNeeded =
-            resolutionMode !is ResolutionMode.ReceiverResolution && resolutionMode !is ResolutionMode.ContextDependent
+            resolutionMode !is ResolutionMode.ReceiverResolution &&
+                    (resolutionMode !is ResolutionMode.ContextDependent ||
+                            // TODO check why we need a special case for ContextDependent.Delegate
+                            resolutionMode is ResolutionMode.ContextDependent.Delegate)
 
         val integerOperatorCall = buildIntegerLiteralOperatorCall {
             source = originalCall.source
@@ -1075,17 +1078,14 @@ open class FirExpressionsResolveTransformer(transformer: FirAbstractBodyResolveT
 
         transformedLHS?.let { callableReferenceAccess.replaceExplicitReceiver(transformedLHS) }
 
-        return when (data) {
+        // TODO check why we need a special case for ContextDependent.Delegate
-            is ResolutionMode.ContextDependent -> {
+        return if (data is ResolutionMode.ContextDependent && data !is ResolutionMode.ContextDependent.Delegate) {
-                context.storeCallableReferenceContext(callableReferenceAccess)
+            context.storeCallableReferenceContext(callableReferenceAccess)
-                callableReferenceAccess
+            callableReferenceAccess
-            }
+        } else {
-
+            components.syntheticCallGenerator.resolveCallableReferenceWithSyntheticOuterCall(
-            else -> {
+                callableReferenceAccess, data.expectedType, resolutionContext,
-                components.syntheticCallGenerator.resolveCallableReferenceWithSyntheticOuterCall(
+            ) ?: callableReferenceAccess
-                    callableReferenceAccess, data.expectedType, resolutionContext,
-                ) ?: callableReferenceAccess
-            }
         }.also {
             dataFlowAnalyzer.exitCallableReference(it)
         }
@@ -1636,7 +1636,7 @@ open class FirExpressionsResolveTransformer(transformer: FirAbstractBodyResolveT
 
     override fun transformArrayOfCall(arrayOfCall: FirArrayOfCall, data: ResolutionMode): FirStatement =
         whileAnalysing(session, arrayOfCall) {
-            if (data is ResolutionMode.ContextDependent) {
+            if (data is ResolutionMode.ContextDependent.Default) {
                 arrayOfCall.transformChildren(transformer, data)
                 arrayOfCall
             } else if (data is ResolutionMode.WithExpectedType && !data.expectedTypeRef.coneType.isPrimitiveOrUnsignedArray) {