From 1f9b179749dfdf3c9f0c103d83f3b9ef9613f50a Mon Sep 17 00:00:00 2001
From: Kai Reinhard <K.Reinhard@micromata.de>
Date: Sun, 16 Dec 2018 17:34:58 +0000
Subject: [PATCH] Filtering (text search and max result size) works now for file lists.

---
 borgbutler-core/src/main/java/de/micromata/borgbutler/cache/ArchiveFilelistCache.java |   26 ++++++++++++++++++--------
 1 files changed, 18 insertions(+), 8 deletions(-)

diff --git a/borgbutler-core/src/main/java/de/micromata/borgbutler/cache/ArchiveFilelistCache.java b/borgbutler-core/src/main/java/de/micromata/borgbutler/cache/ArchiveFilelistCache.java
index 9d066c0..7bf10a7 100644
--- a/borgbutler-core/src/main/java/de/micromata/borgbutler/cache/ArchiveFilelistCache.java
+++ b/borgbutler-core/src/main/java/de/micromata/borgbutler/cache/ArchiveFilelistCache.java
@@ -2,6 +2,7 @@
 
 import de.micromata.borgbutler.config.BorgRepoConfig;
 import de.micromata.borgbutler.data.Archive;
+import de.micromata.borgbutler.data.FileSystemFilter;
 import de.micromata.borgbutler.json.borg.BorgFilesystemItem;
 import de.micromata.borgbutler.utils.ReplaceUtils;
 import lombok.Getter;
@@ -54,19 +55,26 @@
     /**
      * Will load and touch the archive file if exist. The file will be touched (last modified time will be set to now)
      * for pruning oldest cache files. The last modified time will be the time of the last usage.
+     *
      * @param repoConfig
      * @param archive
+     * @param filter     If given, only file items matching this filter are returned.
      * @return
      */
-    public List<BorgFilesystemItem> load(BorgRepoConfig repoConfig, Archive archive, int maxSize) {
+    public List<BorgFilesystemItem> load(BorgRepoConfig repoConfig, Archive archive, FileSystemFilter filter) {
         File file = getFile(repoConfig, archive);
         if (!file.exists()) {
             return null;
         }
-        return load(file, maxSize);
+        return load(file, filter);
     }
 
-    public List<BorgFilesystemItem> load(File file, int maxSize) {
+    /**
+     * @param file
+     * @param filter If given, only file items matching this filter are returned.
+     * @return
+     */
+    public List<BorgFilesystemItem> load(File file, FileSystemFilter filter) {
         if (file.exists() == false) {
             log.error("File '" + file.getAbsolutePath() + "' doesn't exist. Can't get archive content files.");
             return null;
@@ -86,14 +94,16 @@
                 return null;
             }
             int size = (Integer) obj;
-            if (maxSize > 0 && maxSize < size) {
-                size = maxSize;
-            }
+            int maxSize = filter != null ? filter.getMaxResultSize() : -1;
             list = new ArrayList<>();
+            int counter = 0;
             for (int i = 0; i < size; i++) {
                 obj = inputStream.readObject();
                 if (obj instanceof BorgFilesystemItem) {
-                    list.add((BorgFilesystemItem) obj);
+                    if (filter == null || filter.matches(((BorgFilesystemItem) obj))) {
+                        list.add((BorgFilesystemItem) obj);
+                        if (maxSize > 0 && counter++ >= maxSize) break;
+                    }
                 } else {
                     log.error("Can't load archive content. FilesystemItem expected, but received: "
                             + (obj != null ? obj.getClass() : "null")
@@ -187,7 +197,7 @@
 
     File getFile(BorgRepoConfig repoConfig, Archive archive) {
         return new File(cacheDir, ReplaceUtils.encodeFilename(CACHE_ARCHIVE_LISTS_BASENAME + archive.getTime()
-                + "-" + repoConfig.getRepo() + "-" + archive.getName() + CACHE_FILE_GZIP_EXTENSION,
+                        + "-" + repoConfig.getRepo() + "-" + archive.getName() + CACHE_FILE_GZIP_EXTENSION,
                 true));
     }
 

--
Gitblit v1.10.0