Open txt file from directory passed as runtime arg

Nianna · Oct 20, 2024 · 06d6226 · 06d6226
1 parent e99701c
commit 06d6226
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/src/main/java/com/github/nianna/karedi/controller/RootController.java b/src/main/java/com/github/nianna/karedi/controller/RootController.java
@@ -21,6 +21,7 @@
 import javafx.scene.layout.BorderPane;
 
 import java.io.File;
+import java.util.Arrays;
 import java.util.Optional;
 
 public class RootController implements Controller {
@@ -211,9 +212,26 @@ public void onStageShown() {
     private Optional<File> findTxtFileToOpenFromRunParams() {
         return KarediApp.getInstance().getParameters().getRaw()
                 .stream()
-                .filter(param -> param.endsWith(".txt"))
                 .findFirst()
-				.map(File::new);
+                .map(File::new)
+                .filter(File::exists)
+                .flatMap(this::getTxtFileFromFile);
+    }
+
+    private Optional<File> getTxtFileFromFile(File file) {
+        if (isATxtFile(file)) {
+            return Optional.of(file);
+        }
+        if (file.isDirectory()) {
+            File[] txtFiles = file.listFiles(this::isATxtFile);
+            if (txtFiles != null) {
+                return Arrays.stream(txtFiles)
+                        .filter(txtFile -> file.getName().equalsIgnoreCase(Utils.trimExtension(txtFile.getName())))
+                        .findFirst()
+                        .or(() -> Arrays.stream(txtFiles).findFirst());
+            }
+        }
+        return Optional.empty();
     }
 
     private class EditLyricsAction extends KarediAction {