wallet_reset_ac.sh

#!/usr/bin/env bash

##
## Decker (c) 2018-2020
##

# How to use?

# This is small script for wallet reset automation. Before use you should send all mined funds
# from node to an other address or cold wallet using ./sendawaynn.sh or ./sendawaynn_auto.sh
# script to leave only iguana outputs (this is not mandatory, but recommended). Also, you should
# have normal utxo count, "normal" mean - to be able to send it in one single transaction. For example,
# if you have more than 1000 utxos - tx will be oversized. Optimal count of uxos to don't break
# max single tx. size limit (100000 bytes) is 100-200 utxos.

# After you start this script it will automatically do following steps:

# 1. Wait for daemon start if it's not started.
# 2. Get pubkey and privkey for given NN_ADDRESS below from daemon.
# 3. Send all your balance in one tx to NN_ADDRESS. If daemon can't send all balance (for example, 
#    in case of oversized tx - script will just exit on this step, nothing hurt)
# 4. Wait for 1 confirmation and save height of block in which this tx is included.
# 5. Create new z-address and grab privkey from it (needed to the trick with rescan).
# 6. Stop daemon and make sure it really stopped.
# 7. Copy current wallet.dat in $HOME/.komodo as backup_%date%_%time%.dat and remove wallet.dat after.
# 8. Start daemon with -gen -notary -pubkey="$NN_PUBKEY" & args .
# 9. Wait for daemon start .
# 10. Imports your NN privkey and rescan from height from step 4.
# 
# Done. 

# If you wish to continue work on this, PRs in repo are welcome.

komodo_cli="$HOME/komodo/src/komodo-cli"
komodo_daemon="$HOME/komodo/src/komodod"

source $HOME/komodo/src/pubkey.txt
echo Pub: ${pubkey}
NN_ADDRESS=$(${komodo_cli} decodescript "21${pubkey}ac" | jq -r .addresses[0])
echo NN Address: ${NN_ADDRESS}

# you'll need only to set NN_ADDRESS, other needed info such as pubkey and privkey
# script will get automatically from daemon

# --------------------------------------------------------------------------
function init_colors () 
{
RESET="\033[0m"
BLACK="\033[30m"    
RED="\033[31m"      
GREEN="\033[32m"    
YELLOW="\033[33m"   
BLUE="\033[34m"     
MAGENTA="\033[35m"  
CYAN="\033[36m"     
WHITE="\033[37m"    
}

# --------------------------------------------------------------------------
function log_print()
{
   datetime=$(date '+%Y-%m-%d %H:%M:%S')
   echo [$datetime] $1
}

# --------------------------------------------------------------------------
function wait_for_daemon()
{

#if [[ ! -z $1 && $1 != "KMD" ]]
if [ ! -z $1 ] && [ $1 != "KMD" ]
then
    coin=$1
    asset=" -ac_name=$1"
else
    coin="KMD"
    asset=""
fi

i=0
while ! $komodo_cli $asset getinfo >/dev/null 2>&1
do 
   i=$((i+1))
   log_print "Waiting for daemon start $coin ($i)"
   sleep 1
   # TODO: in case if daemon start too long, for example, more than 5-7 mins. we should exit from script
done
}

# --------------------------------------------------------------------------
function stop_daemon()
{

#if [[ ! -z $1 && $1 != "KMD" ]]
if [ ! -z $1 ] && [ $1 != "KMD" ]
then
    coin=$1
    asset=" -ac_name=$1"
else
    coin="KMD"
    asset=""
fi

i=0
$komodo_cli $asset stop

if [ $coin == "KMD" ]
then
ddatadir=$HOME/.komodo
else
ddatadir=$HOME/.komodo/$coin
fi

while [ -f $ddatadir/komodod.pid ]
do 
   i=$((i+1))
   log_print "Waiting for daemon $coin stop ($i)"
   sleep 1
done

while [ ! -z $(lsof -Fp $ddatadir/.lock | head -1 | cut -c 2-) ]
do 
   i=$((i+1))
   log_print "Waiting for .lock release by $coin  ($i)"
   sleep 1
done

}

# --------------------------------------------------------------------------
function send_balance()
{
    #if [[ ! -z $1 && $1 != "KMD" ]]
    if [ ! -z $1 ] && [ $1 != "KMD" ]
    then
        coin=$1
        asset=" -ac_name=$1"
    else
    coin="KMD"
        asset=""
    fi

    #echo $komodo_cli $asset getbalance
    BALANCE=$($komodo_cli $asset getbalance 2>/dev/null)
    ERRORLEVEL=$?
    
    if [ "$ERRORLEVEL" -eq "0" ] && [ "$BALANCE" != "0.00000000" ]; then
        message=$(echo -e "(${GREEN}$coin${RESET}) $BALANCE")
        log_print "$message"
    else
        BALANCE="0.00000000"
	    message=$(echo -e "(${RED}$coin${RESET}) $BALANCE")
        log_print "$message"
        # exit 
        # https://stackoverflow.com/questions/1378274/in-a-bash-script-how-can-i-exit-the-entire-script-if-a-certain-condition-occurs
        # If exit <x> is used instead, when the script is invoked with source, it will result in exiting the shell that started the script, but an executable script will just terminate, as expected.
        return 1 2> /dev/null || exit 1
    fi

    # sendtoaddress
    #$komodo_cli $asset sendtoaddress $NN_ADDRESS $BALANCE "" "" true

    # redirected stderr to stdout
    RESULT=$($komodo_cli $asset sendtoaddress $NN_ADDRESS $BALANCE "" "" true 2>&1)
    ERRORLEVEL=$?
    if [ "$ERRORLEVEL" -ne "0" ]; then
	log_print "tx $RESULT"
    # exit
    return 1 2> /dev/null || exit 1
    fi
    log_print "txid: $RESULT"

    i=0
    confirmations=0
    while [ "$confirmations" -eq "0" ]
    do
    confirmations=$($komodo_cli $asset gettransaction $RESULT | jq .confirmations)
    i=$((i+1))
    log_print "Waiting for confirmations ($i).$confirmations"
    sleep 10
    done

    #if [ "$confirmations" -le "0" ]; then
    #    # in case of invalid (orphaned) block mined, sent tx can have negative number of confirmations (!)
    #    message=$(echo -e "${RED}error:${RESET} txid.$RESULT have ${RED}$confirmations${RESET} conf., probably invalid block was mined!")
    #    log_print "$message"
    #    # exit
    #    # try to determine current block height and decrement it by 100 for future rescan, when we will import z_key
    #    blockhash=$($komodo_cli $asset getbestblockhash)
    #    height=$($komodo_cli $asset getblock $blockhash | jq .height)
    #    height=$((height-100))
    #    blockhash=$($komodo_cli $asset getblockhash $height)
    #else
        blockhash=$($komodo_cli $asset gettransaction $RESULT | jq -r .blockhash)
        height=$($komodo_cli $asset getblock $blockhash | jq .height)
    #fi

}

# --------------------------------------------------------------------------
function reset_wallet() {

    if [ ! -z $1 ] && [ $1 != "KMD" ]
    then
        coin=$1
        asset=" -ac_name=$1"
    else
        coin="KMD"
        asset=""
    fi

    log_print "Start reset ($coin) ..."

    wait_for_daemon $coin

    log_print "Gathering pubkey ..."
    NN_PUBKEY=$($komodo_cli $asset validateaddress $NN_ADDRESS | jq -r .pubkey)
    if [ -z $NN_PUBKEY ]
    then
        log_print "Failed to obtain pubkey. Exit"
        # exit
        return 1 2> /dev/null || exit 1
    else
        log_print "Pubkey is $NN_PUBKEY"
    fi

    log_print "Gathering privkey ..."
    NN_PRIVKEY=$($komodo_cli $asset dumpprivkey $NN_ADDRESS)
    if [ -z $NN_PRIVKEY ]
    then
        log_print "Failed to obtain privkey. Exit"
        # exit
        return 1 2> /dev/null || exit 1
    else
        log_print "Privkey is obtained"
    fi

    # disable generate to avoid daemon crash during multiple "error adding notary vin" messages
    $komodo_cli $asset setgenerate false

    send_balance $coin
    log_print "ht.$height ($blockhash)"

    NN_ZADDRESS=$($komodo_cli $asset z_getnewaddress)
    NN_ZKEY=$($komodo_cli $asset z_exportkey $NN_ZADDRESS)
    log_print "New z-address $NN_ZADDRESS"

    if [ $coin == "KMD" ]
    then
        daemon_args=$(ps -fC komodod | grep -v -- "-ac_name=" | grep -Po "komodod .*" | sed 's/komodod//g')
    else
        daemon_args=$(ps -fC komodod | grep -- "-ac_name=$coin" | grep -Po "komodod .*" | sed 's/komodod//g')
    fi

    # if previous time daemon launched with reindex or rescan, we don't need to re-launch daemon with same params,
    # so, we just remove them from launch string.

    daemon_args=$(echo $daemon_args | sed -e "s/-reindex//g")
    daemon_args=$(echo $daemon_args | sed -e "s/-rescan//g")

    log_print "($coin) Args: \"$daemon_args\""

    # TODO: check args, if we can't get arg and can't start daemon, don't need to stop it (!)

    log_print "Stopping daemon ... "
    stop_daemon $coin
    log_print "Removing old wallet ... "

    wallet_file=backup_$(date '+%Y_%m_%d_%H%M%S').dat
    
    if [ $coin == "KMD" ]
    then
        cp $HOME/.komodo/wallet.dat $HOME/.komodo/$wallet_file
        rm $HOME/.komodo/wallet.dat
    else
        cp $HOME/.komodo/$coin/wallet.dat $HOME/.komodo/$coin/$wallet_file
        rm $HOME/.komodo/$coin/wallet.dat
    fi
    
    sleep 5
    log_print "Starting daemon ($coin) ... "

    # *** STARTING DAEMON ***
    $komodo_daemon $daemon_args &
    
    #$komodo_daemon -gen -notary -pubkey="$NN_PUBKEY" &
    
    wait_for_daemon $coin
    log_print "Importing private key ... "
    $komodo_cli $asset importprivkey $NN_PRIVKEY "" false
    log_print "Rescanning from ht.$height ... "
    $komodo_cli $asset z_importkey "$NN_ZKEY" "yes" $height
    log_print "Done reset ($coin)"

}

# Main

curdir=$(pwd)

init_colors

# Unlock all lockunspents to make sure we can access all our utxo!
$komodo_cli lockunspent true $($komodo_cli listlockunspent | jq -c .)
# reset_wallet KMD

reset_wallet AXO
reset_wallet BET
reset_wallet BOTS
reset_wallet BTCH
reset_wallet CCL
reset_wallet COQUICASH
reset_wallet CRYPTO
reset_wallet DEX
reset_wallet ETOMIC
reset_wallet HODL
reset_wallet ILN
reset_wallet JUMBLR
reset_wallet K64
reset_wallet KOIN
reset_wallet KSB
reset_wallet KV
reset_wallet MESH
reset_wallet MGW
reset_wallet MORTY
reset_wallet MSHARK
reset_wallet NINJA
reset_wallet OOT
reset_wallet OUR
reset_wallet PANGEA
reset_wallet PGT
reset_wallet REVS
reset_wallet RFOX
reset_wallet RICK
reset_wallet SEC
reset_wallet SUPERNET
reset_wallet WLC21
reset_wallet ZEXO
reset_wallet ZILLA
reset_wallet STBL