#!/bin/bash

# Exit script if command fails
# -u stops the script on unset variables
# -e stops the script on errors
# -o pipefail stops the script if a pipe fails
set -e

# Get script name
SCRIPT=$(basename "$0")

# Display Help
Help() {
    echo
    echo "$SCRIPT"
    echo
    echo "Description: Backup Odoo database from container."
    echo "Syntax: $SCRIPT [-d|-c|-o|help]"
    echo "Example: $SCRIPT -d erp -c odoo06 -o tmp/erp.zip"
    echo "options:"
    echo "  -d    Database name. Defaults to 'odoo'."
    echo "  -c    Docker container. Defaults to 'odoo'."
    echo "  -o    Output directory with or without filename. Defaults to '/var/tmp'."
    echo "  help  Show $SCRIPT manual."
    echo
}

# Show help and exit
if [[ $1 == 'help' ]]; then
    Help
    exit
fi

# Process params
while getopts ":d: :c: :o:" opt; do
    case $opt in
        c) CONTAINER="$OPTARG";;
        d) DATABASE="$OPTARG";;
        o) OUTPUT="$OPTARG";;
        \?) echo "Invalid option -$OPTARG" >&2
        Help
        exit;;
    esac
done

# Fallback to environment vars and default values
: ${DATABASE:='odoo'}
: ${CONTAINER:='odoo'}
: ${OUTPUT:='/var/tmp'}

# Check if dir var is file or folder
if [ "${OUTPUT: -4}" = ".zip" ];then
    DIR=`dirname "$OUTPUT"`
    FILE=`basename "$OUTPUT"`
else
    DIR=$OUTPUT
fi

# Ensure output directory exists
mkdir -p $DIR

# Ensure file name is set
if [ -z "$FILE" ];then
  FILE="$DATABASE.zip"
fi

# Set file path
FILEPATH="${DIR}/${FILE}"

echo "Backup database $DATABASE to $FILEPATH ..."

# Copy filestore
echo "Remove and copy filestore."
rm -rf "$DIR/filestore"
docker cp "$CONTAINER:/var/lib/odoo/filestore/$DATABASE" "$DIR/filestore"

# Create pgpass file
echo -e "\e[1A\e[KCreate pgpass file."
docker exec -e DATABASE=$DATABASE $CONTAINER /bin/bash -c 'echo "$HOST:5432:$DATABASE:$USER:$PASSWORD" > ~/.pgpass'
docker exec -e DATABASE=$DATABASE $CONTAINER /bin/bash -c 'chmod 600 ~/.pgpass'

# Dump database
echo -e "\e[1A\e[KDump database."
docker exec -e DATABASE=$DATABASE $CONTAINER /bin/bash -c 'pg_dump -h $HOST -U $USER $DATABASE' > "$DIR/dump.sql"

# Zip everything
echo -e "\e[1A\e[KZip filestore and dump to $FILE."
rm -f "$FILEPATH"
cd "$DIR"
zip -q -r "$FILE" ./dump.sql ./filestore

# Remove source files
echo -e "\e[1A\e[KCleanup source files."
rm -rf dump.sql filestore

# Notify if backup has finished
echo "The Odoo backup has finished: $FILEPATH"
