module Gem::QueryUtils

# File lib/rubygems/query_utils.rb, line 13
def add_query_options
  add_option("-i", "--[no-]installed",
             "Check for installed gem") do |value, options|
    options[:installed] = value
  end

  add_option("-I", "Equivalent to --no-installed") do |_value, options|
    options[:installed] = false
  end

  add_version_option command, "for use with --installed"

  add_option("-d", "--[no-]details",
             "Display detailed information of gem(s)") do |value, options|
    options[:details] = value
  end

  add_option("--[no-]versions",
             "Display only gem names") do |value, options|
    options[:versions] = value
    options[:details] = false unless value
  end

  add_option("-a", "--all",
             "Display all gem versions") do |value, options|
    options[:all] = value
  end

  add_option("-e", "--exact",
             "Name of gem(s) to query on matches the",
             "provided STRING") do |value, options|
    options[:exact] = value
  end

  add_option("--[no-]prerelease",
             "Display prerelease versions") do |value, options|
    options[:prerelease] = value
  end

  add_local_remote_options
end

# File lib/rubygems/query_utils.rb, line 59
def execute
  gem_names = if args.empty?
    [options[:name]]
  else
    options[:exact] ? args.map {|arg| /\A#{Regexp.escape(arg)}\Z/ } : args.map {|arg| /#{arg}/i }
  end

  terminate_interaction(check_installed_gems(gem_names)) if check_installed_gems?

  gem_names.each {|n| show_gems(n) }
end

# File lib/rubygems/query_utils.rb, line 109
def args
  options[:args].to_a
end

# File lib/rubygems/query_utils.rb, line 73
def check_installed_gems(gem_names)
  exit_code = 0

  if args.empty? && !gem_name?
    alert_error "You must specify a gem name"
    exit_code = 4
  elsif gem_names.count > 1
    alert_error "You must specify only ONE gem!"
    exit_code = 4
  else
    installed = installed?(gem_names.first, options[:version])
    installed = !installed unless options[:installed]

    say(installed)
    exit_code = 1 unless installed
  end

  exit_code
end

# File lib/rubygems/query_utils.rb, line 93
def check_installed_gems?
  !options[:installed].nil?
end

# File lib/rubygems/query_utils.rb, line 113
def display_header(type)
  if (ui.outs.tty? && Gem.configuration.verbose) || both?
    say
    say "*** #{type} GEMS ***"
    say
  end
end

# File lib/rubygems/query_utils.rb, line 231
def entry_details(entry, detail_tuple, specs, platforms, platform_ruby_abis)
  return unless options[:details]

  name_tuple, spec = detail_tuple

  spec = spec.fetch_spec(name_tuple)if spec.respond_to?(:fetch_spec)

  entry << "\n"

  if ruby_abi_metadata?(platform_ruby_abis)
    spec_platform_ruby_abis entry, platforms, platform_ruby_abis
  else
    spec_platforms entry, platforms
  end
  spec_authors     entry, spec
  spec_homepage    entry, spec
  spec_license     entry, spec
  spec_loaded_from entry, spec, specs
  spec_summary     entry, spec
end

# File lib/rubygems/query_utils.rb, line 252
def entry_versions(entry, name_tuples, platforms, platform_ruby_abis, specs)
  return unless options[:versions]

  list =
    if platforms.empty? || options[:details]
      name_tuples.map(&:version).uniq
    elsif ruby_abi_metadata?(platform_ruby_abis)
      platforms.sort.reverse.flat_map do |version, pls|
        out = version_label(version, specs)
        labels = version_platform_labels(version, pls, platform_ruby_abis, label_platform: true)
        labels.empty? ? [out] : labels.map {|label| "#{out} #{label}" }
      end
    else
      platforms.sort.reverse.map do |version, pls|
        out = version_label(version, specs)
        labels = version_platform_labels(version, pls, platform_ruby_abis)
        labels.empty? ? out : "#{out} #{labels.join(" ")}"
      end
    end

  use_multiline_separator = !options[:details] && ruby_abi_metadata?(platform_ruby_abis) && list.length > 1
  separator = use_multiline_separator ? "\n#{" " * (entry.first.length + 2)}" : ", "
  entry << " (#{list.join separator})"
end

# File lib/rubygems/query_utils.rb, line 97
def gem_name?
  !options[:name].nil?
end

# File lib/rubygems/query_utils.rb, line 183
def installed?(name, req = Gem::Requirement.default)
  Gem::Specification.any? {|s| s.name =~ name && req =~ s.version }
end

# File lib/rubygems/query_utils.rb, line 308
def make_entry(entry_tuples, platforms, platform_ruby_abis)
  detail_tuple = entry_tuples.first

  name_tuples, specs = entry_tuples.flatten.partition do |item|
    Gem::NameTuple === item
  end

  entry = [name_tuples.first.name]

  entry_versions(entry, name_tuples, platforms, platform_ruby_abis, specs)
  entry_details(entry, detail_tuple, specs, platforms, platform_ruby_abis)

  entry.join
end

# File lib/rubygems/query_utils.rb, line 187
def output_query_results(spec_tuples)
  output = []
  versions = Hash.new {|h,name| h[name] = [] }

  spec_tuples.each do |spec_tuple, source|
    versions[spec_tuple.name] << [spec_tuple, source]
  end

  versions = versions.sort_by do |(n,_),_|
    n.downcase
  end

  output_versions output, versions

  say output.join(options[:details] ? "\n\n" : "\n")
end

# File lib/rubygems/query_utils.rb, line 204
def output_versions(output, versions)
  versions.each do |_gem_name, matching_tuples|
    matching_tuples = matching_tuples.sort_by {|n,_| n.version }.reverse

    platforms = Hash.new {|h,version| h[version] = [] }
    platform_ruby_abis = Hash.new {|h,version| h[version] = Hash.new {|hh,platform| hh[platform] = [] } }

    matching_tuples.each do |n, _|
      platforms[n.version] << n.platform if n.platform
      platform_ruby_abis[n.version][n.platform] << n.ruby_abi if n.ruby_abi
    end

    seen = {}

    matching_tuples.delete_if do |n,_|
      if seen[n.version]
        true
      else
        seen[n.version] = true
        false
      end
    end

    output << clean_text(make_entry(matching_tuples, platforms, platform_ruby_abis))
  end
end

# File lib/rubygems/query_utils.rb, line 101
def prerelease
  options[:prerelease]
end

# File lib/rubygems/query_utils.rb, line 287
def ruby_abi_metadata?(platform_ruby_abis)
  platform_ruby_abis.values.any? do |ruby_abis_by_platform|
    ruby_abis_by_platform.values.any?(&:any?)
  end
end

# File lib/rubygems/query_utils.rb, line 122
def show_gems(name)
  show_local_gems(name)  if local?
  show_remote_gems(name) if remote?
end

# File lib/rubygems/query_utils.rb, line 127
def show_local_gems(name, req = Gem::Requirement.default)
  display_header("LOCAL")

  specs = Gem::Specification.find_all do |s|
    name_matches = name ? s.name =~ name : true
    version_matches = show_prereleases? || !s.version.prerelease?

    name_matches && version_matches
  end.uniq(&:full_name)

  spec_tuples = specs.map do |spec|
    [spec.name_tuple, spec]
  end

  output_query_results(spec_tuples)
end

# File lib/rubygems/query_utils.rb, line 105
def show_prereleases?
  prerelease.nil? || prerelease
end

# File lib/rubygems/query_utils.rb, line 144
def show_remote_gems(name)
  display_header("REMOTE")

  fetcher = Gem::SpecFetcher.fetcher

  spec_tuples = if name.nil?
    fetcher.detect(specs_type) { true }
  else
    matching_tuples = fetcher.detect(specs_type) do |name_tuple|
      name === name_tuple.name && options[:version].satisfied_by?(name_tuple.version)
    end

    if args.empty?
      matching_tuples
    else
      fetcher.decode_content_addressable_tuples(matching_tuples, latest: specs_type == :latest)
    end
  end

  output_query_results(spec_tuples)
end

# File lib/rubygems/query_utils.rb, line 323
def spec_authors(entry, spec)
  authors = "Author#{spec.authors.length > 1 ? "s" : ""}: ".dup
  authors << spec.authors.join(", ")
  entry << format_text(authors, 68, 4)
end

# File lib/rubygems/query_utils.rb, line 329
def spec_homepage(entry, spec)
  return if spec.homepage.nil? || spec.homepage.empty?

  entry << "\n" << format_text("Homepage: #{spec.homepage}", 68, 4)
end

# File lib/rubygems/query_utils.rb, line 335
def spec_license(entry, spec)
  return if spec.license.nil? || spec.license.empty?

  licenses = "License#{spec.licenses.length > 1 ? "s" : ""}: ".dup
  licenses << spec.licenses.join(", ")
  entry << "\n" << format_text(licenses, 68, 4)
end

# File lib/rubygems/query_utils.rb, line 343
def spec_loaded_from(entry, spec, specs)
  return unless spec.loaded_from

  if specs.length == 1
    default = spec.default_gem? ? " (default)" : nil
    entry << "\n" << "    Installed at#{default}: #{spec.base_dir}"
  else
    label = "Installed at"
    specs.each do |s|
      version = s.version.to_s
      default = s.default_gem? ? ", default" : ""
      entry << "\n" << "    #{label} (#{version}#{default}): #{s.base_dir}"
      label = " " * label.length
    end
  end
end

# File lib/rubygems/query_utils.rb, line 385
def spec_platform_ruby_abis(entry, platforms, platform_ruby_abis)
  entry << "    Platforms:\n"

  platforms.sort.each do |version, pls|
    labels = version_platform_labels(version, pls, platform_ruby_abis)
    next if labels.empty?

    if platforms.length == 1
      labels.each do |label|
        entry << "        #{label}\n"
      end
    else
      label = "        #{version}: "
      data = format_text labels.join(", "), 68, label.length
      data[0, label.length] = label
      entry << data << "\n"
    end
  end
end

# File lib/rubygems/query_utils.rb, line 360
def spec_platforms(entry, platforms)
  platforms = platforms.transform_values(&:uniq)
  non_ruby = platforms.any? do |_, pls|
    pls.any? {|pl| pl != Gem::Platform::RUBY }
  end

  return unless non_ruby

  if platforms.length == 1
    title = platforms.values.first.length == 1 ? "Platform" : "Platforms"
    entry << "    #{title}: #{platforms.values.sort.join(", ")}\n"
  else
    entry << "    Platforms:\n"

    sorted_platforms = platforms.sort

    sorted_platforms.each do |version, pls|
      label = "        #{version}: "
      data = format_text pls.sort.join(", "), 68, label.length
      data[0, label.length] = label
      entry << data << "\n"
    end
  end
end

# File lib/rubygems/query_utils.rb, line 405
def spec_summary(entry, spec)
  summary = truncate_text(spec.summary, "the summary for #{spec.full_name}")
  entry << "\n\n" << format_text(summary, 68, 4)
end

# File lib/rubygems/query_utils.rb, line 166
def specs_type
  if options[:all] || options[:version].specific?
    if options[:prerelease]
      :complete
    else
      :released
    end
  elsif options[:prerelease]
    :prerelease
  else
    :latest
  end
end

# File lib/rubygems/query_utils.rb, line 277
def version_label(version, specs)
  out = version.to_s
  return out unless options[:domain] == :local

  default = specs.any? do |s|
    !s.is_a?(Gem::Source) && s.version == version && s.default_gem?
  end
  default ? "default: #{out}" : out
end

# File lib/rubygems/query_utils.rb, line 293
def version_platform_labels(version, platforms, platform_ruby_abis, label_platform: false)
  platforms = platforms.uniq
  return [] if platforms == [Gem::Platform::RUBY]

  platforms = [platforms.delete(Gem::Platform::RUBY), *platforms.sort].compact
  platforms.map do |platform|
    ruby_abis = platform_ruby_abis[version][platform].uniq.sort
    platform_label = label_platform ? "Platform: #{platform}" : platform
    next platform_label if ruby_abis.empty?

    separator = label_platform ? ", " : " "
    "#{platform_label}#{separator}Ruby ABI: #{ruby_abis.join(", ")}"
  end
end