diff --git a/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToElvisIntention.kt b/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToElvisIntention.kt index 8ce9e7435a6..db45470b06a 100644 --- a/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToElvisIntention.kt +++ b/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToElvisIntention.kt @@ -33,7 +33,9 @@ import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode class IfThenToElvisInspection : IntentionBasedInspection( IfThenToElvisIntention::class, { it -> it.isUsedAsExpression(it.analyze(BodyResolveMode.PARTIAL)) } -) +) { + override fun inspectionTarget(element: KtIfExpression) = element.ifKeyword +} class IfThenToElvisIntention : SelfTargetingOffsetIndependentIntention( KtIfExpression::class.java, diff --git a/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToSafeAccessIntention.kt b/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToSafeAccessIntention.kt index eab3fcaa77a..b9cbf698540 100644 --- a/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToSafeAccessIntention.kt +++ b/idea/src/org/jetbrains/kotlin/idea/intentions/branchedTransformations/intentions/IfThenToSafeAccessIntention.kt @@ -17,6 +17,7 @@ package org.jetbrains.kotlin.idea.intentions.branchedTransformations.intentions import com.intellij.openapi.editor.Editor +import com.intellij.psi.PsiElement import org.jetbrains.kotlin.idea.core.replaced import org.jetbrains.kotlin.idea.inspections.IntentionBasedInspection import org.jetbrains.kotlin.idea.intentions.SelfTargetingOffsetIndependentIntention @@ -24,7 +25,9 @@ import org.jetbrains.kotlin.idea.intentions.branchedTransformations.* import org.jetbrains.kotlin.idea.util.application.runWriteAction import org.jetbrains.kotlin.psi.* -class IfThenToSafeAccessInspection : IntentionBasedInspection(IfThenToSafeAccessIntention::class) +class IfThenToSafeAccessInspection : IntentionBasedInspection(IfThenToSafeAccessIntention::class) { + override fun inspectionTarget(element: KtIfExpression) = element.ifKeyword +} class IfThenToSafeAccessIntention : SelfTargetingOffsetIndependentIntention( KtIfExpression::class.java, "Replace 'if' expression with safe access expression"