Refactor ViewAsImageAction to use XDebuggerTreeBackendOnlyActionBase

Author: srwi

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

@@ -10,20 +10,19 @@ 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.XDebuggerTreeBackendOnlyActionBase
 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 = getSelectedPyDebugValue(e.dataContext) ?: return
+        val value = XDebuggerTreeBackendOnlyActionBase.getSelectedValue(e.dataContext) as PyDebugValue? ?: return
 
         if (!checkPythonCompatibility(value.frameAccessor, project)) {
             return
@@ -84,7 +83,7 @@ class ViewAsImageAction : AnAction() {
     override fun update(e: AnActionEvent) {
         super.update(e)
         try {
-            val value = getSelectedPyDebugValue(e.dataContext) ?: throw IllegalStateException("No selected debug value")
+            val value = XDebuggerTreeBackendOnlyActionBase.getSelectedValue(e.dataContext) as PyDebugValue
             val imageProvider = ImageProviderFactory.getImageProvider(value.typeQualifier as String)
             e.presentation.isVisible = imageProvider.typeSupported(value)
             e.presentation.isEnabled = imageProvider.shapeSupported(value)
@@ -100,10 +99,6 @@ 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)
 
@@ -145,38 +140,3 @@ 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)
-    }
-}