Add the onSuccess, onFailure, onProgress methods to AnimatedImage/WebImage, make it possible to notify result

Author: dreampiggy

SDWebImageSwiftUI/Classes/AnimatedImage.swift

@@ -15,6 +15,9 @@ import SDWebImage
 final class AnimatedImageModel : ObservableObject {
     @Published var image: PlatformImage?
     @Published var url: URL?
+    @Published var successBlock: ((PlatformImage, SDImageCacheType) -> Void)?
+    @Published var failureBlock: ((Error) -> Void)?
+    @Published var progressBlock: ((Int, Int) -> Void)?
 }
 
 // Layout Binding Object
@@ -67,7 +70,15 @@ public struct AnimatedImage : ViewRepresentable {
     func updateView(_ view: AnimatedImageViewWrapper, context: ViewRepresentableContext<AnimatedImage>) {
         view.wrapped.image = imageModel.image
         if let url = imageModel.url {
-            view.wrapped.sd_setImage(with: url, placeholderImage: nil, options: webOptions, context: webContext)
+            view.wrapped.sd_setImage(with: url, placeholderImage: nil, options: webOptions, context: webContext, progress: { (receivedSize, expectedSize, _) in
+                self.imageModel.progressBlock?(receivedSize, expectedSize)
+            }) { (image, error, cacheType, _) in
+                if let image = image {
+                    self.imageModel.successBlock?(image, cacheType)
+                } else {
+                    self.imageModel.failureBlock?(error ?? NSError())
+                }
+            }
         }
 
         layoutView(view, context: context)
@@ -179,16 +190,6 @@ public struct AnimatedImage : ViewRepresentable {
         #endif
     }
 
-    public func image(_ image: PlatformImage?) -> Self {
-        imageModel.image = image
-        return self
-    }
-    
-    public func imageUrl(_ url: URL?) -> Self {
-        imageModel.url = url
-        return self
-    }
-    
     public func resizable(
         capInsets: EdgeInsets = EdgeInsets(),
         resizingMode: Image.ResizingMode = .stretch) -> AnimatedImage
@@ -236,6 +237,23 @@ public struct AnimatedImage : ViewRepresentable {
     }
 }
 
+extension AnimatedImage {
+    public func onFailure(perform action: ((Error) -> Void)? = nil) -> AnimatedImage {
+        imageModel.failureBlock = action
+        return self
+    }
+    
+    public func onSuccess(perform action: ((PlatformImage, SDImageCacheType) -> Void)? = nil) -> AnimatedImage {
+        imageModel.successBlock = action
+        return self
+    }
+    
+    public func onProgress(perform action: ((Int, Int) -> Void)? = nil) -> AnimatedImage {
+        imageModel.progressBlock = action
+        return self
+    }
+}
+
 extension AnimatedImage {
     public init(url: URL?, placeholder: PlatformImage? = nil, options: SDWebImageOptions = [], context: [SDWebImageContextOption : Any]? = nil) {
         self.webOptions = options

SDWebImageSwiftUI/Classes/ImageManager.swift

@@ -26,6 +26,9 @@ class ImageManager : ObservableObject {
     var url: URL?
     var options: SDWebImageOptions
     var context: [SDWebImageContextOption : Any]?
+    var successBlock: ((PlatformImage, SDImageCacheType) -> Void)?
+    var failureBlock: ((Error) -> Void)?
+    var progressBlock: ((Int, Int) -> Void)?
 
     init(url: URL?, options: SDWebImageOptions = [], context: [SDWebImageContextOption : Any]? = nil) {
         self.url = url
@@ -34,9 +37,17 @@ class ImageManager : ObservableObject {
     }
 
     func load() {
-        currentOperation = manager.loadImage(with: url, options: options, context: context, progress: nil) { (image, data, error, cacheType, _, _) in
+        currentOperation = manager.loadImage(with: url, options: options, context: context, progress: { [weak self] (receivedSize, expectedSize, _) in
+            self?.progressBlock?(receivedSize, expectedSize)
+        }) { [weak self] (image, data, error, cacheType, _, _) in
+            guard let self = self else {
+                return
+            }
             if let image = image {
                 self.image = image
+                self.successBlock?(image, cacheType)
+            } else {
+                self.failureBlock?(error ?? NSError())
             }
         }
     }

SDWebImageSwiftUI/Classes/WebImage.swift

@@ -82,6 +82,22 @@ extension WebImage {
     }
 }
 
+extension WebImage {
+    public func onFailure(perform action: ((Error) -> Void)? = nil) -> WebImage {
+        self.imageManager.failureBlock = action
+        return self
+    }
+    
+    public func onSuccess(perform action: ((PlatformImage, SDImageCacheType) -> Void)? = nil) -> WebImage {
+        self.imageManager.successBlock = action
+        return self
+    }
+    
+    public func onProgress(perform action: ((Int, Int) -> Void)? = nil) -> WebImage {
+        self.imageManager.progressBlock = action
+        return self
+    }
+}
 
 #if DEBUG
 struct WebImage_Previews : PreviewProvider {