Add onCompletion closure to LazyImage

CHANGELOG.md

@@ -4,7 +4,8 @@
 
 *Mar 25, 2023*
 
-- Add `makeImageView` closure to `LazyImageView` to allow using custom view for rendering images 
+- Add `makeImageView` closure to `LazyImageView` to allow using custom view for rendering images
+- Add `onCompletion` closure to `LazyImage` 
 - Fix an issue with `.videoAssetKey` value missing from `ImageContainer`
 - Fix an issue with `.gif` being encoded as `.jpeg` when `.storeEncodedImages` policy is used 
 

Sources/NukeUI/FetchImage.swift

@@ -74,6 +74,9 @@ public final class FetchImage: ObservableObject, Identifiable {
     /// request. `[]` by default.
     public var processors: [any ImageProcessing] = []
 
+    /// Gets called when the current request is completed.
+    public var onCompletion: ((Result<ImageResponse, Error>) -> Void)?
+
     private var imageTask: ImageTask?
     private var lastResponse: ImageResponse?
     private var cancellable: AnyCancellable?
@@ -163,6 +166,7 @@ public final class FetchImage: ObservableObject, Identifiable {
             self.imageContainer = response.container
         }
         self.result = result
+        self.onCompletion?(result)
     }
 
     // MARK: Load (Async/Await)

Sources/NukeUI/LazyImage.swift

@@ -25,6 +25,7 @@ public struct LazyImage<Content: View>: View {
     private var transaction: Transaction
     private var pipeline: ImagePipeline = .shared
     private var onDisappearBehavior: DisappearBehavior? = .cancel
+    private var onCompletion: ((Result<ImageResponse, Error>) -> Void)?
 
     // MARK: Initializers
 
@@ -112,6 +113,11 @@ public struct LazyImage<Content: View>: View {
         map { $0.onDisappearBehavior = behavior }
     }
 
+    /// Gets called when the current request is completed.
+    public func onCompletion(_ closure: @escaping (Result<ImageResponse, Error>) -> Void) -> Self {
+        map { $0.onCompletion = closure }
+    }
+
     private func map(_ closure: (inout LazyImage) -> Void) -> Self {
         var copy = self
         closure(&copy)
@@ -172,6 +178,7 @@ public struct LazyImage<Content: View>: View {
     private func onAppear() {
         viewModel.transaction = transaction
         viewModel.pipeline = pipeline
+        viewModel.onCompletion = onCompletion
 
         guard viewModel.cachedResponse == nil else { return }
         viewModel.load(context?.request)