Vagrantfile

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

# Define the number of master and worker nodes
# If this number is changed, remember to update setup-hosts.sh script with the new hosts IP details in /etc/hosts of each VM.
NUM_MASTER_NODE = 1
NUM_WORKER_NODE = 2

IP_NW = "192.168.56."
MASTER_IP_START = 1
NODE_IP_START = 2

# All Vagrant configuration is done below. The "2" in Vagrant.configure
# configures the configuration version (we support older styles for
# backwards compatibility). Please don't change it unless you know what
# you're doing.
Vagrant.configure("2") do |config|

  config.vm.boot_timeout = 1200
  # The most common configuration options are documented and commented below.
  # For a complete reference, please see the online documentation at
  # https://docs.vagrantup.com.

  # Every Vagrant development environment requires a box. You can search for
  # boxes at https://vagrantcloud.com/search.
  # config.vm.box = "base"
  config.vm.box = "ubuntu/bionic64"

  # Disable automatic box update checking. If you disable this, then
  # boxes will only be checked for updates when the user runs
  # `vagrant box outdated`. This is not recommended.
  config.vm.box_check_update = false

  # Create a public network, which generally matched to bridged network.
  # Bridged networks make the machine appear as another physical device on
  # your network.
  # config.vm.network "public_network"

  # Share an additional folder to the guest VM. The first argument is
  # the path on the host to the actual folder. The second argument is
  # the path on the guest to mount the folder. And the optional third
  # argument is a set of non-required options.
  # config.vm.synced_folder "../data", "/vagrant_data"

  # Added hack to fix on windows WSL
  config.vm.synced_folder '.', '/vagrant', disabled: true

  config.vm.provider "virtualbox" do |vb|
    vb.customize [ "modifyvm", :id, "--uartmode1", "disconnected" ]
    vb.customize [ "modifyvm", :id, "--hpet", "on" ]
  end

  # Provider-specific configuration so you can fine-tune various
  # backing providers for Vagrant. These expose provider-specific options.
  # Example for VirtualBox:
  #
  # config.vm.provider "virtualbox" do |vb|
  #   # Customize the amount of memory on the VM:
  #   vb.memory = "1024"
  # end
  #
  # View the documentation for the provider you are using for more
  # information on available options.

  # Provision Master Nodes
  (1..NUM_MASTER_NODE).each do |i|
      master_num = format('%02d', i)
      config.vm.define "kubemaster#{master_num}" do |node|
        # Name shown in the GUI
        node.vm.provider "virtualbox" do |vb|
            vb.name = "kubemaster#{master_num}"
            vb.memory = 2048
            vb.cpus = 2
        end
          # set hostname
        node.vm.hostname = "kubemaster#{master_num}"

        # set network ip
        node.vm.network :private_network, ip: IP_NW + "#{MASTER_IP_START + i}"

        #forward ssh port to some local port
        node.vm.network "forwarded_port", guest: 22, host: "#{2710 + i}"

        node.vm.provision "setup-hosts", :type => "shell", :path => "scripts/setup-hosts.sh" do |s|
          s.args = ["enp0s8"] # pass variable to script, gets ip assigned to interface matching the variable
        end

        node.vm.provision "setup-dns", type: "shell", :path => "scripts/update-dns.sh"

      end

  end


  # Provision Worker Nodes
  (1..NUM_WORKER_NODE).each do |i|
    worker_num = format('%02d', i)
    config.vm.define "kubenode#{worker_num}" do |node|
      node.vm.provider "virtualbox" do |vb|
          vb.name = "kubenode#{worker_num}"
          vb.memory = 2048
          vb.cpus = 1
      end
      # set hostname
      node.vm.hostname = "kubenode#{worker_num}"

      # set network ip
      node.vm.network :private_network, ip: IP_NW + "#{NODE_IP_START + i}"
      
      #forward ssh port to some local port
      node.vm.network "forwarded_port", guest: 22, host: "#{2720 + i}"

      node.vm.provision "setup-hosts", :type => "shell", :path => "scripts/setup-hosts.sh" do |s|
        s.args = ["enp0s8"] # pass variable to script, gets ip assigned to interface matching the variable
      end

      node.vm.provision "setup-dns", type: "shell", :path => "scripts/update-dns.sh"
      
      # Provision Ansible on the last node defined. Ansible will run in parallel since the other nodes are already provisioned. 
      if i == NUM_WORKER_NODE
        node.vm.provision "ansible" do |ansible|
          ansible.playbook = "ansible/site.yaml"
          ansible.limit = "all"
          ansible.verbose = "v"
          ansible.groups = {
            "controlplanes" => ["kubemaster0[1:#{NUM_MASTER_NODE}]"],
            "workers" => ["kubenode0[1:#{NUM_WORKER_NODE}]"],
            "all_groups:children" => ["controlplanes", "workers" ]
          }
        end
      end
    end

  end

end