[kotlin] Provide CFG facade for the extract function refactoring

This is the first implementation of a control flow graph facade for the extract function IDE refactoring. The exact contents of 'KtDataFlowExitPointSnapshot' will be refined later. ^KT-65762 Fixed
2024-03-05 01:44:32 +09:00
parent b032e647f7
commit 0fec50135f
165 changed files with 4351 additions and 1 deletions
@@ -0,0 +1,54 @@
+/*
+ * Copyright 2010-2024 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.analysis.api.impl.base.test.cases.components.dataFlowInfoProvider
+
+import com.intellij.openapi.util.TextRange
+import com.intellij.psi.PsiElement
+import com.intellij.psi.util.PsiTreeUtil
+import org.jetbrains.kotlin.analysis.api.KtAnalysisNonPublicApi
+import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.components.stringRepresentation
+import org.jetbrains.kotlin.analysis.test.framework.base.AbstractAnalysisApiBasedTest
+import org.jetbrains.kotlin.analysis.test.framework.services.expressionMarkerProvider
+import org.jetbrains.kotlin.psi.KtExpression
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.psi.psiUtil.startOffset
+import org.jetbrains.kotlin.test.model.TestModule
+import org.jetbrains.kotlin.test.services.TestServices
+import org.jetbrains.kotlin.test.services.assertions
+
+abstract class AbstractExitPointSnapshotTest : AbstractAnalysisApiBasedTest() {
+    override fun doTestByMainFile(mainFile: KtFile, mainModule: TestModule, testServices: TestServices) {
+        val textRange = testServices.expressionMarkerProvider.getSelectedRange(mainFile)
+        val statements = findStatements(mainFile, textRange)
+
+        @OptIn(KtAnalysisNonPublicApi::class)
+        val actualText = analyseForTest(mainFile) {
+            val snapshot = getExitPointSnapshot(statements)
+            stringRepresentation(snapshot)
+        }
+
+        testServices.assertions.assertEqualsToTestDataFileSibling(actualText)
+    }
+
+    private fun findStatements(mainFile: KtFile, textRange: TextRange): List<KtExpression> {
+        var candidate = PsiTreeUtil.findElementOfClassAtOffset(mainFile, textRange.startOffset, KtExpression::class.java, true)
+            ?: error("Cannot find a starting element in range $textRange")
+
+        while (true) {
+            val parent = candidate.parent
+            if (parent is KtExpression && parent.textRange in textRange && parent.startOffset == candidate.startOffset) {
+                candidate = parent
+            } else {
+                break
+            }
+        }
+
+        return generateSequence<PsiElement>(candidate) { it.nextSibling }
+            .filterIsInstance<KtExpression>()
+            .filter { it.textRange in textRange }
+            .toList()
+    }
+}