feat: check for nullness of a PSI file in TestSparkAction.update

JetBrains-Research · Oct 4, 2024 · b40a077 · b40a077
1 parent 797180b
commit b40a077
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/main/kotlin/org/jetbrains/research/testspark/actions/TestSparkAction.kt b/src/main/kotlin/org/jetbrains/research/testspark/actions/TestSparkAction.kt
@@ -74,12 +74,18 @@ class TestSparkAction : AnAction() {
     /**
      * Updates the state of the action based on the provided event.
      *
-     * @param e the AnActionEvent object representing the event
+     * @param e `AnActionEvent` object representing the event
      */
     override fun update(e: AnActionEvent) {
-        val file = e.dataContext.getData(CommonDataKeys.PSI_FILE)!!
+        val file = e.dataContext.getData(CommonDataKeys.PSI_FILE)
+
+        if (file == null) {
+            e.presentation.isEnabledAndVisible = false
+            return
+        }
+
         val psiHelper = PsiHelperProvider.getPsiHelper(file)
-        e.presentation.isEnabledAndVisible = psiHelper != null && psiHelper.availableForGeneration(e)
+        e.presentation.isEnabledAndVisible = (psiHelper != null) && psiHelper.availableForGeneration(e)
     }
 
     /**