Rakefile

require 'bundler/gem_tasks'
require 'rubygems'
require 'bundler/setup'
Bundler::GemHelper.install_tasks

task :default => [:help]

begin
  require 'rspec/core/rake_task'
  RSpec::Core::RakeTask.new(:spec)
rescue LoadError
end

def system_or_die(cmd, fail_msg=nil)
  puts "running: '#{cmd}'"
  res = system(cmd)
  if ! res
    if fail_msg.nil?
      $stderr.puts "Command exited non-zero: #{cmd}"
    else
      $stderr.puts fail_msg
    end
    exit!(1)
  end
end

def get_box_path(provider)
  boxes_dir = File.join(File.dirname(__FILE__), 'spec', 'boxes')
  Dir.mkdir(boxes_dir) if not File.directory?(boxes_dir)
  if provider == 'virtualbox'
    boxurl = 'https://s3.amazonaws.com/puppetlabs-vagrantcloud/centos-7.0-x86_64-virtualbox-puppet-1.0.1.box'
  elsif provider == 'aws'
    boxurl = 'https://github.com/mitchellh/vagrant-aws/raw/master/dummy.box'
  elsif provider == 'vmware_workstation'
    boxurl = 'https://s3.amazonaws.com/puppetlabs-vagrantcloud/centos-7.0-x86_64-vmware-puppet-1.0.1.box'
  else
    STDERR.puts "ERROR: no box URL known for provider '#{provider}'"
    exit!(1)
  end
  box_path = File.join(boxes_dir, "dummy_#{provider}.box")
  if File.exist?(box_path)
    puts "#{provider} box exists at #{box_path}"
  else
    puts "Downloading #{provider} box from #{boxurl} to #{box_path}"
    system_or_die("curl -o #{box_path} '#{boxurl}'")
  end
  return box_path
end

def fix_results_xml()
  # mainly a workaround for https://github.com/sj26/rspec_junit_formatter/issues/31
  return if ! File.exist?('results.xml')
  puts "Munging invalid control chracters in results.xml"
  content = File.read('results.xml')
  content.gsub!(/\033/, "\uFFFD")
  File.open('results.xml', 'w') { |file| file.write(content) }
end

def vmware_workstation_prep()
  # make vmware_workstation provider work in bundler install
  # see: http://www.codingonstilts.com/2013/07/how-to-bundle-exec-vagrant-up-with.html
  plugin_dir = Gem::Specification.find_by_name("vagrant-vmware-workstation").gem_dir
  puts "Found vmware_workstation plugin dir as: #{plugin_dir}"
  loader_link_path = File.join(plugin_dir, "rgloader")
  if not File.exist?(loader_link_path)
    rgloader_path = '/opt/vagrant/embedded/rgloader'
    if not File.exist?(rgloader_path)
      puts "ERROR: could not find rgloader in ~/.vagrant.d"
      exit!(1)
    end
    puts "Found rgloader path as: #{rgloader_path}"
    File.symlink(rgloader_path, loader_link_path)
    puts "Symlinked #{loader_link_path} to #{rgloader_path}"
  end
  ENV['VAGRANT_INSTALLER_EMBEDDED_DIR'] = '/opt/vagrant/embedded'
  lic_link_path = File.join(plugin_dir, "license-vagrant-vmware-workstation.lic")
  if not File.exist?(lic_link_path)
    lic_path = File.join(File.expand_path("~"), ".vagrant.d", "license-vagrant-vmware-workstation.lic")
    if not File.exist?(lic_path)
      puts "ERROR: could not find license file at #{lic_path}"
      exit!(1)
    end
    File.symlink(lic_path, lic_link_path)
    puts "Symlinked #{lic_link_path} to #{lic_path}"
  end
end

desc "Display the list of available rake tasks"
task :help do
  system("rake -T")
end

desc "Cleanup VirtualBox hostonlyifs"
task :clean_vbox do
  puts "Initializing Vagrant ProviderVirtualBox plugin"
  require 'vagrant'
  dm = VagrantPlugins::ProviderVirtualBox::Driver::Meta.new
  puts "Deleting unused VirtualBox host-only networks"
  dm.delete_unused_host_only_networks
end

namespace :acceptance do
  providers = ['virtualbox', 'vmware_workstation']

  # isolate our temp directories so we can easily remove them
  tmp_dir_path = '/tmp/vagrant-r10k-spec'
  Dir.mkdir(tmp_dir_path) unless Dir.exists?(tmp_dir_path)

  providers.each do |prov|
    desc "Run acceptance tests for #{prov}"
    task prov do |task|
      provider = task.name.split(':')[1]
      puts "Running acceptance tests for #{provider}"
      if provider == 'vmware_workstation'
        puts "ERROR: vmware_workstation acceptance tests are broken. Please see https://github.com/jantman/vagrant-r10k/issues/20"
        exit!(1)
        vmware_workstation_prep()
      end
      box_path = get_box_path(provider)
      system_or_die("VS_PROVIDER=#{provider} VS_BOX_PATH=#{box_path} TMPDIR=#{tmp_dir_path} bundle exec vagrant-spec test")
      system("rm -Rf #{tmp_dir_path}/*")
      fix_results_xml()
      if provider == 'virtualbox'
        Rake::Task['clean_vbox'].invoke
      end
    end
  end
end