more config file extensions

metalbear-co · Aug 23, 2023 · 32d5ff0 · 32d5ff0
1 parent 6181c84
commit 32d5ff0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/modules/core/src/main/kotlin/com/metalbear/mirrord/MirrordConfigAPI.kt b/modules/core/src/main/kotlin/com/metalbear/mirrord/MirrordConfigAPI.kt
@@ -158,7 +158,7 @@ class MirrordConfigAPI(private val service: MirrordProjectService) {
     fun getDefaultConfig(): VirtualFile? {
         return getMirrordDir()
             ?.children
-            ?.filter { it.name.endsWith("mirrord.json") }
+            ?.filter { it.name.endsWith("mirrord.json") || it.name.endsWith("mirrord.yaml") || it.name.endsWith("mirrord.toml") }
             ?.minByOrNull { it.name }
     }
 

diff --git a/modules/core/src/main/kotlin/com/metalbear/mirrord/MirrordDropDown.kt b/modules/core/src/main/kotlin/com/metalbear/mirrord/MirrordDropDown.kt
@@ -193,7 +193,11 @@ class MirrordConfigIndex : ScalarIndexExtension<String>() {
 
     override fun getInputFilter(): FileBasedIndex.InputFilter {
         return FileBasedIndex.InputFilter {
-            it.isInLocalFileSystem && !it.isDirectory && it.path.endsWith("mirrord.json")
+            it.isInLocalFileSystem && !it.isDirectory && (
+                it.path.endsWith("mirrord.json") ||
+                    it.path.endsWith("mirrord.yaml") ||
+                    it.path.endsWith("mirrord.toml")
+                )
         }
     }