From 820d8303ba122a944bf40711ccb24ffded603aaf Mon Sep 17 00:00:00 2001
From: Christopher Meiklejohn <christopher.meiklejohn@gmail.com>
Date: Wed, 13 Jul 2011 11:45:17 -0400
Subject: [PATCH] Patch existing find_or_initialize_version_from_spec to make
 size an optional parameter.

---
 app/models/pusher.rb  | 2 +-
 app/models/rubygem.rb | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/app/models/pusher.rb b/app/models/pusher.rb
index f24b3d7340d..5cf331e5d4f 100644
--- 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
index d9cb2f09d31..993666f0abf 100644
--- 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