Added support for SDWebImageAvoidAutoSetImage option to UIButton and …

…highlighted UIImageView
heroims · Sep 23, 2015 · 05ce156 · 05ce156
1 parent 14842f6
commit 05ce156
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/SDWebImage/UIButton+WebCache.m b/SDWebImage/UIButton+WebCache.m
@@ -74,7 +74,12 @@ - (void)sd_setImageWithURL:(NSURL *)url forState:(UIControlState)state placehold
         dispatch_main_sync_safe(^{
             __strong UIButton *sself = wself;
             if (!sself) return;
-            if (image) {
+            if (image && (options & SDWebImageAvoidAutoSetImage) && completedBlock)
+            {
+                completedBlock(image, error, cacheType, url);
+                return;
+            }
+            else if (image) {
                 [sself setImage:image forState:state];
             }
             if (completedBlock && finished) {
@@ -117,7 +122,12 @@ - (void)sd_setBackgroundImageWithURL:(NSURL *)url forState:(UIControlState)state
             dispatch_main_sync_safe(^{
                 __strong UIButton *sself = wself;
                 if (!sself) return;
-                if (image) {
+                if (image && (options & SDWebImageAvoidAutoSetImage) && completedBlock)
+                {
+                    completedBlock(image, error, cacheType, url);
+                    return;
+                }
+                else if (image) {
                     [sself setBackgroundImage:image forState:state];
                 }
                 if (completedBlock && finished) {

diff --git a/SDWebImage/UIImageView+HighlightedWebCache.m b/SDWebImage/UIImageView+HighlightedWebCache.m
@@ -39,7 +39,12 @@ - (void)sd_setHighlightedImageWithURL:(NSURL *)url options:(SDWebImageOptions)op
             dispatch_main_sync_safe (^
                                      {
                                          if (!wself) return;
-                                         if (image) {
+                                         if (image && (options & SDWebImageAvoidAutoSetImage) && completedBlock)
+                                         {
+                                             completedBlock(image, error, cacheType, url);
+                                             return;
+                                         }
+                                         else if (image) {
                                              wself.highlightedImage = image;
                                              [wself setNeedsLayout];
                                          }