VagrantfileAWS

# expfactory-vm, testing
# poldracklab

VAGRANTFILE_API_VERSION = "2"

$script = <<SCRIPT

if [ ! -d $HOME/miniconda2 ]
then
 # install anaconda
 wget http://repo.continuum.io/miniconda/Miniconda-latest-Linux-x86_64.sh -O miniconda.sh
 chmod +x miniconda.sh
 ./miniconda.sh -b
 echo "export PATH=$HOME/miniconda2/bin:\\$PATH" >> .bashrc
 echo "export PATH=$HOME/miniconda2/bin:\\$PATH" >> .env
fi

# install python packages
$HOME/miniconda2/bin/conda update --yes conda
$HOME/miniconda2/bin/pip install setuptools
$HOME/miniconda2/bin/conda install --yes pip numpy scipy nose traits networkx
$HOME/miniconda2/bin/conda install --yes pandas

sudo apt-get update -y > /dev/null
sudo apt-get install -y --force-yes git-core
sudo apt-get install -y --force-yes build-essential libssl-dev libcurl4-gnutls-dev libexpat1-dev gettext unzip
sudo apt-get install -y --force-yes nginx

# Install expfactory
sudo apt-get install -y --force-yes libreadline6 libreadline6-dev
sudo apt-get install -y --force-yes libncurses-dev
$HOME/miniconda2/bin/pip install python-Levenshtein
$HOME/miniconda2/bin/pip install expfactory

# Psiturk
$HOME/miniconda2/bin/pip install psiturk
sudo pip install --upgrade requests
psiturk-setup-example

# Install postgresql (not tested)
sudo apt-get install -y --force-yes postgresql postgresql-contrib
sudo apt-get install -y --force-yes pgadmin3
sudo apt-get install -y --force-yes libpq-dev
sudo -u postgres psql -c "CREATE USER expfactory WITH PASSWORD 'expfactory';"
sudo -u postgres psql -c "CREATE DATABASE expfactory;"
$HOME/miniconda2/bin/pip install psycopg2

# Install MySQL
sudo debconf-set-selections <<< 'mysql-server mysql-server/root_password password expfactory'
sudo debconf-set-selections <<< 'mysql-server mysql-server/root_password_again password expfactory'
sudo apt-get install -y --force-yes libmysqlclient-dev
sudo apt-get install -y --force-yes mysql-server
$HOME/miniconda2/bin/pip install MySQL-python
sudo apt-get install -y --force-yes vim

sudo /etc/init.d/nginx start

# Install expfactory to generate everything else
mkdir -p $HOME/expfactory-build
if [ ! -d $HOME/expfactory-build/expfactory-python ]
then
  git clone https://github.com/expfactory/expfactory-python.git $HOME/expfactory-build/expfactory-python
  cd $HOME/expfactory-build/expfactory-python
  $HOME/miniconda2/bin/pip install -r requirements.txt  
  $HOME/miniconda2/bin/python setup.py install
fi

# Download experiments, battery, vm, doc repos
sudo chmod u+x $HOME/expfactory-build/expfactory-python/script/download_repos.py
$HOME/miniconda2/bin/python $HOME/expfactory-build/expfactory-python/script/download_repos.py $HOME/expfactory-build
sudo mysql -uroot -pexpfactory < $HOME/expfactory-build/vm/script/createdb.sql

# Replace standard index.html with link to expfactory docs
sudo mv $HOME/expfactory-build/vm/static/index.html /var/www/nginx-default/index.html

CUSTOM_INSTALL="False"

if [[ $CUSTOM_INSTALL == "False" ]] ; then
    sudo chmod u+x $HOME/expfactory-build/expfactory-python/script/standard_setup_battery.py 
    $HOME/miniconda2/bin/python $HOME/expfactory-build/expfactory-python/script/standard_setup_battery.py $HOME/expfactory-battery
else

if ! [ -f $HOME/custom_setup_battery.py ]; then
echo """
#!/usr/bin/python
from expfactory.battery import generate
import sys
battery_dest = sys.argv[1]
config = [SUB_CONFIG_SUB]
experiments = [SUB_EXPERIMENTS_SUB]
generate(battery_dest=battery_dest,experiments=experiments,config=config)        
""" >/tmp/aijfaef
sudo cp /tmp/aijfaef $HOME/custom_setup_battery.py
fi

sudo chmod u+x $HOME/custom_setup_battery.py
$HOME/miniconda2/bin/python $HOME/custom_setup_battery.py $HOME/expfactory-battery

fi 

# User should log in to battery folder
cd $HOME/expfactory-battery
echo "Experiment factory battery generation complete!"

SCRIPT

Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
  config.ssh.forward_x11 = true

  config.vm.define :engine do |engine_config|
      #engine_config.vm.box = "aws"
      engine_config.vm.box_url = "https://github.com/mitchellh/vagrant-aws/raw/master/dummy.box"
      engine_config.vm.hostname = 'expfactory-vm'

    engine_config.vm.provider :aws do |aws, override|
        aws.access_key_id = "AWS_ACCESS_KEY"
        aws.secret_access_key = "AWS_SECRET_ACCESS_KEY"
        aws.keypair_name = "AWS_KEYPAIR_NAME"
        aws.region = "us-west-2" 
        aws.ami = "ami-5189a661"
        aws.instance_type = "t2.small"
        aws.security_groups = "AWS_SECURITY_GROUP"
        aws.iam_instance_profile_name = "AWS_IAM_INSTANCE_PROFILE_NAME"
        override.ssh.username = "ubuntu"
        override.ssh.private_key_path = "AWS_PRIVATE_KEY_PAIR"
        override.nfs.functional = false
    end
        engine_config.vm.provision "shell", :privileged => false, inline: $script
  end
end