class Gem::Commands::YankCommand

# File lib/rubygems/commands/yank_command.rb, line 23
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

# File lib/rubygems/commands/yank_command.rb, line 39
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

# File lib/rubygems/commands/yank_command.rb, line 64
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

# File lib/rubygems/commands/yank_command.rb, line 59
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

# File lib/rubygems/commands/yank_command.rb, line 93
def get_platform_from_requirements(requirements)
  Gem.platforms[1].to_s if requirements.key? :added_platform
end

# File lib/rubygems/commands/yank_command.rb, line 87
def get_version_from_requirements(requirements)
  requirements.requirements.first[1].version
rescue
  nil
end

# File lib/rubygems/commands/yank_command.rb, line 71
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