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

# Environment variables:
#
# SKIP_BCC_BUILD: Set to skip the building bcc from source

$ubuntu_18_deps = <<EOF
apt-get -qq update
apt-get -qq install linux-headers-$(uname -r) binutils-dev
apt-get -qq install bison cmake flex g++ git libelf-dev zlib1g-dev libfl-dev systemtap-sdt-dev
apt-get -qq install llvm-7-dev llvm-7-runtime libclang-7-dev clang-7
EOF

$build_bcc = <<EOF
if [ -e /usr/local/lib/libbcc.so ]; then
   echo "libbcc already built, skipping"
   exit 0
fi
git clone https://github.com/iovisor/bcc.git
mkdir -p bcc/build
cd bcc/build
cmake -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=/usr/local ..
make && sudo make install && sudo ldconfig
EOF

Vagrant.configure("2") do |config|
  boxes = {
    'ubuntu-18.04' => {
      'image' => 'ubuntu/bionic64',
      'scripts' => [ $ubuntu_18_deps, ],
    },
    'ubuntu-18.10' => {
      'image' => 'ubuntu/cosmic64',
      'scripts' => [ $ubuntu_18_deps, ],
    },
    'ubuntu-19.04' => {
      'image' => 'ubuntu/disco64',
      'scripts' => [ $ubuntu_18_deps, ],
    },
  }
  boxes.each do | name, params |
    config.vm.define name do |box|
      box.vm.box = params['image']
      box.vm.provider "virtualbox" do |v|
        v.memory = 2048
        v.cpus = 2
      end
      box.vm.provider :libvirt do |v|
        v.memory = 2048
        v.cpus = 2
      end
      (params['scripts'] || []).each do |script|
        box.vm.provision :shell, inline: script
      end
      unless ENV['SKIP_BCC_BUILD']
        box.vm.provision :shell, privileged: false, inline: $build_bcc
      end
      config.vm.post_up_message = <<-HEREDOC
#######
bpftrace source is available in /vagrant
Build command: cmake /vagrant -DCMAKE_INSTALL_PREFIX=/usr/local && make
#######
      HEREDOC
    end
  end
end
