Added count property that reflect actual asset count

Change-Id: I7a377ff53592200ba916b22319c6ca5c35cddea7
tilltue · Jul 6, 2024 · e1e1e51 · e1e1e51
1 parent b5b69f4
commit e1e1e51
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/TLPhotoPicker/Classes/TLAssetsCollection.swift b/TLPhotoPicker/Classes/TLAssetsCollection.swift
@@ -358,6 +358,11 @@ public struct TLAssetsCollection {
             return count + (self.useCameraButton ? 1 : 0)
         }
     }
+    var assetCount: Int {
+        get {
+            return self.fetchResult?.count ?? 0
+        }
+    }
 
     init(collection: PHAssetCollection) {
         self.phAssetCollection = collection

diff --git a/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift b/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift
@@ -611,7 +611,7 @@ extension TLPhotosPickerViewController: TLPhotoLibraryDelegate {
     func loadCompleteAllCollection(collections: [TLAssetsCollection]) {
         self.collections = collections
         self.focusFirstCollection()
-        let isEmpty = !self.collections.contains(where: { $0.count > 0 })
+        let isEmpty = !self.collections.contains(where: { $0.assetCount > 0 })
         self.subTitleStackView.isHidden = isEmpty
         self.emptyView.isHidden = !isEmpty
         self.emptyImageView.isHidden = self.emptyImageView.image == nil