Vagrantfile

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

require 'fileutils'

Vagrant.require_version ">= 1.6.0"

#unless Vagrant.has_plugin?("vagrant-aws") and Vagrant.has_plugin?("vagrant-google") and Vagrant.has_plugin?("vagrant-azure") and Vagrant.has_plugin?("vagrant-digitalocean")
#   puts "-- WARNING --"
#   puts "This Vagrantfile makes use of the vagrant-aws and vagrant-google plugins."
#   puts "These packages are necessary to interact with aws and gce"
#   puts " "
#   puts "execute: \"vagrant plugin install vagrant-aws\" and"
#   puts "execute: \"vagrant plugin install vagrant-google\" to continue."
#   exit
#end

CONFIG = File.join(File.dirname(__FILE__), "config.rb")

#what provider are we executing for?
provider_is_aws  = (!ARGV.nil? && ARGV.join('').include?('provider=aws'))
provider_is_vmware = (!ARGV.nil? && ARGV.join('').include?('provider=vmware'))
provider_is_virtualbox = (!ARGV.nil? && ARGV.join('').include?('provider=virtualbox'))
provider_is_google = (!ARGV.nil? && ARGV.join('').include?('provider=google'))
provider_is_digital_ocean = (!ARGV.nil? && ARGV.join('').include?('provider=digital_ocean'))
provider_is_azure = (!ARGV.nil? && ARGV.join('').include?('provider=azure'))

#complain about missing plugins depending on provider
if provider_is_aws
	 unless Vagrant.has_plugin?("vagrant-aws") 
	  # great plugin routine from https://github.com/WhoopInc/s3auth
  	  # Attempt to install ourself. Bail out on failure so we don't get stuck in an
	  # infinite loop.
	    puts "Did not detect vagrant-aws plugin..."
	    system('vagrant plugin install vagrant-aws') || exit!

	   # Relaunch Vagrant so the plugin is detected. Exit with the same status code.
	   exit system('vagrant', *ARGV)
	   exit
	end
end

if provider_is_vmware
unless Vagrant.has_plugin?("vagrant-vmware-fusion") or Vagrant.has_plugin?("vagrant-vmware-workstation")
            puts "Did not detect vagrant-vmware-fusion or vagrant-vmware-workstation plugin..."
            puts "Install the appropriate plugin (fusion for mac, workstationf or windows) and re-run your comand"
           exit
        end

end

if provider_is_google
	 unless Vagrant.has_plugin?("vagrant-google") 
            puts "Did not detect vagrant-google plugin..."
            system('vagrant plugin install vagrant-google') || exit!
           # Relaunch Vagrant so the plugin is detected. Exit with the same status code.
           exit system('vagrant', *ARGV)
           exit
	end
end

if provider_is_digital_ocean
	 unless Vagrant.has_plugin?("vagrant-digitalocean") 
            puts "Did not detect vagrant-digitalocean plugin..."
            system('vagrant plugin install vagrant-digitalocean') || exit!
           # Relaunch Vagrant so the plugin is detected. Exit with the same status code.
           exit system('vagrant', *ARGV)
           exit
	end
end

if provider_is_azure
	 unless Vagrant.has_plugin?("vagrant-azure") 
            puts "Did not detect vagrant-azure plugin..."
            system('vagrant plugin install vagrant-azure') || exit!

           # Relaunch Vagrant so the plugin is detected. Exit with the same status code.
           exit system('vagrant', *ARGV)
           exit

	end
end


#azure ssh port increment var 
ssh_port = 9000
  
# Defaults Attempt to apply the deprecated environment variable NUM_INSTANCES to
# $num_instances while allowing config.rb to override it
if ENV["NUM_INSTANCES"].to_i > 0 && ENV["NUM_INSTANCES"]
  $num_instances = ENV["NUM_INSTANCES"].to_i
end

if File.exist?(CONFIG)
  require CONFIG
end

Vagrant.configure("2") do |config|

config.vm.box = "coreos-%s" % $update_channel
  config.vm.box_version = ">= 308.0.1"
  config.vm.box_url = "http://%s.release.core-os.net/amd64-usr/current/coreos_production_vagrant.json" % $update_channel
  config.vm.boot_timeout = 1000

  config.vm.provider :vmware_fusion do |vb, override|
    override.vm.box_url = "http://%s.release.core-os.net/amd64-usr/current/coreos_production_vagrant_vmware_fusion.json" % $update_channel
  end

  config.vm.provider :virtualbox do |v|
    # On VirtualBox, we don't have guest additions or a functional vboxsf
    # in CoreOS, so tell Vagrant that so it can be smarter.
    v.check_guest_additions = false
    v.functional_vboxsf     = false
  end
  
	
 ["google"].each do |google|
	config.vm.provider google do |z, override|
        override.vm.box = "gce"
	override.vm.box_url = "https://github.com/mitchellh/vagrant-google/raw/master/google.box"
	override.vm.box_version = ""
	end
     end
     
     ["digital_ocean"].each do |digital_ocean|
	config.vm.provider digital_ocean do |z, override|
        override.vm.box = "digital_ocean"
	override.vm.box_url = "https://github.com/smdahlen/vagrant-digitalocean/raw/master/box/digital_ocean.box"
	override.vm.box_version = ""
	end
     end

    
     ["azure"].each do |azure|
	config.vm.provider azure do |q, override|
        override.vm.box = "azure"
	override.vm.box_url = "https://github.com/msopentech/vagrant-azure/raw/master/dummy.box"
	override.vm.box_version = ""
	end
     end

 ["aws"].each do |aws|
     config.vm.provider aws do |x, override|
     override.vm.box = "dummy"
     override.vm.box_url = "https://github.com/mitchellh/vagrant-aws/raw/master/dummy.box"
     override.vm.box_version = ""
    end
  end

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

  (1..$num_instances).each do |i|
    config.vm.define vm_name = "core-%02d" % i do |config|
      config.vm.hostname = vm_name
      ssh_port = (ssh_port + 1)
#If this is core-01 - make it the weave master - and if vmware or google, use the vmware user-data templates
#because it needs to use the private ip's versus the public ip's
      if config.vm.hostname == "core-01"
	if provider_is_vmware or provider_is_google or provider_is_azure
	theuserdata = File.read("user-data.weavemaster.vmware")
        cloud_config_path = File.join(File.dirname(__FILE__), "user-data.weavemaster.vmware")
	else
	theuserdata = File.read("user-data.weavemaster")
	cloud_config_path = File.join(File.dirname(__FILE__), "user-data.weavemaster")
	end
	else
#otherwise - if its not core-01 - and provider is google or vmware - set the non-master user-data file
        if provider_is_vmware or provider_is_google or provider_is_azure
	theuserdata =  File.read("user-data.vmware")
	cloud_config_path = File.join(File.dirname(__FILE__), "user-data.vmware")
	else
#otherwise - just set it to the user-data file
	theuserdata = File.read("user-data")
	cloud_config_path = File.join(File.dirname(__FILE__), "user-data")
	end
    end

      if $enable_serial_logging
        logdir = File.join(File.dirname(__FILE__), "log")
        FileUtils.mkdir_p(logdir)

        serialFile = File.join(logdir, "%s-serial.txt" % vm_name)
        FileUtils.touch(serialFile)

        config.vm.provider :vmware_fusion do |v, override|
          v.vmx["serial0.present"] = "TRUE"
          v.vmx["serial0.fileType"] = "file"
          v.vmx["serial0.fileName"] = serialFile
          v.vmx["serial0.tryNoRxLoss"] = "FALSE"
        end

        config.vm.provider :virtualbox do |vb, override|
          vb.customize ["modifyvm", :id, "--uart1", "0x3F8", "4"]
          vb.customize ["modifyvm", :id, "--uartmode1", serialFile]
        end
      end

      if $expose_docker_tcp
        config.vm.network "forwarded_port", guest: 2375, host: ($expose_docker_tcp + i - 1), auto_correct: true
      end

      config.vm.provider :vmware_fusion do |vb|
        vb.gui = $vb_gui
      end

      config.vm.provider :virtualbox do |vb|
        vb.gui = $vb_gui
        vb.memory = $vb_memory
        vb.cpus = $vb_cpus
      end

      ip = "172.17.8.#{i+100}"
      config.vm.network :private_network, ip: ip

      #disable synced folders
      config.vm.synced_folder '.', '/vagrant', disabled: true

##do the azure setup
     ["azure"].each do |azure|
	config.vm.provider azure do |a, override|
	 a.mgmt_certificate = ENV['AZURE_MGMT_CERT']
	 a.mgmt_endpoint = ENV['AZURE_MGMT_ENDPOINT']
	 a.subscription_id = ENV['AZURE_SUB_ID']
	 a.storage_acct_name = ENV['AZURE_STORAGE_ACCT']
	 a.vm_image = ENV['AZURE_VM_IMAGE']
         a.vm_user = 'core' # defaults to 'vagrant' if not provided
         a.vm_password = ''
         a.vm_name = config.vm.hostname  
	 a.cloud_service_name = 'gonkulator' 
         a.deployment_name = config.vm.hostname 
         a.vm_location = 'West US' # e.g., West US
         override.ssh.username = 'core' 
	 override.ssh.private_key_path = ENV['AZURE_SSH_PRIV_KEY']
	 a.private_key_file = ENV['AZURE_PRIV_KEY']
	 a.certificate_file = ENV['AZURE_CERT_FILE']
         a.ssh_port = ssh_port
       	 end
       end

 ##do the digital_ocean setup
     ["digital_ocean"].each do |digital_ocean|
	config.vm.provider digital_ocean do |d, override|
   		d.token = ENV['DO_TOKEN']
    		d.image = ENV['DO_IMAGE']
	    	d.region = ENV['DO_REGION']
    		d.size = ENV['DO_SIZE']
		d.root_username = 'core'
		d.private_networking = true 
		override.ssh.username = 'core'
		override.ssh.private_key_path = ENV['DO_OVERRIDE_KEY']
		d.user_data = theuserdata
		d.setup = false
		end
	    end
     
     ##do the google setup
     ["google"].each do |google|
	config.vm.provider google do |g, override|
		g.google_project_id = ENV['GC_PROJECT']
		g.google_client_email = ENV['GC_CLIENT_EMAIL']
		g.google_key_location = ENV['GC_KEY_LOCATION']
		g.machine_type = ENV['GC_MACHINETYPE']
		g.image = ENV['GC_IMAGE']
		g.name = vm_name
		g.metadata = {'user-data' => theuserdata }
		override.ssh.username = "core"
		override.ssh.private_key_path = ENV['GC_OVERRIDE_KEY']
		end
	    end

     ##do the aws setup
     ["aws"].each do |aws|
        config.vm.provider aws do |a, override|
                a.access_key_id = ENV['AWS_KEY']
                a.secret_access_key = ENV['AWS_SECRET']
                a.keypair_name = ENV['AWS_KEYNAME']
		a.region = ENV['AWS_REGION'] 
		a.instance_type = ENV['AWS_INSTANCE']
		a.security_groups =  ENV['AWS_SECURITYGROUP']
		a.ami = ENV['AWS_AMI']
		a.user_data = theuserdata
                override.ssh.private_key_path = ENV['AWS_KEYPATH']
                override.ssh.username = "core"
           end
	end

#If the provider is azure - seed the user_data file into the right director for the default cloudinit unit to pick it up
	if provider_is_azure
	config.vm.provision :shell, :inline => "mkdir -p /var/lib/coreos-install/", :privileged => true
	config.vm.provision :file, :source => "#{cloud_config_path}", :destination => "/tmp/vagrantfile-user-data"
	config.vm.provision :shell, :inline => "mv /tmp/vagrantfile-user-data /var/lib/coreos-install/user_data", :privileged => true
	end
#If the provider is google or aws - do not try to do the file/shell providers
	unless provider_is_aws or provider_is_google or provider_is_digital_ocean or provider_is_azure
        config.vm.provision :file, :source => "#{cloud_config_path}", :destination => "/tmp/vagrantfile-user-data"
        config.vm.provision :shell, :inline => "mv /tmp/vagrantfile-user-data /var/lib/coreos-vagrant/", :privileged => true
	end

    end
  end
end