Current File : //opt/alt/ruby21/lib64/ruby/2.1.0/rubygems/commands/cleanup_command.rb
require 'rubygems/command'
require 'rubygems/dependency_list'
require 'rubygems/uninstaller'
class Gem::Commands::CleanupCommand < Gem::Command
def initialize
super 'cleanup',
'Clean up old versions of installed gems',
:force => false, :install_dir => Gem.dir
add_option('-n', '-d', '--dryrun',
'Do not uninstall gems') do |value, options|
options[:dryrun] = true
end
@candidate_gems = nil
@default_gems = []
@full = nil
@gems_to_cleanup = nil
@original_home = nil
@original_path = nil
@primary_gems = nil
end
def arguments # :nodoc:
"GEMNAME name of gem to cleanup"
end
def defaults_str # :nodoc:
"--no-dryrun"
end
def description # :nodoc:
<<-EOF
The cleanup command removes old versions of gems from GEM_HOME that are not
required to meet a dependency. If a gem is installed elsewhere in GEM_PATH
the cleanup command won't delete it.
If no gems are named all gems in GEM_HOME are cleaned.
EOF
end
def usage # :nodoc:
"#{program_name} [GEMNAME ...]"
end
def execute
say "Cleaning up installed gems..."
if options[:args].empty? then
done = false
last_set = nil
until done do
clean_gems
this_set = @gems_to_cleanup.map { |spec| spec.full_name }.sort
done = this_set.empty? || last_set == this_set
last_set = this_set
end
else
clean_gems
end
say "Clean Up Complete"
if Gem.configuration.really_verbose then
skipped = @default_gems.map { |spec| spec.full_name }
say "Skipped default gems: #{skipped.join ', '}"
end
end
def clean_gems
get_primary_gems
get_candidate_gems
get_gems_to_cleanup
@full = Gem::DependencyList.from_specs
deplist = Gem::DependencyList.new
@gems_to_cleanup.each do |spec| deplist.add spec end
deps = deplist.strongly_connected_components.flatten
@original_home = Gem.dir
@original_path = Gem.path
deps.reverse_each do |spec|
uninstall_dep spec
end
Gem::Specification.reset
end
def get_candidate_gems
@candidate_gems = unless options[:args].empty? then
options[:args].map do |gem_name|
Gem::Specification.find_all_by_name gem_name
end.flatten
else
Gem::Specification.to_a
end
end
def get_gems_to_cleanup
gems_to_cleanup = @candidate_gems.select { |spec|
@primary_gems[spec.name].version != spec.version
}
default_gems, gems_to_cleanup = gems_to_cleanup.partition { |spec|
spec.default_gem?
}
@default_gems += default_gems
@default_gems.uniq!
@gems_to_cleanup = gems_to_cleanup.uniq
end
def get_primary_gems
@primary_gems = {}
Gem::Specification.each do |spec|
if @primary_gems[spec.name].nil? or
@primary_gems[spec.name].version < spec.version then
@primary_gems[spec.name] = spec
end
end
end
def uninstall_dep spec
return unless @full.ok_to_remove?(spec.full_name)
if options[:dryrun] then
say "Dry Run Mode: Would uninstall #{spec.full_name}"
return
end
say "Attempting to uninstall #{spec.full_name}"
uninstall_options = {
:executables => false,
:version => "= #{spec.version}",
}
uninstall_options[:user_install] = Gem.user_dir == spec.base_dir
uninstaller = Gem::Uninstaller.new spec.name, uninstall_options
begin
uninstaller.uninstall
rescue Gem::DependencyRemovalException, Gem::InstallError,
Gem::GemNotInHomeException, Gem::FilePermissionError => e
say "Unable to uninstall #{spec.full_name}:"
say "\t#{e.class}: #{e.message}"
end
ensure
# Restore path Gem::Uninstaller may have changed
Gem.use_paths @original_home, *@original_path
end
end
Mini Shell