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

ENV["LC_ALL"] = "en_US.UTF-8"

Vagrant.require_version ">= 2.0.0"

if ENV['VM_BOX'] == "ubuntu" then
  $vm_box = "bento/ubuntu-16.04"
else
  $vm_box = "centos/7"
end

$instances = 3
$apt_proxy = "http://192.168.205.12:3142"

Vagrant.configure("2") do |config|
  # always use Vagrants insecure key
  config.ssh.insert_key = false
  config.vm.box_check_update = false
  config.vm.box = $vm_box
  config.vm.synced_folder ".", "/vagrant", disabled: true

  if Vagrant.has_plugin?("vagrant-proxyconf") and $vm_box == "bento/ubuntu-16.04" then
    config.apt_proxy.http = $apt_proxy || ""
    config.apt_proxy.https = "DIRECT"
  end

  if $vm_box == "bento/ubuntu-16.04" then
    config.vm.provision "file", source: "apt_sources.list", destination: "/tmp/sources.list"
    config.vm.provision "shell", inline: "sudo mv -f /tmp/sources.list /etc/apt/sources.list"
    config.vm.provision "shell", inline: "sudo apt-get update"
    config.vm.provision "shell", inline: "sudo apt-get install -y htop di nmon"
    config.vm.provision "shell", inline: "sudo apt-get remove -y snapd"
    config.vm.provision "shell", inline: "sudo apt-get autoremove -y"
    config.vm.provision "shell", inline: "sudo apt-get autoclean"
    config.vm.provision "shell", inline: "byobu-launcher-install"
  elsif $vm_box == "centos/7" then
    config.vm.provision "shell", inline: "yum install -y epel-release"
    config.vm.provision "shell", inline: "yum install -y nano htop byobu nmon telnet net-tools bind-utils"
    config.vm.provision "shell", inline: "byobu-launcher-install"
  end

  if Vagrant.has_plugin?("vagrant-proxyconf")
    config.proxy.http     = ENV['HTTP_PROXY'] || ENV['http_proxy'] || ""
    config.proxy.https    = ENV['HTTPS_PROXY'] || ENV['https_proxy'] || ""
    config.proxy.no_proxy = $no_proxy
  end

  if Vagrant.has_plugin?("vagrant-vbguest") then
    config.vbguest.auto_update = false
  end

  (1..$instances).each do |instance_id|
    if instance_id <= 3
      $vm_name = "es-master-#{instance_id.to_s.rjust(2, '0')}"
    elsif instance_id <= 6
      $vm_name = "es-hot-#{instance_id.to_s.rjust(2, '0')}"
    elsif instance_id <= 8
      $vm_name = "es-warm-#{instance_id.to_s.rjust(2, '0')}"
    elsif instance_id <= 11
      $vm_name = "redis-#{instance_id.to_s.rjust(2, '0')}"
    elsif instance_id <= 13
      $vm_name = "logstash-#{instance_id.to_s.rjust(2, '0')}"
    elsif instance_id == 14
      $vm_name = "kibana-#{instance_id.to_s.rjust(2, '0')}"
    end

    config.vm.define vm_name = $vm_name do |config|
      config.vm.hostname = vm_name
      config.vm.network "private_network", ip: "172.28.128.1#{instance_id.to_s.rjust(2, '0')}"

      config.vm.provider "virtualbox" do |vb|
        vb.customize ["modifyvm", :id, "--natdnshostresolver1", "off"]
        vb.customize ["modifyvm", :id, "--natdnsproxy1", "off"]
        vb.name = vm_name
        if instance_id <= 3 then
          vb.memory = "8192"
          vb.cpus = "4"
        elsif instance_id > 3 then
          vb.memory = "2048"
          vb.cpus = "1"
        end
    end

      # if instance_id == $instances
      #   config.vm.provision "ansible" do |ansible|
      #     if $instances == 3
      #       ansible.groups = {
      #         "esMasters" => ["es-master-0[1:3]"],
      #         "esHots" => "",
      #         "esWarms" => "",
      #         "redis" => "",
      #         "logstash" => ["es-master-0[1:3]"],
      #         "kibana" => ["es-master-0[1:3]"],
      #         "elasticsearch:children" => ["esMasters","esHots","esWarms"],
      #         "esDatas:children" => ["esHots","esWarms"]
      #       }
      #     end
      #     if $instances == 14
      #       ansible.groups = {
      #         "esMasters" => ["es-master-[01:03]"],
      #         "esHots" => ["es-hot-[04:06]"],
      #         "esWarms" => ["es-warm-[07:08]"],
      #         "redis" => ["redis-[09:11]"],
      #         "logstash" => ["logstash-[12:13]"],
      #         "kibana" => ["kibana-14"],
      #         "haproxy:children" => ["kibana"],
      #         "elasticsearch:children" => ["esMasters","esHots","esWarms"],
      #         "esDatas:children" => ["esHots","esWarms"]
      #       }
      #     end
      #     ansible.limit = "all"
      #     ansible.playbook = "play-all.yml"
      #   end
      # end
    end
  end
end