Override update thread

Drjacky · Dec 2, 2023 · dbf5ab2 · dbf5ab2
1 parent 8d1efad
commit dbf5ab2
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/src/main/kotlin/com/github/drjacky/avocado/actions/AvocadoSizeItRightClickAction.kt b/src/main/kotlin/com/github/drjacky/avocado/actions/AvocadoSizeItRightClickAction.kt
@@ -1,16 +1,16 @@
+import com.intellij.openapi.actionSystem.ActionUpdateThread
 import com.intellij.openapi.actionSystem.AnAction
 import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.actionSystem.CommonDataKeys
-import com.intellij.openapi.module.ModuleManager
 import com.intellij.openapi.project.Project
-import com.intellij.openapi.roots.ModuleRootManager
 import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiFile
 import com.intellij.psi.xml.XmlFile
 import java.io.BufferedReader
 import java.io.File
 import java.io.IOException
 import java.io.InputStreamReader
+import javax.swing.SwingUtilities
 
 class AvocadoSizeItRightClickAction : AnAction() {
 
@@ -46,8 +46,15 @@ class AvocadoSizeItRightClickAction : AnAction() {
     }
 
     override fun update(e: AnActionEvent) {
-        val psiFile = e.getData(CommonDataKeys.PSI_FILE)
-        e.presentation.isEnabledAndVisible = isXmlFileInDrawableFolder(psiFile)
+        super.update(e)
+        SwingUtilities.invokeLater {
+            val psiFile = e.getData(CommonDataKeys.PSI_FILE)
+            e.presentation.isEnabledAndVisible = isXmlFileInDrawableFolder(psiFile)
+        }
+    }
+
+    override fun getActionUpdateThread(): ActionUpdateThread {
+        return ActionUpdateThread.EDT
     }
 
     private fun isXmlFileInDrawableFolder(psiFile: PsiFile?): Boolean {