BugFix: /kb or /kb list does not filter out unrelated files.

keuin · Jan 29, 2021 · cc69d50 · cc69d50
1 parent 602fb9e
commit cc69d50
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/java/com/keuin/kbackupfabric/backup/suggestion/BackupNameSuggestionProvider.java b/src/main/java/com/keuin/kbackupfabric/backup/suggestion/BackupNameSuggestionProvider.java
@@ -6,7 +6,9 @@
 
 import java.io.File;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
+import java.util.function.Predicate;
 
 public class BackupNameSuggestionProvider {
 
@@ -34,8 +36,10 @@ public static void updateCandidateList() {
                     File[] files = file.listFiles();
                     if (files == null)
                         return;
-                    for (File f : files)
-                        candidateCacheList.add(f.getName());
+                    Arrays.stream(files).map(File::getName).filter(
+                            ((Predicate<String>) s -> s.toLowerCase().endsWith(".zip"))
+                                    .or(s -> s.toLowerCase().endsWith(".kbi"))
+                    ).forEach(candidateCacheList::add);
                     cacheUpdateTime = (int) System.currentTimeMillis();
                 } catch (NullPointerException ignored) {
                 }