85 lines
2.5 KiB
Ruby
85 lines
2.5 KiB
Ruby
require 'rubygems'
|
|
|
|
class Pry
|
|
module Rubygem
|
|
|
|
class << self
|
|
def installed?(name)
|
|
if Gem::Specification.respond_to?(:find_all_by_name)
|
|
Gem::Specification.find_all_by_name(name).any?
|
|
else
|
|
Gem.source_index.find_name(name).first
|
|
end
|
|
end
|
|
|
|
# Get the gem spec object for the given gem name.
|
|
#
|
|
# @param [String] name
|
|
# @return [Gem::Specification]
|
|
def spec(name)
|
|
specs = if Gem::Specification.respond_to?(:each)
|
|
Gem::Specification.find_all_by_name(name)
|
|
else
|
|
Gem.source_index.find_name(name)
|
|
end
|
|
|
|
first_spec = specs.sort_by { |spec| Gem::Version.new(spec.version) }.last
|
|
|
|
first_spec || raise(CommandError, "Gem `#{name}` not found")
|
|
end
|
|
|
|
# List gems matching a pattern.
|
|
#
|
|
# @param [Regexp] pattern
|
|
# @return [Array<Gem::Specification>]
|
|
def list(pattern = /.*/)
|
|
if Gem::Specification.respond_to?(:each)
|
|
Gem::Specification.select { |spec| spec.name =~ pattern }
|
|
else
|
|
Gem.source_index.gems.values.select { |spec| spec.name =~ pattern }
|
|
end
|
|
end
|
|
|
|
# Completion function for gem-cd and gem-open.
|
|
#
|
|
# @param [String] so_far what the user's typed so far
|
|
# @return [Array<String>] completions
|
|
def complete(so_far)
|
|
if so_far =~ / ([^ ]*)\z/
|
|
self.list(%r{\A#{$2}}).map(&:name)
|
|
else
|
|
self.list.map(&:name)
|
|
end
|
|
end
|
|
|
|
# Installs a gem with all its dependencies.
|
|
#
|
|
# @param [String] name
|
|
# @return [void]
|
|
def install(name)
|
|
require 'rubygems/dependency_installer'
|
|
gem_config = Gem.configuration['gem']
|
|
gemrc_opts = (gem_config.nil? ? "" : gem_config.split(' '))
|
|
destination = if gemrc_opts.include?('--user-install')
|
|
Gem.user_dir
|
|
elsif File.writable?(Gem.dir)
|
|
Gem.dir
|
|
else
|
|
Gem.user_dir
|
|
end
|
|
installer = Gem::DependencyInstaller.new(install_dir: destination)
|
|
installer.install(name)
|
|
rescue Errno::EACCES
|
|
raise CommandError,
|
|
"Insufficient permissions to install #{green(name)}."
|
|
rescue Gem::GemNotFoundException
|
|
raise CommandError,
|
|
"Gem #{green(name)} not found. Aborting installation."
|
|
else
|
|
Gem.refresh
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|