Display duration when usedPrefetch enable.

tilltue · Jan 31, 2018 · e4b7649 · e4b7649
1 parent 08bfbcf
commit e4b7649
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift b/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift
@@ -766,6 +766,10 @@ extension TLPhotosPickerViewController: UICollectionViewDelegate,UICollectionVie
                     DispatchQueue.main.async {
                         if self.requestIds[indexPath] != nil {
                             cell?.imageView?.image = image
+                            if self.allowedVideo {
+                                cell?.durationView?.isHidden = asset.type != .video
+                                cell?.duration = asset.type == .video ? phAsset.duration : nil
+                            }
                             if complete {
                                 self.requestIds.removeValue(forKey: indexPath)
                             }