#!/bin/bash

#set -o xtrace
set -u
set -o pipefail

RESTORE="false"
OUTPUT=""
while getopts ':r' OPT; do
  case $OPT in
    r) RESTORE="true"; OUTPUT="${OUTPUT}  Option: RESTORE OPERATION";;
    \?) usage; exit;;
  esac
done
shift $((OPTIND -1))

if [[ $RESTORE == "true" ]] && [[ ! -f docker-restore.sh ]]; then
  echo "Problem: docker-restore.sh missing!"
  exit 1
fi

usage () { 
  printf "Usage: vvv.sh <project> <profile> [ <profile> ... ]\n"
  printf "   Project = Compose project folder.\n"
  printf "   Profile(s) = profiles to stop and then restart after.\n"
}

scripts () {
  for MODE in backup restore; do
    FILE="docker-${MODE}.sh"
    SCRIPT="#!/bin/bash"$'\n'
    SCRIPT="${SCRIPT}sudo docker run --rm \\"$'\n'
    SCRIPT="${SCRIPT}  --name volumerize \\"$'\n'
    SCRIPT="${SCRIPT}  -e TZ=Australia/Perth \\"$'\n'
    SCRIPT="${SCRIPT}  -e VOLUMERIZE_SOURCE=/source \\"$'\n'
    SCRIPT="${SCRIPT}  -e VOLUMERIZE_TARGET=file:///backup \\"$'\n'
    SCRIPT="${SCRIPT}  -v backup_${PROJECT}-cache:/volumerize-cache \\"$'\n'
    if [[ $MODE == "restore" ]]; then
      SCRIPT="${SCRIPT}  -v backup_${PROJECT}-volume:/backup:ro \\"$'\n'
      for VOL in $VOLUMES; do
        SCRIPT="${SCRIPT}  -v ${VOL}:/source/${VOL} \\"$'\n'
      done
      SCRIPT="${SCRIPT}  blacklabelops/volumerize:1.7.1 restore"$'\n'$'\n'
    elif [[ $MODE == "backup" ]]; then
      SCRIPT="${SCRIPT}  -v backup_${PROJECT}-volume:/backup \\"$'\n'
      for VOL in $VOLUMES; do
        SCRIPT="${SCRIPT}  -v ${VOL}:/source/${VOL}:ro \\"$'\n'
      done
      SCRIPT="${SCRIPT}  blacklabelops/volumerize:1.7.1 backup"$'\n'$'\n'
    else
      printf "Fatal error 1.\n"
      exit 1
    fi
    printf "${SCRIPT}" > "${FILE}"
    chmod +x "${FILE}"
  done
}

if [[ $# -lt 2  ]]; then
  usage
  exit 1
fi

PROJECT="$1"
if [[ ! -d "../${PROJECT}" ]]; then
  printf "Folder ${1} does not exist!\n"
  exit 1
fi

VOLUMES="$(sudo ls /var/lib/docker/volumes |grep "${PROJECT}_")"
if [[ -z "$VOLUMES" ]]; then
  printf "There are no volumes found to backup/restore!\n"
  exit 1
fi

pushd "../${PROJECT}" > /dev/null || exit 1
if [[ ! $RESTORE == "true" ]]; then
  scripts
fi
shift

for PRO in $@; do
  COMMAND="sudo docker compose --profile=${PRO} stop"
  printf "${COMMAND}\n"
  ${COMMAND}
  if ! ${COMMAND}; then
    printf "Services 'docker compose stop' error."
    exit 1
  fi
  COMMAND="sudo docker compose --profile=${PRO} rm -f"
  printf "${COMMAND}\n"
  ${COMMAND}
  if ! ${COMMAND}; then
    printf "Services 'docker compose rm' error."
    exit 1
  fi
done

if [[ $RESTORE == "true" ]]; then
  "./docker-restore.sh"
else
  "./docker-backup.sh"
fi
  
for PRO in $@; do
  COMMAND="sudo docker compose --profile=${PRO} up -d"
  printf "${COMMAND}\n"
  if ! ${COMMAND}; then
    printf "Services 'up' error."
    exit 1
  fi
done

popd > /dev/null || exit 1