#!/usr/bin/env bash
set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
CURRENT_BACKUP_VERSION=1
set -e; case "$1" in
  backup:export)
    OUTPUT_FILE="$2"
    BACKUP_DIR="$DOKKU_ROOT"

    BACKUP_TMP_DIR=$(mktemp -d)
    BACKUP_TMP_FILE="$BACKUP_TMP_DIR/backup.tar"

    pluginhook backup-export 1 $BACKUP_DIR | tar -cf $BACKUP_TMP_FILE --files-from -

    pushd $BACKUP_DIR > /dev/null
    ls -d */ | grep -oE '[^/]+' > "$BACKUP_TMP_DIR/.dokku_backup_apps"
    popd > /dev/null

    # we want to insert the files in the root of the tar
    pushd $BACKUP_TMP_DIR > /dev/null
    echo $CURRENT_BACKUP_VERSION > .dokku_backup_version
    tar --append -f $BACKUP_TMP_FILE .dokku_backup_version
    tar --append -f $BACKUP_TMP_FILE .dokku_backup_apps
    popd > /dev/null

    # if no file specified, output to stdout
    if [[ -z $OUTPUT_FILE ]]; then
        cat $BACKUP_TMP_FILE
    else
        mv $BACKUP_TMP_FILE $OUTPUT_FILE
    fi

    rm -rf $BACKUP_TMP_DIR
    ;;

  backup:import)
    INPUT_FILE="$2"
    [[ -z $INPUT_FILE ]] && INPUT_FILE="-"

    BACKUP_TMP_DIR=$(mktemp -d)

    tar xf $INPUT_FILE --directory=$BACKUP_TMP_DIR

    if [[ ! -f $BACKUP_TMP_DIR/.dokku_backup_version ]]; then
        echo "Unable to determine backup version"
        exit 1
    fi

    VERSION=$(< $BACKUP_TMP_DIR/.dokku_backup_version)
    if [[ $VERSION -ne 1 ]]; then
        echo "Unknown format version $VERSION"
        exit 1
    fi

    echo "Importing a version $VERSION backup..."

    BACKUP_ROOT="$BACKUP_TMP_DIR""$DOKKU_ROOT"
    TARGET_DIR="$DOKKU_ROOT"

    # create all the app directories
    while read app; do mkdir "$TARGET_DIR/$app"; echo "Imported $app"; done < $BACKUP_TMP_DIR/.dokku_backup_apps > /dev/null

    # have the plugins import their stuff
    pluginhook backup-import $VERSION "$BACKUP_ROOT" $TARGET_DIR "$BACKUP_TMP_DIR/.dokku_backup_apps"

    rm -rf $BACKUP_TMP_DIR

    echo "Import complete."
    ;;

  help)
    cat && cat<<EOF
    backup:export [file]                            Export dokku configuration files
    backup:import [file]                            Import dokku configuration files
EOF
    ;;

esac
cat
