JezK
Edit File: yank_command.rb
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: <<-EOF The yank command removes a gem you pushed to a server from the server's index. Note that if you push a gem to rubygems.org the yank command does not prevent other people from downloading the gem via the download link. Once you have pushed a gem several downloads will happen automatically via the webhooks. If you accidentally pushed passwords or other sensitive data you will need to change them immediately and yank your gem. If you are yanking a gem due to intellectual property reasons contact http://help.rubygems.org for permanant removal. Be sure to mention this as the reason for the removal request. EOF 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', 'Remove a pushed gem from the index' 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