update_tools_for_ubuntu.sh

#! /bin/bash

################################################
######## Parsing Command Line Arguments ########
################################################

OS_CODENAME=$(lsb_release --codename --short)
PIHOME=/home/ubuntu
DEXTER=Dexter
LIB=lib
DEXTER_PATH=$PIHOME/$DEXTER/$LIB/$DEXTER
DEXTER_SCRIPT=$DEXTER_PATH/script_tools
REPO_PACKAGE=dexter-autodetection-and-i2c-mutex

# called way down bellow
check_if_run_with_pi() {
  ## if not running with the pi user then exit
  if [ $(id -ur) -ne $(id -ur pi) ]; then
    echo "script_tools installer script must be run with \"pi\" user. Exiting."
    exit 4
  fi
}

parse_cmdline_arguments() {
  # the following option is required should the python package be installed
  # by default, the python package are not installed
  installpythonpkg=false

  # the following 3 options are mutually exclusive
  systemwide=true
  userlocal=false
  envlocal=false
  usepython3exec=false

  # the following 2 options can be used together
  updatedebs=false
  installdebs=false

  # the following option tells which branch has to be used
  selectedbranch="master" # set to master by default

  # iterate through bash arguments
  for i; do
    case "$i" in
      --install-python-package)
        installpythonpkg=true
        ;;
      --user-local)
        userlocal=true
        systemwide=false
        ;;
      --env-local)
        envlocal=true
        systemwide=false
        ;;
      --system-wide)
        ;;
      --update-aptget)
        updatedebs=true
        ;;
      --install-deb-deps)
        installdebs=true
        ;;
      --use-python3-exe-too)
        usepython3exec=true
        ;;
      develop|feature/*|hotfix/*|fix/*|DexterOS*|v*)
        selectedbranch="$i"
        ;;
    esac
  done


  # exit if git is not installed
  if [[ $installdebs = "false" ]]; then
    command -v git >/dev/null 2>&1 || { echo "This script requires \"git\" but it's not installed. Use \"--install-deb-deps\" option. Aborting." >&2; exit 1; }
  fi

  # exit if python/python3 are not installed in the current environment
  if [[ $installpythonpkg = "true" ]]; then
    command -v python >/dev/null 2>&1 || { echo "Executable \"python\" couldn't be found. Aborting." >&2; exit 2; }
    if [[ $usepython3exec = "true" ]]; then
      command -v python3 >/dev/null 2>&1 || { echo "Executable \"python3\" couldn't be found. Aborting." >&2; exit 3; }
    fi
  fi

  pushd $PIHOME > /dev/null
  result=${PWD##*/}

  echo "Updating script_tools for $selectedbranch branch with the following options:"
  echo "--install-python-package=$installpythonpkg"
  echo "--system-wide=$systemwide"
  echo "--user-local=$userlocal"
  echo "--env-local=$envlocal"
  echo "--use-python3-exe-too=$usepython3exec"
  echo "--update-aptget=$updatedebs"
  echo "--install-deb-deps=$installdebs"

  # create folders recursively if they don't exist already
  # can't use <<functions_library.sh>> here because there's no
  # cloned script_tools yet at this part of the install script
  sudo mkdir -p $DEXTER_PATH
  sudo chown ubuntu:ubuntu -R $PIHOME/$DEXTER
  popd > /dev/null
}

################################################
######## Installing Dependencies  ##############
################################################

install_dependencies() {
  if [[ $updatedebs = "true" ]]; then
    # bring in nodejs repo

    # to confirm nodejs is available for the given distribution
    curl -sLf -o /dev/null "https://deb.nodesource.com/node_9.x/dists/$OS_CODENAME/Release"
    ret_val=$?
    if [[ $ret_val -eq 0 ]]; then
      # add gpg key for nodejs
      curl -s https://deb.nodesource.com/gpgkey/nodesource.gpg.key | sudo apt-key add -
      # add nodejs to apt-get source list
      sudo sh -c "echo 'deb https://deb.nodesource.com/node_9.x $OS_CODENAME main' > /etc/apt/sources.list.d/nodesource.list"
      sudo sh -c "echo 'deb-src https://deb.nodesource.com/node_9.x $OS_CODENAME main' >> /etc/apt/sources.list.d/nodesource.list"
    else
      echo "Couldn't add Nodejs repo because it's not available for this distribution"
    fi

    sudo apt-get update
  fi
  [[ $installdebs = "true" ]] && sudo apt-get install git build-essential libi2c-dev i2c-tools python-dev python3-dev python-setuptools python3-setuptools python-pip python3-pip libffi-dev -y
}

################################################
######## Cloning script_tools  #################
################################################

clone_scriptools(){
  # it's simpler and more reliable (for now) to just delete the repo and clone a new one
  # otherwise, we'd have to deal with all the intricacies of git
  sudo rm -rf $DEXTER_SCRIPT
  pushd $DEXTER_PATH > /dev/null
  git clone --quiet --depth=1 -b $selectedbranch https://github.com/DexterInd/script_tools.git
  cd $DEXTER_SCRIPT
  # useful in case we need it
  current_branch=$(git branch | grep \* | cut -d ' ' -f2-)
  popd > /dev/null
}

################################################
######## Installing Packages  ##################
################################################

remove_python_packages() {
  # the 1st and only argument
  # takes the name of the package that needs to removed
  rm -f $PIHOME/.pypaths

  # get absolute path to python package
  # saves output to file because we want to have the syntax highlight working
  # does this for both root and the current user because packages can be either system-wide or local
  # later on the strings used with the python command can be put in just one string that gets used repeatedly
  python -c "import pkgutil; import os; \
              eggs_loader = pkgutil.find_loader('$1'); found = eggs_loader is not None; \
              output = os.path.dirname(os.path.realpath(eggs_loader.get_filename('$1'))) if found else ''; print(output);" >> $PIHOME/.pypaths
  sudo python -c "import pkgutil; import os; \
              eggs_loader = pkgutil.find_loader('$1'); found = eggs_loader is not None; \
              output = os.path.dirname(os.path.realpath(eggs_loader.get_filename('$1'))) if found else ''; print(output);" >> $PIHOME/.pypaths
  if [[ $usepython3exec = "true" ]]; then
    python3 -c "import pkgutil; import os; \
                eggs_loader = pkgutil.find_loader('$1'); found = eggs_loader is not None; \
                output = os.path.dirname(os.path.realpath(eggs_loader.get_filename('$1'))) if found else ''; print(output);" >> $PIHOME/.pypaths
    sudo python3 -c "import pkgutil; import os; \
                eggs_loader = pkgutil.find_loader('$1'); found = eggs_loader is not None; \
                output = os.path.dirname(os.path.realpath(eggs_loader.get_filename('$1'))) if found else ''; print(output);" >> $PIHOME/.pypaths
  fi

  # removing eggs for $1 python package
  # ideally, easy-install.pth needs to be adjusted too
  # but pip seems to know how to handle missing packages, which is okay
  while read path;
  do
    if [ ! -z "${path}" -a "${path}" != " " ]; then
      echo "Removing ${path} egg"
      sudo rm -f "${path}"
    fi
  done < $PIHOME/.pypaths
}

install_python_packages() {
  [[ $systemwide = "true" ]] && sudo python setup.py install \
              && [[ $usepython3exec = "true" ]] && sudo python3 setup.py install
  [[ $userlocal = "true" ]] && python setup.py install --user \
              && [[ $usepython3exec = "true" ]] && python3 setup.py install --user
  [[ $envlocal = "true" ]] && python setup.py install \
              && [[ $usepython3exec = "true" ]] && python3 setup.py install
}

install_python_package() {
  if [[ $installpythonpkg = "true" ]]; then
    # remove the python package that resides in this repo
    remove_python_packages $REPO_PACKAGE

    # and install the new one
    pushd $DEXTER_SCRIPT > /dev/null
    install_python_packages
    popd > /dev/null
  fi
}

################################################
######## Aggregating all function calls ########
################################################

# wait on this check until all scripts are ready
#check_if_run_with_pi
parse_cmdline_arguments "$@"
install_dependencies
clone_scriptools
install_python_package

exit 0