Vagrantfile

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

# All Vagrant configuration is done below. The "2" in Vagrant.configure
# configures the configuration version (we support older styles for
# backwards compatibility). Please don't change it unless you know what
# you're doing.
Vagrant.configure(2) do |config|
  # The most common configuration options are documented and commented below.
  # For a complete reference, please see the online documentation at
  # https://docs.vagrantup.com.

  ## Variables (ruby syntax)
  required_plugins = %w(vagrant-r10k vagrant-vbguest vagrant-triggers vagrant-puppet-install)
  plugin_installed = false
  environment      = 'vagrant'

  ## Install Vagrant Plugins
  required_plugins.each do |plugin|
    unless Vagrant.has_plugin? plugin
      system "vagrant plugin install #{plugin}"
      plugin_installed = true
    end
  end

  ## Restart Vagrant: if new plugin installed
  if plugin_installed == true
    exec "vagrant #{ARGV.join(' ')}"
  end

  ## ensure puppet modules directory on the host before 'vagrant up'
  config.trigger.before :up do
    run "mkdir -p puppet/environment/#{environment}/modules"
    run "mkdir -p puppet/environment/#{environment}/modules_contrib"
  end

  ## Every Vagrant development environment requires a box. You can search for
  #  boxes at https://atlas.hashicorp.com/search.
  config.vm.box = 'ubuntu/trusty64'

  ## Ensure puppet installed within guest
  config.puppet_install.puppet_version = '4.3.2'

  ## Create a forwarded port mapping which allows access to a specific port
  #  within the machine from a port on the host machine. In the example below,
  #  accessing "localhost:8080" will access port 80 on the guest machine.
  config.vm.network 'forwarded_port', guest: 5000, host: 8080
  config.vm.network 'forwarded_port', guest: 443, host: 8585

  ## Run r10k
  config.r10k.puppet_dir      = "puppet/environment/#{environment}"
  config.r10k.puppetfile_path = "puppet/environment/#{environment}/Puppetfile"

  ## Custom Manifest: install needed packages
  #
  #  Note: future parser allow array iteration in the puppet manifest
  config.vm.provision 'puppet' do |puppet|
    puppet.environment_path  = 'puppet/environment'
    puppet.environment       = environment
    puppet.manifests_path    = "puppet/environment/#{environment}/manifests"
    puppet.module_path       = [
      "puppet/environment/#{environment}/modules_contrib",
      "puppet/environment/#{environment}/modules",
    ]
    puppet.manifest_file     = 'install_packages.pp'
    puppet.hiera_config_path = 'hiera.yaml'
  end

  ## Custom Manifest: build scikit-learn
  config.vm.provision 'puppet' do |puppet|
    puppet.environment_path  = 'puppet/environment'
    puppet.environment       = environment
    puppet.manifests_path    = "puppet/environment/#{environment}/manifests"
    puppet.module_path       = [
      "puppet/environment/#{environment}/modules_contrib",
      "puppet/environment/#{environment}/modules",
    ]
    puppet.manifest_file     = 'install_sklearn.pp'
    puppet.hiera_config_path = 'hiera.yaml'
  end

  ## Custom Manifest: ensure vagrant-mounted event
  #
  #  Note: future parser allow heredoc syntax in the puppet manifest (since puppet 3.5)
  config.vm.provision 'puppet' do |puppet|
    puppet.environment_path  = 'puppet/environment'
    puppet.environment       = environment
    puppet.manifests_path    = "puppet/environment/#{environment}/manifests"
    puppet.module_path       = [
      "puppet/environment/#{environment}/modules_contrib",
      "puppet/environment/#{environment}/modules",
    ]
    puppet.manifest_file     = 'vagrant_mounted.pp'
    puppet.hiera_config_path = 'hiera.yaml'
  end

  ## Custom Manifest: install redis client / server
  #
  #  Note: future parser allow heredoc syntax in the puppet manifest (since puppet 3.5)
  config.vm.provision 'puppet' do |puppet|
    puppet.environment_path = 'puppet/environment'
    puppet.environment      = environment
    puppet.manifests_path   = "puppet/environment/#{environment}/manifests"
    puppet.module_path      = [
      "puppet/environment/#{environment}/modules_contrib",
      "puppet/environment/#{environment}/modules",
    ]
    puppet.manifest_file    = 'configure_redis.pp'
  end

  ## Custom Manifest: configure system (i.e. system timezone)
  config.vm.provision 'puppet' do |puppet|
    puppet.environment_path = 'puppet/environment'
    puppet.environment      = environment
    puppet.manifests_path   = "puppet/environment/#{environment}/manifests"
    puppet.module_path      = [
      "puppet/environment/#{environment}/modules_contrib",
      "puppet/environment/#{environment}/modules",
    ]
    puppet.manifest_file    = 'configure_system.pp'
  end

  ## Custom Manifest: define webcompilers
  #
  #  Note: future parser allow heredoc sytnax (since puppet 3.5), and allows
  #        array iteration in the puppet manifest.
  config.vm.provision 'puppet' do |puppet|
    puppet.environment_path  = 'puppet/environment'
    puppet.environment       = environment
    puppet.manifests_path    = "puppet/environment/#{environment}/manifests"
    puppet.module_path       = [
      "puppet/environment/#{environment}/modules_contrib",
      "puppet/environment/#{environment}/modules",
    ]
    puppet.manifest_file     = 'compile_asset.pp'
    puppet.hiera_config_path = 'hiera.yaml'
  end

  ## Custom Manifest: install, and configure SQL database
  config.vm.provision 'puppet' do |puppet|
    puppet.environment_path  = 'puppet/environment'
    puppet.environment       = environment
    puppet.manifests_path    = "puppet/environment/#{environment}/manifests"
    puppet.module_path       = [
      "puppet/environment/#{environment}/modules_contrib",
      "puppet/environment/#{environment}/modules",
    ]
    puppet.manifest_file     = 'setup_database.pp'
    puppet.hiera_config_path = 'hiera.yaml'
  end

  ## Custom Manifest: start webserver
  #
  #  Note: future parser allow heredoc syntax in the puppet manifest (since puppet 3.5)
  config.vm.provision 'puppet' do |puppet|
    puppet.environment_path  = 'puppet/environment'
    puppet.environment       = environment
    puppet.manifests_path    = "puppet/environment/#{environment}/manifests"
    puppet.module_path       = [
      "puppet/environment/#{environment}/modules_contrib",
      "puppet/environment/#{environment}/modules",
    ]
    puppet.manifest_file     = 'start_webserver.pp'
    puppet.hiera_config_path = 'hiera.yaml'
  end

  # clean up files on the host after 'vagrant destroy'
  config.trigger.after :destroy do
    run 'rm -Rf log/database'
    run 'rm -Rf log/application'
    run 'rm -Rf log/webcompiler'
    run 'rm -Rf log/webserver'
    run 'rm -Rf build'
    run 'rm -Rf interface/static/css'
    run 'rm -Rf interface/static/img'
    run 'rm -Rf interface/static/js'
    run 'rm -Rf puppet/environment/*/modules_contrib'
    run 'rm -f src/js/.gitignore'
    run 'rm -f src/js/support_vector.js'
  end
end