swim.sh

#!/bin/bash

ROOT_DIR=${PWD}
SERVICES_DIR=${ROOT_DIR}'/services'
UTILS_DIR=${ROOT_DIR}'/utils'
DOS2UNIX="${UTILS_DIR}/dos2unix.exe"
BASE_DIR=${SERVICES_DIR}'/base'
SUBSCRIPTION_MANAGER_DIR=${SERVICES_DIR}"/subscription_manager"
SUBSCRIPTION_MANAGER_DIR_SRC=${SUBSCRIPTION_MANAGER_DIR}"/src"
SWIM_ADSB_DIR=${SERVICES_DIR}"/swim_adsb"
SWIM_ADSB_DIR_SRC=${SWIM_ADSB_DIR}"/src"
SWIM_EXPLORER_DIR=${SERVICES_DIR}"/swim_explorer"
SWIM_EXPLORER_DIR_SRC=${SWIM_EXPLORER_DIR}"/src"
SWIM_USER_CONFIG_DIR=${SERVICES_DIR}"/swim_user_config"
SWIM_USER_CONFIG_DIR_SRC=${SWIM_USER_CONFIG_DIR}"/src"

is_windows() {
  UNAME=$(uname)

  [[ "${UNAME}" != "Linux" ]] && [[ "${UNAME}" != "Darwin" ]]
}

fetch_user_config() {
  if [[ -d ${SWIM_USER_CONFIG_DIR_SRC} ]]
  then
    cd "${SWIM_USER_CONFIG_DIR_SRC}" || exit
    git pull -q --rebase origin master
  else
    git clone -q https://github.com/eurocontrol-swim/swim-user-config.git "${SWIM_USER_CONFIG_DIR_SRC}"
  fi
  cd "${ROOT_DIR}" || exit
}

user_config() {
  # check the prompt argument
  if [[ ${1} == '1' ]]
  then
    P='-p'
  else
    P=''
  fi

  echo "SWIM user configuration..."
  echo -e "=========================="

  ENV_FILE="${ROOT_DIR}/swim.env"

  touch "${ENV_FILE}"

  python "${SWIM_USER_CONFIG_DIR_SRC}/swim_user_config/main.py" -c "${SWIM_USER_CONFIG_DIR}/config.json" -o "${ENV_FILE}" ${P}

  if is_windows
  then
    "${DOS2UNIX}" -q "${ENV_FILE}"
  fi

  while read -r LINE; do export "${LINE}"; done < "${ENV_FILE}"

  rm "${ENV_FILE}"
}


prepare_repos() {
  echo "Preparing Git repositories..."
  echo -e "============================\n"

  echo -n "Preparing subscription-manager..."
  if [[ -d ${SUBSCRIPTION_MANAGER_DIR_SRC} ]]
  then
    cd "${SUBSCRIPTION_MANAGER_DIR_SRC}" || exit
    git pull -q --rebase origin master
  else
    git clone -q https://github.com/eurocontrol-swim/subscription-manager.git "${SUBSCRIPTION_MANAGER_DIR_SRC}"
  fi
  echo "OK"

  echo -n "Preparing swim-adsb..."
  if [[ -d ${SWIM_ADSB_DIR_SRC} ]]
  then
    cd "${SWIM_ADSB_DIR_SRC}" || exit
    git pull -q --rebase origin master
  else
    git clone -q https://github.com/eurocontrol-swim/swim-adsb.git "${SWIM_ADSB_DIR_SRC}"
  fi
  echo "OK"

  echo -n "Preparing swim-explorer..."
  if [[ -d ${SWIM_EXPLORER_DIR_SRC} ]]
  then
    cd "${SWIM_EXPLORER_DIR_SRC}" || exit
    git pull -q --rebase origin master
  else
    git clone -q https://github.com/eurocontrol-swim/swim-explorer.git "${SWIM_EXPLORER_DIR_SRC}"
  fi
  echo "OK"

  echo -e "\n\n"
}

data_provision() {
  echo "Data provisioning to Subscription Manager..."
  echo -e "============================================\n"
  docker-compose run subscription-manager-provision
  echo ""
}

start_services() {
  echo "Starting up SWIM..."
  echo -e "===================\n"
  docker-compose up -d web-server subscription-manager swim-adsb swim-explorer
  echo ""
}

stop_services_with_clean() {
  echo "Stopping SWIM..."
  echo -e "================\n"
  docker-compose down
  echo ""
}

stop_services_with_purge() {
  stop_services_with_clean &&
  docker volume ls -q | grep swim-ti_yp_prototype | xargs -r docker volume rm

  echo ""
}


stop_services() {
  echo "Stopping SWIM..."
  echo -e "================\n"
  docker-compose stop
  echo ""
}

build() {

  echo "Removing old data..."
  echo -e "==================\n"
  # Remove existing volumes
  docker volume ls -q | grep swimtiypprototype | xargs -r docker volume rm

  echo "Building images..."
  echo -e "==================\n"
  # build the base images upon which the swim services will depend on
  cd "${BASE_DIR}" || exit 1

  docker build --no-cache --force-rm -t swim-base -f Dockerfile .

  docker build --no-cache --force-rm -t swim-base.conda -f Dockerfile.conda .

  # Build the rest of the images
  docker-compose build --force-rm

  cd "${ROOT_DIR}" || exit 1
  echo ""

  echo "Removing obsolete docker images..."
  echo -e "==================================\n"
  docker images | grep none | awk '{print $3}' | xargs -r docker rmi
}

status() {
  docker ps
}

usage() {
  echo -e "Usage: swim.sh [COMMAND] [OPTIONS]\n"
  echo "Commands:"
  echo "    user_config             Generates username/password for all the SWIM related users"
  echo "    user_config --prompt    Prompts for username/password for all the SWIM related users"
  echo "    build                   Clones/updates the necessary git repositories and builds the involved docker images"
  echo "    provision               Provisions the Subscription Manager with initial data (users)"
  echo "    start                   Starts up all the SWIM services"
  echo "    stop                    Stops all the services"
  echo "    stop --clean            Stops all the services and cleans up the containers"
  echo "    stop --purge            Stops all the services and cleans up the containers and the volumes"
  echo "    status                  Displays the status of the running containers"
  echo ""
}

ACTION=${1}

case ${ACTION} in
  build)
    # update the repos if they exits othewise clone them
    prepare_repos
    # build the images
    build
    ;;
  start)
    start_services
    ;;
  stop)
    if [[ -n ${2} ]]
    then
      EXTRA=${2}

      case ${EXTRA} in
          --clean)
            stop_services_with_clean
            ;;
          --purge)
            stop_services_with_purge
            ;;
          *)
            echo -e "Invalid argument\n"
            usage
            ;;
        esac
    else
      stop_services
    fi
    ;;
  provision)
    data_provision
    ;;
  status)
    status
    ;;
  user_config)
    # update the swim-user-config repository
    fetch_user_config

    if [[ -n ${2} ]]
    then
      EXTRA=${2}

      case ${EXTRA} in
          --prompt)
            user_config 1
            ;;
          *)
            echo -e "Invalid argument\n"
            usage
            ;;
        esac
    else
      user_config 0
    fi
    ;;
  help)
    usage
    ;;
  *)
    echo -e "Invalid action\n"
    usage
    ;;
esac