From 5a7f74c250f0cd354ec6ca34e315dad0686a7d6d Mon Sep 17 00:00:00 2001
From: Valentin Kipyatkov <valentin@jetbrains.com>
Date: Mon, 13 Apr 2020 09:35:23 +0300
Subject: [PATCH] Extracted method

---
 .../kotlin/idea/slicer/OutflowSlicer.kt       | 51 ++++++++++---------
 1 file changed, 28 insertions(+), 23 deletions(-)

diff --git a/idea/src/org/jetbrains/kotlin/idea/slicer/OutflowSlicer.kt b/idea/src/org/jetbrains/kotlin/idea/slicer/OutflowSlicer.kt
index eb612ca2727..4415cba799d 100644
--- a/idea/src/org/jetbrains/kotlin/idea/slicer/OutflowSlicer.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/slicer/OutflowSlicer.kt
@@ -18,6 +18,7 @@ import org.jetbrains.kotlin.cfg.pseudocode.instructions.jumps.ReturnValueInstruc
 import org.jetbrains.kotlin.descriptors.CallableDescriptor
 import org.jetbrains.kotlin.descriptors.DeclarationDescriptorWithSource
 import org.jetbrains.kotlin.descriptors.FunctionDescriptor
+import org.jetbrains.kotlin.descriptors.ValueParameterDescriptor
 import org.jetbrains.kotlin.idea.caches.resolve.analyze
 import org.jetbrains.kotlin.idea.caches.resolve.resolveToDescriptorIfAny
 import org.jetbrains.kotlin.idea.findUsages.handlers.SliceUsageProcessor
@@ -177,30 +178,10 @@ class OutflowSlicer(
                         if (parameter != null) {
                             parameter.passToProcessorInCallMode(instruction.element)
                         } else {
-                            val function = parameterDescriptor.containingDeclaration as? FunctionDescriptor ?: return@processPseudocodeUsages
+                            val function = parameterDescriptor.containingDeclaration as? FunctionDescriptor
+                                ?: return@processPseudocodeUsages
                             if (function.isImplicitInvokeFunction()) {
-                                val receiverPseudoValue = instruction.receiverValues.entries.singleOrNull()?.key
-                                    ?: return@processPseudocodeUsages
-                                val receiverExpression = receiverPseudoValue.element as? KtExpression ?: return@processPseudocodeUsages
-                                val bindingContext = receiverExpression.analyze(BodyResolveMode.PARTIAL)
-                                var receiverType = bindingContext.getType(receiverExpression)
-                                var receiver: PsiElement = receiverExpression
-                                if (receiverType == null && receiverExpression is KtReferenceExpression) {
-                                    val targetDescriptor = bindingContext[BindingContext.REFERENCE_TARGET, receiverExpression]
-                                    if (targetDescriptor is CallableDescriptor) {
-                                        receiverType = targetDescriptor.returnType
-                                        receiver = targetDescriptor.originalSource.getPsi() ?: return@processPseudocodeUsages
-                                    }
-                                }
-                                if (receiverType == null || !receiverType.isFunctionType) return@processPseudocodeUsages
-                                val isExtension = receiverType.isExtensionFunctionType
-                                val shift = if (isExtension) 1 else 0
-                                val argumentIndex = parameterDescriptor.index - shift
-                                val newMode = if (argumentIndex >= 0)
-                                    mode.withBehaviour(LambdaArgumentInflowBehaviour(argumentIndex))
-                                else
-                                    mode.withBehaviour(LambdaReceiverInflowBehaviour)
-                                receiver.passToProcessor(newMode)
+                                processImplicitInvokeCall(instruction, parameterDescriptor)
                             }
                         }
                     }
@@ -230,6 +211,30 @@ class OutflowSlicer(
         }
     }
 
+    private fun processImplicitInvokeCall(instruction: CallInstruction, parameterDescriptor: ValueParameterDescriptor) {
+        val receiverPseudoValue = instruction.receiverValues.entries.singleOrNull()?.key ?: return
+        val receiverExpression = receiverPseudoValue.element as? KtExpression ?: return
+        val bindingContext = receiverExpression.analyze(BodyResolveMode.PARTIAL)
+        var receiverType = bindingContext.getType(receiverExpression)
+        var receiver: PsiElement = receiverExpression
+        if (receiverType == null && receiverExpression is KtReferenceExpression) {
+            val targetDescriptor = bindingContext[BindingContext.REFERENCE_TARGET, receiverExpression]
+            if (targetDescriptor is CallableDescriptor) {
+                receiverType = targetDescriptor.returnType
+                receiver = targetDescriptor.originalSource.getPsi() ?: return
+            }
+        }
+        if (receiverType == null || !receiverType.isFunctionType) return
+        val isExtension = receiverType.isExtensionFunctionType
+        val shift = if (isExtension) 1 else 0
+        val argumentIndex = parameterDescriptor.index - shift
+        val newMode = if (argumentIndex >= 0)
+            mode.withBehaviour(LambdaArgumentInflowBehaviour(argumentIndex))
+        else
+            mode.withBehaviour(LambdaReceiverInflowBehaviour)
+        receiver.passToProcessor(newMode)
+    }
+
     private fun processDereferenceIfNeeded(
         expression: KtExpression,
         pseudoValue: PseudoValue,