class MiGA::Cli::Action::Init

ask_for_mytaxa() click to toggle source

# File lib/miga/cli/action/init.rb, line 181
def ask_for_mytaxa
  if cli[:mytaxa].nil?
    cli[:mytaxa] =
      cli.ask_user(
        'Should I include MyTaxa modules?',
        'yes', %w(yes no)
      ) == 'yes'
  end
end

check_additional_files(paths) click to toggle source

# File lib/miga/cli/action/init.rb, line 212
def check_additional_files(paths)
  if cli[:mytaxa]
    cli.puts 'Looking for MyTaxa databases:'
    mt = File.dirname paths['MyTaxa']
    cli.print 'Looking for scores... '
    unless Dir.exist?(File.expand_path('db', mt))
      cli.puts "no\nExecute 'python2 #{mt}/utils/download_db.py'"
      exit(1)
    end
    cli.puts 'yes'
    cli.print 'Looking for diamond db... '
    unless File.exist?(File.expand_path('AllGenomes.faa.dmnd', mt))
      cli.puts "no\nDownload "            "'http://enve-omics.ce.gatech.edu/data/public_mytaxa/"            "AllGenomes.faa.dmnd' into #{mt}"
      exit(1)
    end
    cli.puts ''
  end
end

check_configuration_script(rc_fh) click to toggle source

# File lib/miga/cli/action/init.rb, line 139
def check_configuration_script(rc_fh)
  unless File.exist? cli[:config]
    cli[:config] = cli.ask_user(
      'Is there a script I need to load at startup?',
      cli[:config]
    )
  end
  if File.exist? cli[:config]
    cli[:config] = File.expand_path(cli[:config])
    cli.puts "Found bash configuration script: #{cli[:config]}"
    rc_fh.puts "MIGA_STARTUP='#{cli[:config]}'"
    rc_fh.puts '. "$MIGA_STARTUP"'
  else
    cli[:config] = '/dev/null'
  end
  cli.puts ''
end

check_r_packages(paths) click to toggle source

# File lib/miga/cli/action/init.rb, line 233
def check_r_packages(paths)
  cli.puts 'Looking for R packages:'
  %w(ape cluster vegan).each do |pkg|
    cli.print "Testing #{pkg}... "
    if test_r_package(cli, paths, pkg)
      cli.puts 'yes'
    else
      cli.puts 'no, installing'
      cli.print '' + install_r_package(cli, paths, pkg)
      unless test_r_package(cli, paths, pkg)
        raise "Unable to auto-install R package: #{pkg}"
      end
    end
  end
  cli.puts ''
end

check_ruby_gems(paths) click to toggle source

# File lib/miga/cli/action/init.rb, line 250
def check_ruby_gems(paths)
  cli.puts 'Looking for Ruby gems:'
  %w(sqlite3 daemons json).each do |pkg|
    cli.print "Testing #{pkg}... "
    if test_ruby_gem(cli, paths, pkg)
      cli.puts 'yes'
    else
      cli.puts 'no, installing'
      # This hackey mess is meant to ensure the test and installation are done
      # on the configuration Ruby, not on the Ruby currently executing the
      # init action
      cli.print install_ruby_gem(cli, paths, pkg)
      unless test_ruby_gem(cli, paths, pkg)
        raise "Unable to auto-install Ruby gem: #{pkg}"
      end
    end
  end
  cli.puts ''
end

check_software_requirements(rc_fh) click to toggle source

# File lib/miga/cli/action/init.rb, line 157
def check_software_requirements(rc_fh)
  cli.puts 'Looking for requirements:'
  ask_for_mytaxa
  rc_fh.puts 'export MIGA_MYTAXA="no"' unless cli[:mytaxa]
  paths = {}
  rc_fh.puts 'MIGA_PATH=""'
  req_path = File.expand_path('utils/requirements.txt', MiGA.root_path)
  File.open(req_path, 'r') do |fh|
    fh.each_line do |ln|
      next if $. < 3

      r = ln.chomp.split(/\t+/)
      next if r[0] =~ /\(opt\)$/ && !cli[:mytaxa]

      cli.print "Testing #{r[0]}#{" (#{r[3]})" if r[3]}... "
      path = find_software(r[1])
      paths[r[1]] = File.expand_path(r[1], path).shellescape
    end
  end
  rc_fh.puts 'export PATH="$MIGA_PATH$PATH"'
  cli.puts ''
  paths
end

close_rc_file(rc_fh) click to toggle source

# File lib/miga/cli/action/init.rb, line 270
def close_rc_file(rc_fh)
  rc_fh.puts <<~FOOT

    MIGA_CONFIG_VERSION='#{MiGA::MiGA.VERSION}'
    MIGA_CONFIG_LONGVERSION='#{MiGA::MiGA.LONG_VERSION}'
    MIGA_CONFIG_DATE='#{Time.now}'

  FOOT
  rc_fh.close
end

empty_action() click to toggle source

# File lib/miga/cli/action/init.rb, line 61
def empty_action
end

find_software(exec) click to toggle source

# File lib/miga/cli/action/init.rb, line 191
def find_software(exec)
  path = nil
  loop do
    d_path = File.dirname(run_cmd(cli, "which #{exec.shellescape}"))
    if cli[:ask] || d_path == '.'
      path = cli.ask_user('Where can I find it?', d_path, nil, true)
    else
      path = d_path
      cli.puts path
    end
    if File.executable?(File.expand_path(exec, path))
      if d_path != path
        rc_fh.puts "MIGA_PATH=\"#{path}:$MIGA_PATH\" # #{exec}"
      end
      break
    end
    cli.print "I cannot find #{exec} "
  end
  path
end

install_r_package(cli, paths, pkg) click to toggle source

# File lib/miga/cli/action/init.rb, line 80
def install_r_package(cli, paths, pkg)
  r_cmd = "install.packages('#{pkg}', repos='http://cran.rstudio.com/')"
  run_r_cmd(cli, paths, r_cmd)
end

install_ruby_gem(cli, paths, pkg) click to toggle source

# File lib/miga/cli/action/init.rb, line 93
def install_ruby_gem(cli, paths, pkg)
  gem_cmd = "Gem::GemRunner.new.run %w(install --user #{pkg})"
  run_cmd(
    cli,
    "#{paths['ruby'].shellescape} \
      -r rubygems -r rubygems/gem_runner \
      -e #{gem_cmd.shellescape} 2>&1"
  )
end

list_requirements() click to toggle source

# File lib/miga/cli/action/init.rb, line 103
def list_requirements
  if cli.ask_user(
    'Would you like to see all the requirements before starting?',
    'no', %w(yes no)
  ) == 'yes'
    cli.puts ''
    req_path = File.join(MiGA.root_path, 'utils', 'requirements.txt')
    File.open(req_path, 'r') do |fh|
      fh.each_line { |ln| cli.puts ln }
    end
    cli.puts ''
  end
end

open_rc_file() click to toggle source

# File lib/miga/cli/action/init.rb, line 119
def open_rc_file
  rc_path = File.expand_path('.miga_rc', ENV['HOME'])
  if File.exist? rc_path
    if cli.ask_user(
      'I found a previous configuration. Do you want to continue?',
      'yes', %w(yes no)
    ) == 'no'
      cli.puts 'OK, see you soon!'
      exit(0)
    end
  end
  rc_fh = File.open(rc_path, 'w')
  rc_fh.puts <<~BASH
    #!/bin/bash
    # `miga init` made this on #{Time.now}

  BASH
  rc_fh
end

parse_cli() click to toggle source

# File lib/miga/cli/action/init.rb, line 11
def parse_cli
  cli.interactive = true
  cli.defaults = { mytaxa: nil,
                   config: File.expand_path('.miga_modules', ENV['HOME']),
                   ask: false, auto: false, dtype: :bash }
  cli.parse do |opt|
    opt.on(
      '-c', '--config PATH',
      'Path to the Bash configuration file',
      "By default: #{cli[:config]}"
    ) { |v| cli[:config] = v }
    opt.on(
      '--[no-]mytaxa',
      'Should I try setting up MyTaxa its dependencies?',
      'By default: interactive (true if --auto)'
    ) { |v| cli[:mytaxa] = v }
    opt.on(
      '--daemon-type STRING',
      'Type of daemon launcher, one of: bash, ssh, qsub, msub, slurm',
      "By default: interactive (#{cli[:dtype]} if --auto)"
    ) { |v| cli[:dtype] = v.to_sym }
    opt.on(
      '--ask-all',
      'Ask for the location of all software',
      'By default, only the locations missing in PATH are requested'
    ) { |v| cli[:ask] = v }
  end
end

perform() click to toggle source

# File lib/miga/cli/action/init.rb, line 40
  def perform
    cli.puts <<~BANNER
      ===[ Welcome to MiGA, the Microbial Genome Atlas ]===

      I'm the initialization script, and I'll sniff around your computer to
      make sure you have all the requirements for MiGA data processing.

    BANNER
    list_requirements
    rc_fh = open_rc_file
    check_configuration_script rc_fh
    paths = check_software_requirements rc_fh
    check_additional_files paths
    check_r_packages paths
    check_ruby_gems paths
    configure_daemon
    close_rc_file rc_fh
    cli.puts 'Configuration complete. MiGA is ready to work!'
    cli.puts ''
  end

  def empty_action
  end

  def run_cmd(cli, cmd)
    %x. "#{cli[:config]}" && #{cmd}`
  end

  def run_r_cmd(cli, paths, cmd)
    run_cmd(
      cli,
      "echo #{cmd.shellescape} | #{paths['R'].shellescape} --vanilla -q 2>&1"
    )
  end

  def test_r_package(cli, paths, pkg)
    run_r_cmd(cli, paths, "library('#{pkg}')")
    $?.success?
  end

  def install_r_package(cli, paths, pkg)
    r_cmd = "install.packages('#{pkg}', repos='http://cran.rstudio.com/')"
    run_r_cmd(cli, paths, r_cmd)
  end

  def test_ruby_gem(cli, paths, pkg)
    run_cmd(
      cli,
      "#{paths['ruby'].shellescape} -r #{pkg.shellescape} -e '' 2>/dev/null"
    )
    $?.success?
  end

  def install_ruby_gem(cli, paths, pkg)
    gem_cmd = "Gem::GemRunner.new.run %w(install --user #{pkg})"
    run_cmd(
      cli,
      "#{paths['ruby'].shellescape} \
        -r rubygems -r rubygems/gem_runner \
        -e #{gem_cmd.shellescape} 2>&1"
    )
  end

  def list_requirements
    if cli.ask_user(
      'Would you like to see all the requirements before starting?',
      'no', %w(yes no)
    ) == 'yes'
      cli.puts ''
      req_path = File.join(MiGA.root_path, 'utils', 'requirements.txt')
      File.open(req_path, 'r') do |fh|
        fh.each_line { |ln| cli.puts ln }
      end
      cli.puts ''
    end
  end

  private

  def open_rc_file
    rc_path = File.expand_path('.miga_rc', ENV['HOME'])
    if File.exist? rc_path
      if cli.ask_user(
        'I found a previous configuration. Do you want to continue?',
        'yes', %w(yes no)
      ) == 'no'
        cli.puts 'OK, see you soon!'
        exit(0)
      end
    end
    rc_fh = File.open(rc_path, 'w')
    rc_fh.puts <<~BASH
      #!/bin/bash
      # `miga init` made this on #{Time.now}

    BASH
    rc_fh
  end

  def check_configuration_script(rc_fh)
    unless File.exist? cli[:config]
      cli[:config] = cli.ask_user(
        'Is there a script I need to load at startup?',
        cli[:config]
      )
    end
    if File.exist? cli[:config]
      cli[:config] = File.expand_path(cli[:config])
      cli.puts "Found bash configuration script: #{cli[:config]}"
      rc_fh.puts "MIGA_STARTUP='#{cli[:config]}'"
      rc_fh.puts '. "$MIGA_STARTUP"'
    else
      cli[:config] = '/dev/null'
    end
    cli.puts ''
  end

  def check_software_requirements(rc_fh)
    cli.puts 'Looking for requirements:'
    ask_for_mytaxa
    rc_fh.puts 'export MIGA_MYTAXA="no"' unless cli[:mytaxa]
    paths = {}
    rc_fh.puts 'MIGA_PATH=""'
    req_path = File.expand_path('utils/requirements.txt', MiGA.root_path)
    File.open(req_path, 'r') do |fh|
      fh.each_line do |ln|
        next if $. < 3

        r = ln.chomp.split(/\t+/)
        next if r[0] =~ /\(opt\)$/ && !cli[:mytaxa]

        cli.print "Testing #{r[0]}#{" (#{r[3]})" if r[3]}... "
        path = find_software(r[1])
        paths[r[1]] = File.expand_path(r[1], path).shellescape
      end
    end
    rc_fh.puts 'export PATH="$MIGA_PATH$PATH"'
    cli.puts ''
    paths
  end

  def ask_for_mytaxa
    if cli[:mytaxa].nil?
      cli[:mytaxa] =
        cli.ask_user(
          'Should I include MyTaxa modules?',
          'yes', %w(yes no)
        ) == 'yes'
    end
  end

  def find_software(exec)
    path = nil
    loop do
      d_path = File.dirname(run_cmd(cli, "which #{exec.shellescape}"))
      if cli[:ask] || d_path == '.'
        path = cli.ask_user('Where can I find it?', d_path, nil, true)
      else
        path = d_path
        cli.puts path
      end
      if File.executable?(File.expand_path(exec, path))
        if d_path != path
          rc_fh.puts "MIGA_PATH=\"#{path}:$MIGA_PATH\" # #{exec}"
        end
        break
      end
      cli.print "I cannot find #{exec} "
    end
    path
  end

  def check_additional_files(paths)
    if cli[:mytaxa]
      cli.puts 'Looking for MyTaxa databases:'
      mt = File.dirname paths['MyTaxa']
      cli.print 'Looking for scores... '
      unless Dir.exist?(File.expand_path('db', mt))
        cli.puts "no\nExecute 'python2 #{mt}/utils/download_db.py'"
        exit(1)
      end
      cli.puts 'yes'
      cli.print 'Looking for diamond db... '
      unless File.exist?(File.expand_path('AllGenomes.faa.dmnd', mt))
        cli.puts "no\nDownload "            "'http://enve-omics.ce.gatech.edu/data/public_mytaxa/"            "AllGenomes.faa.dmnd' into #{mt}"
        exit(1)
      end
      cli.puts ''
    end
  end

  def check_r_packages(paths)
    cli.puts 'Looking for R packages:'
    %w(ape cluster vegan).each do |pkg|
      cli.print "Testing #{pkg}... "
      if test_r_package(cli, paths, pkg)
        cli.puts 'yes'
      else
        cli.puts 'no, installing'
        cli.print '' + install_r_package(cli, paths, pkg)
        unless test_r_package(cli, paths, pkg)
          raise "Unable to auto-install R package: #{pkg}"
        end
      end
    end
    cli.puts ''
  end

  def check_ruby_gems(paths)
    cli.puts 'Looking for Ruby gems:'
    %w(sqlite3 daemons json).each do |pkg|
      cli.print "Testing #{pkg}... "
      if test_ruby_gem(cli, paths, pkg)
        cli.puts 'yes'
      else
        cli.puts 'no, installing'
        # This hackey mess is meant to ensure the test and installation are done
        # on the configuration Ruby, not on the Ruby currently executing the
        # init action
        cli.print install_ruby_gem(cli, paths, pkg)
        unless test_ruby_gem(cli, paths, pkg)
          raise "Unable to auto-install Ruby gem: #{pkg}"
        end
      end
    end
    cli.puts ''
  end

  def close_rc_file(rc_fh)
    rc_fh.puts <<~FOOT

      MIGA_CONFIG_VERSION='#{MiGA::MiGA.VERSION}'
      MIGA_CONFIG_LONGVERSION='#{MiGA::MiGA.LONG_VERSION}'
      MIGA_CONFIG_DATE='#{Time.now}'

    FOOT
    rc_fh.close
  end
end

run_cmd(cli, cmd) click to toggle source

# File lib/miga/cli/action/init.rb, line 64
def run_cmd(cli, cmd)
  %x. "#{cli[:config]}" && #{cmd}`
end

run_r_cmd(cli, paths, cmd) click to toggle source

# File lib/miga/cli/action/init.rb, line 68
def run_r_cmd(cli, paths, cmd)
  run_cmd(
    cli,
    "echo #{cmd.shellescape} | #{paths['R'].shellescape} --vanilla -q 2>&1"
  )
end

test_r_package(cli, paths, pkg) click to toggle source

# File lib/miga/cli/action/init.rb, line 75
def test_r_package(cli, paths, pkg)
  run_r_cmd(cli, paths, "library('#{pkg}')")
  $?.success?
end

test_ruby_gem(cli, paths, pkg) click to toggle source

# File lib/miga/cli/action/init.rb, line 85
def test_ruby_gem(cli, paths, pkg)
  run_cmd(
    cli,
    "#{paths['ruby'].shellescape} -r #{pkg.shellescape} -e '' 2>/dev/null"
  )
  $?.success?
end

Constants

Public Instance Methods