From 499529bd074e200a0c17b846472de793c3ba045e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Christian=20Vonr=C3=BCti?= <christian@vonrueti.ch>
Date: Sat, 23 Apr 2022 10:14:57 +0200
Subject: [PATCH] Correct strikethrough when unwrapping lambdas KTIJ-21561

closes https://github.com/JetBrains/intellij-community/pull/1996

GitOrigin-RevId: 9a77468e51b338a8ab7d847fba3f80762421004b
---
 .../idea/codeInsight/unwrap/KotlinLambdaUnwrapper.java    | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/codeInsight/unwrap/KotlinLambdaUnwrapper.java b/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/codeInsight/unwrap/KotlinLambdaUnwrapper.java
index 38f55a41ba62..f2f72d4a4974 100644
--- a/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/codeInsight/unwrap/KotlinLambdaUnwrapper.java
+++ b/plugins/kotlin/idea/src/org/jetbrains/kotlin/idea/codeInsight/unwrap/KotlinLambdaUnwrapper.java
@@ -8,6 +8,8 @@ import com.intellij.util.IncorrectOperationException;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.kotlin.psi.*;
 
+import java.util.List;
+
 public class KotlinLambdaUnwrapper extends KotlinUnwrapRemoveBase {
     public KotlinLambdaUnwrapper(String key) {
         super(key);
@@ -53,4 +55,10 @@ public class KotlinLambdaUnwrapper extends KotlinUnwrapRemoveBase {
         context.extractFromBlock(body, enclosingExpression);
         context.delete(enclosingExpression);
     }
+
+    @Override
+    public PsiElement collectAffectedElements(@NotNull PsiElement e, @NotNull List<PsiElement> toExtract) {
+        super.collectAffectedElements(e, toExtract);
+        return getLambdaEnclosingElement((KtLambdaExpression) e);
+    }
 }