1 files changed, 93 insertions, 0 deletions
diff --git a/lib/rubygems/commands/cleanup_command.rb b/lib/rubygems/commands/cleanup_command.rb
new file mode 100644
index 0000000000..f6deac9829
--- /dev/null
+++ b/lib/rubygems/commands/cleanup_command.rb
@@ -0,0 +1,93 @@
+require 'rubygems/command'
+require 'rubygems/source_index'
+require 'rubygems/dependency_list'
+
+module Gem
+  module Commands
+    class CleanupCommand < Command
+      def initialize
+        super(
+          'cleanup',
+          'Clean up old versions of installed gems in the local repository',
+          {
+            :force => false, 
+            :test => false, 
+            :install_dir => Gem.dir
+          })
+        add_option('-d', '--dryrun', "") do |value, options|
+          options[:dryrun] = true
+        end
+      end
+
+      def arguments # :nodoc:
+        "GEMNAME       name of gem to cleanup"
+      end
+
+      def defaults_str # :nodoc:
+        "--no-dryrun"
+      end
+
+      def usage # :nodoc:
+        "#{program_name} [GEMNAME ...]"
+      end
+
+      def execute
+        say "Cleaning up installed gems..."
+        srcindex = Gem::SourceIndex.from_installed_gems
+        primary_gems = {}
+
+        srcindex.each do |name, spec|
+          if primary_gems[spec.name].nil? or primary_gems[spec.name].version < spec.version
+            primary_gems[spec.name] = spec
+          end
+        end
+
+        gems_to_cleanup = []
+
+        unless options[:args].empty? then
+          options[:args].each do |gem_name|
+            specs = Gem.cache.search(/^#{gem_name}$/i)
+            specs.each do |spec|
+              gems_to_cleanup << spec
+            end
+          end
+        else
+          srcindex.each do |name, spec|
+              gems_to_cleanup << spec
+          end
+        end
+
+        gems_to_cleanup = gems_to_cleanup.select { |spec|
+          primary_gems[spec.name].version != spec.version
+        }
+
+        uninstall_command = Gem::CommandManager.instance['uninstall']
+        deplist = DependencyList.new
+        gems_to_cleanup.uniq.each do |spec| deplist.add(spec) end
+
+        deplist.dependency_order.each do |spec|
+          if options[:dryrun] then
+            say "Dry Run Mode: Would uninstall #{spec.full_name}"
+          else
+            say "Attempting uninstall on #{spec.full_name}"
+
+            options[:args] = [spec.name]
+            options[:version] = "= #{spec.version}"
+            options[:executables] = true
+
+            uninstall_command.merge_options(options)
+
+            begin
+              uninstall_command.execute
+            rescue Gem::DependencyRemovalException => ex
+              say "Unable to uninstall #{spec.full_name} ... continuing with remaining gems"
+            end
+          end
+        end
+
+        say "Clean Up Complete"
+      end
+    end
+      
+  end
+end