pry--pry/lib/pry/default_commands/gems.rb

class Pry
  module DefaultCommands

    Gems = Pry::CommandSet.new do

      command "gem-install", "Install a gem and refresh the gem cache." do |gem_name|
        gem_home = Gem.instance_variable_get(:@gem_home)
        output.puts "Attempting to install gem: #{text.bold gem_name}"

        begin
          if File.writable?(gem_home)
            Gem::DependencyInstaller.new.install(gem_name)
            output.puts "Gem #{text.bold gem_name} successfully installed."
          else
            if system("sudo gem install #{gem_name}")
              output.puts "Gem #{text.bold gem_name} successfully installed."
            else
              output.puts "Gem #{text.bold gem_name} could not be installed."
              next
            end
          end
        rescue Gem::GemNotFoundException
          output.puts "Required Gem: #{text.bold gem_name} not found."
          next
        end

        Gem.refresh
        output.puts "Refreshed gem cache."
      end

      command "gem-cd", "Change working directory to specified gem's directory." do |gem|
        if gem
          spec = Gem.source_index.find_name(gem).first
          spec ? Dir.chdir(spec.full_gem_path) : output.puts("Gem `#{gem}` not found.")
        else
          output.puts 'gem-cd requires the name of a gem as an argument.'
        end
      end


      command "gem-list", "List/search installed gems. (Optional parameter: a regexp to limit the search)" do |pattern|
        pattern = Regexp.new pattern.to_s, Regexp::IGNORECASE
        gems = Gem.source_index.gems.values.group_by(&:name)

        gems.each do |gem, specs|
          if gem =~ pattern
            specs.sort! do |a,b| 
              Gem::Version.new(b.version) <=> Gem::Version.new(a.version) 
            end
            
            versions = specs.map.with_index do |spec, index|
              index == 0 ? text.bright_green(spec.version.to_s) : text.green(spec.version.to_s) 
            end

            output.puts "#{text.white gem} (#{versions.join ', '})"
          end
        end
      end

    end
  end
end