SONARKT-400 Migrate LiftReturnStatementCheck to kotlin-analysis-api

Author: Godin

sonar-kotlin-api/src/main/java/org/sonarsource/kotlin/api/checks/ApiExtensions.kt

@@ -21,6 +21,7 @@ import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiWhiteSpace
 import com.intellij.psi.impl.source.tree.LeafPsiElement
 import com.intellij.psi.util.PsiTreeUtil
+import org.jetbrains.kotlin.analysis.api.KaIdeApi
 import org.jetbrains.kotlin.analysis.api.resolution.KaCall
 import org.jetbrains.kotlin.analysis.api.resolution.KaExplicitReceiverValue
 import org.jetbrains.kotlin.analysis.api.resolution.KaFunctionCall
@@ -883,8 +884,9 @@ fun SensorContext.hasCacheEnabled(): Boolean {
         isCacheEnabled
 }
 
-fun KtWhenExpression.isExhaustive(context: KotlinFileContext): Boolean {
-    return entries.any { it.isElse } || context.bindingContext[BindingContext.EXHAUSTIVE_WHEN, this] == true
+@OptIn(KaIdeApi::class)
+fun KtWhenExpression.isExhaustive(): Boolean = withKaSession {
+    return entries.any { it.isElse } || computeMissingCases().isEmpty()
 }
 
 val PropertyDescriptor.unwrappedGetMethod: FunctionDescriptor?

sonar-kotlin-checks/src/main/java/org/sonarsource/kotlin/checks/LiftReturnStatementCheck.kt

@@ -27,7 +27,6 @@ import org.sonarsource.kotlin.api.checks.AbstractCheck
 import org.sonarsource.kotlin.api.checks.isExhaustive
 import org.sonarsource.kotlin.api.frontend.KotlinFileContext
 
-@org.sonarsource.kotlin.api.frontend.K1only
 @Rule(key = "S6510")
 class LiftReturnStatementCheck : AbstractCheck() {
 
@@ -47,7 +46,7 @@ class LiftReturnStatementCheck : AbstractCheck() {
             isReturnOrReturnBlock(it.expression!!)
         }
 
-        if (isAllBranchesReturn && expression.isExhaustive(context)) {
+        if (isAllBranchesReturn && expression.isExhaustive()) {
             reportIssue(expression.whenKeyword, context)
         }
     }