#!/bin/bash

# mynode_quicksync.sh
# Downloads blockchain up through Feb 2019 for MUCH faster syncing
# Seeds afterwards if not disabled
# Dependencies: Must run after mynode script so HDD is mounted

set -e
set -x

source /usr/share/mynode/mynode_config.sh

mkdir -p $QUICKSYNC_DIR
mkdir -p $QUICKSYNC_CONFIG_DIR

cp -f /usr/share/quicksync/settings.json $QUICKSYNC_CONFIG_DIR/settings.json

# Wait a bit for boot to complete
sleep 15s

# Check if quicksync was disabled
while [ -f /home/bitcoin/.mynode/quicksync_disabled ] || [ -f /mnt/hdd/mynode/settings/quicksync_disabled ]; do
    # Pretend quicksync step is complete
    touch $QUICKSYNC_DIR/.quicksync_download_complete
    touch $QUICKSYNC_DIR/.quicksync_complete
    sync
    # Give Bitcoin a couple minutes to start
    sleep 120s
    echo "stable" > $MYNODE_DIR/.mynode_status
    sync
    sleep 10d
done

# Make sure folder exists
mkdir -p $QUICKSYNC_DIR
if [ ! -f $QUICKSYNC_DIR/.quicksync_download_complete ]; then
    echo "quicksync_download" > $MYNODE_DIR/.mynode_status
fi
echo "Starting quicksync..."

# Download finished, but failed during copy, recopy
if [ ! -f $QUICKSYNC_DIR/.quicksync_complete ] && [ -f $QUICKSYNC_DIR/.quicksync_download_complete ]; then
    echo "Quicksync download complete, needs copy"
    /usr/bin/mynode_quicksync_complete.sh
fi
# Check if quicksync was completed
if [ -f $QUICKSYNC_DIR/.quicksync_complete ]; then
    echo "stable" > $MYNODE_DIR/.mynode_status
fi

# Download torrent
rm -rf $QUICKSYNC_DIR/blockchain_temp.torrent
wget -O $QUICKSYNC_DIR/blockchain_temp.torrent $QUICKSYNC_TORRENT_URL
sync
sleep 1

if [ ! -f $QUICKSYNC_DIR/blockchain_temp.torrent ]; then
    echo "Torrent download failed...."
    exit 1
fi

# Check if new torrent is updated
if [ ! -f $QUICKSYNC_DIR/blockchain.torrent ]; then
    cp $QUICKSYNC_DIR/blockchain_temp.torrent $QUICKSYNC_DIR/blockchain.torrent
else
    if [ $IS_RASPI3 -eq 1 ] && [ ! -f $UPLOADER_FILE ]; then
        # Don't help with uploads during normal operation.... too slow
        sleep 1s
    else
        # Run commands as long as torrents are different (last command updates torrent file)
        COMPLETED=0
        if [ -f $QUICKSYNC_DIR/.quicksync_complete ]; then
            COMPLETED=1
        fi
        
        NEW_TORRENT=0
        cmp --silent $QUICKSYNC_DIR/blockchain_temp.torrent $QUICKSYNC_DIR/blockchain.torrent || NEW_TORRENT=1
        if [ $NEW_TORRENT -eq 1 ]; then
            # Delete old QuickSync data+config and start new one
            rm -f $QUICKSYNC_DIR/*
            rm -rf $QUICKSYNC_CONFIG_DIR
            mkdir -p $QUICKSYNC_CONFIG_DIR
            cp -f /usr/share/quicksync/settings.json $QUICKSYNC_CONFIG_DIR/settings.json
            cp $QUICKSYNC_DIR/blockchain_temp.torrent $QUICKSYNC_DIR/blockchain.torrent
            sync

            # If download had been completed
            if [ $COMPLETED -eq 1 ]; then
                touch $QUICKSYNC_DIR/.quicksync_download_complete
                touch $QUICKSYNC_DIR/.quicksync_complete
            fi
            sync
        fi
    fi
fi

# If Quicksync has already completed, let's give BTC some time to start before
# seeding or downloading a more recent copy
if [ ! -f $UPLOADER_FILE ]; then
    if [ -f $QUICKSYNC_DIR/.quicksync_complete ]; then
        /usr/bin/wait_on_bitcoin.sh
        sleep 5m
    fi
fi

# Start torrent
echo "Running torrent..."
transmission-cli \
    --download-dir $QUICKSYNC_DIR \
    --config-dir $QUICKSYNC_CONFIG_DIR \
    --finish=/usr/bin/mynode_quicksync_complete.sh \
    $QUICKSYNC_DIR/blockchain.torrent