module MiGA::Cli::Action::Download::Gtdb

# File lib/miga/cli/action/download/gtdb.rb, line 17
def cli_name_modifiers(opt)
  opt.on(
    '--no-version-name',
    'Do not add sequence version to the dataset name'
  ) { |v| cli[:add_version] = v }
end

# File lib/miga/cli/action/download/gtdb.rb, line 10
def cli_task_flags(opt)
  cli.opt_flag(
    opt, 'reference',
    'Download only reference genomes. By default: download all'
  )
end

# File lib/miga/cli/action/download/gtdb.rb, line 32
def remote_list
  cli.say 'Downloading genome list'
  extra = { sp_reps_only: cli[:reference].to_s }
  json = MiGA::RemoteDataset.download(
    :gtdb, :taxon, cli[:taxon], :genomes, nil, extra
  )
  doc = MiGA::Json.parse(json, contents: true)

  Hash[
    doc.map do |acc|
      [
        remote_row_name(acc),
        {
          ids: [acc], db: :assembly, universe: :gtdb,
          md: {
            type: :genome, gtdb_assembly: acc,
            get_ncbi_taxonomy: cli[:get_ncbi_taxonomy]
          }
        }
      ]
    end
  ]
end

# File lib/miga/cli/action/download/gtdb.rb, line 56
def remote_row_name(asm)
  acc = asm.to_s
  acc.gsub!(/\.\d+\Z/, '') unless cli[:add_version]
  acc.miga_name
end

# File lib/miga/cli/action/download/gtdb.rb, line 24
def sanitize_cli
  cli.ensure_par(taxon: '-T')
  unless cli[:taxon] =~ /^[a-z]__\S+$/
    raise 'Taxon (-T) must be in GTDB format: s__Macondimonas_diazotrophica'
  end
  cli[:save_every] = 1 if cli[:dry]
end