Vagrantfile

#e -*- mode: ruby -*-
# vi: set ft=ruby :

# Vagrantfile API/syntax version. Don't touch unless you know what you're doing!
VAGRANTFILE_API_VERSION = "2"
VAGRANTFILE_LOCAL = 'Vagrantfile.local'

sdist=ENV['CIF_ANSIBLE_SDIST']
es=ENV['CIF_ANSIBLE_ES']
hunter_threads=ENV['CIF_HUNTER_THREADS']
hunter_advanced=ENV['CIF_HUNTER_ADVANCED']
geo_fqdn=ENV['CIF_GATHERER_GEO_FQDN']
csirtg_token=ENV['CSIRTG_TOKEN']
distro=ENV.fetch('CIF_VAGRANT_DISTRO', 'ubuntu')
redhat=0
rhel_user=ENV['RHEL_USER']
rhel_pass=ENV['RHEL_PASSWORD']
es_tests=ENV.fetch('CIF_ELASTICSEARCH_TEST', '0')
es_upsert=ENV.fetch('CIF_STORE_ES_UPSERT_MODE', '0')

redhat=1 if distro == 'redhat'

unless File.directory?('deploymentkit')
    puts "Please unzip the latest release of the deploymentkit before continuing..."
    puts ""
    puts "https://github.com/csirtgadgets/bearded-avenger-deploymentkit/wiki"
    puts ""
    exit
end

$script = <<SCRIPT
export CIF_ANSIBLE_SDIST=#{sdist}
export CIF_ANSIBLE_ES=#{es}
export CIF_ELASTICSEARCH_TEST=#{es_tests}
export CIF_STORE_ES_UPSERT_MODE=#{es_upsert}
export CIF_HUNTER_THREADS=#{hunter_threads}
export CIF_HUNTER_ADVANCED=#{hunter_advanced}
export CIF_GATHERER_GEO_FQDN=#{geo_fqdn}
export CIF_BOOTSTRAP_TEST=1
export CSIRTG_TOKEN=#{csirtg_token}
export RHEL=#{redhat}
export USERNAME=#{rhel_user}
export PASSWORD="#{rhel_pass}"

if [ "${RHEL}" == "1" ]; then
  subscription-manager register --username "${USERNAME}" --password "${PASSWORD}" --auto-attach
  subscription-manager attach
fi

cd /vagrant/deploymentkit
bash easybutton.sh

if [ "${RHEL}" == "1" ]; then
  echo 'unregistering'
  subscription-manager unregister
fi
SCRIPT

Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
  config.vm.provision "shell", inline: $script

  case distro
    when 'redhat'
      config.vm.box = "iamseth/rhel-7.3"
    when 'centos'
       config.vm.box = 'geerlingguy/centos7'
    else
     config.vm.box = 'ubuntu/xenial64'
  end

  #config.vm.network :forwarded_port, guest: 443, host: 8443

  config.vm.provider :virtualbox do |vb|
    vb.customize ["modifyvm", :id, "--cpus", "2", "--ioapic", "on", "--memory", "4096" ]
  end

  if File.file?(VAGRANTFILE_LOCAL)
    external = File.read VAGRANTFILE_LOCAL
    eval external
  end
end