1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/lib/rubygems/commands/push_command.rb
Thomas McDonald 12ac0fa939 [rubygems/rubygems] Simplify host switching logic
This block was a branch and variable heavy way of saying use the
user-provided host if available, or fall-back to the default host if
none of the other options match. IMO the resultant single-if condition
is clearer in intentions

42dbb71cfa
2020-05-08 14:13:29 +09:00

118 lines
2.8 KiB
Ruby

# frozen_string_literal: true
require 'rubygems/command'
require 'rubygems/local_remote_options'
require 'rubygems/gemcutter_utilities'
require 'rubygems/package'
class Gem::Commands::PushCommand < Gem::Command
include Gem::LocalRemoteOptions
include Gem::GemcutterUtilities
def description # :nodoc:
<<-EOF
The push command uploads a gem to the push server (the default is
https://rubygems.org) and adds it to the index.
The gem can be removed from the index and deleted from the server using the yank
command. For further discussion see the help for the yank command.
The push command will use ~/.gem/credentials to authenticate to a server, but you can use the RubyGems environment variable GEM_HOST_API_KEY to set the api key to authenticate.
EOF
end
def arguments # :nodoc:
"GEM built gem to push up"
end
def usage # :nodoc:
"#{program_name} GEM"
end
def initialize
super 'push', 'Push a gem up to the gem server', :host => self.host
@user_defined_host = false
add_proxy_option
add_key_option
add_otp_option
add_option('--host HOST',
'Push to another gemcutter-compatible host',
' (e.g. https://rubygems.org)') do |value, options|
options[:host] = value
@user_defined_host = true
end
@host = nil
end
def execute
gem_name = get_one_gem_name
default_gem_server, push_host = get_hosts_for(gem_name)
@host = if @user_defined_host
options[:host]
elsif default_gem_server
default_gem_server
elsif push_host
push_host
else
options[:host]
end
sign_in @host
send_gem(gem_name)
end
def send_gem(name)
args = [:post, "api/v1/gems"]
gem_data = Gem::Package.new(name)
unless @host
@host = gem_data.spec.metadata['default_gem_server']
end
push_host = nil
if gem_data.spec.metadata.has_key?('allowed_push_host')
push_host = gem_data.spec.metadata['allowed_push_host']
end
@host ||= push_host
# Always include @host, even if it's nil
args += [ @host, push_host ]
say "Pushing gem to #{@host || Gem.host}..."
response = send_push_request(name, args)
with_response response
end
private
def send_push_request(name, args)
rubygems_api_request(*args) do |request|
request.body = Gem.read_binary name
request.add_field "Content-Length", request.body.size
request.add_field "Content-Type", "application/octet-stream"
request.add_field "Authorization", api_key
request.add_field "OTP", options[:otp] if options[:otp]
end
end
def get_hosts_for(name)
gem_metadata = Gem::Package.new(name).spec.metadata
[
gem_metadata["default_gem_server"],
gem_metadata["allowed_push_host"]
]
end
end