Minor. Extract PSICallResolver::createCallableReferenceKotlinCallArgument

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

@@ -383,6 +383,7 @@ class PSICallResolver(
         // todo may be for invoke for case variable + invoke we should create separate dynamicScope(by newCall for invoke)
         override val dynamicScope: MemberScope =
             dynamicCallableDescriptors.createDynamicDescriptorScope(context.call, context.scope.ownerDescriptor)
+
         // same for location
         override val location: LookupLocation = context.call.createLookupLocation()
 
@@ -690,6 +691,25 @@ class PSICallResolver(
             .expandContextForCatchClause(ktExpression)
 
         if (ktExpression is KtCallableReferenceExpression) {
+            return createCallableReferenceKotlinCallArgument(
+                context, ktExpression, startDataFlowInfo, valueArgument, argumentName,
+                outerCallContext
+            )
+        }
+
+        // argumentExpression instead of ktExpression is hack -- type info should be stored also for parenthesized expression
+        val typeInfo = expressionTypingServices.getTypeInfo(argumentExpression, context)
+        return createSimplePSICallArgument(context, valueArgument, typeInfo) ?: createParseErrorElement()
+    }
+
+    private fun createCallableReferenceKotlinCallArgument(
+        context: BasicCallResolutionContext,
+        ktExpression: KtCallableReferenceExpression,
+        startDataFlowInfo: DataFlowInfo,
+        valueArgument: ValueArgument,
+        argumentName: Name?,
+        outerCallContext: BasicCallResolutionContext
+    ): CallableReferenceKotlinCallArgumentImpl {
         checkNoSpread(outerCallContext, valueArgument)
 
         val expressionTypingContext = ExpressionTypingContext.newContext(context)
@@ -731,10 +751,6 @@ class PSICallResolver(
         )
     }
 
-        // argumentExpression instead of ktExpression is hack -- type info should be stored also for parenthesized expression
-        val typeInfo = expressionTypingServices.getTypeInfo(argumentExpression, context)
-        return createSimplePSICallArgument(context, valueArgument, typeInfo) ?: createParseErrorElement()
-    }
 
     private fun BasicCallResolutionContext.expandContextForCatchClause(ktExpression: Any): BasicCallResolutionContext {
         if (ktExpression !is KtExpression) return this