Utility methods for using the RubyGems API.
Add the –key option
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 21 def add_key_option add_option('-k', '--key KEYNAME', Symbol, 'Use the given API key', "from #{Gem.configuration.credentials_path}") do |value,options| options[:key] = value end end
Add the –otp option
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 32 def add_otp_option add_option('--otp CODE', 'Digit code for multifactor authentication', 'You can also use the environment variable GEM_HOST_OTP_CODE') do |value, options| options[:otp] = value end end
The API key from the command options or from the user's configuration.
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 43 def api_key if ENV["GEM_HOST_API_KEY"] ENV["GEM_HOST_API_KEY"] elsif options[:key] 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
The host to connect to either from the RUBYGEMS_HOST environment variable or from the user's configuration
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 66 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
The OTP code from the command options or from the user's configuration.
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 58 def otp options[:otp] || ENV["GEM_HOST_OTP_CODE"] end
Creates an RubyGems API to host
and path
with the given HTTP method
.
If allowed_push_host
metadata is present, then it will only allow that host.
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 85 def rubygems_api_request(method, path, host = nil, allowed_push_host = nil, scope: nil, &block) require 'net/http' self.host = host if host unless self.host alert_error "You must specify a gem server" terminate_interaction(ERROR_CODE) end if allowed_push_host allowed_host_uri = URI.parse(allowed_push_host) host_uri = URI.parse(self.host) unless (host_uri.scheme == allowed_host_uri.scheme) && (host_uri.host == allowed_host_uri.host) alert_error "#{self.host.inspect} is not allowed by the gemspec, which only allows #{allowed_push_host.inspect}" terminate_interaction(ERROR_CODE) end end uri = URI.parse "#{self.host}/#{path}" response = request_with_otp(method, uri, &block) if mfa_unauthorized?(response) ask_otp response = request_with_otp(method, uri, &block) end if api_key_forbidden?(response) update_scope(scope) request_with_otp(method, uri, &block) else response end end
Returns true when the user has enabled multifactor authentication from response
text and no otp provided by options.
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 221 def set_api_key(host, key) if host == Gem::DEFAULT_HOST Gem.configuration.rubygems_api_key = key else Gem.configuration.set_api_key host, key end end
Signs in with the RubyGems API at sign_in_host
and sets the rubygems API key.
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 150 def sign_in(sign_in_host = nil, scope: nil) sign_in_host ||= self.host return if api_key pretty_host = pretty_host(sign_in_host) 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" key_name = get_key_name(scope) scope_params = get_scope_params(scope) response = rubygems_api_request(:post, "api/v1/api_key", sign_in_host, scope: scope) do |request| request.basic_auth email, password request["OTP"] = otp if otp request.body = URI.encode_www_form({ name: key_name }.merge(scope_params)) end with_response response do |resp| say "Signed in with API key: #{key_name}." set_api_key host, resp.body end end
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 124 def update_scope(scope) sign_in_host = self.host pretty_host = pretty_host(sign_in_host) update_scope_params = { scope => true } say "The existing key doesn't have access of #{scope} on #{pretty_host}. Please sign in to update access." email = ask " Email: " password = ask_for_password "Password: " response = rubygems_api_request(:put, "api/v1/api_key", sign_in_host, scope: scope) do |request| request.basic_auth email, password request["OTP"] = otp if otp request.body = URI.encode_www_form({:api_key => api_key }.merge(update_scope_params)) end with_response response do |resp| say "Added #{scope} scope to the existing API key" end end
Retrieves the pre-configured API key key
or terminates interaction with an error.
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 184 def verify_api_key(key) if Gem.configuration.api_keys.key? key 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(ERROR_CODE) end end
If response
is an HTTP Success (2XX) response, yields the response if a block was given or shows the response body to the user.
If the response was not successful, shows an error to the user including the error_prefix
and the response body.
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 200 def with_response(response, error_prefix = nil) case response when Net::HTTPSuccess then if block_given? yield response else say clean_text(response.body) end else message = response.body message = "#{error_prefix}: #{message}" if error_prefix say clean_text(message) terminate_interaction(ERROR_CODE) end end
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 284 def api_key_forbidden?(response) response.kind_of?(Net::HTTPForbidden) && response.body.start_with?("The API key doesn't have access") end
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 240 def ask_otp say 'You have enabled multi-factor authentication. Please enter OTP code.' options[:otp] = ask 'Code: ' end
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 270 def get_key_name(scope) hostname = Socket.gethostname || "unknown-host" user = ENV["USER"] || ENV["USERNAME"] || "unknown-user" ts = Time.now.strftime("%Y%m%d%H%M%S") default_key_name = "#{hostname}-#{user}-#{ts}" key_name = ask "API Key name [#{default_key_name}]: " unless scope if key_name.nil? || key_name.empty? default_key_name else key_name end end
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 253 def get_scope_params(scope) scope_params = {} if scope scope_params = { scope => true } else say "Please select scopes you want to enable for the API key (y/n)" API_SCOPES.each do |scope| selected = ask "#{scope} [y/N]: " scope_params[scope] = true if selected =~ /^[yY](es)?$/ end say "\n" end scope_params end
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 245 def pretty_host(host) if Gem::DEFAULT_HOST == host 'RubyGems.org' else host end end
# File ruby-3.1.2/lib/rubygems/gemcutter_utilities.rb, line 231 def request_with_otp(method, uri, &block) request_method = Net::HTTP.const_get method.to_s.capitalize Gem::RemoteFetcher.fetcher.request(uri, request_method) do |req| req["OTP"] = otp if otp block.call(req) end end