Refactor ViewAsImageAction to use selection compat

Author: srwi

src/main/kotlin/com/github/srwi/pixellens/actions/ViewAsImageAction.kt

@@ -10,19 +10,20 @@ import com.intellij.notification.Notifications
 import com.intellij.openapi.actionSystem.ActionUpdateThread
 import com.intellij.openapi.actionSystem.AnAction
 import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.actionSystem.DataContext
 import com.intellij.openapi.progress.ProgressIndicator
 import com.intellij.openapi.progress.ProgressManager
 import com.intellij.openapi.progress.Task
 import com.intellij.openapi.project.Project
-import com.intellij.xdebugger.impl.ui.tree.actions.XDebuggerTreeActionBase
 import com.jetbrains.python.debugger.PyDebugValue
 import com.jetbrains.python.debugger.PyFrameAccessor
+import java.lang.reflect.Modifier
 import javax.swing.SwingUtilities
 
 class ViewAsImageAction : AnAction() {
     override fun actionPerformed(e: AnActionEvent) {
         val project = e.project ?: return
-        val value = XDebuggerTreeActionBase.getSelectedValue(e.dataContext) as PyDebugValue? ?: return
+        val value = getSelectedPyDebugValue(e.dataContext) ?: return
 
         if (!checkPythonCompatibility(value.frameAccessor, project)) {
             return
@@ -83,7 +84,7 @@ class ViewAsImageAction : AnAction() {
     override fun update(e: AnActionEvent) {
         super.update(e)
         try {
-            val value = XDebuggerTreeActionBase.getSelectedValue(e.dataContext) as PyDebugValue
+            val value = getSelectedPyDebugValue(e.dataContext) ?: throw IllegalStateException("No selected debug value")
             val imageProvider = ImageProviderFactory.getImageProvider(value.typeQualifier as String)
             e.presentation.isVisible = imageProvider.typeSupported(value)
             e.presentation.isEnabled = imageProvider.shapeSupported(value)
@@ -99,6 +100,10 @@ class ViewAsImageAction : AnAction() {
         return if (value.parent == null) value.name else value.evaluationExpression
     }
 
+    private fun getSelectedPyDebugValue(dataContext: DataContext): PyDebugValue? {
+        return XDebuggerSelectionCompat.getSelectedValue(dataContext) as? PyDebugValue
+    }
+
     private fun checkPythonCompatibility(frameAccessor: PyFrameAccessor, project: Project): Boolean {
         val pythonVersion = Python.getInterpreterVersion(frameAccessor)
 
@@ -139,4 +144,39 @@ class ViewAsImageAction : AnAction() {
 
         return isJupyter
     }
-}
+}
+
+private object XDebuggerSelectionCompat {
+    private val candidateClassNames = listOf(
+        "com.intellij.xdebugger.impl.ui.tree.actions.XDebuggerTreeBackendOnlyActionBase",
+        "com.intellij.xdebugger.impl.ui.tree.actions.XDebuggerTreeActionBase",
+    )
+
+    fun getSelectedValue(dataContext: DataContext): Any? {
+        for (className in candidateClassNames) {
+            val value = runCatching { invokeSelectionMethod(className, dataContext) }.getOrNull()
+            if (value != null) {
+                return value
+            }
+        }
+        return null
+    }
+
+    private fun invokeSelectionMethod(className: String, dataContext: DataContext): Any? {
+        val actionClass = Class.forName(className, false, XDebuggerSelectionCompat::class.java.classLoader)
+        val staticMethod = actionClass.methods.firstOrNull {
+            it.name == "getSelectedValue" && it.parameterCount == 1 && Modifier.isStatic(it.modifiers)
+        }
+        if (staticMethod != null) {
+            return staticMethod.invoke(null, dataContext)
+        }
+
+        val companionField = actionClass.getDeclaredField("Companion")
+        val companion = companionField.get(null)
+        val companionMethod = companion.javaClass.methods.firstOrNull {
+            it.name == "getSelectedValue" && it.parameterCount == 1
+        } ?: return null
+
+        return companionMethod.invoke(companion, dataContext)
+    }
+}