Merge pull request #22 from hyperoslo/fix/callback

Add UIImage in callback
hyperoslo · May 31, 2016 · 7e122dc · 7e122dc
2 parents 6152c6b + f800c7b
commit 7e122dc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Source/Extensions/UIImageView+Imaginary.swift b/Source/Extensions/UIImageView+Imaginary.swift
@@ -2,7 +2,7 @@ import UIKit
 
 extension UIImageView {
 
-  public func setImage(URL: NSURL?, placeholder: UIImage? = nil, completion: (() -> ())? = nil) {
+  public func setImage(URL: NSURL?, placeholder: UIImage? = nil, completion: ((UIImage?) -> ())? = nil) {
     image = placeholder
 
     guard let URL = URL else { return }
@@ -20,7 +20,7 @@ extension UIImageView {
       if let image = object {
         dispatch_async(dispatch_get_main_queue()) {
           weakSelf.image = image
-          completion?()
+          completion?(image)
         }
 
         return
@@ -39,7 +39,7 @@ extension UIImageView {
         case let .Success(image):
           Imaginary.transitionClosure(imageView: weakSelf, image: image)
           imageCache.add(key, object: image)
-          completion?()
+          completion?(image)
         default:
           break
         }