# Copyright (c) 2016 Cisco and/or its affiliates.
# 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.

# -*- mode: ruby -*-
# vi: set ts=2 sw=2 sts=2 et ft=ruby :

$user_addition = <<-SHELL
    sudo deluser csit
    sudo adduser --disabled-password --gecos "" csit
    echo csit:csit | sudo chpasswd
    sudo adduser csit vagrant
    id csit
SHELL

$install_prereqs = <<-SHELL
    sudo apt-get -y update
    sudo apt-get -y -f install
    sudo apt-get -y install python-virtualenv python-dev iproute2 debhelper dkms
    sudo update-alternatives --install /bin/sh sh /bin/bash 100
SHELL

$install_vpp = <<-SHELL
    cd /vagrant
    vpp_pkgs="$(echo vpp*.deb)"
    if [ "$vpp_pkgs" != "vpp*.deb" ]; then
      if [ "$(dpkg -l | grep vpp)" != "" ] ; then
        sudo apt-get -y purge vpp\*
      fi
      sudo dpkg -i vpp*.deb
      vppcfg="/etc/vpp/startup.conf"
      sudo rm -f $vppcfg.orig
      sudo cp $vppcfg $vppcfg.orig
      echo -e '\nheapsize 512M' | sudo tee -a $vppcfg
    fi
SHELL


def add_dut(config, name, mgmt_ip, net1, net2)
  config.vm.define name do |node|
    node.vm.box = "puppetlabs/ubuntu-14.04-64-nocm"
    node.vm.hostname = name
    node.vm.provision "shell", inline: $user_addition
    node.vm.provision "shell", inline: $install_prereqs
    node.vm.provision "shell", inline: $install_vpp

    node.vm.network "private_network", ip: mgmt_ip
    node.vm.network "private_network", type: "dhcp", auto_config: false,
        virtualbox__intnet: net1
    node.vm.network "private_network", type: "dhcp", auto_config: false,
        virtualbox__intnet: net2
    node.vm.provider "virtualbox" do |vb|
      vb.memory = "2048"
      vb.customize ["modifyvm", :id, "--nicpromisc3", "allow-all"]
      vb.customize ["modifyvm", :id, "--nicpromisc4", "allow-all"]
    end
  end

end

Vagrant.configure(2) do |config|
  if Vagrant.has_plugin?("vagrant-proxyconf")
    if ENV["http_proxy"]
      config.proxy.http     = ENV["http_proxy"]
    end
    if ENV["https_proxy"]
      config.proxy.https    = ENV["https_proxy"]
    end
    if ENV["no_proxy"]
      config.proxy.no_proxy = ENV["no_proxy"]
    end
  end
  config.vm.define "tg" do |tg|
    tg.vm.box = "puppetlabs/ubuntu-14.04-64-nocm"
    tg.vm.hostname = "tg"

    tg.vm.provision "shell", inline: $user_addition
    tg.vm.provision "shell", inline: $install_prereqs
    tg.vm.network "private_network", ip: '192.168.255.100/24'
    tg.vm.network "private_network", type: "dhcp", auto_config: false,
        virtualbox__intnet: "tg_dut1"
    tg.vm.network "private_network", type: "dhcp", auto_config: false,
        virtualbox__intnet: "tg_dut2"
    tg.vm.provider "virtualbox" do |vb|
      vb.memory = "2048"
      vb.customize ["modifyvm", :id, "--nicpromisc3", "allow-all"]
      vb.customize ["modifyvm", :id, "--nicpromisc4", "allow-all"]
    end

  end

  add_dut(config, "dut1", "192.168.255.101/24", "tg_dut1", "dut1_dut2")
  add_dut(config, "dut2", "192.168.255.102/24", "tg_dut2", "dut1_dut2")
end