Vagrantfile

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

require 'yaml'

default_config_file = File.expand_path('configs/2-controllers-2-computes.yaml',
                                       File.dirname(__FILE__))
config_file = ENV['VAGRANT_CONFIG_FILE'] || default_config_file

config = YAML.load_file(config_file)
vmdefaults = config['vms'].find {|vm| vm['name'] == 'DEFAULT' } || {}
admin_ip = config['networks']['crowbar']['admin_ip']

# Vagrantfile API/syntax version. Don't touch unless you know what you're doing!
vagrantfile_api_version = '2'

Vagrant.configure(vagrantfile_api_version) do |vconfig|
  vconfig.vm.provider :libvirt do |libvirt|
    # libvirt.host = 'localhost'
    # libvirt.username = 'root'
    # libvirt.password = 'linux'
    # libvirt.connect_via_ssh = true
    # libvirt.storage_pool_name = 'default'
  end

  config['vms'].each_with_index do |vmcfg, i|
    node_name   = vmcfg['name']
    next if node_name == 'DEFAULT'

    primary     = vmcfg['primary'] == 'true' ? true : false
    personality = vmcfg['personality']
    ip = personality == 'admin' \
         ? admin_ip
         : config['networks']['crowbar']['pool_start'].
           sub(/\.(\d+)$/) { '.%d' % ($1.to_i + i) }

    vconfig.vm.define node_name, :primary => primary do |node|
      box = vmcfg['box'] || vmdefaults['box']
      node.vm.box = box if box
      node.vm.box_check_update = false

      # Setup NIC for admin network and for Crowbar in general.
      # Vagrant requires the first interface of every VM to be NAT:
      #
      #   https://docs.vagrantup.com/v2/virtualbox/boxes.html
      #
      # which with VirtualBox means it can't communicate with the
      # other VMs.  SUSE OpenStack Cloud needs at least one (other)
      # interface which is able to communicate with other VMs.
      # On VirtualBox this will result in it being made host-only.
      node.vm.network 'private_network',
        ip: ip,
        auto_config: false,
        libvirt__dhcp_enabled: false

      # The override parameter lets us configure global config parameters
      # per provider.
      #
      # See the 'OVERRIDING CONFIGURATION' section of
      # http://docs.vagrantup.com/v2/providers/configuration.html and
      # https://github.com/mitchellh/vagrant/issues/1867 for a full
      # explanation.
      node.vm.provider 'virtualbox' do |provider, override|
        common_provider_settings(provider, vmcfg, vmdefaults)

        # Don't use headless mode
        provider.gui = true

        virtualbox_nics(provider)

        if vmcfg['sbd']
          virtualbox_sbd_disk(provider, vmcfg['sbd'])
          init_sbd_disk(override, '/dev/sdb')
        end
        virtualbox_drbd_disk(provider, node_name) if vmcfg['drbd']
      end

      node.vm.provider 'libvirt' do |provider, override|
        common_provider_settings(provider, vmcfg, vmdefaults)

        provider.volume_cache = 'unsafe'

        if vmcfg['sbd']
          libvirt_sbd_disk(provider)
          init_sbd_disk(override, '/dev/vdb')
        end
        libvirt_drbd_disk(provider, node_name) if vmcfg['drbd']

        provider.nested = true if personality == 'compute'

        libvirt_mgmt_network(config, provider)
      end

      (vmcfg['forward_ports'] || []).each do |forward|
        node.vm.network :forwarded_port,
          host_ip:
            ENV['VAGRANT_INSECURE_PORT_FORWARDS'] =~ /^(y(es)?|true|on)$/i ?
              '*' : '127.0.0.1',
          host: forward['host'],
          guest: forward['guest']
      end

      if personality == 'admin'
        provision_admin(node, admin_ip)
      else
        provision_non_admin(node, node_name, admin_ip)
      end

      if personality == 'controller'
        provision_controller(node, admin_ip)
      end

      # admin.ssh.forward_agent = true

      disable_default_synced_folder(node)
      #admin.vm.synced_folder '/mnt/suse-cloud-7', '/srv/tftpboot/suse-12.2/repos/Cloud', type: 'nfs'
    end
  end
end

def common_provider_settings(provider, vmcfg, vmdefaults)
  provider.memory = vmcfg['ram']  || vmdefaults['ram']
  provider.cpus   = vmcfg['cpus'] || vmdefaults['cpus']
end

def libvirt_drbd_disk(provider, node_name)
  ## create disk for DRBD
  provider.storage :file,
    path: "drbd-#{node_name}.qcow2",
    size: '2100M',
    device: 'vdc',
    cache: 'unsafe'
end

def virtualbox_drbd_disk(provider, node_name)
  ## create disk for DRBD
  provider.customize [
    'createhd',
    '--filename', "drbd-#{node_name}.vmdk",
    '--size', 2100,
    '--format', 'VMDK'
  ]
  provider.customize [
    'storageattach', :id,
    '--storagectl', 'SCSI Controller',
    '--port', 2,
    '--device', 0,
    '--type', 'hdd',
    '--medium', "drbd-#{node_name}.vmdk",
  ]
end

def libvirt_sbd_disk(provider)
  provider.storage :file,
    path: 'sbd.img',
    size: '8M',
    device: 'vdb',
    allow_existing: true,
    shareable: true,
    type: 'raw',
    cache: 'none'
end

def virtualbox_sbd_disk(provider, mode)
  if mode == 'create'
    provider.customize [
      'createhd',
      '--filename', 'sbd.vmdk',
      '--size', 8,
      '--format', 'VMDK',
      '--variant', 'Fixed'
    ]
    provider.customize [ 'modifyhd', 'sbd.vmdk', '--type', 'shareable' ]
  end
  provider.customize [
    'storageattach', :id,
    '--storagectl', 'SCSI Controller',
    '--port', 1,
    '--device', 0,
    '--type', 'hdd',
    '--medium', 'sbd.vmdk',
  ]
end

def init_sbd_disk(override, device)
  # Set up SBD disk
  override.vm.provision 'shell', inline: <<-EOSHELL
    zypper -n in sbd
    /usr/sbin/sbd -d #{device} create
  EOSHELL
end

def libvirt_mgmt_network(config, provider)
  # The vagrant-libvirt provider requires a private management network:
  #
  #   https://github.com/pradels/vagrant-libvirt
  #
  # This defaults to 192.168.121.0/24 but that's a bit too close to
  # conventional OpenStack networks for comfort.
  provider.management_network_address = config['networks']['management']
  provider.management_network_name = 'vagrant-mgmt'
end

def virtualbox_nics(provider)
  # Use AMD instead of Intel NICs to avoid VLAN problems
  provider.customize [ 'modifyvm', :id, '--nictype1', 'Am79C973' ]
  provider.customize [ 'modifyvm', :id, '--nictype2', 'Am79C973' ]
end

def provision_admin(admin, admin_ip)
  files_to_provision = [
    'network.json',
    # Normally Crowbar seizes control of *all* interfaces.  But in the Vagrant
    # case we don't want it to touch eth0, so we need this evil patch:
    'barclamp-network-ignore-eth0.patch',

    # increase SBD timeout to 30 seconds since Vagrant environments
    # can result in very sluggish VMs, especially when low on memory
    'increase-SBD-timeout-30s.patch',

    # fix for https://bugs.launchpad.net/nova/+bug/1691831 is not in a
    # maintenance update yet, apply it manually
    'apply-fix-lp#1691831.patch',

    # handy utility for setting up node aliases in Crowbar
    'setup-node-aliases.sh',

    # sample input files for crowbar_batch
    #'simple-cloud.yaml',
    #'HA-cloud.yaml',
    #'HA-cloud-no-compute.yaml',
    'HA-compute-cloud.yaml',
    'HA-compute-cloud-demo.yaml',
  ].map { |file| 'provisioning/admin/' + file }
  provision_to_tmp(admin, files_to_provision)

  admin.vm.provision 'shell', path: 'provisioning/admin/prep-admin.sh'
  admin.vm.provision 'shell', path: 'provisioning/admin/provision-root-files.sh'

  admin.vm.provision 'shell', path: 'provisioning/admin/switch-vdisks.sh'
  admin.vm.provision 'shell', path: 'provisioning/admin/switch-admin-ip.sh',
    args: admin_ip

  # Automatically install SUSE OpenStack Cloud on first-boot
  admin.vm.provision 'shell', path: 'provisioning/admin/install-suse-cloud.sh'
end

def provision_non_admin(node, node_name, admin_ip)
  node.vm.provision 'shell', path: 'provisioning/non-admin/update-motd'

  node.vm.provision 'shell', path: 'provisioning/non-admin/store-vagrant-name.sh',
    args: node_name

  node.vm.provision 'shell', path: 'provisioning/non-admin/register-with-suse-cloud',
    args: admin_ip

  node.vm.provision 'shell', path: 'provisioning/non-admin/deps-release'
end

def provision_controller(node, admin_ip)
  files_to_provision = [
    # utility to upload Cirros
    'upload-cirros',
    # script to start instance
    'start-testvm',
  ].map { |file| 'provisioning/controller/' + file }
  provision_to_tmp(node, files_to_provision)

  node.vm.provision 'shell', path: 'provisioning/controller/provision-root-files.sh',
    args: admin_ip
end

def provision_to_tmp(node, files_to_provision)
  files_to_provision.each do |source|
    filename = File.basename(source)
    node.vm.provision 'file', source: source, destination: "/tmp/#{filename}"
  end
end

def disable_default_synced_folder(node)
  node.vm.synced_folder '.', '/vagrant', disabled: true
end