Properly parse version from options hash for gem yank

mikel · Mar 1, 2010 · baf3a19 · baf3a19
1 parent d38f025
commit baf3a19
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/gem/lib/rubygems/commands/yank.rb b/gem/lib/rubygems/commands/yank.rb
@@ -26,7 +26,7 @@ def initialize
 
   def execute
     sign_in
-    version = options[:version]
+    version = get_version_from_requirements(options[:version])
     if !version.nil?
       yank_gem(version)
     else
@@ -48,4 +48,13 @@ def yank_gem(version)
 
     say response.body
   end
+
+  private
+    def get_version_from_requirements(requirements)
+      begin
+        requirements.requirements.first[1].version
+      rescue
+        nil
+      end
+    end
 end