add image broken,when set local image fail

veerjainATgmail · Aug 28, 2018 · 947dc0e · 947dc0e
1 parent 4b2330b
commit 947dc0e
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 1 deletion.
diff --git a/iOS/IMUIMessageCollectionView/Assets/IMUIAssets.bundle/image/image-broken.png b/iOS/IMUIMessageCollectionView/Assets/IMUIAssets.bundle/image/image-broken.png
diff --git a/iOS/IMUIMessageCollectionView/Assets/IMUIAssets.bundle/image/image-broken@2x.png b/iOS/IMUIMessageCollectionView/Assets/IMUIAssets.bundle/image/image-broken@2x.png
diff --git a/iOS/IMUIMessageCollectionView/Assets/IMUIAssets.bundle/image/image-broken@3x.png b/iOS/IMUIMessageCollectionView/Assets/IMUIAssets.bundle/image/image-broken@3x.png
diff --git a/iOS/IMUIMessageCollectionView/Views/IMUIImageMessageContentView.swift b/iOS/IMUIMessageCollectionView/Views/IMUIImageMessageContentView.swift
@@ -24,15 +24,18 @@ public class IMUIImageMessageContentView: UIView, IMUIMessageContentViewProtocol
   }
 
   public func layoutContentView(message: IMUIMessageModelProtocol) {
+    imageView.image = nil // reset image
 
     imageView.frame = CGRect(origin: CGPoint.zero, size: message.layout.bubbleContentSize)
     imageView.image = UIImage(contentsOfFile: message.mediaFilePath())
+    if imageView.image == nil && (message.webImageUrl?() == nil || message.webImageUrl?() == "") {
+      self.imageView.image = UIImage.imuiImage(with: "image-broken")
+    }
 
     task?.suspend()
     self.urlString = message.webImageUrl?() ?? ""
     task = IMUIWebImageTaskManager.shared.downloadImage(self.urlString!) { (data, precent, urlString, error) in
       if (error != nil) {
-        print("\(String(describing: error))")
         return
       }