config.rb

require 'erb'

begin
  require 'dotenv'
rescue LoadError
  $stderr.puts "Please install dotenv to load .env:"
  $stderr.puts "  $ vagrant plugin install dotenv"
  exit 1
end

# Size of the CoreOS cluster created by Vagrant
$num_instances=3

if ARGV[0].eql?('up')
  require 'open-uri'
  require 'yaml'

  # Used to fetch a new discovery token for a cluster of size $num_instances
  $new_discovery_url="https://discovery.etcd.io/new?size=#{$num_instances}"
  token = open($new_discovery_url).read

  unless File.exist?(".env")
    $stderr.puts "Please create .env:"
    $stderr.puts "  $ cp .env.sample .env"
    exit 1
  end

  # Create user-data from erb template
  Dotenv.load

  erb = File.open(File.join(File.dirname(__FILE__), "user-data.yml.erb")) { |f| ERB.new(f.read) }

  datadog_enabled = ENV["PAUS_DATADOG_ENABLED"] && ["1", "true"].include?(ENV["PAUS_DATADOG_ENABLED"].downcase)

  [:manager, :replica].each do |instance_type|
    user_data_path = File.join(File.dirname(__FILE__), "user-data-#{instance_type}")
    File.write(user_data_path, erb.result(binding))

    data = YAML.load(IO.readlines(user_data_path)[1..-1].join)
    if data['coreos'].key? 'etcd'
      data['coreos']['etcd']['discovery'] = token
    end
    if data['coreos'].key? 'etcd2'
      data['coreos']['etcd2']['discovery'] = token
    end

    # Fix for YAML.load() converting reboot-strategy from 'off' to `false`
    if data['coreos'].key? 'update'
      if data['coreos']['update'].key? 'reboot-strategy'
        if data['coreos']['update']['reboot-strategy'] == false
          data['coreos']['update']['reboot-strategy'] = 'off'
        end
      end
    end

    yaml = YAML.dump(data)
    File.open(user_data_path, 'w') { |file| file.write("#cloud-config\n\n#{yaml}") }
  end
end

#
# coreos-vagrant is configured through a series of configuration
# options (global ruby variables) which are detailed below. To modify
# these options, first copy this file to "config.rb". Then simply
# uncomment the necessary lines, leaving the $, and replace everything
# after the equals sign..

# Change basename of the VM
# The default value is "core", which results in VMs named starting with
# "core-01" through to "core-${num_instances}".
#$instance_name_prefix="core"

# Change the version of CoreOS to be installed
# To deploy a specific version, simply set $image_version accordingly.
# For example, to deploy version 709.0.0, set $image_version="709.0.0".
# The default value is "current", which points to the current version
# of the selected channel
#$image_version = "current"

# Official CoreOS channel from which updates should be downloaded
$update_channel='beta'

# Log the serial consoles of CoreOS VMs to log/
# Enable by setting value to true, disable with false
# WARNING: Serial logging is known to result in extremely high CPU usage with
# VirtualBox, so should only be used in debugging situations
#$enable_serial_logging=false

# Enable port forwarding of Docker TCP socket
# Set to the TCP port you want exposed on the *host* machine, default is 2375
# If 2375 is used, Vagrant will auto-increment (e.g. in the case of $num_instances > 1)
# You can then use the docker tool locally by setting the following env var:
#   export DOCKER_HOST='tcp://127.0.0.1:2375'
#$expose_docker_tcp=2375

# Enable NFS sharing of your home directory ($HOME) to CoreOS
# It will be mounted at the same path in the VM as on the host.
# Example: /Users/foobar -> /Users/foobar
#$share_home=false

# Customize VMs
#$vm_gui = false
#$vm_memory = 1024
#$vm_cpus = 1

# Share additional folders to the CoreOS VMs
# For example,
# $shared_folders = {'/path/on/host' => '/path/on/guest', '/home/foo/app' => '/app'}
# or, to map host folders to guest folders of the same name,
# $shared_folders = Hash[*['/home/foo/app1', '/home/foo/app2'].map{|d| [d, d]}.flatten]
#$shared_folders = {}

# Enable port forwarding from guest(s) to host machine, syntax is: { 80 => 8080 }, auto correction is enabled by default.
#$forwarded_ports = {}