Kotlin 2.1.10+ prep (#1299)

* Kotlin 2.1.10+ prep

* More

* Formatting

Author: Bencodes

src/main/kotlin/io/bazel/kotlin/plugin/jdeps/k2/JdepsK2Utils.kt

@@ -1,12 +1,26 @@
 package io.bazel.kotlin.plugin.jdeps.k2
 
 import org.jetbrains.kotlin.descriptors.SourceElement
-import org.jetbrains.kotlin.fir.java.JavaBinarySourceElement
 import org.jetbrains.kotlin.load.kotlin.JvmPackagePartSource
 import org.jetbrains.kotlin.load.kotlin.KotlinJvmBinarySourceElement
 import org.jetbrains.kotlin.name.ClassId
 import org.jetbrains.kotlin.serialization.deserialization.descriptors.DeserializedContainerSource
 
+private object RefCache {
+  val jbseClass: Class<*>? by lazy {
+    runCatching {
+      Class.forName("org.jetbrains.kotlin.fir.java.JavaBinarySourceElement")
+    }.getOrNull()
+  }
+
+  val jbseGetJavaClassMethod by lazy {
+    jbseClass
+      ?.runCatching {
+        getMethod("getJavaClass")
+      }?.getOrNull()
+  }
+}
+
 /**
  * Returns whether class is coming from JVM runtime env. There is no need to track these classes.
  *
@@ -25,11 +39,16 @@ internal fun DeserializedContainerSource.classId(): ClassId? =
   }
 
 internal fun SourceElement.binaryClass(): String? =
-  when (this) {
-    is KotlinJvmBinarySourceElement -> binaryClass.location
-    is JvmPackagePartSource -> this.knownJvmBinaryClass?.location
-    is JavaBinarySourceElement -> this.javaClass.virtualFile.path
-    else -> null
+  if (this is KotlinJvmBinarySourceElement) {
+    binaryClass.location
+  } else if (this is JvmPackagePartSource) {
+    this.knownJvmBinaryClass?.location
+  } else if (RefCache.jbseClass != null && RefCache.jbseClass!!.isInstance(this)) {
+    val jClass = RefCache.jbseGetJavaClassMethod!!.invoke(this)
+    val virtualFile = jClass!!.javaClass.getMethod("getVirtualFile").invoke(jClass)
+    virtualFile.javaClass.getMethod("getPath").invoke(virtualFile) as? String
+  } else {
+    null
   }
 
 internal fun DeserializedContainerSource.binaryClass(): String? =