hcoop-backup: Take file from stdin and split it.

[hcoop/scripts.git] / s3.common

# Common functions for dealing with Amazon S3.

# units for BWLIMIT are KB/s
BWLIMIT=325
# units for CHUNKSIZE are MB
CHUNKSIZE=5000

BUCKET=hcoop.net-backups
BACKUPDIR=full
SUBDIR=$(date +%Y.%m.%d)
S3CMD=$(dirname $0)/s3

IFS=$'\n'

export S3_ACCESS_KEY_ID=$(cat ~mwolson_admin/.amazon/access.key)
export S3_SECRET_ACCESS_KEY=~mwolson_admin/.amazon/secret.key

function s3_cmd () {
    # $1: command (get|put|ls|rm)
    # $2: remote file
    # $3: local file
    local cmd=$1
    shift
    local bwarg
    if test "$cmd" = "put"; then
        bwarg="${BWLIMIT}K";
    else
        bwarg=
    fi
    $S3CMD $cmd $BUCKET "$1" "$2" $bwarg
}

function move_over () {
    # Move file to its offsite destination.
    # Expects the file to come from STDIN.
    # $1: filename, $2: relative directory (optional)
    if test -z "$1" || test -n "$3"; then
        echo "Bad programming"
        exit 1
    fi
    local FILE=$1
    local DEST=$BACKUPDIR/$SUBDIR
    if test -n "$2"; then
        DEST=$DEST/$2
    fi
    split -d -b ${CHUNKSIZE}m - ${FILE}.
    for i in ${FILE}.*; do
        s3_cmd put $DEST/$i $i
        rm -f $i
    done
}


function prune_old_backups () {
    # Implement me
    local oldpwd=$PWD
    cd $BACKUPDIR
    find . -mindepth 1 -maxdepth 1 -type d -ctime +7 \
        -execdir rm -fr '{}' \; || true
    rm -rf $SUBDIR
    mkdir -p $SUBDIR
    cd $oldpwd
}