Vagrantfile

require 'json'
require 'thread'

# We need the vagrant-reload plugin
unless Vagrant.has_plugin?("vagrant-reload")
  system "vagrant plugin install vagrant-reload"
  exec "vagrant #{ARGV.join' '}"
end

# Load the state from the state.json file, if possible.
if File.file?("#{ENV['PWD']}/.cache/state.json")
  state = JSON.parse(File.read("#{ENV['PWD']}/.cache/state.json"))

  # ENV can override the state variables
  STACKI_ISO = ENV.fetch('STACKI_ISO', state['STACKI_ISO'])
  PLAYBOOK = ENV.fetch('PLAYBOOK', state['PLAYBOOK'])
  NAME = ENV.fetch('NAME', state['NAME'])
  USE_SRC = ENV.fetch('USE_SRC', state['USE_SRC'])
else
  STACKI_ISO = ENV['STACKI_ISO']
  PLAYBOOK = ENV['PLAYBOOK']
  NAME = ENV['NAME']
  USE_SRC = ENV['USE_SRC']
end

# Make sure Ansible can find our roles
ENV['TEST_FRAMEWORK_ROOT'] = File.dirname(__FILE__)
ENV['ANSIBLE_ROLES_PATH'] = ENV['TEST_FRAMEWORK_ROOT'] + "/provisioning/roles"

# Ansible also needs the path to our STACKI ISO in the ENV
ENV['STACKI_ISO'] = STACKI_ISO

# Try to set the GIT_BRANCH environment variable, if needed
if ENV['GIT_BRANCH'].nil?
  branch = `git symbolic-ref --short HEAD 2>/dev/null`.chomp
  if $?.success?
    ENV['GIT_BRANCH'] = branch
  end
end

Vagrant.configure("2") do |config|
  # Note: This requires Vagrant 2.2.0 or newer
  config.trigger.before :up do |trigger|
    trigger.info = "Acquiring Lock..."
    trigger.ruby do |env, machine|
      # The vagrant-libvirt plugin has a race condition getting
      # virtual bridges and IPs. So, we make sure concurrent processes
      # are spread out with a lock.
      lock = File.open(File.join(env.home_path, "test-framework.lock"), "w+")
      lock.flock(File::LOCK_EX)
      machine.ui.info("Lock acquired")

      Thread.new {
        # Give vagrant 60 seconds to get an IP
        sleep 60
        lock.flock(File::LOCK_UN)
        lock.close()
      }
    end
  end

  config.vm.define "frontend", autostart: false do |config|
    if STACKI_ISO =~  /-sles15\./
      config.vm.box = "stacki/sles-15.1"
      config.vm.box_url = "http://stacki-builds.labs.teradata.com/vagrant-boxes/sles-15.1.json"
    elsif STACKI_ISO =~  /-sles12\./
      config.vm.box = "stacki/sles-12.3"
      config.vm.box_url = "http://stacki-builds.labs.teradata.com/vagrant-boxes/sles-12.3.json"
    elsif STACKI_ISO =~  /-redhat7\./
      config.vm.box = "stacki/centos-7.6"
      config.vm.box_url = "http://stacki-builds.labs.teradata.com/vagrant-boxes/centos-7.6.json"
    end

    config.vm.provider "virtualbox" do |provider, config|
      provider.name = NAME + "_frontend"
      provider.gui = false
      provider.linked_clone = true

      provider.memory = "4098"
      provider.cpus = 4

      config.vm.synced_folder ".cache", "/export/cache", type: "virtualbox"
      config.vm.synced_folder "reports", "/export/reports", type: "virtualbox"
      config.vm.synced_folder "test-suites", "/export/test-suites", type: "virtualbox"

      if USE_SRC == "1"
        config.vm.synced_folder "../", "/export/src/stacki", type: "virtualbox"
      end
    end

    config.vm.provider "libvirt" do |provider, config|
      provider.default_prefix = NAME
      provider.driver = "kvm"
      provider.connect_via_ssh = false

      provider.memory = "4098"
      provider.cpus = 4
      provider.nic_model_type = "e1000"
      provider.random :model => "random"

      config.vm.synced_folder ".cache", "/export/cache", nfs: true, nfs_version: 4, nfs_udp: false
      config.vm.synced_folder "reports", "/export/reports", nfs: true, nfs_version: 4, nfs_udp: false
      config.vm.synced_folder "test-suites", "/export/test-suites", nfs: true, nfs_version: 4, nfs_udp: false

      if USE_SRC == "1"
        config.vm.synced_folder "../", "/export/src/stacki", nfs: true, nfs_version: 4, nfs_udp: false
      end
    end

    config.vm.synced_folder '.', '/vagrant', disabled: true

    config.vm.hostname = "frontend-0-0"
    config.vm.network "private_network",
      ip: "192.168.0.2",
      virtualbox__intnet: NAME,
      libvirt__network_name: NAME,
      libvirt__dhcp_enabled: false,
      libvirt__forward_mode: "veryisolated",
      libvirt__adapter: 1

    config.vm.provision "ansible" do |ansible|
      ansible.playbook = PLAYBOOK
      ansible.become = true
      ansible.compatibility_mode = "2.0"
    end

    config.vm.provision :reload
  end

  (0..2).each do |i|
    config.vm.define "backend-0-#{i}", autostart: false do |config|
      config.vm.box = "stacki/pxe-boot"
      config.vm.box_url = "http://stacki-builds.labs.teradata.com/vagrant-boxes/pxe-boot.json"
      config.vm.boot_timeout = 3600

      config.vm.provider "virtualbox" do |vb, config|
        vb.name = NAME + "_backend-0-#{i}"
        vb.gui = false
        vb.linked_clone = true

        vb.memory = "2048"
        vb.cpus = 2

        vb.check_guest_additions = false
        vb.functional_vboxsf = false

        vb.customize ['modifyvm', :id, '--boot1', 'net']
        vb.customize ['modifyvm', :id, '--boot2', 'disk']
        vb.customize ['modifyvm', :id, '--biospxedebug', 'on']
        vb.customize ['modifyvm', :id, '--cableconnected2', 'on']
        vb.customize ['modifyvm', :id, '--nicbootprio2', '1']
        vb.customize ['modifyvm', :id, "--nictype2", '82540EM']

        config.vm.network "private_network",
          mac: sprintf("5254000000%02X", i+3),
          type: "dhcp",
          virtualbox__intnet: NAME,
          auto_config: false
      end

      config.vm.provider "libvirt" do |provider, config|
        provider.default_prefix = NAME
        provider.driver = "kvm"
        provider.connect_via_ssh = false

        provider.memory = "4098"
        provider.cpus = 2
        provider.disk_bus = "sata"
        provider.nic_model_type = "e1000"
        provider.random :model => "random"

        boot_network = {'network' => NAME}
        provider.boot boot_network
        provider.boot 'hd'
        provider.qemuargs :value => "-boot"
        provider.qemuargs :value => "reboot-timeout=10000"

        config.vm.network "private_network",
          mac: sprintf("5254000000%02X", i+3),
          libvirt__network_name: NAME,
          libvirt__dhcp_enabled: false,
          libvirt__forward_mode: "veryisolated",
          libvirt__adapter: 1,
          auto_config: false
      end

      config.vm.synced_folder '.', '/vagrant', disabled: true
    end
  end

end