/tools/Ruby/lib/ruby/site_ruby/1.8/rubygems/commands/sources_command.rb
http://github.com/agross/netopenspace · Ruby · 136 lines · 105 code · 31 blank · 0 comment · 6 complexity · 139a6e3d5c24ce55af15302dc90c53b9 MD5 · raw file
- require 'rubygems/command'
- require 'rubygems/remote_fetcher'
- require 'rubygems/spec_fetcher'
- require 'rubygems/local_remote_options'
- class Gem::Commands::SourcesCommand < Gem::Command
- include Gem::LocalRemoteOptions
- def initialize
- require 'fileutils'
- super 'sources',
- 'Manage the sources and cache file RubyGems uses to search for gems'
- add_option '-a', '--add SOURCE_URI', 'Add source' do |value, options|
- options[:add] = value
- end
- add_option '-l', '--list', 'List sources' do |value, options|
- options[:list] = value
- end
- add_option '-r', '--remove SOURCE_URI', 'Remove source' do |value, options|
- options[:remove] = value
- end
- add_option '-c', '--clear-all',
- 'Remove all sources (clear the cache)' do |value, options|
- options[:clear_all] = value
- end
- add_option '-u', '--update', 'Update source cache' do |value, options|
- options[:update] = value
- end
- add_proxy_option
- end
- def defaults_str
- '--list'
- end
- def execute
- options[:list] = !(options[:add] ||
- options[:clear_all] ||
- options[:remove] ||
- options[:update])
- if options[:clear_all] then
- path = Gem::SpecFetcher.fetcher.dir
- FileUtils.rm_rf path
- unless File.exist? path then
- say "*** Removed specs cache ***"
- else
- unless File.writable? path then
- say "*** Unable to remove source cache (write protected) ***"
- else
- say "*** Unable to remove source cache ***"
- end
- terminate_interaction 1
- end
- end
- if options[:add] then
- source_uri = options[:add]
- uri = URI.parse source_uri
- begin
- Gem::SpecFetcher.fetcher.load_specs uri, 'specs'
- Gem.sources << source_uri
- Gem.configuration.write
- say "#{source_uri} added to sources"
- rescue URI::Error, ArgumentError
- say "#{source_uri} is not a URI"
- terminate_interaction 1
- rescue Gem::RemoteFetcher::FetchError => e
- say "Error fetching #{source_uri}:\n\t#{e.message}"
- terminate_interaction 1
- end
- end
- if options[:remove] then
- source_uri = options[:remove]
- unless Gem.sources.include? source_uri then
- say "source #{source_uri} not present in cache"
- else
- Gem.sources.delete source_uri
- Gem.configuration.write
- say "#{source_uri} removed from sources"
- end
- end
- if options[:update] then
- fetcher = Gem::SpecFetcher.fetcher
- Gem.sources.each do |update_uri|
- update_uri = URI.parse update_uri
- fetcher.load_specs update_uri, 'specs'
- fetcher.load_specs update_uri, 'latest_specs'
- end
- say "source cache successfully updated"
- end
- if options[:list] then
- say "*** CURRENT SOURCES ***"
- say
- Gem.sources.each do |source|
- say source
- end
- end
- end
- private
- def remove_cache_file(desc, path)
- FileUtils.rm_rf path
- if not File.exist?(path) then
- say "*** Removed #{desc} source cache ***"
- elsif not File.writable?(path) then
- say "*** Unable to remove #{desc} source cache (write protected) ***"
- else
- say "*** Unable to remove #{desc} source cache ***"
- end
- end
- end