Patch existing find_or_initialize_version_from_spec to make size an o…

…ptional parameter.
rubygems · Nov 4, 2012 · 820d830 · 820d830
1 parent 0f4273a
commit 820d830
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/app/models/pusher.rb b/app/models/pusher.rb
@@ -46,7 +46,7 @@ def pull_spec
 
   def find
     @rubygem = Rubygem.find_or_initialize_by_name(spec.name)
-    @version = @rubygem.find_or_initialize_version_from_spec(spec)
+    @version = @rubygem.find_or_initialize_version_from_spec(spec, size)
 
     if @version.new_record?
       true

diff --git a/app/models/rubygem.rb b/app/models/rubygem.rb
@@ -230,9 +230,10 @@ def yank!(version)
     end
   end
 
-  def find_or_initialize_version_from_spec(spec)
-    version = self.versions.find_or_initialize_by_number_and_platform(spec.version.to_s, spec.original_platform.to_s)
+  def find_or_initialize_version_from_spec(spec, size = nil)
+    version         = self.versions.find_or_initialize_by_number_and_platform(spec.version.to_s, spec.original_platform.to_s)
     version.rubygem = self
+    version.size    = size if size
     version
   end