install-igor

#!/bin/bash

INSTALL_ALL=true
bold=$(tput bold)
normal=$(tput sgr0)

USAGE="Usage: install.sh [-server] [-cli] [-web] [-h|-help]
${bold}(must be run as root)${normal}"
HELP="
This script installs or upgrades Igor applications. A first-time install sets
up the initial environment for running them. Ensure it is in the same directory
as the ${bold}igor2.tar.gz${normal} file created with the ${bold}build-igor${normal} script.

By default, all applications are installed or upgraded. To target specific
applications, use the ${bold}-server${normal}, ${bold}-cli${normal}, and/or ${bold}-web${normal} flags.

For igor-server installation, the script will use the OS's package manager to
download npm and sqlite if they are not already present. If the database needs
to be upgraded, run the executable in the db-migrate folder, which is unpacked
during the installation process. The server will fail to start if the required
database version does not match (details will be in the output log).

When installing either igor-server or igor-web, a sub-folder named 'igor-extra'
will be created in your current working directory. This folder contains example
scripts for setting up these applications with logrotate and systemd.

The script creates the 'igor' system user with a home directory and sets the
IGOR_HOME environment variable to this folder (if no prior installation was
performed). This is the default location for files used by the server unless
specified otherwise in its YAML configuration.

*** Default Application Install Locations ***
Servers + Web App: /opt/igor
CLI App:           /usr/local/bin
Config Files:      /etc/igor
"

while [[ $# -gt 0 ]]; do
   arg="$1"
   case $arg in
   -cli)
      INSTALL_CLI=true
      INSTALL_ALL=false
      shift
      ;;
   -server)
      INSTALL_SERVER=true
      INSTALL_ALL=false
      shift
      ;;
   -web)
      INSTALL_WEB=true
      INSTALL_ALL=false
      shift
      ;;
   -all)
      INSTALL_ALL=true
      shift
      ;;
   -h | -help)
      echo "$USAGE"
      echo "$HELP"
      exit 0
      ;;
   -*)
      echo "... unknown flag $1"
      echo "$USAGE"
      exit 1
      ;;
   *)
      echo "... unknown argument $1"
      echo "$USAGE"
      exit 1
      ;;
   esac
done

# Check if the script is being run as root
if [ "$(id -u)" != "0" ]; then
   echo "${bold}This script must be run as root.${normal} See full help for important info." 1>&2
   exit 1
fi

function get_os_type() {
   case $(uname) in
   Linux)
      command -v apt-get >/dev/null && {
         DEBIAN=1
         echo "detected apt-get, assuming Debian-based distro"
         return
      }
      command -v yum >/dev/null && {
         RH=1
         echo "detected yum, assuming RH-based distro"
         return
      }
      ;;
   *)
      echo "Unsupported OS. Did not detect Debian or RedHat distro features."
      exit 1
      ;;
   esac
}

get_os_type

if [[ "$INSTALL_WEB" == true && "$INSTALL_CLI" == true && "$INSTALL_SERVER" == true ]]; then
   INSTALL_ALL=true
fi

# Define variables
USERNAME=igor
HOME_DIR=/home/igor
CLI_DIR=/usr/local/bin
SERVER_DIR=/opt/igor
ETC_DIR=/etc/igor
LOG_DIR=/var/log/igor

if id "$USERNAME" &>/dev/null; then
   echo "System user '$USERNAME' already exists ... skipping creation"
else
   echo "Adding '$USERNAME' as system user"
   if [ $DEBIAN ]; then
      adduser igor --system --group
   fi
   if [ $RH ]; then
      useradd -r -m -s /sbin/nologin igor
   fi
   chmod 750 $HOME_DIR
fi

# Set IGOR_HOME for everyone if setting doesn't already exist
if [ ! -f /etc/profile.d/igor.sh ]; then
   if [ $DEBIAN ]; then
      grep -qxF "export IGOR_HOME=" /etc/environment || echo "export IGOR_HOME=$HOME_DIR" >>/etc/environment
   fi
   if [ $RH ]; then
      grep -qxF "export IGOR_HOME=" $HOME_DIR/.bashrc || echo "export IGOR_HOME=$HOME_DIR" >>$HOME_DIR/.bashrc
   fi
   if [[ ! -f /etc/profile.d/igor.sh ]]; then
      echo "export IGOR_HOME=$HOME_DIR" >/etc/profile.d/igor.sh
   fi
fi

source /etc/profile.d/igor.sh
mkdir -p "$ETC_DIR/certs"
tar -xzvf igor2.tar.gz --transform='s|README-certs.txt|README|' --strip-components=2 -C $ETC_DIR/certs ./other/README-certs.txt
chown igor:igor -R $ETC_DIR
mkdir -p $SERVER_DIR
chown root:root $SERVER_DIR

if [[ $INSTALL_ALL == true || $INSTALL_SERVER == true ]]; then
   echo -e "Installing Igor server\n"
   echo "Starting Igor server package install ..."
   if [ $DEBIAN ]; then
      apt-get install -y nmap sqlite3
   fi
   if [ $RH ]; then
      yum install -y nmap sqlite
   fi
   echo -e "Igor server package installation finished.\n"

   tar -xzf igor2.tar.gz --strip-component=1 -C $SERVER_DIR ./igor-server
   if [ -f "$ETC_DIR/igor-server.yaml" ]; then
      tar -xzvf igor2.tar.gz --transform='s|igor-server.yaml|igor-server.yaml.blank-orig|' --strip-components=2 -C $ETC_DIR ./etc/igor-server.yaml
      echo -e "$ETC_DIR/igor-server.yaml already exists - writing out blank server yaml file for reference\n"
      W_BLANK=true
   else
      tar -xzf igor2.tar.gz --strip-components=2 -C $ETC_DIR ./etc/igor-server.yaml
   fi
   if [ -f "$ETC_DIR/igor-clusters.yaml" ]; then
      echo -e "$ETC_DIR/igor-clusters.yaml already exists - will not be overwritten\n"
   else
      tar -xzf igor2.tar.gz --strip-components=2 -C $ETC_DIR ./etc/igor-clusters.yaml
   fi
   tar -xzf igor2.tar.gz ./db-migrate
fi

if [[ $INSTALL_ALL == true || $INSTALL_WEB == true ]]; then
   echo -e "Installing Igor web server\n"
   tar -xzf igor2.tar.gz --strip-components=1 -C $SERVER_DIR ./igor-web
   if [ -f "$ETC_DIR/igor-web.yaml" ]; then
      tar -xzvf igor2.tar.gz --transform='s|igor-web.yaml|igor-web.yaml.blank-orig|' --strip-components=2 -C $ETC_DIR ./etc/igor-web.yaml
      echo -e "$ETC_DIR/igor-web.yaml already exists - writing out blank web server yaml file for reference\n"
      W_BLANK=true
   else
      tar -xzf igor2.tar.gz --strip-components=2 -C $ETC_DIR ./etc/igor-web.yaml
   fi
   # Attempt to save an existing config.json file and put it back after update
   if [ -f "$SERVER_DIR/web-content/config.json" ]; then
      mv $SERVER_DIR/web-content/config.json /tmp/igorweb.config.json
   fi
   rm -rf $SERVER_DIR/web-content
   tar -xzf igor2.tar.gz --strip-components=1 -C $SERVER_DIR ./web-content
   if [ -f "/tmp/igorweb.config.json" ]; then
      mv /tmp/igorweb.config.json $SERVER_DIR/web-content/config.json
   fi
fi

if [[ $INSTALL_ALL == true || $INSTALL_SERVER == true || $INSTALL_WEB == true ]]; then
   mkdir $LOG_DIR && chown igor:igor $LOG_DIR
   chown root:root -R /opt/igor/* && chmod 755 /opt/igor/*
   chown igor:igor $ETC_DIR/igor-*.yaml && chmod 660 $ETC_DIR/igor-*.yaml && chmod 664 $ETC_DIR/igor-clusters.yaml
   if [ $W_BLANK ]; then
      chown igor:igor $ETC_DIR/igor-*.blank-orig && chmod 664 $ETC_DIR/igor-*.blank-orig
   fi
   tar -xzf igor2.tar.gz ./igor-extra
fi

if [[ $INSTALL_ALL == true || $INSTALL_CLI == true ]]; then
   echo -e "Installing Igor CLI\n"
   tar -xzf igor2.tar.gz --strip-components=1 -C $CLI_DIR ./igor
   chown root:root "$CLI_DIR/igor" && chmod 755 "$CLI_DIR/igor"
   if [ -f "$ETC_DIR/igor.yaml" ]; then
      tar -xzvf igor2.tar.gz --transform='s|igor.yaml|igor.yaml.blank-orig|' --strip-components=2 -C $ETC_DIR ./etc/igor.yaml
      echo "$ETC_DIR/igor.yaml already exists - writing out blank CLI yaml file for reference"
      chown igor:igor $ETC_DIR/igor.yaml.blank-orig && chmod 664 $ETC_DIR/igor.yaml.blank-orig
   else
      tar -xzf igor2.tar.gz --strip-components=2 -C $ETC_DIR ./etc/igor.yaml
      chown igor:igor $ETC_DIR/igor.yaml && chmod 664 $ETC_DIR/igor.yaml
   fi
   $CLI_DIR/igor completion bash >/etc/bash_completion.d/igor
fi