module Gem::GemcutterUtilities

# File lib/rubygems/gemcutter_utilities.rb, line 12
def add_key_option
  add_option('-k', '--key KEYNAME', Symbol,
             'Use the given API key',
             'from ~/.gem/credentials') do |value,options|
    options[:key] = value
  end
end

# File lib/rubygems/gemcutter_utilities.rb, line 20
def api_key
  if options[:key] then
    verify_api_key options[:key]
  elsif Gem.configuration.api_keys.key?(host)
    Gem.configuration.api_keys[host]
  else
    Gem.configuration.rubygems_api_key
  end
end

# File lib/rubygems/gemcutter_utilities.rb, line 60
def host
  configured_host = Gem.host unless
    Gem.configuration.disable_default_gem_server

  @host ||=
    begin
      env_rubygems_host = ENV['RUBYGEMS_HOST']
      env_rubygems_host = nil if
        env_rubygems_host and env_rubygems_host.empty?

      env_rubygems_host|| configured_host
    end
end

# File lib/rubygems/gemcutter_utilities.rb, line 74
def rubygems_api_request(method, path, host = nil, &block)
  require 'net/http'

  self.host = host if host
  unless self.host
    alert_error "You must specify a gem server"
    terminate_interaction 1 # TODO: question this
  end

  uri = URI.parse "#{self.host}/#{path}"

  request_method = Net::HTTP.const_get method.to_s.capitalize

  Gem::RemoteFetcher.fetcher.request(uri, request_method, &block)
end

# File lib/rubygems/gemcutter_utilities.rb, line 30
def sign_in sign_in_host = nil
  sign_in_host ||= self.host
  return if Gem.configuration.rubygems_api_key

  pretty_host = if Gem::DEFAULT_HOST == sign_in_host then
                  'RubyGems.org'
                else
                  sign_in_host
                end

  say "Enter your #{pretty_host} credentials."
  say "Don't have an account yet? " +
      "Create one at #{sign_in_host}/sign_up"

  email    =              ask "   Email: "
  password = ask_for_password "Password: "
  say "\n"

  response = rubygems_api_request(:get, "api/v1/api_key",
                                  sign_in_host) do |request|
    request.basic_auth email, password
  end

  with_response response do |resp|
    say "Signed in."
    Gem.configuration.rubygems_api_key = resp.body
  end
end

# File lib/rubygems/gemcutter_utilities.rb, line 107
def verify_api_key(key)
  if Gem.configuration.api_keys.key? key then
    Gem.configuration.api_keys[key]
  else
    alert_error "No such API key. Please add it to your configuration (done automatically on initial `gem push`)."
    terminate_interaction 1 # TODO: question this
  end
end

# File lib/rubygems/gemcutter_utilities.rb, line 90
def with_response resp, error_prefix = nil
  case resp
  when Net::HTTPSuccess then
    if block_given? then
      yield resp
    else
      say resp.body
    end
  else
    message = resp.body
    message = "#{error_prefix}: #{message}" if error_prefix

    say message
    terminate_interaction 1 # TODO: question this
  end
end