Vagrantfile

Vagrant.configure("2") do |config|
  config.env.enable
  config.vm.synced_folder ".", "/vagrant"
  config.vm.box = ENV["BOX_IMAGE"]
  config.vm.box_version = ENV["BOX_VERSION"]
  config.vm.box_check_update = false
  config.vm.synced_folder ".data/", "/etc/.vagrantdata/"
  
  config.hostmanager.enabled = true
  config.hostmanager.manage_guest = true

  NODEHOSTNAME = ENV["NODE_HOSTNAME"]
  
  if ENV["SETUP_MASTER"]
    config.vm.define ENV["MASTER_HOSTNAME"] do |subconfig|
      subconfig.vm.hostname = ENV["MASTER_HOSTNAME"]
      subconfig.vm.network :private_network, ip: ENV["MASTER_IP"]
      subconfig.vm.provider :virtualbox do |vb|
        vb.customize ["modifyvm", :id, "--cpus", ENV["MASTER_CPU"]]
        vb.customize ["modifyvm", :id, "--memory", ENV["MASTER_MEMORY"]]
      end
      subconfig.vm.provision "ansible_local" do |ansible|
        ansible.playbook = "/etc/k8s-scripts/master.yaml"
        ansible.verbose = "v"
        ansible.extra_vars = {
          kube_version: ENV["KUBE_VERSION"],
          nw_plugin: ENV['NW_PLUGIN'],
          pod_network_cidr: ENV['POD_NW_CIDR'],
          insecure_dashboard: ENV['INSECURE_DASHBOARD'],
          lb_ippool: ENV["LB_IPPOOL"],
          install_helm: ENV["INSTALL_HELM"],
          enable_olm: ENV["ENABLE_OLM"],
        }
      end
    end
  end
  if ENV["SETUP_NODES"]
    (1..( ENV["NODE_COUNT"] ).to_i(10)).each do |i|
      config.vm.define "#{NODEHOSTNAME}-#{i}" do |subconfig|
        subconfig.vm.hostname = "#{NODEHOSTNAME}-#{i}"
        subconfig.vm.network :private_network, ip: ENV["NODE_IP_NW"] + "#{i + 10}"
        subconfig.vm.provider :virtualbox do |vb|
          vb.customize ["modifyvm", :id, "--cpus", ENV["NODE_CPU"]]
          vb.customize ["modifyvm", :id, "--memory", ENV["NODE_MEMORY"]]
        end  
        subconfig.vm.provision "ansible_local" do |ansible|
          ansible.playbook = "/etc/k8s-scripts/worker.yaml"
          ansible.verbose = "v"
        end
      end
    end
  end
end