Keep track of materialised files in storageuseenumerationobserver

Signed-off-by: Claudio Cambra <[email protected]>
nextcloud · Sep 28, 2023 · 777bb9f · 777bb9f
1 parent 5e723f3
commit 777bb9f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/gui/macOS/fileproviderstorageuseenumerationobserver.h b/src/gui/macOS/fileproviderstorageuseenumerationobserver.h
@@ -21,5 +21,6 @@ typedef void(^UsageEnumerationFinishedHandler)(NSError *const error);
 
 @property (readwrite, strong) UsageEnumerationFinishedHandler enumerationFinishedHandler;
 @property (readonly) NSUInteger usage;
+@property (readonly) NSSet<id<NSFileProviderItem>> *materialisedItems;
 
 @end
diff --git a/src/gui/macOS/fileproviderstorageuseenumerationobserver.m b/src/gui/macOS/fileproviderstorageuseenumerationobserver.m
@@ -29,12 +29,17 @@ - (instancetype)init
 // NSFileProviderEnumerationObserver protocol methods
 - (void)didEnumerateItems:(NSArray<id<NSFileProviderItem>> *)updatedItems
 {
+    NSMutableSet<id<NSFileProviderItem>> * const existingItems = self.materialisedItems.mutableCopy;
+
     for (const id<NSFileProviderItem> item in updatedItems) {
         NSLog(@"StorageUseEnumerationObserver: Enumerating %@ with size %llu",
               item.filename, item.documentSize.unsignedLongLongValue);
 
         _usage += item.documentSize.unsignedLongLongValue;
+        [existingItems addObject:item];
     }
+
+    _materialisedItems = existingItems.copy;
 }
 
 - (void)finishEnumeratingWithError:(NSError *)error