Add "Update" action to PythonBytecodeToolWindow toolbar and refactor header UI

Author: meanmail

src/main/kotlin/dev/meanmail/tools/PythonBytecodeTool.kt

@@ -1,5 +1,7 @@
 package dev.meanmail.tools
 
+import com.intellij.openapi.actionSystem.ActionManager
+import com.intellij.openapi.actionSystem.DefaultActionGroup
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.util.Disposer
 import com.intellij.openapi.wm.ToolWindow
@@ -27,6 +29,13 @@ class PythonBytecodeTool : ToolWindowFactory {
         )
         toolWindow.contentManager.addContent(content)
 
+        // Add the update action to the tool window toolbar
+        val actionManager = ActionManager.getInstance()
+        val updateAction = actionManager.getAction("PythonBytecode.UpdateAction")
+        val actionGroup = DefaultActionGroup()
+        actionGroup.add(updateAction)
+        toolWindow.setTitleActions(listOf(updateAction))
+
         // Store the tool window component for later use
         toolWindowComponents[project] = pluginDevToolWindow
 

src/main/kotlin/dev/meanmail/tools/PythonBytecodeToolWindow.kt

@@ -19,7 +19,6 @@ import com.intellij.openapi.fileEditor.FileEditorManager
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.projectRoots.Sdk
 import com.intellij.openapi.roots.ProjectRootManager
-import com.intellij.ui.dsl.builder.panel
 import com.jetbrains.python.sdk.PythonSdkType
 import java.awt.Color
 import java.io.File
@@ -50,18 +49,10 @@ class PythonBytecodeToolWindow(private val project: Project) : Disposable, Pytho
 
     private fun createEditor(): EditorEx {
         val editorFactory = EditorFactory.getInstance()
-        val document = editorFactory.createDocument("Click Update button")
+        val document = editorFactory.createDocument("Click the Update button in the toolbar")
         val editor: EditorEx = editorFactory.createViewer(
             document, project, EditorKind.MAIN_EDITOR
         ) as EditorEx
-        editor.permanentHeaderComponent = panel {
-            row {
-                button("Update") {
-                    updateBytecode()
-                }
-            }
-        }
-        editor.headerComponent = editor.permanentHeaderComponent
 
         return editor
     }

src/main/kotlin/dev/meanmail/tools/UpdateBytecodeAction.kt

@@ -0,0 +1,22 @@
+package dev.meanmail.tools
+
+import com.intellij.openapi.actionSystem.AnAction
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.project.DumbAware
+
+/**
+ * Action to update the Python bytecode display.
+ * This action is available in the Python bytecode tool window.
+ */
+class UpdateBytecodeAction : AnAction(), DumbAware {
+    override fun actionPerformed(e: AnActionEvent) {
+        val project = e.project ?: return
+        val toolWindowComponent = PythonBytecodeTool.getToolWindowComponent(project) ?: return
+        toolWindowComponent.updateBytecode()
+    }
+
+    override fun update(e: AnActionEvent) {
+        // Enable the action only if a project is available
+        e.presentation.isEnabled = e.project != null
+    }
+}

src/main/resources/META-INF/plugin.xml

@@ -42,4 +42,13 @@ First release
                   topic="com.intellij.openapi.fileEditor.FileEditorManagerListener"/>
     </projectListeners>
 
+    <actions>
+        <action id="PythonBytecode.UpdateAction"
+                class="dev.meanmail.tools.UpdateBytecodeAction"
+                text="Update Bytecode"
+                description="Update the Python bytecode display"
+                icon="AllIcons.Actions.Refresh">
+        </action>
+    </actions>
+
 </idea-plugin>