#!/bin/bash

# Script to automatically fetch and clean data on the Grid'5000 shares
# Params: 
# 1: Name for the experiment
# 2: Additional plot tar to include (see ./plots)

set -e

function work_in_progress() {
  echo -ne '\r\033[K'
  sleep 1
  echo -ne '.\r'
  sleep 1
  echo -ne '..\r'
  sleep 1
  echo -ne '...\r'
  sleep 1
}

function backup() {
  if [ -z ${1} ]; then
    echo "Missing experiment name"
    exit 1
  fi
  if [ -z ${2} ]; then
    echo "No plots will be included"
  else
    echo "The file '${2}' will be deleted when this script finishes, do you want to contiue? [y/n]" 
    while true; do
      read -n 1 -s yn
      case $yn in
        [Yy]*) break 2;;
        [Nn]*) exit ;;
        *) echo "Please enter 'y' or 'n'" ;;
      esac
    done
  fi

  # Slugify the name
  NAME=$(\
    echo "$1" | \
    sed -r s/[~\^]+//g | \
    sed -r s/[^a-zA-Z0-9]+/-/g | \
    sed -r s/^-+\|-+$//g | \
    tr A-Z a-z \
  )
  BACKUP_NAME="${NAME}-$(date +%d-%m-%y).tar"
  BACKUP_DIR="archives"
  
  G5K_BACKUP="g5k-$(date +%s).tar.gz"
  
  echo "Creating archive, this can take some time"
  ssh g5k 'rm -rf grenoble/g5k.tar* && find . -maxdepth 1 -type l -exec tar --ignore-failed-read -rf grenoble/g5k.tar -C {} exp-logs exp-data espec-times 2>&1 \; | grep -vi "no such file"' &
  
  PID=$!
  while kill -0 $PID > /dev/null 2>&1; do
    work_in_progress
  done
  
  echo "Compressing archive, this may take some time"
  ssh g5k 'gzip grenoble/g5k.tar' &
  
  PID=$!
  while kill -0 $PID > /dev/null 2>&1; do
    work_in_progress
  done
  
  echo "Copying archive to local machine"
  scp g5k:~/grenoble/g5k.tar.gz ${G5K_BACKUP}
  
  test -d ${BACKUP_DIR} || mkdir ${BACKUP_DIR}
  
  echo "Creating Backup Tar ${BACKUP_DIR}/${BACKUP_NAME}"
  tar -cvf ${BACKUP_DIR}/${BACKUP_NAME} ${G5K_BACKUP} ${2} &
  
  PID=$!
  while kill -0 $PID > /dev/null 2>&1; do
    work_in_progress
  done
  
  rm ${G5K_BACKUP} ${2}
}

function delete() {
  echo "Cleaning up g5k storage"
  ssh g5k 'rm grenoble/g5k.tar.gz || true && $(find . -type l -exec rm -rf {}/exp-logs/ {}/exp-data {}/espec-times \;)' &
  
  PID=$!
  while kill -0 $PID > /dev/null 2>&1; do
    work_in_progress
  done
}

echo "
Make sure that:

- Your ssh key to access the grid is added to the ssh-agent with 'ssh-add <KEY>'
- The config to access the grid is setup as described in:
  https://www.grid5000.fr/w/SSH#Using_SSH_ProxyCommand_feature_to_ease_the_access_to_hosts_inside_Grid.275000
"

if [[ $(ssh-add -L) =~ "The agent has no" ]]; then
  echo "No ssh key added, aborting"
  exit 1
fi
  
  
case $1 in
  -b|--backup)
    backup $2 $3
    ;;
  -d|--delete)
    delete
    ;;
  *)
    echo "Usage: ./persist.sh <OPTIONS>"
    echo ""
    echo "OPTIONS:"
    echo " -b | --backup <experiment-name> <addtitional-plot-archive>"
    echo "    backup exp-data, exp-logs and espec-times from g5k NFS"
    echo " -d | --delete"
    echo "    delete exp-data, exp-logs and espec-times from g5k NFS"
    ;;
esac