Import RubyGems trunk revision 1493.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@13862 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 106 insertions, 0 deletions

diff --git a/lib/rubygems/local_remote_options.rb b/lib/rubygems/local_remote_options.rb
new file mode 100644
index 0000000000..1a5410bef7
--- /dev/null
+++ b/lib/rubygems/local_remote_options.rb
@@ -0,0 +1,106 @@
+#--
+# Copyright 2006 by Chad Fowler, Rich Kilmer, Jim Weirich and others.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#++
+
+require 'rubygems'
+
+# Mixin methods for local and remote Gem::Command options.
+module Gem::LocalRemoteOptions
+
+  # Allows OptionParser to handle HTTP URIs.
+  def accept_uri_http
+    OptionParser.accept URI::HTTP do |value|
+      begin
+        value = URI.parse value
+      rescue URI::InvalidURIError
+        raise OptionParser::InvalidArgument, value
+      end
+
+      raise OptionParser::InvalidArgument, value unless value.scheme == 'http'
+
+      value
+    end
+  end
+
+  # Add local/remote options to the command line parser.
+  def add_local_remote_options
+    add_option(:"Local/Remote", '-l', '--local',
+               'Restrict operations to the LOCAL domain') do |value, options|
+      options[:domain] = :local
+    end
+
+    add_option(:"Local/Remote", '-r', '--remote',
+      'Restrict operations to the REMOTE domain') do |value, options|
+      options[:domain] = :remote
+    end
+
+    add_option(:"Local/Remote", '-b', '--both',
+               'Allow LOCAL and REMOTE operations') do |value, options|
+      options[:domain] = :both
+    end
+
+    add_bulk_threshold_option
+    add_source_option
+    add_proxy_option
+    add_update_sources_option
+  end
+
+  # Add the --bulk-threshold option
+  def add_bulk_threshold_option
+    add_option(:"Local/Remote", '-B', '--bulk-threshold COUNT',
+               "Threshold for switching to bulk",
+               "synchronization (default #{Gem.configuration.bulk_threshold})") do
+      |value, options|
+      Gem.configuration.bulk_threshold = value.to_i
+    end
+  end
+
+  # Add the --http-proxy option
+  def add_proxy_option
+    accept_uri_http
+
+    add_option(:"Local/Remote", '-p', '--[no-]http-proxy [URL]', URI::HTTP,
+               'Use HTTP proxy for remote operations') do |value, options|
+      options[:http_proxy] = (value == false) ? :no_proxy : value
+      Gem.configuration[:http_proxy] = options[:http_proxy]
+    end
+  end
+
+  # Add the --source option
+  def add_source_option
+    accept_uri_http
+
+    add_option(:"Local/Remote", '--source URL', URI::HTTP,
+               'Use URL as the remote source for gems') do |value, options|
+      if options[:added_source] then
+        Gem.sources << value
+      else
+        options[:added_source] = true
+        Gem.sources.replace [value]
+      end
+    end
+  end
+
+  # Add the --source option
+  def add_update_sources_option
+
+    add_option(:"Local/Remote", '-u', '--[no-]update-sources',
+               'Update local source cache') do |value, options|
+      Gem.configuration.update_sources = value
+    end
+  end
+
+  # Is local fetching enabled?
+  def local?
+    options[:domain] == :local || options[:domain] == :both
+  end
+
+  # Is remote fetching enabled?
+  def remote?
+    options[:domain] == :remote || options[:domain] == :both
+  end
+
+end
+