class Gem::Commands::PushCommand

def arguments # :nodoc:
  "GEM       built gem to push up"
end

def description # :nodoc:
  <<-EOF
e push command uploads a gem to the push server (the default is
tps://rubygems.org) and adds it to the index.
e gem can be removed from the index and deleted from the server using the yank
mmand.  For further discussion see the help for the yank command.
e 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 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, scope: get_push_scope
  send_gem(gem_name)
end

def get_attestations_part
  bundles = "[" + options[:attestations].map do |attestation|
    Gem.read_binary(attestation)
  end.join(",") + "]"
  [
    "attestations",
    bundles,
    { content_type: "application/json" },
  ]
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

def get_push_scope
  :push_rubygem
end

def initialize
  super "push", "Push a gem up to the gem server", host: host, attestations: []
  @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
  add_option("--attestation FILE",
              "Push with sigstore attestations") do |value, options|
    options[:attestations] << value
  end
  @host = nil
end

def send_gem(name)
  args = [:post, "api/v1/gems"]
  _, push_host = get_hosts_for(name)
  @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

def send_push_request(name, args)
  scope = get_push_scope
  rubygems_api_request(*args, scope: scope) do |request|
    body = Gem.read_binary name
    if options[:attestations].any?
      request.set_form([
        ["gem", body, { filename: name, content_type: "application/octet-stream" }],
        get_attestations_part,
      ], "multipart/form-data")
    else
      request.body = body
      request.add_field "Content-Type",   "application/octet-stream"
      request.add_field "Content-Length", request.body.size
    end
    request.add_field "Authorization", api_key
  end
end

def usage # :nodoc:
  "#{program_name} GEM"
end