Vagrantfile

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

require 'fileutils'
require 'pathname'

VAGRANT_COMMAND = ARGV[0]

VAGRANTFILE_API_VERSION = "2" if not defined? VAGRANTFILE_API_VERSION

DOCKER_DISK_NAME = "docker_data.vdi"
DOCKER_DISK_SIZE = 100  # unit: GB
DOCKER_DISK_FS = 'btrfs'
DOCKER_DISK_USE_LVM = false
VM_RAM_SIZE = 512
VM_CPU_CORE = 1
VM_NAME = "" #Name that will appear in privsion steps and VirtualBox GUI Window,left empty would be default.
VM_HOSTNAME = "" #host name that will appaer when you ssh into it, will be convert to all lowcase..
VG_BOX_NAME = "centos/7"
VM_IP = 'dhcp'
DOCKER_ENGINE_DAEMON_CONFIG = '-s btrfs --dns 8.8.8.8 --dns 8.8.4.4 -H tcp://0.0.0.0:2375 -H unix:///var/run/docker.sock'

Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|

  #force use virtualbox provider
  config.vm.provider "virtualbox"
  config.vm.provider "parallels"
  config.vm.provider "vmware_fusion"

  if not Vagrant.has_plugin?("vagrant-omnibus") or
     not Vagrant.has_plugin?("vagrant-persistent-storage") or
     not Vagrant.has_plugin?("vagrant-triggers") or
     not Vagrant.has_plugin?("vagrant-vbguest")
    puts ""
    puts "please install required plugins!!!"
    puts ""
    exit
  end

  if Vagrant.has_plugin?("vagrant-cachier")
    # Configure cached packages to be shared between instances of the same base box.
    # More info on http://fgrehm.viewdocs.io/vagrant-cachier/usage
    config.cache.scope = :box
  end

  #auto update virtualbox addition for enabling share folder
  config.vbguest.no_remote = true
  config.vbguest.installer_arguments = %w{--nox11}
  config.vbguest.auto_reboot = true

  config.vm.box = "#{VG_BOX_NAME}"
  config.vm.box_check_update = true

  if config.vm.box.to_s == "centos/7"
    #turn off default rsync sharing in offical CentOS 7 vagrant box
    config.vm.synced_folder '.', '/home/vagrant/sync', disabled: true
    config.vm.synced_folder '.', '/vagrant', disabled: false, type: "virtualbox" 
  end
  #config.ssh.insert_key = false
  if not (VM_NAME.nil? || "#{VM_NAME}".empty?)
    config.vm.define "#{VM_NAME}"
  end
  config.vm.hostname = "#{VM_HOSTNAME}" if not (VM_HOSTNAME.nil? || "#{VM_HOSTNAME}".empty?)
  if "#{VM_IP}" != "dhcp"
    config.vm.network "private_network", :ip => "#{VM_IP}"
  else
    config.vm.network "private_network", type: "dhcp"
  end
  file_disk = "#{DOCKER_DISK_NAME}"
  attach_dir = "disk_data"
  file_path = Pathname.new(attach_dir).join(file_disk)
  file_disk_size = DOCKER_DISK_SIZE * 1024
  FLAG_FILE = Pathname.new(".vagrant").join(".#{VM_NAME}created")

  def setup_and_enable_vg_persistent(config, file_path, file_disk_size)
    config.persistent_storage.enabled = true
    config.persistent_storage.location = "#{file_path}"
    config.persistent_storage.size = file_disk_size
    config.persistent_storage.mountname = 'dockerdata'
    config.persistent_storage.filesystem = "#{DOCKER_DISK_FS}"
    config.persistent_storage.mountpoint = '/var/lib/docker'
    config.persistent_storage.use_lvm = DOCKER_DISK_USE_LVM
  end

  if VAGRANT_COMMAND == "up" and not File.exist?(FLAG_FILE)
    setup_and_enable_vg_persistent(config, file_path, file_disk_size)
    Dir.mkdir(".vagrant") unless File.exists?(".vagrant")
    File.open(FLAG_FILE, "w+") do |f|
      f.write("docker data disk of #{config.vm.hostname} has been created at #{Time.now.strftime("%Y/%m/%d %H:%M:%S")}")
    end
  end

  if VAGRANT_COMMAND == "destroy"
    setup_and_enable_vg_persistent(config, file_path, file_disk_size)
  end

  config.trigger.after :destroy do
    if File.exist?(FLAG_FILE)
      File.delete(FLAG_FILE)
    end
  end

  config.vm.provider "virtualbox" do |vb|
    vb.gui = false
    vb.name = "#{VM_NAME}" if not (VM_NAME.nil? || "#{VM_NAME}".empty?)
    vb.customize ["modifyvm", :id, "--memory", "#{VM_RAM_SIZE}"]
    vb.customize ["modifyvm", :id, "--cpus", "#{VM_CPU_CORE}"]
  end

  config.omnibus.chef_version = "12.10.24"

  config.vm.provision "chef_solo" do |chef|
    if config.vm.box.to_s == "centos/7"
      chef.add_recipe "centos7nic-patch"
      # firewalld built-in in CentOS 7 has a issue with docker:
      # https://github.com/docker/docker/issues/16137
      chef.add_recipe "firewall"
    end

    chef.add_recipe "docker"
    chef.add_recipe "docker::compose"
    chef.add_recipe "docker::machine"

    chef.json = {
      "docker" => {
        'group_members' => ['vagrant'],
        'options' => "#{DOCKER_ENGINE_DAEMON_CONFIG}"
      }
    }
  end

end