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

NODES = 3
DISKS = 3
CACHE = false
LIBVIRT_DISK_CACHE = "default"

Vagrant.configure("2") do |config|
    config.ssh.insert_key = false
    config.vm.box = "centos/7"
    config.vm.box_url = "https://app.vagrantup.com/centos/7"

    # Override
    config.vm.provider :libvirt do |v,override|
        override.vm.synced_folder '.', '/home/vagrant/sync', disabled: true

        # change cpu mode to passthrough as workaround, refer bugs:
        #https://bugzilla.redhat.com/show_bug.cgi?id=1467599
        #https://bugzilla.redhat.com/show_bug.cgi?id=1386223#c10
        #vagrant-libvirt/vagrant-libvirt#667
        v.cpu_mode = 'host-passthrough'
    end

    # Make kube master
    config.vm.define :master do |master|
        master.vm.network :private_network, ip: "192.168.10.90"
        master.vm.host_name = "master"

        master.vm.provider :virtualbox do |vb|
            vb.memory = 1024
            vb.cpus = 2
        end
        master.vm.provider :libvirt do |lv|
            lv.memory = 1024
            lv.cpus = 2
            lv.volume_cache = LIBVIRT_DISK_CACHE
        end

        # View the documentation for the provider you're using for more
        # information on available options.
        master.vm.provision :ansible do |ansible|
            ansible.verbose = true
            ansible.limit = "all"
            ansible.playbook = "site.yml"
            ansible.groups = {
                "master" => ["master"],
                "nodes" => (0..NODES-1).map {|j| "node#{j}"},
            }
            ansible.extra_vars = {
                "vagrant_home"  => ENV['VAGRANT_HOME'] ? ENV['VAGRANT_HOME'] : "~/.vagrant.d",
                "vagrant_cache" => ENV['VAGRANT_CACHE'] ? ENV['VAGRANT_CACHE'] : CACHE,
                "vagrant_master"  => "192.168.10.90"
            }
        end

    end

    # Make the glusterfs cluster, each with DISKS number of drives
    (0..NODES-1).each do |i|
        config.vm.define "node#{i}" do |node|
            node.vm.hostname = "node#{i}"
            node.vm.network :private_network, ip: "192.168.10.10#{i}"

            # Settings for Virtualbox
            node.vm.provider :virtualbox do |vb|
                unless File.exist?("disk-#{i}-0.vdi")
                    vb.customize ["storagectl", :id,"--name", "VboxSata", "--add", "sata"]
                end
            end

            (0..DISKS-1).each do |d|
                node.vm.provider :virtualbox do |vb|
                    unless File.exist?("disk-#{i}-#{d}.vdi")
                        vb.customize [ "createmedium", "--filename", "disk-#{i}-#{d}.vdi", "--size", 1024*1024 ]
                    end
                    vb.customize [ "storageattach", :id, "--storagectl", "VboxSata", "--port", 3+d, "--device", 0, "--type", "hdd", "--medium", "disk-#{i}-#{d}.vdi" ]
                    vb.memory = 1024
                    vb.cpus = 2
                end
                node.vm.provider :libvirt do  |lv|
                    driverletters = ('b'..'z').to_a
                    lv.storage :file, :device => "vd#{driverletters[d]}", :path => "atomic-disk-#{i}-#{d}.disk", :size => '1024G'
                    lv.memory = 1024
                    lv.cpus =2
                    lv.volume_cache = LIBVIRT_DISK_CACHE
                end
            end
        end
    end
end