ros_install.bash

#!/bin/bash
set -e # exit on first error

update_rosdep() {
  # Prepare rosdep to install dependencies.
  if [ ! -d /etc/ros/rosdep ]; then
    echo "Initializing rosdep ..."
    sudo rosdep init >/dev/null
  fi
  echo "Updating rosdep ..."
  rosdep update >/dev/null
  echo "Done updating rosdep."
}

config_bashrc() {
  echo "source /opt/ros/$ROS_DISTRO/setup.bash" >>~/.bashrc

  echo "Enter the path to your catkin workspace. Press enter to accept the default location '$CATKIN_DIR' "
  read custom_catkin_ws

  if [-z "$custom_catkin_ws" ]; then
    CATKIN_DIR=custom_catkin_ws
  fi

  BASHRC_PATH_STRING="source $CATKIN_DIR/devel/setup.bash"
  ROS_PACKAGE_PATH_STRING="ROS_PACKAGE_PATH=$CATKIN_DIR/src:/opt/ros/$ROS_DISTRO/share:$ROS_PACKAGE_PATH"

  echo "adding the following to your bashrc:"
  echo "  $BASHRC_PATH_STRING"
  echo "  $ROS_PACKAGE_PATH_STRING"

  echo "$BASHRC_PATH_STRING" >>~/.bashrc
  echo "$ROS_PACKAGE_PATH_STRING" >>~/.bashrc

  echo "Done configuring bashrc."
  echo "Sourcing bashrc"
  source "/home/$USER/.bashrc"
}

sudo sh -c 'echo "deb http://packages.ros.org/ros/ubuntu $(lsb_release -sc) main" > /etc/apt/sources.list.d/ros-latest.list'
sudo apt install curl
curl -s https://raw.githubusercontent.com/ros/rosdistro/master/ros.asc | sudo apt-key add -

echo "Updating package lists ..."
sudo apt update

echo "Installing ROS $ROS_DISTRO ..."

sudo sh -c 'echo "deb http://packages.ros.org/ros/ubuntu $(lsb_release -sc) main" > /etc/apt/sources.list.d/ros-latest.list'

sudo apt install curl
curl -s https://raw.githubusercontent.com/ros/rosdistro/master/ros.asc | sudo apt-key add -
sudo apt update
sudo apt install ros-$ROS_DISTRO-desktop

echo "Installing ROS $ROS_DISTRO ..."
yes | sudo apt install $PYTHON_VERSION-rosinstall $PYTHON_VERSION-catkin-pkg $PYTHON_VERSION-rosdep $PYTHON_VERSION-wstool \
ros-$ROS_DISTRO-catkin ros-$ROS_DISTRO-desktop ros-$ROS_DISTRO-pcl-ros \
ros-$ROS_DISTRO-image-transport ros-$ROS_DISTRO-image-transport-plugins \
ros-$ROS_DISTRO-libg2o ros-$ROS_DISTRO-geographic-msgs \
ros-$ROS_DISTRO-tf2-geometry-msgs >/dev/null


# ROS environment setup
echo "Setting up ROS environment..."
source /opt/ros/$ROS_DISTRO/setup.bash

echo "Would you like to configure your bashrc to automatically source setup files and set ROS_PACKAGE_PATH? (y/n)"
echo "(If you have already added it manually or in previous install, enter n)"

while read ans; do
  case "$ans" in
  y)
    config_bashrc
    update_rosdep
    break
    ;;
  n)
    update_rosdep
    break
    ;;
  *) echo "Invalid input (y/n):" ;;
  esac
done