diff options
Diffstat (limited to 'lib/rubygems/commands/yank_command.rb')
-rw-r--r-- | lib/rubygems/commands/yank_command.rb | 98 |
1 files changed, 98 insertions, 0 deletions
diff --git a/lib/rubygems/commands/yank_command.rb b/lib/rubygems/commands/yank_command.rb new file mode 100644 index 0000000000..da0cf7ad3b --- /dev/null +++ b/lib/rubygems/commands/yank_command.rb @@ -0,0 +1,98 @@ +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 + |