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

# Environment variables:
#
# SKIP_BCC_BUILD: Set to skip the building bcc from source
# LLVM_VERSION: The LLVM version to install

llvm_version = ENV["LLVM_VERSION"] || 12

$ubuntu_deps = <<EOF
wget https://apt.llvm.org/llvm.sh
bash ./llvm.sh "$LLVM_VERSION"
apt-get -qq update
apt-get -qq install linux-headers-$(uname -r) binutils-dev python
apt-get -qq install asciidoctor bison cmake flex g++ git libelf-dev zlib1g-dev libfl-dev systemtap-sdt-dev "libclang-${LLVM_VERSION}-dev" libcereal-dev libpcap-dev
apt-get -qq install --no-install-recommends pkg-config
EOF

$fedora_deps = <<EOF
dnf builddep -q -y bpftrace
dnf install -q -y git cereal-devel libpcap-devel
EOF

$centos_deps = <<EOF
dnf install -q -y bison flex cmake elfutils-libelf-devel zlib-devel gcc gcc-c++ python3
dnf install -q -y clang-devel llvm-devel binutils-devel kernel-devel binutils cereal-devel libpcap-dev
ln -s `which python3` /usr/local/bin/python
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
git checkout v0.22.0
cmake .. -DCMAKE_BUILD_TYPE=Debug -DCMAKE_INSTALL_PREFIX=/usr/local \
  -DENABLE_EXAMPLES=0 -DENABLE_TESTS=0 -DENABLE_MAN=0 \
  -DENABLE_LLVM_SHARED=1
make && sudo make install && sudo ldconfig
EOF

$build_libbpf = <<EOF
if [ -e /usr/local/lib/libbpf.so ]; then
   echo "libbpf already built, skipping"
   exit 0
fi
git clone https://github.com/libbpf/libbpf.git
cd libbpf/src
make
sudo PREFIX=/usr/local/ LIBDIR=/usr/local/lib make install install_uapi_headers
EOF

Vagrant.configure("2") do |config|
  boxes = {
    'ubuntu-16.04'     => {
      'image'          => 'ubuntu/xenial64',
      'scripts'        => [ $ubuntu_deps, ],
    },
    'ubuntu-18.04'     => {
      'image'          => 'ubuntu/bionic64',
      'scripts'        => [ $ubuntu_deps, ],
    },
    'ubuntu-20.04'     => {
      'image'          => 'ubuntu/focal64',
      'scripts'        => [ $ubuntu_deps, ],
      'fix_console'    => 0,
    },
    'ubuntu-21.04'     => {
      'image'          => 'ubuntu/hirsute64',
      'scripts'        => [ $ubuntu_deps, ],
      'fix_console'    => 0,
    },
    'ubuntu-21.10'     => {
      'image'          => 'ubuntu/impish64',
      'scripts'        => [ $ubuntu_deps, ],
      'fix_console'    => 0,
    },
    'fedora-34'        => {
      'image'          => 'fedora/34-cloud-base',
      'scripts'        => [ $fedora_deps, ],
      'skip_bcc_build' => 1
    },
    'centos-8'         => {
      'image'          => 'generic/centos8',
      'scripts'        => [ $centos_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
        if params['fix_console'] == 1
          v.customize ["modifyvm", :id, "--uart1", "0x3F8", "4"]
          v.customize ["modifyvm", :id, "--uartmode1", "file", "./#{name}_ttyS0.log"]
        end
      end
      box.vm.provider :libvirt do |v|
        v.memory = 2048
        v.cpus = 2
      end
      box.vm.synced_folder ".", "/vagrant", disabled: false
      (params['scripts'] || []).each do |script|
        box.vm.provision :shell, inline: script, env: { LLVM_VERSION: llvm_version }
      end
      unless ENV['SKIP_BCC_BUILD'] || (params['skip_bcc_build'] == 1)
        box.vm.provision :shell, privileged: false, inline: $build_bcc
      end
      unless ENV['SKIP_LIBBPF_BUILD'] || (params['skip_libbpf_build'] == 1)
        box.vm.provision :shell, privileged: false, inline: $build_libbpf
      end
      config.vm.post_up_message = <<-HEREDOC
#######
bpftrace source is available in /vagrant
Build command: mkdir build && cd build && cmake /vagrant -DVENDOR_GTEST=1 && make
#######
      HEREDOC
    end
  end
end
