Let asset support remote url.

tilltue · May 31, 2018 · 1c4bc55 · 1c4bc55
1 parent 75ab30e
commit 1c4bc55
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 2 deletions.
diff --git a/TLPhotoPicker/Classes/TLAssetsCollection.swift b/TLPhotoPicker/Classes/TLAssetsCollection.swift
@@ -54,6 +54,8 @@ public class TLPHAsset {
         }
     }
 
+    public var remoteURL: URL?
+
     public func extType() -> ImageExtType {
         var ext = ImageExtType.png
         if let fileName = self.originalFileName, let extention = URL(string: fileName)?.pathExtension.lowercased() {
@@ -209,6 +211,11 @@ public class TLPHAsset {
         self.init(asset: nil)
         self.fullResolutionImage = image
     }
+
+    public convenience init(url: URL) {
+        self.init(asset: nil)
+        self.remoteURL = url
+    }
 }
 
 extension TLPHAsset: Equatable {

diff --git a/TLPhotoPicker/Classes/TLPhotoCollectionViewCell.swift b/TLPhotoPicker/Classes/TLPhotoCollectionViewCell.swift
@@ -49,6 +49,9 @@ open class TLPhotoCollectionViewCell: UICollectionViewCell {
             self.selectedView?.layer.borderColor = self.configure.selectedColor.cgColor
             self.orderBgView?.backgroundColor = self.configure.selectedColor
             self.videoIconImageView?.image = self.configure.videoIcon
+            if self.imageView?.image == nil {
+                self.imageView?.image = self.configure.placeholderIcon
+            }
         }
     }
 
@@ -162,5 +165,6 @@ open class TLPhotoCollectionViewCell: UICollectionViewCell {
         self.durationView?.backgroundColor = UIColor(red: 0, green: 0, blue: 0, alpha: 0.6)
         self.selectedHeight?.constant = 10
         self.selectedAsset = false
+        self.imageView?.image = self.configure.placeholderIcon
     }
 }
diff --git a/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift b/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift
@@ -810,8 +810,19 @@ extension TLPhotosPickerViewController: UICollectionViewDelegate,UICollectionVie
         } else {
             cell.indicator?.stopAnimating()
         }
-        if let customCollection = customCollection, collection == customCollection, let image = asset.fullResolutionImage {
-            cell.imageView?.image = image
+        if let customCollection = customCollection, collection == customCollection {
+            if let image = asset.fullResolutionImage {
+                cell.imageView?.image = image
+            } else if let url = asset.remoteURL {
+                cell.imageView?.image = self.placeholderThumbnail
+                DispatchQueue.global().async {
+                    if let data = try? Data(contentsOf: url), let image = UIImage(data: data) {
+                        DispatchQueue.main.async {
+                            cell.imageView?.image = image
+                        }
+                    }
+                }
+            }
         } else if let phAsset = asset.phAsset {
             if self.usedPrefetch {
                 let options = PHImageRequestOptions()