make plugin ready for K2 mode

- starting from 242, the plugin works over AnalysisAPI which hides all details about kotlin engine used inside kotlin plugin itself

Author: akozlova

build.gradle.kts

@@ -111,17 +111,11 @@ val runWithCustomSandbox by intellijPlatformTesting.runIde.registering {
    }
 }
 
-//tasks.named<RunIdeTask>("runIde") {
-//   jvmArgumentProviders += CommandLineArgumentProvider {
-//      listOf("-Didea.kotlin.plugin.use.k2=true")
-//   }
-//}
-
-//tasks.test {
-//   jvmArgumentProviders += CommandLineArgumentProvider {
-//      listOf("-Didea.kotlin.plugin.use.k2=true")
-//   }
-//}
+val runWithK2Mode by intellijPlatformTesting.runIde.registering {
+   task {
+      jvmArgs = listOf("-Didea.kotlin.plugin.use.k2=true")
+   }
+}
 
 intellijPlatform {
    buildSearchableOptions = false

src/IC-243/kotlin/io/kotest/plugin/intellij/psi/superClasses.kt

@@ -0,0 +1,32 @@
+package io.kotest.plugin.intellij.psi
+
+import org.jetbrains.kotlin.analysis.api.analyze
+import org.jetbrains.kotlin.analysis.api.permissions.KaAllowAnalysisOnEdt
+import org.jetbrains.kotlin.analysis.api.permissions.allowAnalysisOnEdt
+import org.jetbrains.kotlin.analysis.api.types.symbol
+import org.jetbrains.kotlin.name.FqName
+import org.jetbrains.kotlin.name.StandardClassIds
+import org.jetbrains.kotlin.psi.KtClassOrObject
+
+/**
+ * Recursively returns the list of classes and interfaces extended or implemented by the class.
+ */
+@OptIn(KaAllowAnalysisOnEdt::class)
+fun KtClassOrObject.getAllSuperClasses(): List<FqName> {
+   return superTypeListEntries.mapNotNull { it.typeReference }
+      .flatMap { ref ->
+         // SurroundSelectionWithFunctionIntention.isAvailable is called in EDT before the intention is applied
+         // unfortunately API to avoid this was introduced in 23.2 only
+         // this we need to move intentions to the facade or accept EDT here until 23.2- are still supported
+         allowAnalysisOnEdt {
+            analyze(this) {
+               val kaType = ref.type
+               val superTypes = (kaType.allSupertypes(false) + kaType).toList()
+               superTypes.mapNotNull {
+                  val classId = it.symbol?.classId?.takeIf { id -> id != StandardClassIds.Any }
+                  classId?.asSingleFqName()
+               }
+            }
+         }
+      }
+}