1 files changed, 149 insertions, 0 deletions
diff --git a/lib/rubygems/commands/update_command.rb b/lib/rubygems/commands/update_command.rb
new file mode 100644
index 0000000000..e17ba2516a
--- /dev/null
+++ b/lib/rubygems/commands/update_command.rb
@@ -0,0 +1,149 @@
+require 'rubygems/command'
+require 'rubygems/install_update_options'
+require 'rubygems/local_remote_options'
+require 'rubygems/source_info_cache'
+require 'rubygems/version_option'
+
+module Gem
+  module Commands
+    class UpdateCommand < Command
+
+      include Gem::InstallUpdateOptions
+      include Gem::LocalRemoteOptions
+      include Gem::VersionOption
+
+      def initialize
+        super(
+          'update',
+          'Update the named gems (or all installed gems) in the local repository',
+          {
+            :generate_rdoc => true, 
+            :generate_ri => true, 
+            :force => false, 
+            :test => false,
+            :install_dir => Gem.dir
+          })
+
+        add_install_update_options
+
+        add_option('--system',
+          'Update the RubyGems system software') do |value, options|
+          options[:system] = value
+        end
+
+        add_local_remote_options
+
+        add_platform_option
+      end
+
+      def arguments # :nodoc:
+        "GEMNAME       name of gem to update"
+      end
+
+      def defaults_str # :nodoc:
+        "--rdoc --ri --no-force --no-test\n" +
+        "--install-dir #{Gem.dir}"
+      end
+
+      def usage # :nodoc:
+        "#{program_name} GEMNAME [GEMNAME ...]"
+      end
+
+      def execute
+        if options[:system] then
+          say "Updating RubyGems..."
+
+          unless options[:args].empty? then
+            fail "No gem names are allowed with the --system option"
+          end
+
+          options[:args] = ["rubygems-update"]
+        else
+          say "Updating installed gems..."
+        end
+
+        hig = highest_installed_gems = {}
+
+        Gem::SourceIndex.from_installed_gems.each do |name, spec|
+          if hig[spec.name].nil? or hig[spec.name].version < spec.version
+            hig[spec.name] = spec
+          end
+        end
+
+        remote_gemspecs = Gem::SourceInfoCache.search(//)
+
+        gems_to_update = if options[:args].empty? then
+                           which_to_update(highest_installed_gems, remote_gemspecs)
+                         else
+                           options[:args]
+                         end
+
+        options[:domain] = :remote # install from remote source
+
+        # HACK use the real API
+        install_command = Gem::CommandManager.instance['install']
+
+        gems_to_update.uniq.sort.each do |name|
+          say "Attempting remote update of #{name}"
+          options[:args] = [name]
+          options[:ignore_dependencies] = true # HACK skip seen gems instead
+          install_command.merge_options(options)
+          install_command.execute
+        end
+
+        if gems_to_update.include?("rubygems-update") then
+          latest_ruby_gem = remote_gemspecs.select { |s|
+            s.name == 'rubygems-update'
+          }.sort_by { |s|
+            s.version
+          }.last
+
+          say "Updating version of RubyGems to #{latest_ruby_gem.version}"
+          installed = do_rubygems_update(latest_ruby_gem.version.to_s)
+
+          say "RubyGems system software updated" if installed
+        else
+          say "Gems: [#{gems_to_update.uniq.sort.collect{|g| g.to_s}.join(', ')}] updated"
+        end
+      end
+
+      def do_rubygems_update(version_string)
+        args = []
+        args.push '--prefix', Gem.prefix unless Gem.prefix.nil?
+        args << '--no-rdoc' unless options[:generate_rdoc]
+        args << '--no-ri' unless options[:generate_ri]
+
+        update_dir = File.join(Gem.dir, 'gems',
+                               "rubygems-update-#{version_string}")
+
+        success = false
+
+        Dir.chdir update_dir do
+          say "Installing RubyGems #{version_string}"
+          setup_cmd = "#{Gem.ruby} setup.rb #{args.join ' '}"
+
+          # Make sure old rubygems isn't loaded
+          if Gem.win_platform? then
+            system "set RUBYOPT= & #{setup_cmd}"
+          else
+            system "RUBYOPT=\"\" #{setup_cmd}"
+          end
+        end
+      end
+
+      def which_to_update(highest_installed_gems, remote_gemspecs)
+        result = []
+        highest_installed_gems.each do |l_name, l_spec|
+          highest_remote_gem =
+            remote_gemspecs.select  { |spec| spec.name == l_name }.
+                            sort_by { |spec| spec.version }.
+                            last
+          if highest_remote_gem and l_spec.version < highest_remote_gem.version
+            result << l_name
+          end
+        end
+        result
+      end
+    end
+  end
+end