yank_command.rb   [plain text]

require 'rubygems/command'
require 'rubygems/local_remote_options'
require 'rubygems/version_option'
require 'rubygems/gemcutter_utilities'

class Gem::Commands::YankCommand < Gem::Command
  include Gem::LocalRemoteOptions
  include Gem::VersionOption
  include Gem::GemcutterUtilities

  def description # :nodoc:
    'Remove a specific gem version release from RubyGems.org'
  end

  def arguments # :nodoc:
    "GEM       name of gem"
  end

  def usage # :nodoc:
    "#{program_name} GEM -v VERSION [-p PLATFORM] [--undo] [--key KEY_NAME]"
  end

  def initialize
    super 'yank', description

    add_version_option("remove")
    add_platform_option("remove")

    add_option('--undo') do |value, options|
      options[:undo] = true
    end

    add_option('-k', '--key KEY_NAME',
               'Use API key from your gem credentials file') do |value, options|
      options[:key] = value
    end
  end

  def execute
    sign_in

    version   = get_version_from_requirements(options[:version])
    platform  = get_platform_from_requirements(options)
    api_key   = Gem.configuration.rubygems_api_key
    api_key   = Gem.configuration.api_keys[options[:key].to_sym] if options[:key]

    if version then
      if options[:undo] then
        unyank_gem(version, platform, api_key)
      else
        yank_gem(version, platform, api_key)
      end
    else
      say "A version argument is required: #{usage}"
      terminate_interaction
    end
  end

  def yank_gem(version, platform, api_key)
    say "Yanking gem from #{self.host}..."
    yank_api_request(:delete, version, platform, "api/v1/gems/yank", api_key)
  end

  def unyank_gem(version, platform, api_key)
    say "Unyanking gem from #{host}..."
    yank_api_request(:put, version, platform, "api/v1/gems/unyank", api_key)
  end

  private

  def yank_api_request(method, version, platform, api, api_key)
    name = get_one_gem_name
    response = rubygems_api_request(method, api) do |request|
      request.add_field("Authorization", api_key)

      data = {
        'gem_name' => name,
        'version' => version,
      }
      data['platform'] = platform if platform

      request.set_form_data data
    end
    say response.body
  end

  def get_version_from_requirements(requirements)
    requirements.requirements.first[1].version
  rescue
    nil
  end

  def get_platform_from_requirements(requirements)
    Gem.platforms[1].to_s if requirements.key? :added_platform
  end

end