fix(HomeLastPicCollectionViewCell): Refresh during multi selection deselects all

kDrive/UI/Controller/Files/File List/FileListViewController.swift

@@ -635,13 +635,20 @@ class FileListViewController: UICollectionViewController, SwipeActionCollectionV
 
         let cell = collectionView.dequeueReusableCell(type: cellType, for: indexPath) as! FileCollectionViewCell
         let file = displayedFiles[indexPath.row]
+        let multipleSelectionEnabled = viewModel.multipleSelectionViewModel?.isMultipleSelectionEnabled ?? false
 
         cell.initStyle(isFirst: file.isFirstInList, isLast: file.isLastInList)
         cell.configureWith(
             driveFileManager: viewModel.driveFileManager,
             file: file,
-            selectionMode: viewModel.multipleSelectionViewModel?.isMultipleSelectionEnabled == true
+            selectionMode: multipleSelectionEnabled
         )
+
+        if multipleSelectionEnabled,
+           let multipleSelectionViewModel = viewModel.multipleSelectionViewModel {
+            cell.isSelected = multipleSelectionViewModel.selectedItems.contains(file)
+        }
+
         cell.delegate = self
         if ReachabilityListener.instance.currentStatus == .offline && !file.isDirectory && !file.isAvailableOffline {
             cell.setEnabled(false)

kDrive/UI/Controller/Menu/PhotoList/PhotoListViewController.swift

@@ -299,11 +299,23 @@ final class PhotoListViewController: FileListViewController {
             return cell
         }
 
+        let multipleSelectionEnabled: Bool
+        let isSelected: Bool
+        if let multipleSelectionViewModel = viewModel.multipleSelectionViewModel {
+            multipleSelectionEnabled = multipleSelectionViewModel.isMultipleSelectionEnabled
+            isSelected = multipleSelectionViewModel.selectedItems.contains { $0.uid == file.uid }
+        } else {
+            multipleSelectionEnabled = false
+            isSelected = false
+        }
+
         cell.configureWith(
             file: file,
             roundedCorners: false,
-            selectionMode: viewModel.multipleSelectionViewModel?.isMultipleSelectionEnabled == true
+            selectionMode: multipleSelectionEnabled,
+            isSelected: isSelected
         )
+
         return cell
     }
 

kDrive/UI/View/Home/HomeLastPicCollectionViewCell.swift

@@ -31,6 +31,10 @@ class HomeLastPicCollectionViewCell: UICollectionViewCell {
     private var thumbnailDownloadTask: Kingfisher.DownloadTask?
 
     override var isSelected: Bool {
+        willSet {
+            // NOTE: DifferenceKit reset the selected flag mid air. Investigating
+            print("•• isSelected changed: \(newValue) fid:\(file?.id)")
+        }
         didSet {
             configureForSelection()
         }
@@ -77,9 +81,10 @@ class HomeLastPicCollectionViewCell: UICollectionViewCell {
         contentInsetView.cornerRadius = UIConstants.cornerRadius
     }
 
-    func configureWith(file: File, roundedCorners: Bool = true, selectionMode: Bool = false) {
+    func configureWith(file: File, roundedCorners: Bool = true, selectionMode: Bool = false, isSelected: Bool = false) {
         self.selectionMode = selectionMode
         self.file = file
+
         checkmarkImage.isHidden = !selectionMode
         darkLayer.isHidden = false
         thumbnailDownloadTask = file.getThumbnail { [weak self, fileId = file.id] image, isThumbnail in
@@ -94,7 +99,8 @@ class HomeLastPicCollectionViewCell: UICollectionViewCell {
             contentInsetView.cornerRadius = UIConstants.cornerRadius
         }
         videoData.isHidden = !(file.uti.conforms(to: .video) || file.uti.conforms(to: .movie))
-        configureForSelection()
+
+        self.isSelected = isSelected
     }
 
     private func configureForSelection() {