feat: done preview

Author: baronha

example/src/App.js

@@ -45,7 +45,7 @@ export default function App() {
 
   const onPicker = async () => {
     try {
-      const singleSelectedMode = true;
+      const singleSelectedMode = false;
 
       const response = await openPicker({
         selectedAssets: images,

ios/CustomPhotoPicker/Cell.swift

@@ -31,7 +31,7 @@ class Cell: TLPhotoCollectionViewCell {
 
     // Xử lý sự kiện Long Press
     @objc private func handleLongPress(_ gestureRecognizer: UILongPressGestureRecognizer) {
-        guard gestureRecognizer.state == .began
+        guard gestureRecognizer.state == .began && !self.isCameraCell
         else {
             return
         }

ios/CustomPhotoPicker/CustomPhotoPickerViewController.swift

@@ -10,37 +10,11 @@ import Foundation
 import Photos
 import TLPhotoPicker
 
-class CustomPhotoPickerViewController: TLPhotosPickerViewController, ViewerControllerDataSource {
+class CustomPhotoPickerViewController: TLPhotosPickerViewController {
     var dismissPhotoPicker: ((_ withTLPHAssets: [TLPHAsset]) -> Void)?
 
     var viewerController: ViewerController?
 
-    func numberOfItemsInViewerController(_: ViewerController) -> Int {
-        var count = 0
-
-        for section in 0 ..< collectionView.numberOfSections {
-            count += collectionView.numberOfItems(inSection: section)
-        }
-
-        return count
-    }
-
-    func viewerController(_: ViewerController, viewableAt indexPath: IndexPath) -> Viewable {
-        let viewable = PreviewItem(id: UUID().uuidString)
-
-        if let cell = collectionView?.cellForItem(at: indexPath) as? Cell, let placeholder = cell.imageView?.image, let asset = cell.asset {
-            viewable.assetID = asset.localIdentifier
-
-            if asset.duration > 0 {
-                viewable.type = .video
-            }
-
-            viewable.placeholder = placeholder
-        }
-
-        return viewable
-    }
-
     override func viewDidLoad() {
         super.viewDidLoad()
         if config.isPreview {
@@ -69,11 +43,11 @@ class CustomPhotoPickerViewController: TLPhotosPickerViewController, ViewerContr
                     let footerView = PreviewFooterView()
                     footerView.viewDelegate = self
 
-                    self.viewerController?.delegate = self
-
                     self.viewerController!.footerView = footerView
                 }
 
+                self.viewerController?.delegate = self
+
                 self.present(self.viewerController!, animated: true, completion: nil)
             }
         }
@@ -113,6 +87,34 @@ class CustomPhotoPickerViewController: TLPhotosPickerViewController, ViewerContr
     }
 }
 
+extension CustomPhotoPickerViewController: ViewerControllerDataSource {
+    func numberOfItemsInViewerController(_: ViewerController) -> Int {
+        var count = 0
+
+        for section in 0 ..< collectionView.numberOfSections {
+            count += collectionView.numberOfItems(inSection: section)
+        }
+
+        return count
+    }
+
+    func viewerController(_: ViewerController, viewableAt indexPath: IndexPath) -> Viewable {
+        let viewable = PreviewItem(id: UUID().uuidString)
+
+        if let cell = collectionView?.cellForItem(at: indexPath) as? Cell, let placeholder = cell.imageView?.image, let asset = cell.asset {
+            viewable.assetID = asset.localIdentifier
+
+            if asset.duration > 0 {
+                viewable.type = .video
+            }
+
+            viewable.placeholder = placeholder
+        }
+
+        return viewable
+    }
+}
+
 extension CustomPhotoPickerViewController: PreviewHeaderViewDelegate {
     func headerView(_: PreviewHeaderView, didPressClearButton _: UIButton) {
         self.viewerController?.dismiss(nil)
@@ -235,11 +237,22 @@ extension CustomPhotoPickerViewController: PreviewFooterViewDelegate {
 
 extension CustomPhotoPickerViewController: ViewerControllerDelegate {
     func viewerController(_ viewerController: ViewerController, didChangeFocusTo indexPath: IndexPath) {
-        guard let footerView = viewerController.footerView as? PreviewFooterView else { return }
+        if indexPath.row == 0 {
+            let cell = self.collectionView.cellForItem(at: indexPath) as? TLPhotoCollectionViewCell
+
+            if cell?.isCameraCell == true {
+                viewerController.dismiss(nil)
+                return
+            }
+        }
+
+        guard let footerView = viewerController.footerView as? PreviewFooterView
+        else { return }
 
         guard let button = footerView.selectButton else { return }
 
-        guard let cell = self.collectionView.cellForItem(at: indexPath) as? TLPhotoCollectionViewCell, let localID = cell.asset?.localIdentifier else { return }
+        guard let cell = self.collectionView.cellForItem(at: indexPath) as? TLPhotoCollectionViewCell, let localID = cell.asset?.localIdentifier
+        else { return }
 
         guard let asset = TLPHAsset.asset(with: localID) else { return }
 

ios/Preview/PreviewItem.swift

@@ -27,14 +27,12 @@ class PreviewItem: Viewable {
     }
 
     func media(_ completion: @escaping (_ image: UIImage?, _ error: NSError?) -> Void) {
-        if let assetID = self.assetID {
+        if let assetID = assetID {
             if let asset = PHAsset.fetchAssets(withLocalIdentifiers: [assetID], options: nil).firstObject {
                 PreviewItem.image(for: asset) { image in
                     completion(image, nil)
                 }
             }
-        } else {
-            completion(self.placeholder, nil)
         }
     }