#!/bin/bash

source /usr/share/mynode/mynode_config.sh
source /usr/share/mynode/mynode_functions.sh
source /usr/share/mynode/mynode_app_versions.sh

set -x
set -e

# Make sure time is in the log
date

# Mark we are upgrading
echo "upgrading" > $MYNODE_STATUS_FILE

# Shut down main services to save memory and CPU
/usr/bin/mynode_stop_critical_services.sh

# Check if upgrades use tor
TORIFY=""
if [ -f /mnt/hdd/mynode/settings/torify_apt_get ]; then
    TORIFY="torify"
fi

# Delete ramlog to prevent ram issues (remake necessary folders)
rm -rf /var/log/*
mkdir -p /var/log/nginx
if [ $IS_RASPI = 1 ]; then
    log2ram write || true
    log2ram stop || true
fi

# Skip base upgrades if we are doing an app install / uninstall
if ! skip_base_upgrades ; then

    # Create any necessary users
    useradd -m -s /bin/bash joinmarket || true

    # Setup bitcoin user folders
    mkdir -p /home/bitcoin/.mynode/
    chown -R bitcoin:bitcoin /home/bitcoin/.mynode/

    # User updates and settings
    adduser admin bitcoin
    grep "joinmarket" /etc/sudoers || (echo 'joinmarket ALL=(ALL) NOPASSWD:ALL' | EDITOR='tee -a' visudo)

    # Migrate from version file to version+install combo
    /usr/bin/mynode_migrate_version_files.sh


    # Stop and disable any old services
    systemctl disable https || true
    systemctl stop https || true
    systemctl disable tls_proxy || true
    systemctl stop tls_proxy || true

    # Create dhparam.pem (do before dpkg configure since its needed for nginx)
    /usr/bin/mynode_gen_dhparam.sh


    # Check if any dpkg installs have failed and correct
    dpkg --configure -a


    # Add sources
    apt-get -y install apt-transport-https
    DEBIAN_VERSION=$(lsb_release -c | awk '{ print $2 }')
    # Tor (arm32 support was dropped)
    if [ $IS_64_BIT = 1 ]; then
        grep -qxF "deb https://deb.torproject.org/torproject.org ${DEBIAN_VERSION} main" /etc/apt/sources.list  || echo "deb https://deb.torproject.org/torproject.org ${DEBIAN_VERSION} main" >> /etc/apt/sources.list
        grep -qxF "deb-src https://deb.torproject.org/torproject.org ${DEBIAN_VERSION} main" /etc/apt/sources.list  || echo "deb-src https://deb.torproject.org/torproject.org ${DEBIAN_VERSION} main" >> /etc/apt/sources.list
    fi
    # Raspbian mirrors
    #if [ $IS_RASPI = 1 ]; then
    #    grep -qxF "deb http://plug-mirror.rcac.purdue.edu/raspbian/ ${DEBIAN_VERSION} main" /etc/apt/sources.list  || echo "deb http://plug-mirror.rcac.purdue.edu/raspbian/ ${DEBIAN_VERSION} main" >> /etc/apt/sources.list
    #    grep -qxF "deb http://mirrors.ocf.berkeley.edu/raspbian/raspbian ${DEBIAN_VERSION} main" /etc/apt/sources.list  || echo "deb http://mirrors.ocf.berkeley.edu/raspbian/raspbian ${DEBIAN_VERSION} main" >> /etc/apt/sources.list
    #fi

    # Import Keys
    set +e
    curl https://keybase.io/roasbeef/pgp_keys.asc | gpg --import
    curl https://keybase.io/bitconner/pgp_keys.asc | gpg --import
    curl https://keybase.io/guggero/pgp_keys.asc | gpg --import # Pool
    curl https://raw.githubusercontent.com/JoinMarket-Org/joinmarket-clientserver/master/pubkeys/AdamGibson.asc | gpg --import
    gpg --keyserver hkp://keyserver.ubuntu.com --recv-keys 01EA5486DE18A882D4C2684590C8019E36C2E964
    gpg --keyserver hkp://keyserver.ubuntu.com --recv-keys E777299FC265DD04793070EB944D35F9AC3DB76A # Bitcoin - Michael Ford (fanquake)
    curl https://keybase.io/suheb/pgp_keys.asc | gpg --import
    curl https://samouraiwallet.com/pgp.txt | gpg --import # two keys from Samourai team
    gpg  --keyserver hkp://keyserver.ubuntu.com --recv-keys DE23E73BFA8A0AD5587D2FCDE80D2F3F311FD87E #loopd
    $TORIFY curl https://deb.torproject.org/torproject.org/A3C4F0F979CAA22CDBA8F512EE8CBC9E886DDD89.asc | gpg --import  # tor
    gpg --export A3C4F0F979CAA22CDBA8F512EE8CBC9E886DDD89 | apt-key add -                                       # tor
    set -e


    # Check for updates (might auto-install all updates later)
    export DEBIAN_FRONTEND=noninteractive
    $TORIFY apt-get update --allow-releaseinfo-change

    # Freeze any packages we don't want to update
    if [ $IS_X86 = 1 ]; then
        apt-mark hold grub*
    fi
    apt-mark hold redis-server

    # Upgrade packages
    $TORIFY apt-get -y upgrade

    # Install any new software
    $TORIFY apt-get -y install apt-transport-https
    $TORIFY apt-get -y install fonts-dejavu
    $TORIFY apt-get -y install pv sysstat network-manager unzip pkg-config libfreetype6-dev libpng-dev
    $TORIFY apt-get -y install libatlas-base-dev libffi-dev libssl-dev glances python3-bottle
    $TORIFY apt-get -y -qq install apt-transport-https ca-certificates
    $TORIFY apt-get -y install libgmp-dev automake libtool libltdl-dev libltdl7
    $TORIFY apt-get -y install xorg chromium openbox lightdm openjdk-11-jre libevent-dev ncurses-dev
    $TORIFY apt-get -y install libudev-dev libusb-1.0-0-dev python3-venv gunicorn sqlite3 libsqlite3-dev
    $TORIFY apt-get -y install torsocks python3-requests libsystemd-dev libjpeg-dev zlib1g-dev psmisc
    $TORIFY apt-get -y install hexyl libbz2-dev liblzma-dev netcat-openbsd hdparm iotop

    # Install device specific packages
    if [ $IS_X86 = 1 ]; then
        $TORIFY apt-get -y install cloud-init
    fi

    # Make sure some software is removed
    apt-get -y purge ntp # (conflicts with systemd-timedatectl)
    apt-get -y purge chrony # (conflicts with systemd-timedatectl)


    # Install nginx
    mkdir -p /var/log/nginx || true
    $TORIFY apt-get -y install nginx || true
    # Install may fail, so we need to edit the default config file and reconfigure
    rm -f /etc/nginx/modules-enabled/50-mod-* || true
    echo "" > /etc/nginx/sites-available/default
    dpkg --configure -a

    # Install any pip software
    pip2 install tzupdate virtualenv pysocks redis qrcode image subprocess32 --no-cache-dir


    # Update Python3 to 3.7.X
    PYTHON_VERSION=3.7.9
    CURRENT_PYTHON3_VERSION=$(python3 --version)
    if [[ "$CURRENT_PYTHON3_VERSION" != *"Python ${PYTHON_VERSION}"* ]]; then
        mkdir -p /opt/download
        cd /opt/download
        rm -rf Python-*

        wget https://www.python.org/ftp/python/${PYTHON_VERSION}/Python-${PYTHON_VERSION}.tar.xz -O python.tar.xz
        tar xf python.tar.xz

        cd Python-*
        ./configure
        make -j4
        make install
        cd ~
    else
        echo "Python up to date"
    fi


    # Install any pip3 software
    pip3 install --upgrade pip setuptools wheel
    pip3 install gnureadline --no-cache-dir
    pip3 install docker-compose --no-cache-dir
    pip3 install pipenv --no-cache-dir
    pip3 install bcrypt --no-cache-dir
    pip3 install pysocks --no-cache-dir
    pip3 install redis --no-cache-dir
    pip3 install systemd --no-cache-dir

    # Update Node
    if [ -f /etc/apt/sources.list.d/nodesource.list ]; then
        CURRENT_NODE_VERSION=$(cat /etc/apt/sources.list.d/nodesource.list)
        if [[ "$CURRENT_NODE_VERSION" != *"node_${NODE_JS_VERSION}"* ]]; then
            # Upgrade node
            curl -sL https://deb.nodesource.com/setup_${NODE_JS_VERSION} | bash -
            apt-get install -y nodejs

            # Mark apps using node as needing re-install
            rm -f /home/bitcoin/.mynode/lndhub_version
            rm -f /home/bitcoin/.mynode/sphinxrelay_version
            rm -f /home/bitcoin/.mynode/thunderhub_version
            rm -f /home/bitcoin/.mynode/corsproxy_version
            rm -f /home/bitcoin/.mynode/rtl_version
            rm -f /home/bitcoin/.mynode/caravan_version
            rm -f /home/bitcoin/.mynode/btcrpcexplorer_version
            rm -f /home/bitcoin/.mynode/bos_version
        else
            echo "Node version match"
        fi
    else
        echo "No node apt sources file?"
    fi

    # Update NPM (Node Package Manager)
    npm install -g npm@$NODE_NPM_VERSION
    
    # Install Docker
    if [ ! -f /usr/bin/docker ]; then
        rm -f /tmp/docker_install.sh
        wget https://get.docker.com -O /tmp/docker_install.sh
        sed -i 's/sleep 20/sleep 1/' /tmp/docker_install.sh
        /bin/bash /tmp/docker_install.sh
    fi

    # Use systemd for managing Docker
    rm -f /etc/init.d/docker
    rm -f /etc/systemd/system/multi-user.target.wants/docker.service
    systemctl -f enable docker.service

    groupadd docker || true
    usermod -aG docker admin
    usermod -aG docker bitcoin
    usermod -aG docker root


    # Install node packages

fi

# Install LNDManage
pip3 install lndmanage==$LNDMANAGE_VERSION --no-cache-dir
echo $LNDMANAGE_VERSION > $LNDMANAGE_VERSION_FILE


# Upgrade BTC
echo "Upgrading BTC..."
ARCH="UNKNOWN"
if [ $IS_RASPI = 1 ]; then
    ARCH="arm-linux-gnueabihf"
    if [ $IS_RASPI4_ARM64 = 1 ]; then
        ARCH="aarch64-linux-gnu"
    fi
elif [ $IS_ROCK64 = 1 ] || [ $IS_ROCKPRO64 = 1 ]; then
    ARCH="aarch64-linux-gnu"
elif [ $IS_X86 = 1 ]; then
    ARCH="x86_64-linux-gnu"
else
    echo "Unknown Bitcoin Version"
    exit 1
fi
BTC_UPGRADE_URL=https://bitcoincore.org/bin/bitcoin-core-$BTC_VERSION/bitcoin-$BTC_VERSION-$ARCH.tar.gz
BTC_UPGRADE_SHA256SUM_URL=https://bitcoincore.org/bin/bitcoin-core-$BTC_VERSION/SHA256SUMS
BTC_UPGRADE_SHA256SUM_ASC_URL=https://bitcoincore.org/bin/bitcoin-core-$BTC_VERSION/SHA256SUMS.asc
CURRENT=""
if [ -f $BTC_VERSION_FILE ]; then
    CURRENT=$(cat $BTC_VERSION_FILE)
fi
if [ "$CURRENT" != "$BTC_VERSION" ]; then
    # Download and install Bitcoin
    rm -rf /opt/download
    mkdir -p /opt/download
    cd /opt/download

    wget $BTC_UPGRADE_URL
    wget $BTC_UPGRADE_SHA256SUM_URL -O SHA256SUMS
    wget $BTC_UPGRADE_SHA256SUM_ASC_URL -O SHA256SUMS.asc

    sha256sum --ignore-missing --check SHA256SUMS
    if [ $? == 0 ]; then
        gpg --verify SHA256SUMS.asc SHA256SUMS |& grep "gpg: Good signature"
        if [ $? == 0 ]; then
            # Install Bitcoin
            tar -xvf bitcoin-$BTC_VERSION-$ARCH.tar.gz
            mv bitcoin-$BTC_VERSION bitcoin
            install -m 0755 -o root -g root -t /usr/local/bin bitcoin/bin/*

            # Mark current version
            echo $BTC_VERSION > $BTC_VERSION_FILE
        else
            echo "ERROR UPGRADING BITCOIN - GPG FAILED"
        fi
    else
        echo "ERROR UPGRADING BITCOIN - SHASUM FAILED"
    fi
fi

# Upgrade LND
echo "Upgrading LND..."
LND_ARCH="lnd-linux-armv7"
if [ $IS_X86 = 1 ]; then
    LND_ARCH="lnd-linux-amd64"
fi
if [ $IS_RASPI4_ARM64 = 1 ]; then
    LND_ARCH="lnd-linux-arm64"
fi
LND_UPGRADE_URL=https://github.com/lightningnetwork/lnd/releases/download/$LND_VERSION/$LND_ARCH-$LND_VERSION.tar.gz
CURRENT=""
if [ -f $LND_VERSION_FILE ]; then
    CURRENT=$(cat $LND_VERSION_FILE)
fi
if [ "$CURRENT" != "$LND_VERSION" ]; then
    # Download and install LND
    rm -rf /opt/download
    mkdir -p /opt/download
    cd /opt/download

    wget $LND_UPGRADE_URL
    wget $LND_UPGRADE_MANIFEST_URL -O manifest.txt
    wget $LND_UPGRADE_MANIFEST_SIG_URL -O manifest.txt.sig

    gpg --verify manifest.txt.sig manifest.txt
    if [ $? == 0 ]; then
        # Install LND
        tar -xzf lnd-*.tar.gz
        mv $LND_ARCH-$LND_VERSION lnd
        install -m 0755 -o root -g root -t /usr/local/bin lnd/*

        # Mark current version
        echo $LND_VERSION > $LND_VERSION_FILE
    else
        echo "ERROR UPGRADING LND - GPG FAILED"
    fi
fi

# Upgrade Loop
echo "Upgrading loop..."
LOOP_ARCH="loop-linux-armv7"
if [ $IS_X86 = 1 ]; then
    LOOP_ARCH="loop-linux-amd64"
fi
if [ $IS_RASPI4_ARM64 = 1 ]; then
    LOOP_ARCH="loop-linux-arm64"
fi
LOOP_UPGRADE_URL=https://github.com/lightninglabs/loop/releases/download/$LOOP_VERSION/$LOOP_ARCH-$LOOP_VERSION.tar.gz
CURRENT=""
if [ -f $LOOP_VERSION_FILE ]; then
    CURRENT=$(cat $LOOP_VERSION_FILE)
fi
if [ "$CURRENT" != "$LOOP_VERSION" ]; then
    # Download and install Loop
    rm -rf /opt/download
    mkdir -p /opt/download
    cd /opt/download

    wget $LOOP_UPGRADE_URL
    wget $LOOP_UPGRADE_MANIFEST_URL -O manifest.txt
    wget $LOOP_UPGRADE_MANIFEST_SIG_URL -O manifest.txt.sig

    gpg --verify manifest.txt.sig manifest.txt
    if [ $? == 0 ]; then
        # Install Loop
        tar -xzf loop-*.tar.gz
        mv $LOOP_ARCH-$LOOP_VERSION loop
        install -m 0755 -o root -g root -t /usr/local/bin loop/*

        # Mark current version
        echo $LOOP_VERSION > $LOOP_VERSION_FILE
    else
        echo "ERROR UPGRADING LOOP - GPG FAILED"
    fi
fi

# Upgrade Pool
echo "Upgrading pool..."
POOL_ARCH="pool-linux-armv7"
if [ $IS_X86 = 1 ]; then
    POOL_ARCH="pool-linux-amd64"
fi
if [ $IS_RASPI4_ARM64 = 1 ]; then
    POOL_ARCH="pool-linux-arm64"
fi
POOL_UPGRADE_URL=https://github.com/lightninglabs/pool/releases/download/$POOL_VERSION/$POOL_ARCH-$POOL_VERSION.tar.gz
CURRENT=""
if [ -f $POOL_VERSION_FILE ]; then
    CURRENT=$(cat $POOL_VERSION_FILE)
fi
if [ "$CURRENT" != "$POOL_VERSION" ]; then
    # Download and install pool
    rm -rf /opt/download
    mkdir -p /opt/download
    cd /opt/download

    wget $POOL_UPGRADE_URL
    wget $POOL_UPGRADE_MANIFEST_URL -O manifest.txt
    wget $POOL_UPGRADE_MANIFEST_SIG_URL -O manifest.txt.sig

    gpg --verify manifest.txt.sig manifest.txt
    if [ $? == 0 ]; then
        # Install Pool
        tar -xzf pool-*.tar.gz
        mv $POOL_ARCH-$POOL_VERSION pool
        install -m 0755 -o root -g root -t /usr/local/bin pool/*

        # Mark current version
        echo $POOL_VERSION > $POOL_VERSION_FILE
    else
        echo "ERROR UPGRADING POOL - GPG FAILED"
    fi
fi

# Upgrade Lightning Terminal
echo "Upgrading lit..."
LIT_ARCH="lightning-terminal-linux-armv7"
if [ $IS_X86 = 1 ]; then
    LIT_ARCH="lightning-terminal-linux-amd64"
fi
if [ $IS_RASPI4_ARM64 = 1 ]; then
    LIT_ARCH="lightning-terminal-linux-arm64"
fi
LIT_UPGRADE_URL=https://github.com/lightninglabs/lightning-terminal/releases/download/$LIT_VERSION/$LIT_ARCH-$LIT_VERSION.tar.gz
CURRENT=""
if [ -f $LIT_VERSION_FILE ]; then
    CURRENT=$(cat $LIT_VERSION_FILE)
fi
if [ "$CURRENT" != "$LIT_VERSION" ]; then
    # Download and install lit
    rm -rf /opt/download
    mkdir -p /opt/download
    cd /opt/download

    wget $LIT_UPGRADE_URL
    wget $LIT_UPGRADE_MANIFEST_URL -O manifest.txt
    wget $LIT_UPGRADE_MANIFEST_SIG_URL  -O manifest.txt.sig

    gpg --verify manifest.txt.sig manifest.txt
    if [ $? == 0 ]; then
        # Install lit
        tar -xzf lightning-terminal-*.tar.gz
        mv $LIT_ARCH-$LIT_VERSION lightning-terminal
        install -m 0755 -o root -g root -t /usr/local/bin lightning-terminal/lit*

        # Mark current version
        echo $LIT_VERSION > $LIT_VERSION_FILE
    else
        echo "ERROR UPGRADING LIT - GPG FAILED"
    fi
fi

# Install LndHub
if should_install_app "lndhub" ; then
    LNDHUB_UPGRADE_URL=https://github.com/BlueWallet/LndHub/archive/$LNDHUB_VERSION.tar.gz
    CURRENT=""
    if [ -f $LNDHUB_VERSION_FILE ]; then
        CURRENT=$(cat $LNDHUB_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$LNDHUB_VERSION" ]; then
        cd /opt/mynode
        rm -rf LndHub

        wget $LNDHUB_UPGRADE_URL
        tar -xzf ${LNDHUB_VERSION}.tar.gz
        rm -f ${LNDHUB_VERSION}.tar.gz
        mv LndHub-* LndHub
        chown -R bitcoin:bitcoin LndHub

        cd LndHub
        sudo -u bitcoin npm install --only=production
        sudo -u bitcoin ln -s /home/bitcoin/.lnd/tls.cert tls.cert
        sudo -u bitcoin ln -s /home/bitcoin/.lnd/data/chain/bitcoin/mainnet/admin.macaroon admin.macaroon

        echo $LNDHUB_VERSION > $LNDHUB_VERSION_FILE
    fi
    cd ~
fi


# Install Caravan
if should_install_app "caravan" ; then
    CARAVAN_UPGRADE_URL=https://github.com/unchained-capital/caravan/archive/$CARAVAN_VERSION.tar.gz
    CURRENT=""
    if [ -f $CARAVAN_VERSION_FILE ]; then
        CURRENT=$(cat $CARAVAN_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$CARAVAN_VERSION" ] || [ ! -f $CARAVAN_SETTINGS_UPDATE_FILE ]; then
        cd /opt/mynode
        rm -rf caravan

        rm -f caravan.tar.gz
        wget $CARAVAN_UPGRADE_URL -O caravan.tar.gz
        tar -xzf caravan.tar.gz
        rm -f caravan.tar.gz
        mv caravan-* caravan
        chown -R bitcoin:bitcoin caravan

        cd caravan
        sudo -u bitcoin npm install --only=production
        echo $CARAVAN_VERSION > $CARAVAN_VERSION_FILE
        touch $CARAVAN_SETTINGS_UPDATE_FILE
    fi
    cd ~
fi


# Install cors proxy (my fork)
CORSPROXY_UPGRADE_URL=https://github.com/tehelsper/CORS-Proxy/archive/$CORSPROXY_VERSION.tar.gz
CURRENT=""
if [ -f $CORSPROXY_VERSION_FILE ]; then
    CURRENT=$(cat $CORSPROXY_VERSION_FILE)
fi
if [ "$CURRENT" != "$CORSPROXY_VERSION" ]; then
    cd /opt/mynode
    rm -rf corsproxy

    rm -f corsproxy.tar.gz
    wget $CORSPROXY_UPGRADE_URL -O corsproxy.tar.gz
    tar -xzf corsproxy.tar.gz
    rm -f corsproxy.tar.gz
    mv CORS-* corsproxy

    cd corsproxy
    npm install
    echo $CORSPROXY_VERSION > $CORSPROXY_VERSION_FILE
fi
cd ~


# Upgrade electrs
# ELECTRS_UPGRADE_URL=https://github.com/romanz/electrs/archive/$ELECTRS_VERSION.tar.gz
# CURRENT=""
# if [ -f $ELECTRS_VERSION_FILE ]; then
#     CURRENT=$(cat $ELECTRS_VERSION_FILE)
# fi
# if [ "$CURRENT" != "$ELECTRS_VERSION" ]; then
#     cd /opt/mynode
#     rm -rf electrs

#     rm -f electrs.tar.gz
#     wget $ELECTRS_UPGRADE_URL -O electrs.tar.gz
#     tar -xzf electrs.tar.gz
#     rm -f electrs.tar.gz
#     mv electrs-* electrs

#     cd electrs
#     cargo build --release
#     install -g root -o root target/release/electrs /usr/bin/electrs
#     echo $ELECTRS_VERSION > $ELECTRS_VERSION_FILE
# fi
# cd ~
echo $ELECTRS_VERSION > $ELECTRS_VERSION_FILE


# Install recent version of secp256k1
echo "Installing secp256k1..."
if [ ! -f /usr/include/secp256k1_ecdh.h ]; then
    rm -rf /tmp/secp256k1
    cd /tmp/
    git clone https://github.com/bitcoin-core/secp256k1.git
    cd secp256k1

    ./autogen.sh
    ./configure
    make
    make install
    cp -f include/* /usr/include/
fi

# Upgrade JoinMarket (legacy)
if should_install_app "joinmarket" ; then
    echo "Upgrading JoinMarket..." # Old
    if [ $IS_RASPI = 1 ] || [ $IS_X86 = 1 ]; then
        JOINMARKET_UPGRADE_URL=https://github.com/JoinMarket-Org/joinmarket-clientserver/archive/$JOINMARKET_VERSION.tar.gz
        CURRENT=""
        if [ -f $JOINMARKET_VERSION_FILE ]; then
            CURRENT=$(cat $JOINMARKET_VERSION_FILE)
        fi
        if [ "$CURRENT" != "$JOINMARKET_VERSION" ]; then
            # Download and build JoinMarket
            cd /opt/mynode

            # Backup old version in case config / wallet was stored within folder
            if [ ! -d /opt/mynode/jm_backup ] && [ -d /opt/mynode/joinmarket-clientserver ]; then
                cp -R /opt/mynode/joinmarket-clientserver /opt/mynode/jm_backup
                chown -R bitcoin:bitcoin /opt/mynode/jm_backup
            fi

            rm -rf joinmarket-clientserver

            sudo -u bitcoin wget $JOINMARKET_UPGRADE_URL -O joinmarket.tar.gz
            sudo -u bitcoin tar -xvf joinmarket.tar.gz
            sudo -u bitcoin rm joinmarket.tar.gz
            mv joinmarket-clientserver-* joinmarket-clientserver

            cd joinmarket-clientserver

            # Apply Patch to fix cryptography dependency
            #sed -i "s/'txtorcon', 'pyopenssl'/'txtorcon', 'cryptography==3.3.2', 'pyopenssl'/g" jmdaemon/setup.py || true

            # Install
            yes | ./install.sh --without-qt

            echo $JOINMARKET_VERSION > $JOINMARKET_VERSION_FILE
        fi
    fi
fi

# Upgrade JoininBox
echo "Upgrading JoinInBox..."
if should_install_app "joininbox" ; then
    if [ $IS_RASPI = 1 ] || [ $IS_X86 = 1 ]; then
        JOININBOX_UPGRADE_URL=https://github.com/openoms/joininbox/archive/$JOININBOX_VERSION.tar.gz
        CURRENT=""
        if [ -f $JOININBOX_VERSION_FILE ]; then
            CURRENT=$(cat $JOININBOX_VERSION_FILE)
        fi
        if [ "$CURRENT" != "$JOININBOX_VERSION" ]; then
            # Download and build JoinInBox
            cd /home/joinmarket
            
            # Delete all non-hidden files
            rm -rf *
            rm -rf joininbox-*

            sudo -u joinmarket wget $JOININBOX_UPGRADE_URL -O joininbox.tar.gz
            sudo -u joinmarket tar -xvf joininbox.tar.gz
            sudo -u joinmarket rm joininbox.tar.gz
            mv joininbox-* joininbox

            chmod -R +x ./joininbox/
            sudo -u joinmarket cp -rf ./joininbox/scripts/* .

            # Install
            sudo -u joinmarket bash -c "cd /home/joinmarket/; ./install.joinmarket.sh install" || true

            echo $JOININBOX_VERSION > $JOININBOX_VERSION_FILE
        fi
    fi
fi

# Install Whirlpool
if should_install_app "whirlpool" ; then
    WHIRLPOOL_UPGRADE_URL=https://code.samourai.io/whirlpool/whirlpool-client-cli/uploads/$WHIRLPOOL_UPLOAD_FILE_ID/whirlpool-client-cli-$WHIRLPOOL_VERSION-run.jar
    WHIRLPOOL_SIG_URL=https://code.samourai.io/whirlpool/whirlpool-client-cli/uploads/$WHIRLPOOL_UPLOAD_SIG_ID/whirlpool-client-cli-$WHIRLPOOL_VERSION-run.jar.sig.asc
    CURRENT=""
    if [ -f $WHIRLPOOL_VERSION_FILE ]; then
        CURRENT=$(cat $WHIRLPOOL_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$WHIRLPOOL_VERSION" ]; then
        sudo -u bitcoin mkdir -p /opt/mynode/whirlpool
        cd /opt/mynode/whirlpool
        sudo rm -rf *.jar
        sudo -u bitcoin wget -O whirlpool.jar $WHIRLPOOL_UPGRADE_URL

        #wget -O whirlpool.asc $WHIRLPOOL_SIG_URL
        cp -f /usr/share/whirlpool/whirlpool.asc whirlpool.asc
        gpg --verify whirlpool.asc

        echo $WHIRLPOOL_VERSION > $WHIRLPOOL_VERSION_FILE
    fi
fi


# Upgrade RTL
if should_install_app "rtl" ; then
    RTL_UPGRADE_URL=https://github.com/Ride-The-Lightning/RTL/archive/$RTL_VERSION.tar.gz
    RTL_UPGRADE_ASC_URL=https://github.com/Ride-The-Lightning/RTL/releases/download/$RTL_VERSION/$RTL_VERSION.tar.gz.asc
    CURRENT=""
    if [ -f $RTL_VERSION_FILE ]; then
        CURRENT=$(cat $RTL_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$RTL_VERSION" ]; then
        cd /opt/mynode
        rm -rf RTL

        sudo -u bitcoin wget $RTL_UPGRADE_URL -O RTL.tar.gz
        #sudo -u bitcoin wget $RTL_UPGRADE_ASC_URL -O RTL.tar.gz.asc

        #gpg --verify RTL.tar.gz.asc RTL.tar.gz
        #if [ $? == 0 ]; then
        if [ true ]; then
            sudo -u bitcoin tar -xvf RTL.tar.gz
            sudo -u bitcoin rm RTL.tar.gz
            sudo -u bitcoin mv RTL-* RTL
            cd RTL
            sudo -u bitcoin NG_CLI_ANALYTICS=false npm install --only=production

            echo $RTL_VERSION > $RTL_VERSION_FILE
        else
            echo "ERROR UPGRADING RTL - GPG FAILED"
        fi
    fi
fi

# Upgrade BTC RPC Explorer
if should_install_app "btcrpcexplorer" ; then
    BTCRPCEXPLORER_UPGRADE_URL=https://github.com/janoside/btc-rpc-explorer/archive/$BTCRPCEXPLORER_VERSION.tar.gz
    CURRENT=""
    if [ -f $BTCRPCEXPLORER_VERSION_FILE ]; then
        CURRENT=$(cat $BTCRPCEXPLORER_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$BTCRPCEXPLORER_VERSION" ]; then
        cd /opt/mynode
        rm -rf btc-rpc-explorer
        sudo -u bitcoin wget $BTCRPCEXPLORER_UPGRADE_URL -O btc-rpc-explorer.tar.gz
        sudo -u bitcoin tar -xvf btc-rpc-explorer.tar.gz
        sudo -u bitcoin rm btc-rpc-explorer.tar.gz
        sudo -u bitcoin mv btc-rpc-* btc-rpc-explorer
        cd btc-rpc-explorer
        sudo -u bitcoin npm install --only=production

        echo $BTCRPCEXPLORER_VERSION > $BTCRPCEXPLORER_VERSION_FILE
    fi
fi


# Upgrade LNBits
if should_install_app "lnbits" ; then
    # Find URL by going to https://github.com/lnbits/lnbits/releases and finding the exact commit for the mynode tag
    LNBITS_UPGRADE_URL=https://github.com/lnbits/lnbits/archive/$LNBITS_VERSION.tar.gz
    CURRENT=""
    if [ -f $LNBITS_VERSION_FILE ]; then
        CURRENT=$(cat $LNBITS_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$LNBITS_VERSION" ]; then
        cd /opt/mynode
        rm -rf lnbits
        sudo -u bitcoin wget $LNBITS_UPGRADE_URL -O lnbits.tar.gz
        sudo -u bitcoin tar -xvf lnbits.tar.gz
        sudo -u bitcoin rm lnbits.tar.gz
        sudo -u bitcoin mv lnbits-* lnbits
        cd lnbits

        # Copy over config file
        cp /usr/share/mynode/lnbits.env /opt/mynode/lnbits/.env
        chown bitcoin:bitcoin /opt/mynode/lnbits/.env

        # Install lnbits
        sudo -u bitcoin python3 -m venv lnbits_venv
        sudo -u bitcoin ./lnbits_venv/bin/pip install -r requirements.txt
        sudo -u bitcoin ./lnbits_venv/bin/quart assets
        sudo -u bitcoin ./lnbits_venv/bin/quart migrate

        echo $LNBITS_VERSION > $LNBITS_VERSION_FILE
    fi
fi


# Upgrade Specter Desktop
if should_install_app "specter" ; then
    CURRENT=""
    if [ -f $SPECTER_VERSION_FILE ]; then
        CURRENT=$(cat $SPECTER_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$SPECTER_VERSION" ]; then
        cd /opt/mynode
        rm -rf specter
        mkdir -p specter
        chown -R bitcoin:bitcoin specter
        cd specter

        # Make venv
        if [ ! -d env ]; then
            sudo -u bitcoin python3 -m venv env
        fi
        source env/bin/activate
        pip3 install ecdsa===0.13.3
        pip3 install cryptoadvance.specter===$SPECTER_VERSION --upgrade
        deactivate

        echo $SPECTER_VERSION > $SPECTER_VERSION_FILE
    fi
fi


# Upgrade Thunderhub
if should_install_app "thunderhub" ; then
    THUNDERHUB_UPGRADE_URL=https://github.com/apotdevin/thunderhub/archive/$THUNDERHUB_VERSION.tar.gz
    CURRENT=""
    if [ -f $THUNDERHUB_VERSION_FILE ]; then
        CURRENT=$(cat $THUNDERHUB_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$THUNDERHUB_VERSION" ]; then
        cd /opt/mynode
        rm -rf thunderhub
        sudo -u bitcoin wget $THUNDERHUB_UPGRADE_URL -O thunderhub.tar.gz
        sudo -u bitcoin tar -xvf thunderhub.tar.gz
        sudo -u bitcoin rm thunderhub.tar.gz
        sudo -u bitcoin mv thunderhub-* thunderhub
        cd thunderhub

        sudo -u bitcoin npm install # --only=production # (can't build with only production)
        sudo -u bitcoin npm run build
        sudo -u bitcoin npx next telemetry disable

        # Setup symlink to service files
        rm -f /opt/mynode/thunderhub/.env.local
        sudo ln -s /mnt/hdd/mynode/thunderhub/.env.local /opt/mynode/thunderhub/.env.local

        echo $THUNDERHUB_VERSION > $THUNDERHUB_VERSION_FILE
    fi
fi


# Install LND Connect
LNDCONNECTARCH="lndconnect-linux-armv7"
if [ $IS_X86 = 1 ]; then
    LNDCONNECTARCH="lndconnect-linux-amd64"
fi
LNDCONNECT_UPGRADE_URL=https://github.com/LN-Zap/lndconnect/releases/download/$LNDCONNECT_VERSION/$LNDCONNECTARCH-$LNDCONNECT_VERSION.tar.gz
CURRENT=""
if [ -f $LNDCONNECT_VERSION_FILE ]; then
    CURRENT=$(cat $LNDCONNECT_VERSION_FILE)
fi
if [ "$CURRENT" != "$LNDCONNECT_VERSION" ]; then
    rm -rf /opt/download
    mkdir -p /opt/download
    cd /opt/download
    wget $LNDCONNECT_UPGRADE_URL -O lndconnect.tar.gz
    tar -xvf lndconnect.tar.gz
    rm lndconnect.tar.gz
    mv lndconnect-* lndconnect
    install -m 0755 -o root -g root -t /usr/local/bin lndconnect/*

    echo $LNDCONNECT_VERSION > $LNDCONNECT_VERSION_FILE
fi


# Install ngrok for debugging
if [ ! -f /usr/bin/ngrok  ]; then
    cd /tmp
    rm -rf /tmp/ngrok*
    NGROK_URL=https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-arm.zip
    if [ $IS_X86 = 1 ]; then
        NGROK_URL=https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-386.zip
    fi
    wget $NGROK_URL
    unzip ngrok-*.zip
    cp ngrok /usr/bin/
fi


# Upgrade CKbunker
if should_install_app "ckbunker" ; then
    CKBUNKER_UPGRADE_URL=https://github.com/Coldcard/ckbunker/archive/$CKBUNKER_VERSION.tar.gz
    CURRENT=""
    if [ -f $CKBUNKER_VERSION_FILE ]; then
        CURRENT=$(cat $CKBUNKER_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$CKBUNKER_VERSION" ]; then
        cd /opt/mynode
        rm -rf ckbunker

        sudo -u bitcoin wget $CKBUNKER_UPGRADE_URL -O ckbunker.tar.gz
        sudo -u bitcoin tar -xvf ckbunker.tar.gz
        sudo -u bitcoin rm ckbunker.tar.gz
        sudo -u bitcoin mv ckbunker-* ckbunker
        cd ckbunker

        # Make venv
        if [ ! -d env ]; then
            sudo -u bitcoin python3 -m venv env
        fi
        source env/bin/activate
        pip3 install -r requirements.txt
        pip3 install --editable .
        deactivate

        echo $CKBUNKER_VERSION > $CKBUNKER_VERSION_FILE
    fi
fi


# Upgrade Sphinx Relay
if should_install_app "sphinxrelay" ; then
    SPHINXRELAY_UPGRADE_URL=https://github.com/stakwork/sphinx-relay/archive/$SPHINXRELAY_VERSION.tar.gz
    CURRENT=""
    if [ -f $SPHINXRELAY_VERSION_FILE ]; then
        CURRENT=$(cat $SPHINXRELAY_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$SPHINXRELAY_VERSION" ]; then
        cd /opt/mynode
        rm -rf sphinxrelay

        sudo -u bitcoin wget $SPHINXRELAY_UPGRADE_URL -O sphinx-relay.tar.gz
        sudo -u bitcoin tar -xvf sphinx-relay.tar.gz
        sudo -u bitcoin rm sphinx-relay.tar.gz
        sudo -u bitcoin mv sphinx-relay-* sphinxrelay
        cd sphinxrelay

        sudo -u bitcoin npm install

        echo $SPHINXRELAY_VERSION > $SPHINXRELAY_VERSION_FILE
    fi
fi


# Upgrade pyblock
if should_install_app "pyblock" ; then
    PYBLOCK_UPGRADE_URL=https://github.com/curly60e/pyblock/archive/refs/tags/$PYBLOCK_VERSION.tar.gz
    CURRENT=""
    if [ -f $PYBLOCK_VERSION_FILE ]; then
        CURRENT=$(cat $PYBLOCK_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$PYBLOCK_VERSION" ]; then
        cd /opt/mynode
        rm -rf pyblock

        sudo -u bitcoin wget $PYBLOCK_UPGRADE_URL -O pyblock.tar.gz
        sudo -u bitcoin tar -xvf pyblock.tar.gz
        sudo -u bitcoin rm pyblock.tar.gz
        sudo -u bitcoin mv pyblock-* pyblock
        cd pyblock

        # Make venv
        if [ ! -d env ]; then
            sudo -u bitcoin python3 -m venv env
        fi
        source env/bin/activate
        pip3 install -r requirements.txt
        deactivate

        # Copy default settings files
        cp -f /usr/share/pyblock/bclock.conf bclock.conf
        cp -f /usr/share/pyblock/blndconnect.conf blndconnect.conf
        chown bitcoin:bitcoin bclock.conf
        chown bitcoin:bitcoin blndconnect.conf

        echo $PYBLOCK_VERSION > $PYBLOCK_VERSION_FILE
    fi
fi


# Upgrade WARden
if should_install_app "warden" ; then
    WARDEN_UPGRADE_URL=https://github.com/pxsocs/warden/archive/refs/tags/$WARDEN_VERSION.tar.gz
    CURRENT=""
    if [ -f $WARDEN_VERSION_FILE ]; then
        CURRENT=$(cat $WARDEN_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$WARDEN_VERSION" ]; then
        cd /opt/mynode
        rm -rf warden

        sudo -u bitcoin wget $WARDEN_UPGRADE_URL -O warden.tar.gz
        sudo -u bitcoin tar -xvf warden.tar.gz
        sudo -u bitcoin rm warden.tar.gz
        sudo -u bitcoin mv warden-* warden
        cd warden

        # Make venv
        if [ ! -d env ]; then
            sudo -u bitcoin python3 -m venv env
        fi
        source env/bin/activate
        pip3 install -r requirements.txt
        deactivate

        echo $WARDEN_VERSION > $WARDEN_VERSION_FILE
    fi
fi


# Upgrade Balance of Satoshis
if should_install_app "bos" ; then
    CURRENT=""
    if [ -f $BOS_VERSION_FILE ]; then
        CURRENT=$(cat $BOS_VERSION_FILE)
    fi
    if [ "$CURRENT" != "$BOS_VERSION" ]; then
        npm install -g balanceofsatoshis@$BOS_VERSION

        echo $BOS_VERSION > $BOS_VERSION_FILE
    fi
fi





# Upgrade Tor
#rm -f /usr/local/bin/tor || true
#TOR_VERSION=$(tor --version)
#if [[ "$TOR_VERSION" != *"Tor version 0.4"* ]]; then
#    $TORIFY apt-get remove -y tor
#    $TORIFY apt-get install -y tor
#fi


# Enable fan control
if [ $IS_ROCKPRO64 = 1 ]; then
    systemctl enable fan_control
fi


# Update nginx conf file
cp -f /usr/share/mynode/nginx.conf /etc/nginx/nginx.conf


# Cleanup MOTD
rm -f /etc/update-motd.d/10-armbian-header || true
rm -f /etc/update-motd.d/30-armbian-sysinfo || true
rm -f /etc/update-motd.d/35-armbian-tips || true
rm -f /etc/update-motd.d/40-armbian-updates || true
rm -f /etc/update-motd.d/41-armbian-config || true
rm -f /etc/update-motd.d/98-armbian-autoreboot-warn || true


# Random Cleanup
if [ -f /etc/apt/sources.list.d/vscode.list ]; then
    sed -i "s/^deb/#deb/g" /etc/apt/sources.list.d/vscode.list
fi
if [ -f /etc/apt/trusted.gpg.d/microsoft.gpg ]; then
    rm /etc/apt/trusted.gpg.d/microsoft.gpg
fi


# Clean apt-cache
apt-get clean

# Enable any new/required services
systemctl enable check_in
systemctl enable background
systemctl enable docker
systemctl enable bitcoin
systemctl enable seed_bitcoin_peers
systemctl enable lnd
systemctl enable lit
systemctl enable firewall
systemctl enable invalid_block_check
systemctl enable usb_driver_check
systemctl enable docker_images
systemctl enable glances
systemctl enable webssh2
systemctl enable tor
systemctl enable loop
systemctl enable pool
systemctl enable rotate_logs
systemctl enable corsproxy_btcrpc

# Disable any old services
systemctl disable bitcoind || true
systemctl disable poold || true
systemctl disable loopd || true
systemctl disable btc_rpc_explorer || true
systemctl disable mempoolspace || true
systemctl disable hitch || true
systemctl disable mongodb || true
systemctl disable lnd_admin || true
systemctl disable lnd_unlock || true
systemctl disable dhcpcd || true
rm /etc/systemd/system/bitcoind.service || true
rm /etc/systemd/system/btc_rpc_explorer.service || true
rm /etc/systemd/system/mempoolspace.service || true
rm /etc/systemd/system/poold.service || true
rm /etc/systemd/system/loopd.service || true

# Reload service settings
systemctl daemon-reload

# Sync FS
sync

# Done!
echo "UPGRADE COMPLETE!!!"
date