only shows option for clj files & works for cljc/cljs

s-ajensen · Apr 25, 2024 · 5e413ae · 5e413ae
1 parent f874afa
commit 5e413ae
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 1 deletion.
diff --git a/.idea/artifacts/spec_mate.xml b/.idea/artifacts/spec_mate.xml
diff --git a/.idea/gradle.xml b/.idea/gradle.xml
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
diff --git a/src/main/kotlin/io/sajensen/specmate/GoToTestAction.kt b/src/main/kotlin/io/sajensen/specmate/GoToTestAction.kt
@@ -5,6 +5,7 @@ import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.fileEditor.FileEditorManager
 import com.intellij.openapi.ui.Messages
 import com.intellij.openapi.vfs.LocalFileSystem
+import com.intellij.openapi.vfs.VirtualFile
 
 class GoToTestAction : AnAction("Test") {
 
@@ -13,7 +14,7 @@ class GoToTestAction : AnAction("Test") {
         val currentFile = e.getData(com.intellij.openapi.actionSystem.CommonDataKeys.VIRTUAL_FILE)
 
         if (project != null && currentFile != null) {
-            val testFilePath = currentFile.path.replaceFirst("src", "spec").replace(".clj", "_spec.clj")
+            val testFilePath = getSpecPath(currentFile)
             val testFile = LocalFileSystem.getInstance().findFileByPath(testFilePath)
 
             if (testFile != null) {
@@ -28,4 +29,20 @@ class GoToTestAction : AnAction("Test") {
             }
         }
     }
+
+    private fun getSpecPath(file: VirtualFile): String {
+        val basePath = file.path.replaceFirst("src", "spec").substringBeforeLast(".")
+        val extension = file.extension
+        return basePath.plus("_spec.").plus(extension)
+    }
+
+    override fun update(e: AnActionEvent) {
+        super.update(e)
+
+        val project = e.project
+        val file = e.getData(com.intellij.openapi.actionSystem.CommonDataKeys.VIRTUAL_FILE)
+        val isClojureFile = file?.extension in listOf("clj", "cljc", "cljs")
+
+        e.presentation.isEnabledAndVisible = project != null && isClojureFile
+    }
 }