.check_training_config.bash

#!/bin/bash

TARGET_BRANCH=kinetic  # industrial_training git branch
ROS_RELEASE=kinetic    # ROS release version

#=======================================================================
# verify that PC configuration matches requirements for training class
#=======================================================================

function print_result() {
  if [ $? -eq 0 ]; then   # check command result
    echo -e "\e[00;32m[OK]\e[00m"
  else
    echo -e "\e[00;31m[FAIL]\e[00m"
  fi
}

function check_internet() {
  echo "Checking internet connection... "
  printf "  - %-30s" "google.com:"
  print_result $(ping -q -c1 google.com &> /dev/null)
  printf "  - %-30s" "training wiki:"
  print_result $(/usr/bin/wget -q -O /dev/null https://github.com/ros-industrial/industrial_training/wiki)
 
} #end check_internet()

function check_repo() {
  echo "Checking git repo status... "
  DIR=$(dirname "${BASH_SOURCE[0]}")
  printf "  - %-30s" "git repo exists:"
  print_result $(cd $DIR && git status &> /dev/null)
  printf "  - %-30s" "active branch:"
  ACTIVE_BRANCH=$(cd $DIR && git rev-parse --abbrev-ref HEAD)
  print_result $([ "$ACTIVE_BRANCH" == "$TARGET_BRANCH" ])

  # attempt to checkout correct branch, if needed
  if [ "$ACTIVE_BRANCH" != "$TARGET_BRANCH" ]; then
    printf "      %-28s" "attempting to fix branch:"
    print_result $(cd $DIR && git checkout -q $TARGET_BRANCH)
  fi

  printf "  - %-30s" "repo version:"
  REMOTE_URL=https://github.com/ros-industrial/industrial_training.git
  REMOTE_GIT=$(git ls-remote -q $REMOTE_URL $TARGET_BRANCH 2> /dev/null | cut -c1-6)
  LOCAL_GIT=$(cd $DIR && git rev-parse HEAD | cut -c1-6)
  print_result $([ "$REMOTE_GIT" == "$LOCAL_GIT" ])

  # attempt to update repo, if needed
  if [ "$REMOTE_GIT" != "$LOCAL_GIT" ]; then
    printf "      %-28s" "attempting to update repo:"
    print_result $(cd $DIR && git pull -q $REMOTE_URL)
    printf "      %-28s" "re-check repo version:"
    LOCAL_GIT=$(cd $DIR && git rev-parse HEAD | cut -c1-6)
    print_result $([ "$REMOTE_GIT" == "$LOCAL_GIT" ])
    [ "$REMOTE_GIT" != "$LOCAL_GIT" ] && printf "      remote: %s   local: %s\n" "$REMOTE_GIT" "$LOCAL_GIT"
  fi

} #end check_repo()

function check_deb() {
  printf "  - %-30s" "$1:"
  print_result $(dpkg-query -s $1 &> /dev/null)
}

function check_debs() {
  echo "Checking debian packages... "
  check_deb git
  check_deb meld
  check_deb ros-$ROS_RELEASE-desktop
  check_deb ros-$ROS_RELEASE-perception
  check_deb ros-$ROS_RELEASE-moveit
  check_deb ros-$ROS_RELEASE-industrial-core
  check_deb python-catkin-tools
  check_deb ros-$ROS_RELEASE-openni-launch
  check_deb ros-$ROS_RELEASE-openni-camera
  check_deb ros-$ROS_RELEASE-openni2-launch
  check_deb ros-$ROS_RELEASE-openni2-launch
  check_deb build-essential
  check_deb libfontconfig1
  check_deb mesa-common-dev
  check_deb libglu1-mesa-dev
  check_deb pcl-tools
}

function check_bashrc() {
  echo "Checking .bashrc... "
  printf "  - %-30s" "\$ROS_ROOT:"
  if [ -z ${ROS_ROOT+x} ]; then
	print_result $(false)
  else
	print_result $([ $ROS_ROOT == "/opt/ros/$ROS_RELEASE/share/ros" ])
  fi  
}

function check_qtc() {
  echo "Checking for QTCreator w/ ROS plugin... "
  printf "  - %-30s" "qtcreator-ros:"
  print_result $( which qtcreator-ros )
}


#---------------------------------------
# run the actual tests
#---------------------------------------

check_internet
check_repo
check_debs
check_qtc
check_bashrc