#/bin/bash

echo

case "$1" in

  "clean" )
    echo "Cleanup docker, remove FLARECAST containers"
    TO_DELETE=$(docker ps -qa)
    if [ "$2" == "--keep_db" ]
    then
      DB_CONTAINER_ID=$(docker ps -f name=db -qa)
      echo "keep database container "$DB_CONTAINER_ID
      TO_DELETE=$(docker ps -qa | grep -v $DB_CONTAINER_ID)
    fi
    docker rm -f $TO_DELETE
    if [ "$2" == "--all" ]
    then
      echo "Remove all FLARECAST images"
      docker rmi $(docker images -f label='eu.flarecast.type' -q)
    fi
  ;;
  
  "update" )
    echo "Update infrastructure to latest version"
    curl -k "https://dev.flarecast.eu/stash/projects/INFRA/repos/dev-infra/browse/docker-compose.yml?raw" > docker-compose.yml
    docker-compose pull
    docker pull flarecast/swpcimport
    docker pull flarecast/all-feature-extraction
  ;;

  "run" )
    echo "Run flarecast infrastructure"
    echo "loading database"
    docker-compose up -d db_service
    echo
    echo "wait 10 s..."
    sleep 10
    echo
    echo "start infrastructure"
    docker-compose up
  ;;

  * )
      echo "usage: infrastructure [clean [--all | --keep_db] | update | run]"
      echo
      echo "clean: Removes all docker containers"
      echo "     --keep_db: Removes all docker containers except database container"
      echo "     --all: Removes all docker containers and images"
      echo "update: Downloads newest docker images for the flarecast infrastructure"
      echo "run: Runs the flarecast infrastructure"
  ;;

esac

echo