backup

#!/bin/bash

if [ -z "$AWS_ACCESS_KEY_ID" ]; then
  echo "AWS_ACCESS_KEY_ID must be set"
fi

if [ -z "$AWS_SECRET_ACCESS_KEY" ]; then
  echo "AWS_SECRET_ACCESS_KEY must be set"
fi

if [ -z "$S3BUCKET" ]; then
  echo "S3BUCKET must be set"
fi

if [ -z "$DATEFORMAT" ]; then
  DATEFORMAT='%Y%m%d_%H%M%S'
fi

if [ -z "$FILEPREFIX" ]; then
  FILEPREFIX=''
fi

if [ "$1" == "backup" ]; then
  echo "Starting backup..."

  DATE=$(date +$DATEFORMAT)
  FILENAME=$FILEPREFIX$DATE.tar.gz
  FILE=/backup/backup-$FILENAME

  mongodump --quiet -h $MONGO_PORT_27017_TCP_ADDR -p $MONGO_PORT_27017_TCP_PORT
  tar -zcvf $FILE dump/
  aws s3api put-object --bucket $S3BUCKET --key $FILENAME --body $FILE
  echo "Cleaning up..."
  rm -rf dump/ $FILE
elif [ "$1" == "list" ]; then
        echo "Listing backups..."

        aws s3api list-objects --bucket $S3BUCKET --query 'Contents[].{Key: Key, Size: Size}' --output table
else
  echo "Starting restore"

  FILE=$1

  aws s3api get-object --bucket $S3BUCKET --key $FILE /backup/$FILE
  tar -zxvf /backup/$FILE -C /backup
  mongorestore -h $MONGO_PORT_27017_TCP_ADDR -p $MONGO_PORT_27017_TCP_PORT dump/
  echo "Cleaning up..."
  rm -rf dump/ /backup/$FILE
fi