Fix #409

JetBrains · Oct 23, 2023 · a597be8 · a597be8
1 parent 2cfa461
commit a597be8
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/main/kotlin/org/arend/project/ArendProjectImportProvider.kt b/src/main/kotlin/org/arend/project/ArendProjectImportProvider.kt
@@ -1,6 +1,8 @@
 package org.arend.project
 
+import com.intellij.openapi.project.Project
 import com.intellij.openapi.vfs.VirtualFile
+import com.intellij.openapi.vfs.isFile
 import com.intellij.projectImport.ProjectImportBuilder
 import com.intellij.projectImport.ProjectImportProvider
 import org.arend.util.FileUtils
@@ -9,6 +11,16 @@ class ArendProjectImportProvider : ProjectImportProvider() {
     override fun doGetBuilder(): ProjectImportBuilder<*> =
         ProjectImportBuilder.EXTENSIONS_POINT_NAME.findExtensionOrFail(ArendProjectImportBuilder::class.java)
 
+    override fun canImport(fileOrDirectory: VirtualFile, project: Project?): Boolean {
+        return if (fileOrDirectory.isDirectory && fileOrDirectory.findChild(FileUtils.LIBRARY_CONFIG_FILE) != null) {
+            true
+        } else if (fileOrDirectory.isFile) {
+            canImportFromFile(fileOrDirectory)
+        } else {
+            false
+        }
+    }
+
     override fun canImportFromFile(file: VirtualFile) =
         file.name == FileUtils.LIBRARY_CONFIG_FILE