# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# Verify whether required plugins are installed.
required_plugins = [ "vagrant-disksize" ]
required_plugins.each do |plugin|
  if not Vagrant.has_plugin?(plugin)
    raise "The vagrant plugin #{plugin} is required. Please run `vagrant plugin install #{plugin}`"
  end
end

Vagrant.configure(2) do |config|

  # Configure all VM specs.
  config.vm.provider "virtualbox" do |v|
    v.memory = 8192
    v.cpus = 8
  end

  # Configure the disk size.
  disk_size = "160GB"

  # The below options are good enough for the 'mini' flavor but
  # make sure you export the variables to match the flavor you want to test.
  config.vm.define "ubuntu1604" do |xenial|
    xenial.vm.box = "ubuntu/xenial64"
    xenial.disksize.size = disk_size
    config.vm.provision "shell"do |s|
      s.privileged = false
      s.inline = <<-SHELL
          cd /vagrant
          export XCI_FLAVOR=${XCI_FLAVOR:-mini}
          export VM_CPU=${VM_CPU:-2}
          export VM_DISK=${VM_DISK:-40}
          export VM_MEMORY_SIZE=${VM_MEMORY_SIZE:-2048}
          export VM_DOMAIN_TYPE=qemu
          [[ ! -e ${HOME}/.ssh/id_rsa ]] && ssh-keygen -q -P '' -f ${HOME}/.ssh/id_rsa
          ./xci-deploy.sh
      SHELL
      s.env = {
          "XCI_FLAVOR" => "#{ENV['XCI_FLAVOR']}",
          "VM_CPU" => "#{ENV['VM_CPU']}",
          "VM_DISK" => "#{ENV['VM_DISK']}",
          "VM_MEMORY_SIZE" => "#{ENV['VM_MEMORY_SIZE']}"
      }
    end
  end

  config.vm.define "centos7" do |centos7|
    centos7.vm.box = "centos/7"
    centos7.disksize.size = disk_size
    # The CentOS build does not have growroot, so we
    # have to do it ourselves.
    config.vm.provision "shell" do |s|
      s.privileged = false
      s.inline = <<-SHELL
          cd /vagrant
          PART_START=$(parted /dev/sda --script unit MB print | awk '/^ 3 / {print $3}')
          parted /dev/sda --script unit MB mkpart primary ${PART_START} 100%
          parted /dev/sda --script set 4 lvm on
          pvcreate /dev/sda4
          vgextend VolGroup00 /dev/sda4
          lvextend -l +100%FREE /dev/mapper/VolGroup00-LogVol00
          xfs_growfs /dev/mapper/VolGroup00-LogVol00
          export XCI_FLAVOR=${XCI_FLAVOR:-mini}
          export VM_CPU=${VM_CPU:-2}
          export VM_DISK=${VM_DISK:-40}
          export VM_MEMORY_SIZE=${VM_MEMORY_SIZE:-2048}
          export VM_DOMAIN_TYPE=qemu
          [[ ! -e ${HOME}/.ssh/id_rsa ]] && ssh-keygen -q -P '' -f ${HOME}/.ssh/id_rsa
          ./xci-deploy.sh
      SHELL
      s.env = {
          "XCI_FLAVOR" => "#{ENV['XCI_FLAVOR']}",
          "VM_CPU" => "#{ENV['VM_CPU']}",
          "VM_DISK" => "#{ENV['VM_DISK']}",
          "VM_MEMORY_SIZE" => "#{ENV['VM_MEMORY_SIZE']}"
      }
    end
  end
end