Top_Level_Vagrantfile

# System for quickly and painlessly provisioning Couchbase Server virtual machines across multiple Couchbase versions and OS's.
# See README.md for usage instructions

### Variable declarations - FEEL FREE TO EDIT THESE ###
begin
ip_addresses = { # Values for both OS's and Couchbase versions that are cat'd together to form a full ip address
  "centos5"  => 110,
  "centos6"  => 111,
  "centos7"  => 112,
  "centos6u4" => 113,
  "debian7"  => 120,
  "debian8"  => 121,
  "opensuse11" => 130,
  "opensuse12" => 131,
  "ubuntu10" => 140,
  "ubuntu12" => 141,
  "ubuntu14" => 142,
  "ubuntu16" => 143,
  "windows"  => 150,

  "1.8.1"    => 51,

  "2.0.1"    => 56,
  "2.1.1"    => 61,
  "2.2.0"    => 65,
  "2.5.0"    => 70,
  "2.5.1"    => 71,
  "2.5.2"    => 72,

  "3.0.0"    => 80,
  "3.0.1"    => 81,
  "3.0.2"    => 82,
  "3.0.3"    => 83,

  "3.1.0"    => 90,
  "3.1.1"    => 91,
  "3.1.2"    => 92,
  "3.1.3"    => 93,
  "3.1.4"    => 94,
  "3.1.5"    => 95,
  "3.1.6"    => 96,

  "4.0.0"    => 100,
  "4.1.0"    => 110,
  "4.1.1"    => 111,
  "4.1.2"    => 112,

  "4.5.0"    => 150,
  "4.5.1"    => 151,

  "4.6.0" => 160,
  "4.6.1" => 161,
  "4.6.2" => 162,
  "4.6.3" => 163,
  "4.6.4" => 164,
  "watson-testing" => 165,

  "5.0.0" => 170,
  "5.0.1" => 171,

  "5.1.0" => 175,
  "spock-testing" => 176,

  "vulcan-testing" => 180,

  "cbdev"    => 200,
  "perfrunner" => 210,
}
vagrant_boxes = { # Vagrant Cloud base boxes for each operating system
  "ubuntu10" => {"box_name" => "ubuntu-server-10044-x64-vbox4210",
                 "box_url"  => "http://puppet-vagrant-boxes.puppetlabs.com/ubuntu-server-10044-x64-vbox4210.box"
               },
  "ubuntu12" => "hashicorp/precise64",
  "ubuntu14" => "ubuntu/trusty64",
  "ubuntu16" => "ubuntu/xenial64",
  "debian7"  => "cargomedia/debian-7-amd64-default",
  "centos5"  => {"box_name" => "centos5u8_x64",
                 "box_url"  => "https://dl.dropbox.com/u/17738575/CentOS-5.8-x86_64.box"
               },
  "centos6"  => {"box_name" => "puppetlabs/centos-6.6-64-puppet",
                 "box_version" => "1.0.1"
                },
  "centos6u4" => {"box_name" => "hansode/centos-6.4-x86_64",
                  "box_version" => "0.2.0"
                 },
  "centos7"  => { "box_name" => "puppetlabs/centos-7.0-64-puppet",
                  "box_version" => "1.0.1"
                },
  "windows"  => "emyl/win2008r2",
  "opensuse11"  => "minesense/opensuse11.1",
  "opensuse12"   => {"box_name" => "opensuse-12.3-64",
                 "box_url" => "http://sourceforge.net/projects/opensusevagrant/files/12.3/opensuse-12.3-64.box/download",
                },
  "debian8"  => "lazyfrosch/debian-8-jessie-amd64-puppet",
}

# Collect the names of the working directory and its parent (os and cb version)
operating_system = File.basename(Dir.getwd)
version  ||= File.basename(File.expand_path('..'))

# Couchbase Server Version download links
couchbase_releases = "http://packages.couchbase.com/releases"

# Requires VPN
latest_builds =  "http://172.23.120.24/builds/latestbuilds/couchbase-server/"
watson_build_num= "4602"
watson_builds =  "#{latest_builds}/watson/#{watson_build_num}/"
spock_build_num= "5555"
spock_builds =  "#{latest_builds}/spock/#{spock_build_num}/"
vulcan_build_num= "2211"
vulcan_builds =  "#{latest_builds}/vulcan/#{vulcan_build_num}/"

couchbase_download_links = {
  "1.8.1" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_x86_64_#{version}",
  "2.0.1" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_x86_64_#{version}",
  "2.1.1" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_x86_64_#{version}",
  "2.2.0" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64",
  "2.5.0" => {
              "centos5"  => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64_openssl098",
              "centos6"  => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64",
              "ubuntu10" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64_openssl098",
              "ubuntu12" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64",
  },
  "2.5.1" => {
              "centos5"  => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64_openssl098",
              "centos6"  => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64",
              "ubuntu10" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64_openssl098",
              "ubuntu12" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64",
  },
  "2.5.2" => {
              "centos5"  => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64_openssl098",
              "centos6"  => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64",
              "ubuntu10" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64_openssl098",
              "ubuntu12" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64",
  },
  "watson-testing" => {
              "centos6"    => "#{watson_builds}/couchbase-server-enterprise-4.6.4-#{watson_build_num}-centos6.x86_64",
              "centos7"    => "#{watson_builds}/couchbase-server-enterprise-4.6.4-#{watson_build_num}-centos7.x86_64",
              "debian7"    => "#{watson_builds}/couchbase-server-enterprise_4.6.4-#{watson_build_num}-debian7_amd64",
              "debian8"    => "#{watson_builds}/couchbase-server-enterprise_4.6.4-#{watson_build_num}-debian8_amd64",
              "opensuse11" => "#{watson_builds}/couchbase-server-enterprise-4.6.4-#{watson_build_num}-suse11.x86_64",
              "ubuntu12"   => "#{watson_builds}/couchbase-server-enterprise_4.6.4-#{watson_build_num}-ubuntu12.04_amd64",
              "ubuntu14"   => "#{watson_builds}/couchbase-server-enterprise_4.6.4-#{watson_build_num}-ubuntu14.04_amd64",
  },
  "spock-testing" => {
              "centos6"    => "#{spock_builds}/couchbase-server-enterprise-5.1.0-#{spock_build_num}-centos6.x86_64",
              "centos7"    => "#{spock_builds}/couchbase-server-enterprise-5.1.0-#{spock_build_num}-centos7.x86_64",
              "debian7"    => "#{spock_builds}/couchbase-server-enterprise_5.1.0-#{spock_build_num}-debian7_amd64",
              "debian8"    => "#{spock_builds}/couchbase-server-enterprise_5.1.0-#{spock_build_num}-debian8_amd64",
              "opensuse11" => "#{spock_builds}/couchbase-server-enterprise-5.1.0-#{spock_build_num}-suse11.x86_64",
              "opensuse12" => "#{spock_builds}/couchbase-server-enterprise-5.1.0-#{spock_build_num}-suse12.x86_64",
              "ubuntu14"   => "#{spock_builds}/couchbase-server-enterprise_5.1.0-#{spock_build_num}-ubuntu14.04_amd64",
              "ubuntu16"   => "#{spock_builds}/couchbase-server-enterprise_5.1.0-#{spock_build_num}-ubuntu16.04_amd64",
  },
  "vulcan-testing" => {
              "centos6"    => "#{vulcan_builds}/couchbase-server-enterprise-5.5.0-#{vulcan_build_num}-centos6.x86_64",
              "centos7"    => "#{vulcan_builds}/couchbase-server-enterprise-5.5.0-#{vulcan_build_num}-centos7.x86_64",
              "debian8"    => "#{vulcan_builds}/couchbase-server-enterprise_5.5.0-#{vulcan_build_num}-debian8_amd64",
              "debian9"    => "#{vulcan_builds}/couchbase-server-enterprise_5.5.0-#{vulcan_build_num}-debian9_amd64",
              "opensuse11" => "#{vulcan_builds}/couchbase-server-enterprise-5.5.0-#{vulcan_build_num}-suse11.x86_64",
              "opensuse12" => "#{vulcan_builds}/couchbase-server-enterprise-5.5.0-#{vulcan_build_num}-suse12.x86_64",
              "ubuntu14"   => "#{vulcan_builds}/couchbase-server-enterprise_5.5.0-#{vulcan_build_num}-ubuntu14.04_amd64",
              "ubuntu16"   => "#{vulcan_builds}/couchbase-server-enterprise_5.5.0-#{vulcan_build_num}-ubuntu16.04_amd64",
  },
  "centos6"    => "#{couchbase_releases}/#{version}/couchbase-server-enterprise-#{version}-centos6.x86_64",
  "centos7"    => "#{couchbase_releases}/#{version}/couchbase-server-enterprise-#{version}-centos7.x86_64",
  "debian7"    => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}-debian7_amd64",
  "debian8"    => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}-debian8_amd64",
  "debian9"    => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}-debian9_amd64",
  "opensuse11" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise-#{version}-suse11.x86_64",
  "opensuse12" => "#{couchbase_releases}/#{version}/couchbase-server-enterprise-#{version}-suse12.x86_64",
  "ubuntu12"   => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}-ubuntu12.04_amd64",
  "ubuntu14"   => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}-ubuntu14.04_amd64",
  "ubuntu16"   => "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}-ubuntu16.04_amd64",
}

default_number_of_nodes = 4
default_RAM_in_MB = 1024
default_number_of_cpus = 1

## Edit these to allow other machines in your network to access the VMs

# Set to true to activate non-host access
#  make sure no one else will be impacted by this, as VMs will have forced ip that could collide
#  (typically you should be close to alone on the LAN, with a few machines, ie at home)
# alternatively set use_dhcp to true to avoid this (at the cost of not knowing the IP in advance)
public_lan = false
use_dhcp = false

# Name of the host endpoint to serve as bridge to local network
#  (if not found vagrant will ask the user for each node)
default_bridge = ["wlan0", "wlp1s0"]

# Base for IP in public network. %d replaced by node number, eg "192.168.1.10%d" to get 101, 102, ...
#  (once again, be careful of potential ip collisions!)
public_ip_base = "192.168.1.10%d"


### DO NOT EDIT BELOW THIS LINE

# Number of nodes to provision
unless ENV['VAGRANT_NODES'].nil? || ENV['VAGRANT_NODES'] == 0
  num_nodes = ENV['VAGRANT_NODES'].to_i
else
  if num_nodes.nil?
    num_nodes = default_number_of_nodes
  end
end

unless ENV['VAGRANT_CPUS'].nil? || ENV['VAGRANT_CPUS'] == 0
  num_cpus = ENV['VAGRANT_CPUS'].to_i
else
  if num_cpus.nil?
    num_cpus = default_number_of_cpus
  end
end

unless ENV['VAGRANT_RAM'].nil? || ENV['VAGRANT_RAM'] == 0
  ram_in_MB = ENV['VAGRANT_RAM'].to_i
else
  if ram_in_MB.nil?
    ram_in_MB = default_RAM_in_MB
  end
end

unless ENV['VAGRANT_VPN'].nil?
  vpn = "on"
else
  vpn = "off"
end

# Check to see if a custom download location has been given, if not use a default value (2.5.0 style)
unless ENV['VAGRANT_CB_DOWNLOAD'].nil?
  url = ENV['VAGRANT_CB_DOWNLOAD']
else
  if couchbase_download_links.has_key?(version)
    if couchbase_download_links[version].is_a?(String)
      url = couchbase_download_links[version]
    elsif couchbase_download_links[version].has_key?(operating_system)
      url = couchbase_download_links[version][operating_system]
    end
  elsif couchbase_download_links.has_key?(operating_system)
    url = couchbase_download_links[operating_system]
  end
  url ||= "#{couchbase_releases}/#{version}/couchbase-server-enterprise_#{version}_x86_64"
end

puppet_location ||= "../.."

# Check to see if a custom ip address has been given, if not generate one
if (defined?(ip)).nil?
  ip_address = "10." + String(ip_addresses[operating_system]) + "." + String(ip_addresses[version]) + ".10%d"
end

# Generate a hostname template
hostname = "#{version.gsub '.', ''}-#{operating_system}.vagrants"
if hostname =~ /^[0-9]/
  hostname.prepend("cb")
end
hostname.prepend("node%d-")

# Check to see if the vagrant command given was 'up', if so print a handy dialogue
if ARGV[0] == "up" && !ARGV[1]
  puts "\e[32m=== Upping #{num_nodes} node(s) on #{operating_system} and cb version #{version} ==="
end

### Start the vagrant configuration ###
Vagrant.configure("2") do |config|

  # Define VM properties for each node (for both virtualbox and
  # libvirt providers).
  config.vm.provider :virtualbox do |vb|
    vb.memory = ram_in_MB
    vb.cpus = num_cpus
    vb.customize ["modifyvm", :id, "--ioapic", "on"]
    vb.customize ["modifyvm", :id, "--natdnshostresolver1", "#{vpn}"]
    vb.linked_clone = true if Vagrant::VERSION >= "1.8.0"
    vb.destroy_unused_network_interfaces = true
  end
  config.vm.provider :libvirt do |libvirt|
    libvirt.memory = ram_in_MB
    libvirt.cpus = num_cpus
  end

  config.vm.synced_folder ENV['HOME'], "/vmhost_home/"

  # Define the vagrant box download location
  if !(vagrant_boxes[operating_system]["box_url"].nil?)
    config.vm.box_url = vagrant_boxes[operating_system]["box_url"]
  end

  # Define the vagrant box name
  if !(vagrant_boxes[operating_system]["box_name"].nil?)
    box_name = vagrant_boxes[operating_system]["box_name"]
  else
    box_name = vagrant_boxes[operating_system]
  end

  # Define the box version if specified - default to most recent
  if !(vagrant_boxes[operating_system]["box_version"].nil?)
    box_version = vagrant_boxes[operating_system]["box_version"]
  end

  # Check to see if the VM is not running Windows and provision with puppet
  if !(operating_system.include?("win"))
    if (operating_system.include?("ubuntu16"))
      config.vm.provision "shell", inline: "apt-get install -y puppet"
    end
    # Provision the server itself with puppet
    config.vm.provision "puppet" do |puppet|
      puppet.manifests_path = puppet_location # Define a custom location and name for the puppet file
      puppet.manifest_file = "puppet.pp"
      puppet.facter = { # Pass variables to puppet
        "version" => version, # Couchbase Server version
        "url" => url, # Couchbase download location
      }
    end
  end

  # Provision Config for each of the nodes
  1.upto(num_nodes) do |num|
    config.vm.define "node#{num}" do |node|
      node.vm.box = box_name
      if !(box_version.nil?)
        node.vm.box_version = box_version
      end
      if Vagrant.has_plugin?("vagrant-cachier")
        # Configure cached packages to be shared between instances of the same base box.
        config.cache.scope = :box
      end
      if public_lan && use_dhcp
        node.vm.network :public_network, :bridge => default_bridge
        puts "Public LAN ip obtained via DHCP, find it by connecting to the node: vagrant ssh node#{num}"
      elsif public_lan
       node.vm.network :public_network, :bridge => default_bridge, :ip =>  public_ip_base % num
       puts "Public LAN ip : #{public_ip_base % num}"
      else
        node.vm.network :private_network, :ip => ip_address % num
        if Vagrant.has_plugin?("landrush")
          puts "Private network (host only) : http://#{hostname % num}:8091/"
        else
          puts "Private network (host only) : http://#{ip_address % num}:8091/"
        end
      end
      node.vm.hostname = hostname % num
      node.vm.provider "virtualbox" do |v|
        v.name = "Couchbase Server #{version} #{operating_system.gsub '/', '_'} Node #{num}"
        if(operating_system.include?("win")) # If the VM is running Windows it will start with a GUI
          v.gui = true
        end
      end
      if Vagrant.has_plugin?("landrush")
        node.landrush.enabled = true
        node.landrush.tld = "vagrants"
      end
      # Postfix a random value to hostname to uniquify it.
      node.vm.provider "libvirt" do |v|
        v.random_hostname = true
      end
    end
  end

  if ARGV[0] == "up" && !ARGV[1]
    if public_lan && use_dhcp
      puts "\e[32m=== Upping #{num_nodes} node(s) on public LAN via DHCP ==="
    elsif public_lan
      puts "\e[32m=== Upping #{num_nodes} node(s) on public LAN IPs #{public_ip_base.sub('%d','')}{1..#{num_nodes}} ==="
    else
      puts "\e[32m=== Upping #{num_nodes} node(s) on IPs #{ip_address.sub('%d','')}{1..#{num_nodes}} ==="
    end
  end

end
rescue
end