backup.sh

#!/bin/sh

DIR=$1
echo "Using directory $DIR"
CMD=$2
# This script is based on https://forum.dfinity.org/t/canister-backup/11777/2
QU=../qu/target/release/qu

set -e

mkdir -p $DIR

restore() {
    FILE="$1"
    echo "Restoring $FILE..."
    $QU raw $(cat .dfx/local/canister_ids.json | jq -r ".beacon.local") "stable_mem_write" --args-file "$FILE" | $QU send --yes --raw - > /dev/null
}

if [ "$CMD" == "restore" ]; then
    export IC_URL=http://localhost:8080
    PAGE=0
    while true; do
        for _ in {1..10}; do
            FILE="$DIR/page$PAGE.bin"
            restore $FILE &
            PAGE=$((PAGE + 1))
        done
        wait
        if [ "$(stat -f%z $FILE)" == "18" ]; then break; fi
    done
    echo "Restoring heap..."
    dfx canister call beacon stable_to_heap
    exit 0
fi

fetch() {
    FILE="$1"
    $QU raw "srn4v-3aaaa-aaaar-qaftq-cai" "stable_mem_read" --args "($PAGE:nat64)" --query |\
        $QU send --yes --raw - > $FILE
}

git rev-parse HEAD > $DIR/commit.txt

PAGE=0
while true; do
    for _ in {1..10}; do
        FILE="$DIR/page$PAGE.bin"
        echo "Fetching page $PAGE..."
        fetch $FILE &
        PAGE=$((PAGE + 1))
    done
    wait
    if [ "$(stat -f%z $FILE)" == "18" ]; then break; fi
done