motion-prime/elements/image.rb in motion-prime-1.0.5 vs motion-prime/elements/image.rb in motion-prime-1.0.6

@@ -1,15 +1,48 @@
 module MotionPrime
   class ImageElement < BaseElement
     after_render :fetch_image
+
     def view_class
       "UIImageView"
     end
 
     def fetch_image
       return unless computed_options[:url]
       raise "You must set default image for `#{name}`" unless computed_options[:default]
-      view.setImageWithURL NSURL.URLWithString(computed_options[:url]),
-                        placeholderImage: computed_options[:default].uiimage
+
+      view.setImage(computed_options[:default].uiimage)
+      ref_key = allocate_strong_references
+      BW::Reactor.schedule do
+        manager = SDWebImageManager.sharedManager
+        manager.downloadWithURL(computed_options[:url],
+          options: 0,
+          progress: lambda{ |r_size, e_size|  },
+          completed: lambda{ |image, error, type, finished|
+            unless image
+              release_strong_references(ref_key)
+              return
+            end
+
+            if computed_options[:post_process].present?
+              image = computed_options[:post_process][:method].to_proc.call(computed_options[:post_process][:target], image)
+            end
+
+            self.performSelectorOnMainThread :set_image, withObject: image, waitUntilDone: true
+            release_strong_references(ref_key)
+          }
+        )
+      end
+    end
+
+    def set_image(*args)
+      self.view.setImage(args)
+    end
+
+    def strong_references
+      # .compact() is required here, otherwise screen will not be released
+      refs = [section, (section.collection_section if section.respond_to?(:cell_section_name))].compact
+      refs += section.strong_references if section
+      refs
     end
   end
 end
\ No newline at end of file