check-setup.sh

#!/bin/bash

short_system=$(uname -s)
sys_vagrant="0"
sys_cygwin="0"
sys_osx="0"

mongo_missing="0"
node_missing="0"
heroku_missing="0"
npm_missing="0"

# set this to the number of the current lab
cur_lab=6

system=$(uname -a | cut -d' ' -f1,2)
if [ "$system" == "Linux precise32" ] || [ "$system" == "Linux vagrant-ubuntu-trusty-64" ]
then
  sys_vagrant="1"  
  echo "Running on Vagrant guest"
  
  user=$(whoami)
  
  if [ "$user" != "root" ]
  then
    echo "ERROR: You must run this script with sudo. Enter the following and hit enter."
    echo "sudo bash lab6/check-setup.sh"
    exit
  fi
  
elif [ $short_system == "Darwin"  ]
then
  sys_osx="1"
  echo "Running on Mac OSX"
else
  sys_cygwin="1"
  echo "Running on Windows"
fi

if [ "$sys_vagrant" == "1" ]
then
# on vagrant guest

  required_pkg=( "mongo" "heroku" "node" "npm")

  all_present="1"

  for i in ${required_pkg[@]}
  do
    binloc="$(which $i)"
    if [ "${#binloc}" == "0" ]
    then
      echo "You don't have $i"
      all_present="0"
      if [ "$i" == "mongo" ]
      then
        mongo_missing="1"
      elif [ "$i" == "heroku" ]
      then
        heroku_missing="1"
      elif [ "$i" == "node" ]
      then
        node_missing="1"
      elif [ "$i" == "npm" ]
      then
        npm_missing="1"
      fi
    fi
  done
  
  if [ "$mongo_missing" == "1" ]
  then
    echo "Installing MongoDB..."
    mongo_res=$(/home/vagrant/introHCI/mongo.sh)

    mongo_loc=$(which mongo)
    if [ "${#mongo_loc}" == "0" ]
    then
        echo "Auto install failed."
    else
        echo "Auto install succeeded"
    fi
  else
    echo "Ensuring mongo data directory is set up."
      mkdir -p /data/db;
    chown vagrant /data/db;
  fi
  
  if [ "$heroku_missing" == "1" ]
  then
    heroku_res=$(echo "Installing Heroku Toolbelt...";
    /home/vagrant/introHCI/heroku.sh)
    heroku_loc=$(which heroku)
    if [ "${#heroku_loc}" == "0" ]
    then
        echo "Auto install failed."
    else
        echo "Auto install succeeded"
    fi
  fi
  
  if [ "$node_missing" == "1" ]
  then
    echo "Installing nodejs"    
    node_res=$(/home/vagrant/introHCI/nodejs.sh)
    node_loc=$(which node)
    if [ "${#node_loc}" == "0" ]
    then
        echo "Auto install failed."
    else
        echo "Auto install succeeded"
    fi
  fi
  
  if [ "$npm_missing" == "1" ]
  then
    echo "Installing npm"  
    npm_res=$(/home/vagrant/introHCI/nodejs.sh)
    npm_loc=$(which npm)
    if [ "${#npm_loc}" == "0" ]
    then
        echo "Auto install failed."
    else
        echo "Auto install succeeded"
    fi
  fi

  # current lab hardcoded
  node_status=$(cd lab4;npm ls 2>&1)

  if [[ $node_status == *"UNMET DEPENDENCY"* ]]
  then
    echo "FAIL: Node is missing packages"
    echo "Attempting to repair."
    install_status=$(cd lab6; npm -y install --no-bin-links)

    node_status=$(cd lab6;npm ls 2>&1)
  
    if [[ $node_status != *"UNMET DEPENDENCY"* ]]
    then
      echo "PASS: Repair successful. All node packages installed."
    fi
  fi

  # change ssh timeout to fix disconnect issues

  ssh_result=$(grep "Setup SSH timeouts" /etc/ssh/sshd_config | wc -l | xargs)

  if [ $ssh_result != "1" ]
  then
    echo "Patching ssh timeout configuration."

    echo -e "\n# Setup SSH timeouts\nClientAliveInterval 30\nClientAliveCountMax 4" >> /etc/ssh/sshd_config 
    echo -e "\n# Setup SSH timeouts\nServerAliveInterval 30\nServerAliveCountMax 4" >> /etc/ssh/ssh_config 
    /etc/init.d/ssh restart > /dev/null

  fi

  if [ $all_present == "1" ]
  then
    echo "PASS: Vagrant is correctly set up."
  fi


else

  if [ "$sys_osx" == "1" ]
  then
  #on osx host system
    dirloc="$(pwd)"

    IFS=/ read -a dirarr <<< "$dirloc"
    if [ "${dirarr[4]}" != "introHCI" ]
    then
      echo "FAIL: Either you are not running this script in the introHCI directory or your directory is named incorrectly."
    else
      echo "PASS: introHCI directory named and positioned correctly"
    fi

  elif [ "$sys_cygwin" == "1"  ]
  then
    dirloc="$(pwd)"

    IFS=/ read -a dirarr <<< "$dirloc"
    if [ "${dirarr[5]}" != "introHCI" ]
    then
      echo "FAIL: Either you are not running this script in the introHCI directory or your directory is named incorrectly."
    else
      echo "PASS: introHCI directory named and positioned correctly"
    fi
  fi

  
  vagrant_check=$(grep MSB Vagrantfile | wc -l | xargs)

  if [ $vagrant_check == "4" ]
  then
    echo "PASS: You are using the correct Vagrantfile"
  else
    echo "FAIL: CSE170 Vagrantfile not found. Are you running this in the introHCI directory?"
  fi

  missing_dirs="0"
  hcidirs=$(ls)

  # current lab hardcoded
  for i in {1..6} 
  do
    target_dir="lab$i"
  if [[ $hcidirs == *"$target_dir"* ]]
    then
      echo "Found $target_dir"
    else
      echo "ERROR: Cannot find $target_dir"
      missing_dirs="1"
    fi
  done

  if [ $missing_dirs == "1" ]
  then
    echo "FAIL: Your introHCI directory is missing the above lab folders."
  else
    echo "PASS: All required lab directories present."
  fi

fi