feat: handle order button select view

baronha · baronha · commit 56260ed25491 · 2023-09-15T23:24:36.000+07:00
diff --git a/ios/CustomPhotoPicker/CustomPhotoPickerViewController.swift b/ios/CustomPhotoPicker/CustomPhotoPickerViewController.swift
@@ -7,6 +7,7 @@
 //
 
 import Foundation
+import Photos
 import TLPhotoPicker
 
 class CustomPhotoPickerViewController: TLPhotosPickerViewController, ViewerControllerDataSource {
@@ -64,6 +65,8 @@ class CustomPhotoPickerViewController: TLPhotosPickerViewController, ViewerContr
                     let footerView = PreviewFooterView()
                     footerView.viewDelegate = self
 
+                    self.viewerController?.delegate = self
+
                     self.viewerController!.footerView = footerView
                 }
 
@@ -96,10 +99,6 @@ class CustomPhotoPickerViewController: TLPhotosPickerViewController, ViewerContr
         }
     }
 
-    func cellOnLongPress(_: Cell) {
-        //
-    }
-
     override func viewWillAppear(_ animated: Bool) {
         super.viewWillAppear(animated)
         self.navigationController?.setNavigationBarHidden(true, animated: true)
@@ -123,7 +122,112 @@ extension CustomPhotoPickerViewController: PreviewHeaderViewDelegate, PreviewFoo
         }
     }
 
-    func footerView(_: PreviewFooterView, didPressSelectButton _: UIButton) {
-        self.viewerController?.dismiss(nil)
+    private func canSelect(phAsset: PHAsset) -> Bool {
+        if let closure = self.canSelectAsset {
+            return closure(phAsset)
+        } else if let delegate = self.delegate {
+            return delegate.canSelectAsset(phAsset: phAsset)
+        }
+        return true
+    }
+
+    private func getSelectedAssets(_ asset: TLPHAsset) -> TLPHAsset? {
+        if let index = self.selectedAssets.firstIndex(where: { $0.phAsset == asset.phAsset }) {
+            return self.selectedAssets[index]
+        }
+        return nil
+    }
+
+    private func orderUpdateCells() {
+        let visibleIndexPaths = self.collectionView.indexPathsForVisibleItems.sorted(by: { $0.row < $1.row })
+
+        for indexPath in visibleIndexPaths {
+            guard let cell = self.collectionView.cellForItem(at: indexPath) as? TLPhotoCollectionViewCell,
+                  let localID = cell.asset?.localIdentifier,
+                  let asset = TLPHAsset.asset(with: localID) else { continue }
+
+            if let selectedAsset = getSelectedAssets(asset) {
+                cell.selectedAsset = true
+                cell.orderLabel?.text = "\(selectedAsset.selectedOrder)"
+            } else {
+                cell.selectedAsset = false
+            }
+        }
+    }
+
+    func footerView(_: PreviewFooterView, didPressSelectButton button: SelectButton) {
+        let generator = UIImpactFeedbackGenerator(style: .medium)
+        generator.impactOccurred()
+
+        if let indexPath = self.viewerController?.currentIndexPath {
+            guard let cell = self.collectionView.cellForItem(at: indexPath) as? TLPhotoCollectionViewCell, let localID = cell.asset?.localIdentifier else { return }
+
+            guard var asset = TLPHAsset.asset(with: localID), let phAsset = asset.phAsset else { return }
+
+            if let index = selectedAssets.firstIndex(where: { $0.phAsset == asset.phAsset }) {
+                // deselect
+                logDelegate?.deselectedPhoto(picker: self, at: indexPath.row)
+                selectedAssets.remove(at: index)
+                #if swift(>=4.1)
+                    selectedAssets = selectedAssets.enumerated().compactMap { offset, asset -> TLPHAsset? in
+                        var asset = asset
+                        asset.selectedOrder = offset + 1
+                        return asset
+                    }
+                #else
+                    selectedAssets = selectedAssets.enumerated().flatMap { offset, asset -> TLPHAsset? in
+                        var asset = asset
+                        asset.selectedOrder = offset + 1
+                        return asset
+                    }
+                #endif
+                cell.selectedAsset = false
+                button.selectedAsset = false
+                self.orderUpdateCells()
+
+            } else {
+                // select
+                logDelegate?.selectedPhoto(picker: self, at: indexPath.row)
+                guard !maxCheck(), self.canSelect(phAsset: phAsset) else { return }
+
+                asset.selectedOrder = selectedAssets.count + 1
+                selectedAssets.append(asset)
+                cell.selectedAsset = true
+                button.selectedAsset = true
+                cell.orderLabel?.text = "\(asset.selectedOrder)"
+                button.setTitle("\(asset.selectedOrder)", for: .normal)
+            }
+        }
+    }
+}
+
+extension CustomPhotoPickerViewController: ViewerControllerDelegate {
+    func viewerController(_ viewerController: ViewerController, didChangeFocusTo indexPath: IndexPath) {
+        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 var asset = TLPHAsset.asset(with: localID) else { return }
+
+        if let index = selectedAssets.firstIndex(where: { $0.phAsset == asset.phAsset }) {
+            button.selectedAsset = true
+            button.setTitle("\(index + 1)", for: .normal)
+        } else {
+            button.selectedAsset = false
+        }
+    }
+
+    func viewerControllerDidDismiss(_: ViewerController) {
+        //
+    }
+
+    func viewerController(_: ViewerController, didFailDisplayingViewableAt _: IndexPath, error _: NSError) {
+        //
+    }
+
+    func viewerController(_: ViewerController, didLongPressViewableAt _: IndexPath) {
+        //
     }
 }
diff --git a/ios/Preview/PreviewFooter.swift b/ios/Preview/PreviewFooter.swift
@@ -9,64 +9,73 @@ import TLPhotoPicker
 import UIKit
 
 protocol PreviewFooterViewDelegate: AnyObject {
-    func footerView(_ headerView: PreviewFooterView, didPressSelectButton button: UIButton)
+    func footerView(_ headerView: PreviewFooterView, didPressSelectButton button: SelectButton)
 }
 
-class PreviewFooterView: TLPhotoCollectionViewCell {
-    weak var viewDelegate: PreviewFooterViewDelegate?
-
-    lazy var selectButton: UIButton = {
-        let button = UIButton(type: .system)
+class SelectButton: UIButton {
+    override init(frame: CGRect) {
+        super.init(frame: frame)
 
-        button.addTarget(self, action: #selector(PreviewFooterView.selectAction(button:)), for: .touchUpInside)
-        button.frame = CGRect(x: 0, y: 0, width: 24, height: 24)
-//
-//        button.layer.cornerRadius = 12
-//        button.clipsToBounds = true
+        self.frame.size = .init(width: 24, height: 24)
 
-        button.backgroundColor = config.selectedColor
+        backgroundColor = .clear
 
-        button.layer.borderWidth = 1.5
-        button.layer.borderColor = UIColor.white.cgColor
+        layer.borderWidth = 2
+        layer.borderColor = UIColor.white.cgColor
 
-        button.layer.cornerRadius = 16
-        button.clipsToBounds = true
-        button.contentEdgeInsets = UIEdgeInsets(top: 8, left: 8, bottom: 8, right: 8)
+        layer.cornerRadius = 16
+        clipsToBounds = true
+        contentEdgeInsets = UIEdgeInsets(top: 4, left: 4, bottom: 4, right: 4)
 
-        button.setTitle("1", for: .normal)
-        button.setTitleColor(.white, for: .normal)
+        setTitleColor(.white, for: .normal)
+        titleLabel?.font = .boldSystemFont(ofSize: 16)
 
-        button.translatesAutoresizingMaskIntoConstraints = false
+        translatesAutoresizingMaskIntoConstraints = false
+    }
 
-        // Tạo constraints để cố định vị trí của tiêu đề ở giữa tuyệt đối (absolute center)
+    @objc open var selectedAsset: Bool = false {
+        willSet(newValue) {
+            backgroundColor = config.selectedColor
+            if !newValue {
+                backgroundColor = .clear
+                setTitle("", for: .normal)
+            }
+        }
+    }
 
-//        let label = UILabel()
-//        label.text = "2"
-//        label.textColor = UIColor.white // Tuỳ chỉnh màu văn bản theo ý muốn
-//        label.textAlignment = .center
-//        label.frame = button.bounds // Đặt kích thước và vị trí của UILabel bằng bằng với UIButton
-//        button.addSubview(label)
+    @available(*, unavailable)
+    required init?(coder _: NSCoder) {
+        fatalError("init(coder:) has not been implemented")
+    }
+}
 
-        return button
-    }()
+class PreviewFooterView: UIView {
+    weak var viewDelegate: PreviewFooterViewDelegate?
+    var selectButton: SelectButton!
 
     override init(frame: CGRect) {
         super.init(frame: frame)
+
         let stackView = UIStackView()
+
+        selectButton = SelectButton()
+
+        selectButton.addTarget(self, action: #selector(PreviewFooterView.selectAction(button:)), for: .touchUpInside)
+
         stackView.translatesAutoresizingMaskIntoConstraints = false
         stackView.axis = .horizontal
         stackView.alignment = .trailing
         addSubview(stackView)
 
-        self.selectButton.translatesAutoresizingMaskIntoConstraints = false
-        stackView.addArrangedSubview(self.selectButton)
+        selectButton.translatesAutoresizingMaskIntoConstraints = false
+        stackView.addArrangedSubview(selectButton)
 
-        self.selectButton.centerXAnchor.constraint(equalTo: stackView.centerXAnchor).isActive = true
-        self.selectButton.centerYAnchor.constraint(equalTo: stackView.centerYAnchor).isActive = true
+        selectButton.centerXAnchor.constraint(equalTo: stackView.centerXAnchor).isActive = true
+        selectButton.centerYAnchor.constraint(equalTo: stackView.centerYAnchor).isActive = true
 
         // (Tuỳ chọn) Đặt constraints cho chiều rộng và chiều cao của UIButton
-        self.selectButton.widthAnchor.constraint(equalToConstant: 32).isActive = true
-        self.selectButton.heightAnchor.constraint(equalToConstant: 32).isActive = true
+        selectButton.widthAnchor.constraint(equalToConstant: 32).isActive = true
+        selectButton.heightAnchor.constraint(equalToConstant: 32).isActive = true
 
         NSLayoutConstraint.activate([
             stackView.trailingAnchor.constraint(equalTo: trailingAnchor, constant: -20),
@@ -82,7 +91,7 @@ class PreviewFooterView: TLPhotoCollectionViewCell {
         super.layoutSubviews()
     }
 
-    @objc func selectAction(button: UIButton) {
-        self.viewDelegate?.footerView(self, didPressSelectButton: button)
+    @objc func selectAction(button: SelectButton) {
+        viewDelegate?.footerView(self, didPressSelectButton: button)
     }
 }
diff --git a/ios/Preview/PreviewHeader.swift b/ios/Preview/PreviewHeader.swift
@@ -25,16 +25,17 @@ class PreviewHeaderView: UIView {
 
         button.setTitle(config.doneTitle, for: .normal)
         button.setTitleColor(.white, for: .normal)
-        button.titleLabel?.font = UIFont.boldSystemFont(ofSize: UIFont.systemFontSize)
+        button.titleLabel?.font = UIFont.boldSystemFont(ofSize: UIFont.labelFontSize)
         button.addTarget(self, action: #selector(PreviewHeaderView.doneAction(button:)), for: .touchUpInside)
-        button.backgroundColor = config.selectedColor
 
-        button.contentEdgeInsets = UIEdgeInsets(top: 8, left: 14, bottom: 8, right: 14)
+//        button.backgroundColor = config.selectedColor
 
-        button.layer.cornerRadius = 6.0
+//        button.contentEdgeInsets = UIEdgeInsets(top: 8, left: 14, bottom: 8, right: 14)
+//
+//        button.layer.cornerRadius = 6.0
 
-        button.setContentHuggingPriority(.required, for: .vertical)
-        button.setContentCompressionResistancePriority(.required, for: .vertical)
+//        button.setContentHuggingPriority(.required, for: .vertical)
+//        button.setContentCompressionResistancePriority(.required, for: .vertical)
 
         return button
     }()
diff --git a/ios/Viewer/ViewerController.swift b/ios/Viewer/ViewerController.swift
@@ -1,12 +1,12 @@
 import CoreData
 import UIKit
 
-public protocol ViewerControllerDataSource: class {
+public protocol ViewerControllerDataSource: AnyObject {
     func numberOfItemsInViewerController(_ viewerController: ViewerController) -> Int
     func viewerController(_ viewerController: ViewerController, viewableAt indexPath: IndexPath) -> Viewable
 }
 
-public protocol ViewerControllerDelegate: class {
+public protocol ViewerControllerDelegate: AnyObject {
     func viewerController(_ viewerController: ViewerController, didChangeFocusTo indexPath: IndexPath)
     func viewerControllerDidDismiss(_ viewerController: ViewerController)
     func viewerController(_ viewerController: ViewerController, didFailDisplayingViewableAt indexPath: IndexPath, error: NSError)
